Fixed getMigration filetype for built versions

ilovepixelart · Dec 10, 2024 · 06e3663 · 06e3663
1 parent e84f5e9
commit 06e3663
Showing 1 changed file with 18 additions and 15 deletions.
diff --git a/src/migrator.ts b/src/migrator.ts
@@ -356,21 +356,24 @@ class Migrator {
    * @private
    * @async
    */
-  private async getMigrations(): Promise<{
-    migrationsInDb: IMigration[]
-    migrationsInFs: IFileMigration[]
-  }> {
-    const files = fs.readdirSync(this.migrationsPath)
-    const migrationsInDb = await this.migrationModel.find({}).exec()
-    const migrationsInFs = files
-      .filter((filename) => /^\d{13,}-/.test(filename) && filename.endsWith('.ts'))
-      .map((filename) => {
-        const [time] = filename.split('-')
-        const timestamp = Number.parseInt(time ?? '')
-        const createdAt = new Date(timestamp)
-        const existsInDatabase = migrationsInDb.some((migration) => filename === migration.filename)
-        return { createdAt, filename, existsInDatabase }
-      })
+private async getMigrations(): Promise<{
+  migrationsInDb: IMigration[]
+  migrationsInFs: IFileMigration[]
+}> {
+  const files = fs.readdirSync(this.migrationsPath)
+  const migrationsInDb = await this.migrationModel.find({}).exec()
+
+  const fileExtensions = ['.ts', '.js'];
+
+  const migrationsInFs = files
+    .filter((filename) => /^\d{13,}-/.test(filename) && fileExtensions.some(ext => filename.endsWith(ext)))
+    .map((filename) => {
+      const [time] = filename.split('-')
+      const timestamp = Number.parseInt(time ?? '')
+      const createdAt = new Date(timestamp)
+      const existsInDatabase = migrationsInDb.some((migration) => filename === migration.filename)
+      return { createdAt, filename, existsInDatabase }
+    })
 
     return { migrationsInDb, migrationsInFs }
   }