From bc4a3c3b4453d3c7c0caad40879e766228a60274 Mon Sep 17 00:00:00 2001
From: Christopher Kolstad <chriswk@getunleash.ai>
Date: Tue, 22 Feb 2022 15:10:06 +0100
Subject: [PATCH] fix: cleanup migrations after the 3.13.0 bug

---
 .../20210428103922-patch-role-table.js        | 24 +++++++++
 .../20210428103924-patch-admin-role-name.js   | 15 ++++++
 .../20210428103924-patch-admin_role.js        | 27 ++++++++++
 .../20210428103924-patch-role_permissions.js  | 54 +++++++++++++++++++
 4 files changed, 120 insertions(+)
 create mode 100644 src/migrations/20210428103922-patch-role-table.js
 create mode 100644 src/migrations/20210428103924-patch-admin-role-name.js
 create mode 100644 src/migrations/20210428103924-patch-admin_role.js
 create mode 100644 src/migrations/20210428103924-patch-role_permissions.js

diff --git a/src/migrations/20210428103922-patch-role-table.js b/src/migrations/20210428103922-patch-role-table.js
new file mode 100644
index 0000000000..717f444e60
--- /dev/null
+++ b/src/migrations/20210428103922-patch-role-table.js
@@ -0,0 +1,24 @@
+'use strict';
+
+const async = require('async');
+
+function resolveRoleName(permissions) {
+    if (!permissions || permissions.length === 0) {
+        return 'Viewer';
+    }
+    if (permissions.includes('ADMIN')) {
+        return 'Admin';
+    }
+    return 'Editor';
+}
+
+exports.up = function (db, cb) {
+    db.runSql(
+        'ALTER TABLE roles ADD COLUMN IF NOT EXISTS project text', cb
+    );
+};
+
+exports.down = function (db, cb) {
+    // We can't just remove roles for users as we don't know if there has been any manual additions.
+    cb();
+};
diff --git a/src/migrations/20210428103924-patch-admin-role-name.js b/src/migrations/20210428103924-patch-admin-role-name.js
new file mode 100644
index 0000000000..be57aa0f96
--- /dev/null
+++ b/src/migrations/20210428103924-patch-admin-role-name.js
@@ -0,0 +1,15 @@
+'use strict';
+
+exports.up = function (db, cb) {
+    db.runSql(
+        `
+        UPDATE roles set name='Admin' where name='Super User';
+        `,
+        cb,
+    );
+};
+
+exports.down = function (db, cb) {
+    // We can't just remove roles for users as we don't know if there has been any manual additions.
+    cb();
+};
diff --git a/src/migrations/20210428103924-patch-admin_role.js b/src/migrations/20210428103924-patch-admin_role.js
new file mode 100644
index 0000000000..891e6f02ab
--- /dev/null
+++ b/src/migrations/20210428103924-patch-admin_role.js
@@ -0,0 +1,27 @@
+'use strict';
+
+exports.up = function (db, cb) {
+    db.runSql(
+        `
+        DO $$
+        declare
+        begin
+                WITH admin AS (
+                    SELECT * FROM roles WHERE name in ('Admin', 'Super User') LIMIT 1
+                )
+                INSERT into role_user(role_id, user_id)
+                VALUES 
+                    ((select id from admin), (select id FROM users where username='admin' LIMIT 1)); 
+
+        EXCEPTION WHEN OTHERS THEN
+            raise notice 'Ignored';
+        end;
+        $$;`,
+        cb,
+    );
+};
+
+exports.down = function (db, cb) {
+    // We can't just remove roles for users as we don't know if there has been any manual additions.
+    cb();
+};
diff --git a/src/migrations/20210428103924-patch-role_permissions.js b/src/migrations/20210428103924-patch-role_permissions.js
new file mode 100644
index 0000000000..57cf744373
--- /dev/null
+++ b/src/migrations/20210428103924-patch-role_permissions.js
@@ -0,0 +1,54 @@
+'use strict';
+
+exports.up = function (db, cb) {
+    db.runSql(
+        `
+        DO $$
+        declare
+        begin
+                WITH editor AS (
+                    SELECT * FROM roles WHERE name in ('Editor', 'Regular') LIMIT 1
+                )
+
+                INSERT INTO role_permission(role_id, project, permission)
+                VALUES
+                    ((SELECT id from editor), '', 'CREATE_STRATEGY'),
+                    ((SELECT id from editor), '', 'UPDATE_STRATEGY'),
+                    ((SELECT id from editor), '', 'DELETE_STRATEGY'),
+        
+                    ((SELECT id from editor), '', 'UPDATE_APPLICATION'),
+        
+                    ((SELECT id from editor), '', 'CREATE_CONTEXT_FIELD'),
+                    ((SELECT id from editor), '', 'UPDATE_CONTEXT_FIELD'),
+                    ((SELECT id from editor), '', 'DELETE_CONTEXT_FIELD'),
+                    
+                    ((SELECT id from editor), '', 'CREATE_PROJECT'),
+        
+                    ((SELECT id from editor), '', 'CREATE_ADDON'),
+                    ((SELECT id from editor), '', 'UPDATE_ADDON'),
+                    ((SELECT id from editor), '', 'DELETE_ADDON'),
+                
+                    ((SELECT id from editor), 'default', 'UPDATE_PROJECT'),
+                    ((SELECT id from editor), 'default', 'DELETE_PROJECT'),
+                    ((SELECT id from editor), 'default', 'CREATE_FEATURE'),
+                    ((SELECT id from editor), 'default', 'UPDATE_FEATURE'),
+                    ((SELECT id from editor), 'default', 'DELETE_FEATURE');
+                
+                -- Clean up duplicates
+                DELETE FROM role_permission p1
+                USING           role_permission p2
+                WHERE  p1.created_at < p2.created_at   -- select the "older" ones
+                AND  p1.project    = p2.project       -- list columns that define duplicates
+                AND  p1.permission = p2.permission;
+        EXCEPTION WHEN OTHERS THEN
+            raise notice 'Ignored';
+        end;
+        $$;`,
+        cb,
+    );
+};
+
+exports.down = function (db, cb) {
+    // We can't just remove roles for users as we don't know if there has been any manual additions.
+    cb();
+};