From 7382967613df32b5ed7f7ad4fa882baaf40e1c00 Mon Sep 17 00:00:00 2001
From: Fabien O'Carroll <fabien@allou.is>
Date: Wed, 4 Sep 2019 12:11:53 +0800
Subject: [PATCH] Added createColumnMigration helper

no-issue
---
 core/server/data/schema/commands.js | 37 ++++++++++++++++++++++++++++-
 1 file changed, 36 insertions(+), 1 deletion(-)

diff --git a/core/server/data/schema/commands.js b/core/server/data/schema/commands.js
index c4c7a402b8..f5ce16ef05 100644
--- a/core/server/data/schema/commands.js
+++ b/core/server/data/schema/commands.js
@@ -133,6 +133,40 @@ function checkTables(transaction) {
     }
 }
 
+const createLog = type => msg => common.logging[type](msg);
+
+function createColumnMigration(...migrations) {
+    async function runColumnMigration(conn, migration) {
+        const {
+            table,
+            column,
+            dbIsInCorrectState,
+            operation,
+            operationVerb,
+            columnDefinition
+        } = migration;
+
+        const hasColumn = await conn.schema.hasColumn(table, column);
+        const isInCorrectState = dbIsInCorrectState(hasColumn);
+
+        const log = createLog(isInCorrectState ? 'info' : 'warn');
+
+        log(`${operationVerb} ${table}.${column}`);
+
+        if (!isInCorrectState) {
+            await operation(table, column, conn, columnDefinition);
+        }
+    }
+
+    return async function columnMigration(options) {
+        const conn = options.transacting || options.connection;
+
+        for (const migration of migrations) {
+            await runColumnMigration(conn, migration);
+        }
+    };
+}
+
 module.exports = {
     checkTables: checkTables,
     createTable: createTable,
@@ -143,5 +177,6 @@ module.exports = {
     dropUnique: dropUnique,
     addColumn: addColumn,
     dropColumn: dropColumn,
-    getColumns: getColumns
+    getColumns: getColumns,
+    createColumnMigration
 };