851e585b9b
Add ISC license headers to all .go, .ts, .tsx, and .sql files using each file's git history to determine the correct copyright year or year range. Trademarked icons (brand logos, vendor logos, compliance framework logos) are excluded. Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>
112 lines
3.6 KiB
SQL
112 lines
3.6 KiB
SQL
-- Copyright (c) 2025-2026 Probo Inc <hello@getprobo.com>.
|
|
--
|
|
-- Permission to use, copy, modify, and/or distribute this software for any
|
|
-- purpose with or without fee is hereby granted, provided that the above
|
|
-- copyright notice and this permission notice appear in all copies.
|
|
--
|
|
-- THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
|
|
-- REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
-- AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
-- INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
-- LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
|
|
-- OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
-- PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
-- Rename control_state ENUM to mitigation_state
|
|
-- We need to create a new type and update all references since
|
|
-- PostgreSQL doesn't support renaming enum types directly
|
|
CREATE TYPE mitigation_state AS ENUM (
|
|
'NOT_STARTED',
|
|
'IN_PROGRESS',
|
|
'NOT_APPLICABLE',
|
|
'IMPLEMENTED'
|
|
);
|
|
|
|
-- Create new mitigation_importance type
|
|
CREATE TYPE mitigation_importance AS ENUM (
|
|
'MANDATORY',
|
|
'PREFERRED',
|
|
'ADVANCED'
|
|
);
|
|
|
|
-- Rename controls table to mitigations, adding the new columns with the new types
|
|
CREATE TABLE mitigations (
|
|
id TEXT PRIMARY KEY,
|
|
tenant_id TEXT NOT NULL,
|
|
framework_id TEXT NOT NULL,
|
|
name TEXT NOT NULL,
|
|
description TEXT NOT NULL,
|
|
content_ref TEXT NOT NULL,
|
|
category TEXT NOT NULL,
|
|
state mitigation_state NOT NULL,
|
|
importance mitigation_importance NOT NULL,
|
|
version INTEGER NOT NULL,
|
|
standards TEXT[] NOT NULL,
|
|
created_at TIMESTAMP WITH TIME ZONE NOT NULL,
|
|
updated_at TIMESTAMP WITH TIME ZONE NOT NULL
|
|
);
|
|
|
|
-- Copy data from controls to mitigations table
|
|
INSERT INTO mitigations (
|
|
id,
|
|
tenant_id,
|
|
framework_id,
|
|
name,
|
|
description,
|
|
content_ref,
|
|
category,
|
|
state,
|
|
importance,
|
|
version,
|
|
standards,
|
|
created_at,
|
|
updated_at
|
|
)
|
|
SELECT
|
|
id,
|
|
tenant_id,
|
|
framework_id,
|
|
name,
|
|
description,
|
|
content_ref,
|
|
category,
|
|
(state::TEXT)::mitigation_state,
|
|
(importance::TEXT)::mitigation_importance,
|
|
version,
|
|
standards,
|
|
created_at,
|
|
updated_at
|
|
FROM controls;
|
|
|
|
-- Update tasks to reference mitigations instead of controls
|
|
ALTER TABLE tasks RENAME COLUMN control_id TO mitigation_id;
|
|
|
|
-- Update foreign key constraint
|
|
ALTER TABLE tasks DROP CONSTRAINT fk_tasks_control_id;
|
|
ALTER TABLE tasks ADD CONSTRAINT fk_tasks_mitigation_id
|
|
FOREIGN KEY (mitigation_id) REFERENCES mitigations(id) ON DELETE CASCADE;
|
|
|
|
-- If we have any control_state_transitions table, rename it
|
|
-- Since the original migration file 20250310T161900Z.sql dropped it, this might not be necessary,
|
|
-- but including for completeness
|
|
DO $$
|
|
BEGIN
|
|
IF EXISTS (
|
|
SELECT FROM information_schema.tables
|
|
WHERE table_name = 'control_state_transitions'
|
|
) THEN
|
|
ALTER TABLE control_state_transitions RENAME TO mitigation_state_transitions;
|
|
ALTER TABLE mitigation_state_transitions RENAME COLUMN control_id TO mitigation_id;
|
|
ALTER TABLE mitigation_state_transitions
|
|
DROP CONSTRAINT IF EXISTS control_state_transitions_control_id_fkey;
|
|
ALTER TABLE mitigation_state_transitions
|
|
ADD CONSTRAINT mitigation_state_transitions_mitigation_id_fkey
|
|
FOREIGN KEY (mitigation_id) REFERENCES mitigations(id);
|
|
END IF;
|
|
END $$;
|
|
|
|
-- Drop old controls table and enum types after migration is complete
|
|
DROP TABLE controls;
|
|
DROP TYPE control_state;
|
|
DROP TYPE control_importance;
|