From be9af2074463a0cbcbe04288a55d3d2ddccbb4cb Mon Sep 17 00:00:00 2001
From: VoxBorealis <uusitalo.y@gmail.com>
Date: Thu, 20 Jul 2023 10:24:15 +0300
Subject: [PATCH] make fkey constraint migration more dynamic

---
 ...0230717105644-clean-up-fkey-constraints.js | 41 ++++++++++---------
 1 file changed, 21 insertions(+), 20 deletions(-)

diff --git a/app/src/migrations/20230717105644-clean-up-fkey-constraints.js b/app/src/migrations/20230717105644-clean-up-fkey-constraints.js
index fe62f7f..d3c435e 100644
--- a/app/src/migrations/20230717105644-clean-up-fkey-constraints.js
+++ b/app/src/migrations/20230717105644-clean-up-fkey-constraints.js
@@ -2,8 +2,20 @@
 
 module.exports = {
   async up(queryInterface, Sequelize) {
+    const removeConstraints = async (foreignKeys, target) => {
+      for (const foreignKey of foreignKeys) {
+        const { tableName, constraintName } = foreignKey;
+        if (constraintName.includes(target)) {
+          await queryInterface.removeConstraint(tableName, constraintName);
+        }
+      }
+    };
+
     // Clean up ScheduledMessage constraints to Jobs.
-    await queryInterface.removeConstraint("ScheduledMessages", "ScheduledMessages_channelId_fkey");
+    const scheduledMessageForeignKeys = await queryInterface.getForeignKeyReferencesForTable(
+      "ScheduledMessages",
+    );
+    await removeConstraints(scheduledMessageForeignKeys, "channelId");
     await queryInterface.changeColumn("ScheduledMessages", "channelId", {
       type: Sequelize.INTEGER,
       references: {
@@ -14,10 +26,8 @@ module.exports = {
     });
 
     // Clean up DefaultOffice constraints.
-    await queryInterface.removeConstraint("People", "People_DefaultOffice_fkey");
-    await queryInterface.removeConstraint("People", "People_DefaultOffice_fkey1");
-    await queryInterface.removeConstraint("People", "People_DefaultOffice_fkey2");
-    await queryInterface.removeConstraint("People", "People_DefaultOffice_fkey3");
+    const defaultOfficeForeignKeys = await queryInterface.getForeignKeyReferencesForTable("People");
+    await removeConstraints(defaultOfficeForeignKeys, "DefaultOffice");
     await queryInterface.changeColumn("People", "DefaultOffice", {
       type: Sequelize.INTEGER,
       references: {
@@ -27,10 +37,8 @@ module.exports = {
       onDelete: "SET NULL",
     });
     // Clean up Signups constraints.
-    await queryInterface.removeConstraint("Signups", "Signups_OfficeId_fkey");
-    await queryInterface.removeConstraint("Signups", "Signups_OfficeId_fkey1");
-    await queryInterface.removeConstraint("Signups", "Signups_OfficeId_fkey2");
-    await queryInterface.removeConstraint("Signups", "Signups_OfficeId_fkey3");
+    const signupsForeignkeys = await queryInterface.getForeignKeyReferencesForTable("Signups");
+    await removeConstraints(signupsForeignkeys, "OfficeId");
     await queryInterface.changeColumn("Signups", "OfficeId", {
       type: Sequelize.INTEGER,
       references: {
@@ -41,9 +49,10 @@ module.exports = {
     });
 
     // Clean up Defaultsignups constraints.
-    await queryInterface.removeConstraint("Defaultsignups", "Defaultsignups_OfficeId_fkey");
-    await queryInterface.removeConstraint("Defaultsignups", "Defaultsignups_OfficeId_fkey1");
-    await queryInterface.removeConstraint("Defaultsignups", "Defaultsignups_OfficeId_fkey2");
+    const defaultsignupsForeignKeys = await queryInterface.getForeignKeyReferencesForTable(
+      "Defaultsignups",
+    );
+    await removeConstraints(defaultsignupsForeignKeys, "OfficeId");
     await queryInterface.changeColumn("Defaultsignups", "OfficeId", {
       type: Sequelize.INTEGER,
       references: {
@@ -66,8 +75,6 @@ module.exports = {
       });
     };
     await addOldDefaultsignupsConstraint();
-    await addOldDefaultsignupsConstraint();
-    await addOldDefaultsignupsConstraint();
 
     await queryInterface.removeConstraint("Signups", "Signups_OfficeId_fkey");
     const addOldSignupsConstraint = async () => {
@@ -80,9 +87,6 @@ module.exports = {
       });
     };
     await addOldSignupsConstraint();
-    await addOldSignupsConstraint();
-    await addOldSignupsConstraint();
-    await addOldSignupsConstraint();
 
     const addOldDefaultOfficeConstraint = async () => {
       await queryInterface.changeColumn("People", "DefaultOffice", {
@@ -95,9 +99,6 @@ module.exports = {
     };
     await queryInterface.removeConstraint("People", "People_DefaultOffice_fkey");
     await addOldDefaultOfficeConstraint();
-    await addOldDefaultOfficeConstraint();
-    await addOldDefaultOfficeConstraint();
-    await addOldDefaultOfficeConstraint();
 
     await queryInterface.removeConstraint("ScheduledMessages", "ScheduledMessages_channelId_fkey");