rake-db 2.0.1 → 2.0.2

package/dist/index.esm.js

@@ -1,24 +1,24 @@
-import { toArray, Adapter, quote, isRaw, getRaw, getColumnTypes, columnTypes, getTableData, ColumnType, Operators, resetTableData, emptyObject, TransactionAdapter } from 'pqb';
+import { toArray, Adapter, quote, isRaw, getRaw, getColumnTypes, columnTypes, getTableData, ColumnType, Operators, resetTableData, emptyObject, TransactionAdapter, logParamToLogObject } from 'pqb';
 import Enquirer from 'enquirer';
 import path from 'path';
 import { readdir, mkdir, writeFile } from 'fs/promises';
 import { singular } from 'pluralize';
 
-var __defProp$4 = Object.defineProperty;
+var __defProp$3 = Object.defineProperty;
 var __defProps$3 = Object.defineProperties;
 var __getOwnPropDescs$3 = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols$4 = Object.getOwnPropertySymbols;
-var __hasOwnProp$4 = Object.prototype.hasOwnProperty;
-var __propIsEnum$4 = Object.prototype.propertyIsEnumerable;
-var __defNormalProp$4 = (obj, key, value) => key in obj ? __defProp$4(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues$4 = (a, b) => {
+var __getOwnPropSymbols$3 = Object.getOwnPropertySymbols;
+var __hasOwnProp$3 = Object.prototype.hasOwnProperty;
+var __propIsEnum$3 = Object.prototype.propertyIsEnumerable;
+var __defNormalProp$3 = (obj, key, value) => key in obj ? __defProp$3(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues$3 = (a, b) => {
   for (var prop in b || (b = {}))
-    if (__hasOwnProp$4.call(b, prop))
-      __defNormalProp$4(a, prop, b[prop]);
-  if (__getOwnPropSymbols$4)
-    for (var prop of __getOwnPropSymbols$4(b)) {
-      if (__propIsEnum$4.call(b, prop))
-        __defNormalProp$4(a, prop, b[prop]);
+    if (__hasOwnProp$3.call(b, prop))
+      __defNormalProp$3(a, prop, b[prop]);
+  if (__getOwnPropSymbols$3)
+    for (var prop of __getOwnPropSymbols$3(b)) {
+      if (__propIsEnum$3.call(b, prop))
+        __defNormalProp$3(a, prop, b[prop]);
     }
   return a;
 };
@@ -31,7 +31,12 @@ const migrationConfigDefaults = {
       require("ts-node").register({ compilerOptions: { module: "CommonJS" } });
     }
     require(path2);
-  }
+  },
+  log: true,
+  logger: console
+};
+const getMigrationConfigWithDefaults = (config) => {
+  return __spreadValues$3(__spreadValues$3({}, migrationConfigDefaults), config);
 };
 const getDatabaseAndUserFromOptions = (options) => {
   if (options.connectionString) {
@@ -59,9 +64,9 @@ const setAdapterOptions = (options, set) => {
     if (set.password !== void 0) {
       url.password = set.password;
     }
-    return __spreadProps$3(__spreadValues$4({}, options), { connectionString: url.toString() });
+    return __spreadProps$3(__spreadValues$3({}, options), { connectionString: url.toString() });
   } else {
-    return __spreadValues$4(__spreadValues$4({}, options), set);
+    return __spreadValues$3(__spreadValues$3({}, options), set);
   }
 };
 const askAdminCredentials = async () => {
@@ -94,7 +99,9 @@ const setAdminCredentialsToOptions = async (options) => {
 const createSchemaMigrations = async (db, config) => {
   try {
     await db.query(
-      `CREATE TABLE "${config.migrationsTable}" ( version TEXT NOT NULL )`
+      `CREATE TABLE ${quoteTable(
+        config.migrationsTable
+      )} ( version TEXT NOT NULL )`
     );
     console.log("Created versions table");
   } catch (err) {
@@ -170,6 +177,14 @@ const joinWords = (...words) => {
 const joinColumns = (columns) => {
   return columns.map((column) => `"${column}"`).join(", ");
 };
+const quoteTable = (table) => {
+  const index = table.indexOf(".");
+  if (index !== -1) {
+    return `"${table.slice(0, index)}"."${table.slice(index + 1)}"`;
+  } else {
+    return `"${table}"`;
+  }
+};
 
 const execute = async (options, sql) => {
   const db = new Adapter(options);
@@ -279,17 +294,17 @@ change(async (db) => {`;
   if (rest) {
     if (first === "create" || first === "drop") {
       content += `
-  db.${first === "create" ? "createTable" : "dropTable"}('${rest}', (t) => ({`;
+  await db.${first === "create" ? "createTable" : "dropTable"}('${rest}', (t) => ({`;
       content += makeColumnsContent(args);
       content += "\n  }));";
     } else if (first === "change") {
       content += `
-  db.changeTable('${rest}', (t) => ({`;
+  await db.changeTable('${rest}', (t) => ({`;
       content += "\n  }));";
     } else if (first === "add" || first === "remove") {
       const table = first === "add" ? getTextAfterTo(rest) : getTextAfterFrom(rest);
       content += `
-  db.changeTable(${table ? `'${table}'` : "tableName"}, (t) => ({`;
+  await db.changeTable(${table ? `'${table}'` : "tableName"}, (t) => ({`;
       content += makeColumnsContent(args, first);
       content += "\n  }));";
     }
@@ -330,21 +345,21 @@ const setCurrentMigrationUp = (up) => {
 };
 const getCurrentPromise = () => currentPromise;
 
-var __defProp$3 = Object.defineProperty;
+var __defProp$2 = Object.defineProperty;
 var __defProps$2 = Object.defineProperties;
 var __getOwnPropDescs$2 = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols$3 = Object.getOwnPropertySymbols;
-var __hasOwnProp$3 = Object.prototype.hasOwnProperty;
-var __propIsEnum$3 = Object.prototype.propertyIsEnumerable;
-var __defNormalProp$3 = (obj, key, value) => key in obj ? __defProp$3(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues$3 = (a, b) => {
+var __getOwnPropSymbols$2 = Object.getOwnPropertySymbols;
+var __hasOwnProp$2 = Object.prototype.hasOwnProperty;
+var __propIsEnum$2 = Object.prototype.propertyIsEnumerable;
+var __defNormalProp$2 = (obj, key, value) => key in obj ? __defProp$2(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues$2 = (a, b) => {
   for (var prop in b || (b = {}))
-    if (__hasOwnProp$3.call(b, prop))
-      __defNormalProp$3(a, prop, b[prop]);
-  if (__getOwnPropSymbols$3)
-    for (var prop of __getOwnPropSymbols$3(b)) {
-      if (__propIsEnum$3.call(b, prop))
-        __defNormalProp$3(a, prop, b[prop]);
+    if (__hasOwnProp$2.call(b, prop))
+      __defNormalProp$2(a, prop, b[prop]);
+  if (__getOwnPropSymbols$2)
+    for (var prop of __getOwnPropSymbols$2(b)) {
+      if (__propIsEnum$2.call(b, prop))
+        __defNormalProp$2(a, prop, b[prop]);
     }
   return a;
 };
@@ -385,7 +400,7 @@ const addColumnIndex = (indexes, key, item) => {
   if (item.data) {
     if (item.data.index) {
       indexes.push({
-        columns: [__spreadProps$2(__spreadValues$3({}, item.data.index), { column: key })],
+        columns: [__spreadProps$2(__spreadValues$2({}, item.data.index), { column: key })],
         options: item.data.index
       });
     }
@@ -415,7 +430,9 @@ const constraintToSql = (tableName, up, foreignKey) => {
   )}) ${referencesToSql(table, foreignKey.foreignColumns, foreignKey.options)}`;
 };
 const referencesToSql = (table, columns, foreignKey) => {
-  const sql = [`REFERENCES "${table}"(${joinColumns(columns)})`];
+  const sql = [
+    `REFERENCES ${quoteTable(table)}(${joinColumns(columns)})`
+  ];
   if (foreignKey.match) {
     sql.push(`MATCH ${foreignKey.match.toUpperCase()}`);
   }
@@ -444,7 +461,7 @@ const migrateIndex = (state, up, { columns, options }) => {
   if (options.unique) {
     sql.push("UNIQUE");
   }
-  sql.push(`INDEX "${indexName}" ON "${state.tableName}"`);
+  sql.push(`INDEX "${indexName}" ON ${quoteTable(state.tableName)}`);
   if (options.using) {
     sql.push(`USING ${options.using}`);
   }
@@ -486,7 +503,9 @@ const migrateIndex = (state, up, { columns, options }) => {
 const migrateComments = async (state, comments) => {
   for (const { column, comment } of comments) {
     await state.migration.query(
-      `COMMENT ON COLUMN "${state.tableName}"."${column}" IS ${quote(comment)}`
+      `COMMENT ON COLUMN ${quoteTable(state.tableName)}."${column}" IS ${quote(
+        comment
+      )}`
     );
   }
 };
@@ -498,8 +517,9 @@ const primaryKeyToSql = (primaryKey) => {
   )})`;
 };
 const getPrimaryKeysOfTable = async (db, tableName) => {
-  const { rows } = await db.query({
-    text: `SELECT
+  const { rows } = await db.query(
+    {
+      text: `SELECT
   pg_attribute.attname AS name,
   format_type(pg_attribute.atttypid, pg_attribute.atttypmod) AS type
 FROM pg_index, pg_class, pg_attribute, pg_namespace
@@ -511,26 +531,29 @@ WHERE
   pg_attribute.attrelid = pg_class.oid AND
   pg_attribute.attnum = any(pg_index.indkey) AND
   indisprimary`,
-    values: [tableName]
-  });
+      values: [tableName]
+    },
+    db.types,
+    void 0
+  );
   return rows;
 };
 
-var __defProp$2 = Object.defineProperty;
+var __defProp$1 = Object.defineProperty;
 var __defProps$1 = Object.defineProperties;
 var __getOwnPropDescs$1 = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols$2 = Object.getOwnPropertySymbols;
-var __hasOwnProp$2 = Object.prototype.hasOwnProperty;
-var __propIsEnum$2 = Object.prototype.propertyIsEnumerable;
-var __defNormalProp$2 = (obj, key, value) => key in obj ? __defProp$2(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues$2 = (a, b) => {
+var __getOwnPropSymbols$1 = Object.getOwnPropertySymbols;
+var __hasOwnProp$1 = Object.prototype.hasOwnProperty;
+var __propIsEnum$1 = Object.prototype.propertyIsEnumerable;
+var __defNormalProp$1 = (obj, key, value) => key in obj ? __defProp$1(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues$1 = (a, b) => {
   for (var prop in b || (b = {}))
-    if (__hasOwnProp$2.call(b, prop))
-      __defNormalProp$2(a, prop, b[prop]);
-  if (__getOwnPropSymbols$2)
-    for (var prop of __getOwnPropSymbols$2(b)) {
-      if (__propIsEnum$2.call(b, prop))
-        __defNormalProp$2(a, prop, b[prop]);
+    if (__hasOwnProp$1.call(b, prop))
+      __defNormalProp$1(a, prop, b[prop]);
+  if (__getOwnPropSymbols$1)
+    for (var prop of __getOwnPropSymbols$1(b)) {
+      if (__propIsEnum$1.call(b, prop))
+        __defNormalProp$1(a, prop, b[prop]);
     }
   return a;
 };
@@ -548,16 +571,19 @@ const createJoinTable = async (migration, up, tables, options, fn) => {
     return createTable(migration, up, tableName, options, () => ({}));
   }
   const tablesWithPrimaryKeys = await Promise.all(
-    tables.map(
-      async (table) => [
-        table,
-        await getPrimaryKeysOfTable(migration, table).then(
-          (items) => items.map((item) => __spreadProps$1(__spreadValues$2({}, item), {
-            joinedName: joinWords(singular(table), item.name)
-          }))
-        )
-      ]
-    )
+    tables.map(async (table) => {
+      const primaryKeys = await getPrimaryKeysOfTable(migration, table).then(
+        (items) => items.map((item) => __spreadProps$1(__spreadValues$1({}, item), {
+          joinedName: joinWords(singular(table), item.name)
+        }))
+      );
+      if (!primaryKeys.length) {
+        throw new Error(
+          `Primary key for table ${quoteTable(table)} is not defined`
+        );
+      }
+      return [table, primaryKeys];
+    })
   );
   return createTable(migration, up, tableName, options, (t) => {
     const result = {};
@@ -593,7 +619,7 @@ const createTable = async (migration, up, tableName, options, fn) => {
   if (!up) {
     const { dropMode } = options;
     await migration.query(
-      `DROP TABLE "${tableName}"${dropMode ? ` ${dropMode}` : ""}`
+      `DROP TABLE ${quoteTable(tableName)}${dropMode ? ` ${dropMode}` : ""}`
     );
     return;
   }
@@ -622,7 +648,7 @@ const createTable = async (migration, up, tableName, options, fn) => {
   ${constraintToSql(state.tableName, up, foreignKey)}`);
   });
   await migration.query({
-    text: `CREATE TABLE "${tableName}" (${lines.join(",")}
+    text: `CREATE TABLE ${quoteTable(tableName)} (${lines.join(",")}
 )`,
     values: state.values
   });
@@ -631,7 +657,7 @@ const createTable = async (migration, up, tableName, options, fn) => {
   await migrateComments(state, state.comments);
   if (options.comment) {
     await migration.query(
-      `COMMENT ON TABLE "${tableName}" IS ${quote(options.comment)}`
+      `COMMENT ON TABLE ${quoteTable(tableName)} IS ${quote(options.comment)}`
     );
   }
 };
@@ -743,7 +769,7 @@ const changeTable = async (migration, up, tableName, options, fn) => {
   });
   if (state.alterTable.length) {
     await migration.query(
-      `ALTER TABLE "${tableName}"
+      `ALTER TABLE ${quoteTable(tableName)}
   ${state.alterTable.join(",\n  ")}`
     );
   }
@@ -762,7 +788,7 @@ const changeActions = {
       value = Array.isArray(comment) ? comment[0] : null;
     }
     return migration.query(
-      `COMMENT ON TABLE "${tableName}" IS ${quote(value)}`
+      `COMMENT ON TABLE ${quoteTable(tableName)} IS ${quote(value)}`
     );
   },
   add(state, up, key, item, options) {
@@ -856,26 +882,36 @@ const getRawOrValue = (item, values) => {
   return typeof item === "object" && item && isRaw(item) ? getRaw(item, values) : quote(item);
 };
 
-var __defProp$1 = Object.defineProperty;
-var __getOwnPropSymbols$1 = Object.getOwnPropertySymbols;
-var __hasOwnProp$1 = Object.prototype.hasOwnProperty;
-var __propIsEnum$1 = Object.prototype.propertyIsEnumerable;
-var __defNormalProp$1 = (obj, key, value) => key in obj ? __defProp$1(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues$1 = (a, b) => {
+var __defProp = Object.defineProperty;
+var __defProps = Object.defineProperties;
+var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
   for (var prop in b || (b = {}))
-    if (__hasOwnProp$1.call(b, prop))
-      __defNormalProp$1(a, prop, b[prop]);
-  if (__getOwnPropSymbols$1)
-    for (var prop of __getOwnPropSymbols$1(b)) {
-      if (__propIsEnum$1.call(b, prop))
-        __defNormalProp$1(a, prop, b[prop]);
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
     }
   return a;
 };
+var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
 class Migration extends TransactionAdapter {
-  constructor(tx, up) {
+  constructor(tx, up, options) {
     super(tx.pool, tx.client, tx.types);
     this.up = up;
+    this.log = logParamToLogObject(options.logger || console, options.log);
+  }
+  async query(query, types = this.types, log = this.log) {
+    return wrapWithLog(log, query, () => super.query(query, types));
+  }
+  async arrays(query, types = this.types, log = this.log) {
+    return wrapWithLog(log, query, () => super.arrays(query, types));
   }
   createTable(tableName, cbOrOptions, cb) {
     const options = typeof cbOrOptions === "function" ? {} : cbOrOptions;
@@ -903,7 +939,7 @@ class Migration extends TransactionAdapter {
   }
   async renameTable(from, to) {
     const [table, newName] = this.up ? [from, to] : [to, from];
-    await this.query(`ALTER TABLE "${table}" RENAME TO "${newName}"`);
+    await this.query(`ALTER TABLE ${quoteTable(table)} RENAME TO "${newName}"`);
   }
   addColumn(tableName, columnName, fn) {
     return addColumn(this, this.up, tableName, columnName, fn);
@@ -950,6 +986,22 @@ class Migration extends TransactionAdapter {
       [from]: t.rename(to)
     }));
   }
+  createSchema(schemaName) {
+    return createSchema(this, this.up, schemaName);
+  }
+  dropSchema(schemaName) {
+    return createSchema(this, !this.up, schemaName);
+  }
+  createExtension(name, options = {}) {
+    return createExtension(this, this.up, name, __spreadProps(__spreadValues({}, options), {
+      checkExists: options.ifNotExists
+    }));
+  }
+  dropExtension(name, options = {}) {
+    return createExtension(this, !this.up, name, __spreadProps(__spreadValues({}, options), {
+      checkExists: options.ifExists
+    }));
+  }
   async tableExists(tableName) {
     return queryExists(this, {
       text: `SELECT 1 FROM "information_schema"."tables" WHERE "table_name" = $1`,
@@ -969,26 +1021,62 @@ class Migration extends TransactionAdapter {
     });
   }
 }
+const wrapWithLog = async (log, query, fn) => {
+  if (!log) {
+    return fn();
+  } else {
+    const sql = typeof query === "string" ? { text: query, values: [] } : query.values ? query : __spreadProps(__spreadValues({}, query), { values: [] });
+    const logData = log.beforeQuery(sql);
+    try {
+      const result = await fn();
+      log.afterQuery(sql, logData);
+      return result;
+    } catch (err) {
+      log.onError(err, sql, logData);
+      throw err;
+    }
+  }
+};
 const addColumn = (migration, up, tableName, columnName, fn) => {
   return changeTable(migration, up, tableName, {}, (t) => ({
     [columnName]: t.add(fn(t))
   }));
 };
 const addIndex = (migration, up, tableName, columns, options) => {
-  return changeTable(migration, up, tableName, {}, (t) => __spreadValues$1({}, t.add(t.index(columns, options))));
+  return changeTable(migration, up, tableName, {}, (t) => __spreadValues({}, t.add(t.index(columns, options))));
 };
 const addForeignKey = (migration, up, tableName, columns, foreignTable, foreignColumns, options) => {
-  return changeTable(migration, up, tableName, {}, (t) => __spreadValues$1({}, t.add(t.foreignKey(columns, foreignTable, foreignColumns, options))));
+  return changeTable(migration, up, tableName, {}, (t) => __spreadValues({}, t.add(t.foreignKey(columns, foreignTable, foreignColumns, options))));
 };
 const addPrimaryKey = (migration, up, tableName, columns, options) => {
-  return changeTable(migration, up, tableName, {}, (t) => __spreadValues$1({}, t.add(t.primaryKey(columns, options))));
+  return changeTable(migration, up, tableName, {}, (t) => __spreadValues({}, t.add(t.primaryKey(columns, options))));
+};
+const createSchema = (migration, up, schemaName) => {
+  if (up) {
+    return migration.query(`CREATE SCHEMA "${schemaName}"`);
+  } else {
+    return migration.query(`DROP SCHEMA "${schemaName}"`);
+  }
+};
+const createExtension = (migration, up, name, options) => {
+  if (!up) {
+    return migration.query(
+      `DROP EXTENSION${options.checkExists ? " IF EXISTS" : ""} "${name}"${options.cascade ? " CASCADE" : ""}`
+    );
+  }
+  return migration.query(
+    `CREATE EXTENSION${options.checkExists ? " IF NOT EXISTS" : ""} "${name}"${options.schema ? ` SCHEMA "${options.schema}"` : ""}${options.version ? ` VERSION '${options.version}'` : ""}${options.cascade ? " CASCADE" : ""}`
+  );
 };
 const queryExists = (db, sql) => {
   return db.query(sql).then(({ rowCount }) => rowCount > 0);
 };
 
-const migrateOrRollback = async (options, config, up) => {
+const migrateOrRollback = async (options, config, args, up) => {
+  var _a;
   const files = await getMigrationFiles(config, up);
+  const argCount = parseInt(args[0]);
+  let count = isNaN(argCount) ? up ? Infinity : 1 : argCount;
   for (const opts of toArray(options)) {
     const db = new Adapter(opts);
     const migratedVersions = await getMigratedVersionsMap(db, config);
@@ -997,7 +1085,10 @@ const migrateOrRollback = async (options, config, up) => {
         if (up && migratedVersions[file.version] || !up && !migratedVersions[file.version]) {
           continue;
         }
+        if (count-- <= 0)
+          break;
         await processMigration(db, up, file, config);
+        (_a = config.logger) == null ? void 0 : _a.log(`${file.path} ${up ? "migrated" : "rolled back"}`);
       }
     } finally {
       await db.destroy();
@@ -1006,7 +1097,7 @@ const migrateOrRollback = async (options, config, up) => {
 };
 const processMigration = async (db, up, file, config) => {
   await db.transaction(async (tx) => {
-    const db2 = new Migration(tx, up);
+    const db2 = new Migration(tx, up, config);
     setCurrentMigration(db2);
     setCurrentMigrationUp(up);
     config.requireTs(file.path);
@@ -1020,18 +1111,20 @@ const processMigration = async (db, up, file, config) => {
 };
 const saveMigratedVersion = async (db, version, config) => {
   await db.query(
-    `INSERT INTO "${config.migrationsTable}" VALUES ('${version}')`
+    `INSERT INTO ${quoteTable(config.migrationsTable)} VALUES ('${version}')`
   );
 };
 const removeMigratedVersion = async (db, version, config) => {
   await db.query(
-    `DELETE FROM "${config.migrationsTable}" WHERE version = '${version}'`
+    `DELETE FROM ${quoteTable(
+      config.migrationsTable
+    )} WHERE version = '${version}'`
   );
 };
 const getMigratedVersionsMap = async (db, config) => {
   try {
     const result = await db.arrays(
-      `SELECT * FROM "${config.migrationsTable}"`
+      `SELECT * FROM ${quoteTable(config.migrationsTable)}`
     );
     return Object.fromEntries(result.rows.map((row) => [row[0], true]));
   } catch (err) {
@@ -1042,114 +1135,26 @@ const getMigratedVersionsMap = async (db, config) => {
     throw err;
   }
 };
-const migrate = (options, config) => migrateOrRollback(options, config, true);
-const rollback = (options, config) => migrateOrRollback(options, config, false);
+const migrate = (options, config, args) => migrateOrRollback(options, config, args, true);
+const rollback = (options, config, args) => migrateOrRollback(options, config, args, false);
 
-var __defProp = Object.defineProperty;
-var __defProps = Object.defineProperties;
-var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols = Object.getOwnPropertySymbols;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __propIsEnum = Object.prototype.propertyIsEnumerable;
-var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues = (a, b) => {
-  for (var prop in b || (b = {}))
-    if (__hasOwnProp.call(b, prop))
-      __defNormalProp(a, prop, b[prop]);
-  if (__getOwnPropSymbols)
-    for (var prop of __getOwnPropSymbols(b)) {
-      if (__propIsEnum.call(b, prop))
-        __defNormalProp(a, prop, b[prop]);
-    }
-  return a;
+const rakeDb = async (options, partialConfig = {}, args = process.argv.slice(2)) => {
+  const config = getMigrationConfigWithDefaults(partialConfig);
+  const command = args[0].split(":")[0];
+  if (command === "create") {
+    await createDb(options, config);
+  } else if (command === "drop") {
+    await dropDb(options);
+  } else if (command === "migrate") {
+    await migrate(options, config, args.slice(1));
+  } else if (command === "rollback") {
+    await rollback(options, config, args.slice(1));
+  } else if (command === "g" || command === "generate") {
+    await generate(config, args.slice(1));
+  } else {
+    console.log(`Usage: rake-db [command] [arguments]`);
+  }
 };
-var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
-jest.mock("../common", () => __spreadProps(__spreadValues({}, jest.requireActual("../common")), {
-  getMigrationFiles: jest.fn(),
-  createSchemaMigrations: jest.fn()
-}));
-const options = { connectionString: "postgres://user@localhost/dbname" };
-const files = [
-  { path: "file1", version: "1" },
-  { path: "file2", version: "2" },
-  { path: "file3", version: "3" }
-];
-const getMigratedVersionsArrayMock = jest.fn();
-Adapter.prototype.arrays = getMigratedVersionsArrayMock;
-const queryMock = jest.fn();
-Adapter.prototype.query = queryMock;
-Adapter.prototype.transaction = (cb) => {
-  return cb({});
-};
-const transactionQueryMock = jest.fn();
-Migration.prototype.query = transactionQueryMock;
-const requireTsMock = jest.fn();
-const config = __spreadProps(__spreadValues({}, migrationConfigDefaults), {
-  requireTs: requireTsMock
-});
-describe("migrateOrRollback", () => {
-  beforeEach(() => {
-    jest.clearAllMocks();
-  });
-  describe("migrate", () => {
-    it("should work properly", async () => {
-      getMigrationFiles.mockReturnValueOnce(files);
-      getMigratedVersionsArrayMock.mockResolvedValueOnce({ rows: [["1"]] });
-      queryMock.mockReturnValueOnce(void 0);
-      requireTsMock.mockResolvedValue(void 0);
-      await migrate(options, config);
-      expect(getMigrationFiles).toBeCalledWith(config, true);
-      expect(requireTsMock).toBeCalledWith("file2");
-      expect(requireTsMock).toBeCalledWith("file3");
-      expect(transactionQueryMock).toBeCalledWith(
-        `INSERT INTO "schemaMigrations" VALUES ('2')`
-      );
-      expect(transactionQueryMock).toBeCalledWith(
-        `INSERT INTO "schemaMigrations" VALUES ('3')`
-      );
-    });
-    it("should create migrations table if it not exist", async () => {
-      getMigrationFiles.mockReturnValueOnce([]);
-      getMigratedVersionsArrayMock.mockRejectedValueOnce({ code: "42P01" });
-      createSchemaMigrations.mockResolvedValueOnce(void 0);
-      await migrate(options, config);
-      expect(getMigrationFiles).toBeCalledWith(config, true);
-      expect(createSchemaMigrations).toBeCalled();
-      expect(requireTsMock).not.toBeCalled();
-      expect(transactionQueryMock).not.toBeCalled();
-    });
-  });
-  describe("rollback", () => {
-    it("should work properly", async () => {
-      getMigrationFiles.mockReturnValueOnce(files.reverse());
-      getMigratedVersionsArrayMock.mockResolvedValueOnce({
-        rows: [["1"], ["2"]]
-      });
-      queryMock.mockReturnValueOnce(void 0);
-      requireTsMock.mockResolvedValue(void 0);
-      await rollback(options, config);
-      expect(getMigrationFiles).toBeCalledWith(config, false);
-      expect(requireTsMock).toBeCalledWith("file2");
-      expect(requireTsMock).toBeCalledWith("file1");
-      expect(transactionQueryMock).toBeCalledWith(
-        `DELETE FROM "schemaMigrations" WHERE version = '2'`
-      );
-      expect(transactionQueryMock).toBeCalledWith(
-        `DELETE FROM "schemaMigrations" WHERE version = '1'`
-      );
-    });
-    it("should create migrations table if it not exist", async () => {
-      getMigrationFiles.mockReturnValueOnce([]);
-      getMigratedVersionsArrayMock.mockRejectedValueOnce({ code: "42P01" });
-      createSchemaMigrations.mockResolvedValueOnce(void 0);
-      await rollback(options, config);
-      expect(getMigrationFiles).toBeCalledWith(config, false);
-      expect(createSchemaMigrations).toBeCalled();
-      expect(requireTsMock).not.toBeCalled();
-      expect(transactionQueryMock).not.toBeCalled();
-    });
-  });
-});
 
-export { Migration, change, createDb, dropDb, generate };
+export { Migration, change, createDb, dropDb, generate, migrate, rakeDb, rollback };
 //# sourceMappingURL=index.esm.js.map