From c7242971e33be97e8bb2de48d7d7c24c838d566e Mon Sep 17 00:00:00 2001
From: OG T <ogt@WOOOMacMiniM4.local>
Date: Tue, 5 May 2026 14:08:19 +0800
Subject: [PATCH] fix(aiops): align incidents schema with autoheal model

---
 .../031_fix_incidents_autoheal_schema.sql     | 47 +++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 migrations/031_fix_incidents_autoheal_schema.sql

diff --git a/migrations/031_fix_incidents_autoheal_schema.sql b/migrations/031_fix_incidents_autoheal_schema.sql
new file mode 100644
index 0000000..e9c210d
--- /dev/null
+++ b/migrations/031_fix_incidents_autoheal_schema.sql
@@ -0,0 +1,47 @@
+-- Migration 031: align incidents table with AutoHeal ORM
+-- Created: 2026-05-05 Asia/Taipei
+--
+-- Context:
+-- - Existing production DB was created from migration 013 with error_traceback,
+--   playbook_id and severity VARCHAR(5).
+-- - database/autoheal_models.py now writes traceback_str, matched_playbook_id
+--   and severity values such as "medium".
+-- - Keep the old columns for backward compatibility and add the ORM columns
+--   additively so running services do not need a destructive migration.
+
+BEGIN;
+
+ALTER TABLE incidents
+    ADD COLUMN IF NOT EXISTS traceback_str TEXT;
+
+ALTER TABLE incidents
+    ADD COLUMN IF NOT EXISTS matched_playbook_id INTEGER;
+
+ALTER TABLE incidents
+    ALTER COLUMN severity TYPE VARCHAR(20);
+
+UPDATE incidents
+SET traceback_str = error_traceback
+WHERE traceback_str IS NULL
+  AND error_traceback IS NOT NULL;
+
+UPDATE incidents
+SET matched_playbook_id = playbook_id
+WHERE matched_playbook_id IS NULL
+  AND playbook_id IS NOT NULL;
+
+DO $$
+BEGIN
+    IF NOT EXISTS (
+        SELECT 1
+        FROM pg_constraint
+        WHERE conname = 'incidents_matched_playbook_id_fkey'
+    ) THEN
+        ALTER TABLE incidents
+            ADD CONSTRAINT incidents_matched_playbook_id_fkey
+            FOREIGN KEY (matched_playbook_id)
+            REFERENCES playbooks(id);
+    END IF;
+END $$;
+
+COMMIT;