feat(storage): add "end" method to storage for cleaning up resources when commands are finished

.changeset/calm-grapes-arrive.md

@@ -0,0 +1,5 @@
+---
+'@emigrate/plugin-tools': minor
+---
+
+Add "end" method to storage plugins so they can cleanup resources when a command is finished

.changeset/eight-moles-tell.md

@@ -0,0 +1,5 @@
+---
+'@emigrate/mysql': patch
+---
+
+Fix issue with closing the connection pool too early

.changeset/healthy-flies-divide.md

@@ -0,0 +1,5 @@
+---
+'@emigrate/storage-fs': minor
+---
+
+Implement an empty "end" method for cleaning up

.changeset/large-plants-smoke.md

@@ -0,0 +1,5 @@
+---
+'@emigrate/cli': patch
+---
+
+Handle storage initialization errors in the "list" and "remove" commands

.changeset/thick-forks-knock.md

@@ -0,0 +1,5 @@
+---
+'@emigrate/cli': minor
+---
+
+Call storage.end() to cleanup resources when a command has finished

packages/cli/src/cli.ts

@@ -230,7 +230,7 @@ Examples:
 
   try {
     const { default: listCommand } = await import('./commands/list.js');
-    await listCommand({ directory, storage, reporter });
+    process.exitCode = await listCommand({ directory, storage, reporter });
   } catch (error) {
     if (error instanceof ShowUsageError) {
       console.error(error.message, '\n');
@@ -306,7 +306,7 @@ Examples:
 
   try {
     const { default: removeCommand } = await import('./commands/remove.js');
-    await removeCommand({ directory, storage, reporter, force }, positionals[0] ?? '');
+    process.exitCode = await removeCommand({ directory, storage, reporter, force }, positionals[0] ?? '');
   } catch (error) {
     if (error instanceof ShowUsageError) {
       console.error(error.message, '\n');

packages/cli/src/commands/list.ts

@@ -2,10 +2,11 @@ import process from 'node:process';
 import path from 'node:path';
 import { getOrLoadReporter, getOrLoadStorage } from '@emigrate/plugin-tools';
 import { type MigrationMetadataFinished } from '@emigrate/plugin-tools/types';
-import { BadOptionError, MigrationHistoryError, MissingOptionError } from '../errors.js';
+import { BadOptionError, MigrationHistoryError, MissingOptionError, StorageInitError } from '../errors.js';
 import { type Config } from '../types.js';
 import { withLeadingPeriod } from '../with-leading-period.js';
 import { getMigrations } from '../get-migrations.js';
+import { exec } from '../exec.js';
 
 const lazyDefaultReporter = async () => import('../reporters/default.js');
 
@@ -21,7 +22,6 @@ export default async function listCommand({ directory, reporter: reporterConfig,
     throw new BadOptionError('storage', 'No storage found, please specify a storage using the storage option');
   }
 
-  const storage = await storagePlugin.initializeStorage();
   const reporter = await getOrLoadReporter([reporterConfig ?? lazyDefaultReporter]);
 
   if (!reporter) {
@@ -33,6 +33,14 @@ export default async function listCommand({ directory, reporter: reporterConfig,
 
   await reporter.onInit?.({ command: 'list', cwd, dry: false, directory });
 
+  const [storage, storageError] = await exec(async () => storagePlugin.initializeStorage());
+
+  if (storageError) {
+    await reporter.onFinished?.([], new StorageInitError('Could not initialize storage', { cause: storageError }));
+
+    return 1;
+  }
+
   const migrationFiles = await getMigrations(cwd, directory);
 
   let migrationHistoryError: MigrationHistoryError | undefined;
@@ -82,4 +90,8 @@ export default async function listCommand({ directory, reporter: reporterConfig,
   }
 
   await reporter.onFinished?.(finishedMigrations, migrationHistoryError);
+
+  await storage.end();
+
+  return migrationHistoryError ? 1 : 0;
 }

packages/cli/src/commands/remove.ts

@@ -7,10 +7,12 @@ import {
   MissingArgumentsError,
   MissingOptionError,
   OptionNeededError,
+  StorageInitError,
 } from '../errors.js';
 import { type Config } from '../types.js';
 import { getMigration } from '../get-migration.js';
 import { getDuration } from '../get-duration.js';
+import { exec } from '../exec.js';
 
 type ExtraFlags = {
   force?: boolean;
@@ -37,7 +39,6 @@ export default async function removeCommand(
     throw new BadOptionError('storage', 'No storage found, please specify a storage using the storage option');
   }
 
-  const storage = await storagePlugin.initializeStorage();
   const reporter = await getOrLoadReporter([reporterConfig ?? lazyDefaultReporter]);
 
   if (!reporter) {
@@ -47,6 +48,16 @@ export default async function removeCommand(
     );
   }
 
+  const [storage, storageError] = await exec(async () => storagePlugin.initializeStorage());
+
+  if (storageError) {
+    await reporter.onFinished?.([], new StorageInitError('Could not initialize storage', { cause: storageError }));
+
+    return 1;
+  }
+
+  await reporter.onInit?.({ command: 'remove', cwd, dry: false, directory });
+
   const migrationFile = await getMigration(cwd, directory, name, !force);
 
   const finishedMigrations: MigrationMetadataFinished[] = [];
@@ -59,16 +70,14 @@ export default async function removeCommand(
     }
 
     if (migrationHistoryEntry.status === 'done' && !force) {
-      throw new OptionNeededError(
+      removalError = new OptionNeededError(
         'force',
         `The migration "${migrationFile.name}" is not in a failed state. Use the "force" option to force its removal`,
       );
-    }
+    } else {
-
       historyEntry = migrationHistoryEntry;
     }
-
+  }
-  await reporter.onInit?.({ command: 'remove', cwd, dry: false, directory });
 
   await reporter.onMigrationRemoveStart?.(migrationFile);
 
@@ -107,4 +116,8 @@ export default async function removeCommand(
   }
 
   await reporter.onFinished?.(finishedMigrations, removalError);
+
+  await storage.end();
+
+  return removalError ? 1 : 0;
 }

packages/cli/src/commands/up.test.ts

@@ -218,6 +218,7 @@ function getStorage(historyEntries: Array<string | MigrationHistoryEntry>) {
     remove: mock.fn(),
     onSuccess: mock.fn(),
     onError: mock.fn(),
+    end: mock.fn(),
   };
 
   return storage;

packages/cli/src/commands/up.ts

@@ -15,11 +15,13 @@ import {
   MigrationRunError,
   MissingOptionError,
   StorageInitError,
+  toError,
 } from '../errors.js';
 import { type Config } from '../types.js';
 import { withLeadingPeriod } from '../with-leading-period.js';
 import { getMigrations as getMigrationsOriginal, type GetMigrationsFunction } from '../get-migrations.js';
 import { getDuration } from '../get-duration.js';
+import { exec } from '../exec.js';
 
 type ExtraFlags = {
   cwd?: string;
@@ -30,29 +32,6 @@ type ExtraFlags = {
 const lazyDefaultReporter = async () => import('../reporters/default.js');
 const lazyPluginLoaderJs = async () => import('../plugin-loader-js.js');
 
-const toError = (error: unknown) => (error instanceof Error ? error : new Error(String(error)));
-
-type Fn<Args extends any[], Result> = (...args: Args) => Result;
-type Result<T> = [value: T, error: undefined] | [value: undefined, error: Error];
-
-/**
- * Execute a function and return a result tuple
- *
- * This is a helper function to make it easier to handle errors without the extra nesting of try/catch
- */
-const exec = async <Args extends any[], Return extends Promise<any>>(
-  fn: Fn<Args, Return>,
-  ...args: Args
-): Promise<Result<Awaited<Return>>> => {
-  try {
-    const result = await fn(...args);
-
-    return [result, undefined];
-  } catch (error) {
-    return [undefined, toError(error)];
-  }
-};
-
 export default async function upCommand({
   storage: storageConfig,
   reporter: reporterConfig,
@@ -165,6 +144,8 @@ export default async function upCommand({
         new BadOptionError('plugin', `No loader plugin found for file extension: ${extension}`),
       );
 
+      await storage.end();
+
       return 1;
     }
   }
@@ -191,6 +172,8 @@ export default async function upCommand({
 
     await reporter.onFinished?.([...failedEntries, ...finishedMigrations], error);
 
+    await storage.end();
+
     return failedEntries.length > 0 ? 1 : 0;
   }
 
@@ -207,6 +190,8 @@ export default async function upCommand({
 
     await reporter.onFinished?.([], toError(error));
 
+    await storage.end();
+
     return 1;
   }
 
@@ -226,7 +211,7 @@ export default async function upCommand({
     process.off('SIGINT', cleanup);
     process.off('SIGTERM', cleanup);
 
-    cleaningUp = storage.unlock(lockedMigrationFiles);
+    cleaningUp = storage.unlock(lockedMigrationFiles).then(async () => storage.end());
 
     return cleaningUp;
   };

packages/cli/src/errors.ts

@@ -2,6 +2,8 @@ import { type MigrationHistoryEntry, type MigrationMetadata } from '@emigrate/pl
 
 const formatter = new Intl.ListFormat('en', { style: 'long', type: 'disjunction' });
 
+export const toError = (error: unknown) => (error instanceof Error ? error : new Error(String(error)));
+
 export class EmigrateError extends Error {
   constructor(
     public code: string,

packages/cli/src/exec.ts

@@ -0,0 +1,22 @@
+import { toError } from './errors.js';
+
+type Fn<Args extends any[], Result> = (...args: Args) => Result;
+type Result<T> = [value: T, error: undefined] | [value: undefined, error: Error];
+
+/**
+ * Execute a function and return a result tuple
+ *
+ * This is a helper function to make it easier to handle errors without the extra nesting of try/catch
+ */
+export const exec = async <Args extends any[], Return extends Promise<any>>(
+  fn: Fn<Args, Return>,
+  ...args: Args
+): Promise<Result<Awaited<Return>>> => {
+  try {
+    const result = await fn(...args);
+
+    return [result, undefined];
+  } catch (error) {
+    return [undefined, toError(error)];
+  }
+};

packages/mysql/src/index.ts

@@ -167,6 +167,10 @@ export const createMysqlStorage = ({ table = defaultTable, connection }: MysqlSt
 
       try {
         await initializeTable(pool, table);
+      } catch (error) {
+        await pool.end();
+        throw error;
+      }
 
       const storage: Storage = {
         async lock(migrations) {
@@ -219,12 +223,12 @@ export const createMysqlStorage = ({ table = defaultTable, connection }: MysqlSt
         async onError(migration, error) {
           await finishMigration(pool, table, { ...migration, status: 'failed', error });
         },
+        async end() {
+          await pool.end();
+        },
       };
 
       return storage;
-      } finally {
-        await pool.end();
-      }
     },
   };
 };

packages/plugin-tools/src/types.ts

@@ -74,6 +74,12 @@ export type Storage = {
    * @param error The error that caused the migration to fail.
    */
   onError(migration: MigrationMetadataFinished, error: SerializedError): Promise<void>;
+  /**
+   * Called when the command is finished or aborted (e.g. by a SIGTERM or SIGINT signal).
+   *
+   * Use this to clean up any resources like database connections or file handles.
+   */
+  end(): Promise<void>;
 };
 
 export type EmigrateStorage = {

packages/storage-fs/src/index.ts

@@ -112,6 +112,9 @@ export default function storageFs({ filename }: StorageFsOptions): EmigrateStora
         async onError(migration, error) {
           await update(migration.name, 'failed', error);
         },
+        async end() {
+          // Nothing to do
+        },
       };
     },
   };