Add pgInstallSchemaFromString helper

voxpelli · Aug 17, 2023 · 66f689b · 66f689b
1 parent 2853a7c
commit 66f689b
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 10 deletions.
diff --git a/index.d.ts b/index.d.ts
@@ -23,6 +23,7 @@ export {
 
 export {
   installSchemaFromString,
+  pgInstallSchemaFromString,
 } from './lib/schema-helper.js';
 
 export {

diff --git a/index.js b/index.js
@@ -14,6 +14,7 @@ export {
 
 export {
   installSchemaFromString,
+  pgInstallSchemaFromString,
 } from './lib/schema-helper.js';
 
 export {

diff --git a/lib/schema-helper.js b/lib/schema-helper.js
@@ -11,22 +11,33 @@ function getTablesFromString (createTablesString) {
   });
 }
 
+/**
+ * @param {import('./advanced-types.js').DefineUmzeptionContexts['pg']} context
+ * @param {string|Promise<string>} createTablesString
+ * @returns {Promise<void>}
+ */
+export async function pgInstallSchemaFromString (context, createTablesString) {
+  const tables = getTablesFromString(await createTablesString);
+
+  await context.value.transact(async client => {
+    for (const table of tables) {
+      await client.query(table);
+    }
+  });
+}
+
 /**
  * @param {string|Promise<string>} createTablesString
  * @returns {import('umzug').MigrationFn<import('./advanced-types.js').AnyUmzeptionContext>}
  */
 export function installSchemaFromString (createTablesString) {
   return async ({ context }) => {
-    if (context.type !== 'pg') {
-      throw new Error(`Unsupported context type: ${context.type}`);
+    switch (context.type) {
+      case 'pg':
+        await pgInstallSchemaFromString(context, createTablesString);
+        break;
+      default:
+        throw new Error(`Unsupported context type: ${context.type}`);
     }
-
-    const tables = getTablesFromString(await createTablesString);
-
-    await context.value.transact(async client => {
-      for (const table of tables) {
-        await client.query(table);
-      }
-    });
   };
 }