aboviq/emigrate
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

feat(cli): support NodeJS callback style migration functions

Browse source
This commit is contained in:
Joakim Carlstein 2023-11-16 11:24:00 +01:00
parent 0c49249bd9
commit 8dadfe9a5b
3 changed files with 25 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

22
packages/cli/src/plugin-loader-js.ts
View file

@ -1,5 +1,21 @@
import { promisify } from 'node:util';
import { type LoaderPlugin } from '@emigrate/plugin-tools/types';
// eslint-disable-next-line @typescript-eslint/ban-types
const promisifyIfNeeded = <T extends Function>(fn: T) => {
if (fn.length === 0) {
return fn;
}
if (fn.length === 1) {
return promisify(fn);
}
throw new Error(
`Unexpected arguments length of migration function, expected 0 or 1 argument but got: ${fn.length} arguments`,
);
};
const loaderJs: LoaderPlugin = {
loadableExtensions: ['.js', '.cjs', '.mjs'],
async loadMigration(migration) {
@ -7,7 +23,7 @@ const loaderJs: LoaderPlugin = {
if (typeof migrationModule === 'function') {
return async () => {
await migrationModule();
await promisifyIfNeeded(migrationModule)();
};
}
@ -19,7 +35,7 @@ const loaderJs: LoaderPlugin = {
) {
const migrationFunction = migrationModule.default;
return async () => {
await migrationFunction();
await promisifyIfNeeded(migrationFunction)();
};
}
@ -31,7 +47,7 @@ const loaderJs: LoaderPlugin = {
) {
const migrationFunction = migrationModule.up;
return async () => {
await migrationFunction();
await promisifyIfNeeded(migrationFunction)();
};
}