fix(medusa): migrate with feature flag (#6771)

* fix(medusa): migrate with feature flag * fix * Create small-guests-whisper.md
medusajs · Mar 21, 2024 · 205573f · 205573f
1 parent 8f5d94b
commit 205573f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/.changeset/small-guests-whisper.md b/.changeset/small-guests-whisper.md
@@ -0,0 +1,5 @@
+---
+"@medusajs/medusa": patch
+---
+
+fix(medusa): ensure feature flags are loaded properly in migrate command
diff --git a/packages/medusa/src/commands/migrate.js b/packages/medusa/src/commands/migrate.js
@@ -72,9 +72,8 @@ const main = async function ({ directory }) {
   args.shift()
   args.shift()
 
-  const featureFlagRouter = featureFlagLoader(configModule)
   const configModule = configModuleLoader(directory)
-  const dataSource = await getDataSource(directory)
+  const featureFlagRouter = featureFlagLoader(configModule)
 
   if (args[0] === "run") {
     if (featureFlagRouter.isFeatureEnabled(MedusaV2Flag.key)) {
@@ -91,6 +90,7 @@ const main = async function ({ directory }) {
         { registerInContainer: false }
       )
     } else {
+      const dataSource = await getDataSource(directory)
       await dataSource.runMigrations()
       await dataSource.destroy()
       await runIsolatedModulesMigration(configModule)
@@ -101,11 +101,13 @@ const main = async function ({ directory }) {
 
     Logger.info("Migrations completed.")
   } else if (args[0] === "revert") {
+    const dataSource = await getDataSource(directory)
     await dataSource.undoLastMigration({ transaction: "all" })
     await dataSource.destroy()
     await revertIsolatedModulesMigration(configModule)
     Logger.info("Migrations reverted.")
   } else if (args[0] === "show") {
+    const dataSource = await getDataSource(directory)
     const unapplied = await dataSource.showMigrations()
     Logger.info(unapplied)
     await dataSource.destroy()