@event-driven-io/pongo 0.17.0-alpha.6 → 0.17.0-beta.10

package/dist/cli.cjs

@@ -5,7 +5,7 @@
 
 
 
-var _chunkAV4SHJQBcjs = require('./chunk-AV4SHJQB.cjs');
+var _chunkTP73JUIXcjs = require('./chunk-TP73JUIX.cjs');
 
 // src/cli.ts
 var _commander = require('commander');
@@ -92,7 +92,7 @@ var parseDefaultDbSchema = (imported) => {
   if (!imported.default.schema.dbs) {
     return missingDbs;
   }
-  const dbs = _chunkAV4SHJQBcjs.objectEntries.call(void 0, imported.default.schema.dbs).map((db) => db[1]);
+  const dbs = _chunkTP73JUIXcjs.objectEntries.call(void 0, imported.default.schema.dbs).map((db) => db[1]);
   const defaultDb = dbs.find((db) => db.name === void 0);
   if (!defaultDb) {
     return missingDefaultDb;
@@ -100,11 +100,11 @@ var parseDefaultDbSchema = (imported) => {
   if (!defaultDb.collections) {
     return missingCollections;
   }
-  const collections = _chunkAV4SHJQBcjs.objectEntries.call(void 0, defaultDb.collections).map((col) => col[1]);
+  const collections = _chunkTP73JUIXcjs.objectEntries.call(void 0, defaultDb.collections).map((col) => col[1]);
   if (collections.length === 0) {
     return missingCollections;
   }
-  return _chunkAV4SHJQBcjs.toDbSchemaMetadata.call(void 0, defaultDb);
+  return _chunkTP73JUIXcjs.toDbSchemaMetadata.call(void 0, defaultDb);
 };
 var configCommand = new (0, _commander.Command)("config").description(
   "Manage Pongo configuration"
@@ -141,17 +141,28 @@ configCommand.command("sample").description("Generate or print sample configurat
 });
 
 // src/commandLine/migrate.ts
-var _dumbo = require('@event-driven-io/dumbo');
 
 
 
 
-var _pg = require('@event-driven-io/dumbo/pg');
+
+var _dumbo = require('@event-driven-io/dumbo');
 
 var migrateCommand = new (0, _commander.Command)("migrate").description(
   "Manage database migrations"
 );
 migrateCommand.command("run").description("Run database migrations").option(
+  "-dbt, --database-type <string>",
+  "Database type that should be used for connection (e.g., PostgreSQL or SQLite)",
+  void 0
+).option(
+  "-drv, --database-driver <string>",
+  'Database driver that should be used for connection (e.g., "pg" for PostgreSQL, "sqlite3" for SQLite)'
+).option(
+  "-dbn, --database-name <string>",
+  "Database name to connect to",
+  void 0
+).option(
   "-cs, --connection-string <string>",
   "Connection string for the database"
 ).option(
@@ -162,8 +173,9 @@ migrateCommand.command("run").description("Run database migrations").option(
   },
   []
 ).option("-f, --config <path>", "Path to configuration file with Pongo config").option("-dr, --dryRun", "Perform dry run without commiting changes", false).action(async (options) => {
-  const { collection, dryRun } = options;
+  const { collection, dryRun, databaseName, databaseDriver } = options;
   const connectionString = _nullishCoalesce(options.connectionString, () => ( process.env.DB_CONNECTION_STRING));
+  const databaseType = _nullishCoalesce(options.databaseType, () => ( _dumbo.parseConnectionString.call(void 0, connectionString).databaseType));
   let collectionNames;
   if (!connectionString) {
     console.error(
@@ -182,18 +194,29 @@ migrateCommand.command("run").description("Run database migrations").option(
     );
     process.exit(1);
   }
-  const pool = _pg.dumbo.call(void 0, { connectionString });
-  const migrations = collectionNames.flatMap(
-    (collectionsName) => _chunkAV4SHJQBcjs.pongoCollectionSchemaComponent.call(void 0, collectionsName).migrations({
-      connector: "PostgreSQL:pg"
-      // TODO: Provide connector here
-    })
-  );
-  await _pg.runPostgreSQLMigrations.call(void 0, pool, migrations, {
+  const driverType = `${databaseType}:${databaseDriver}`;
+  const migrations = getMigrations({
+    driverType,
+    connectionString,
+    databaseName,
+    collectionNames
+  });
+  const pool = _dumbo.dumbo.call(void 0, { connectionString, driverType });
+  await _dumbo.runSQLMigrations.call(void 0, pool, migrations, {
     dryRun
   });
 });
 migrateCommand.command("sql").description("Generate SQL for database migration").option(
+  "-dbt, --database-type <string>",
+  "Database type that should be used for connection (e.g., PostgreSQL or SQLite)"
+).option(
+  "-drv, --database-driver <string>",
+  'Database driver that should be used for connection (e.g., "pg" for PostgreSQL, "sqlite3" for SQLite)'
+).option(
+  "-dbn, --database-name <string>",
+  "Database name to connect to",
+  void 0
+).option(
   "-col, --collection <name>",
   "Specify the collection name",
   (value, previous) => {
@@ -201,7 +224,7 @@ migrateCommand.command("sql").description("Generate SQL for database migration")
   },
   []
 ).option("-f, --config <path>", "Path to configuration file with Pongo config").option("--print", "Print the SQL to the console (default)", true).action(async (options) => {
-  const { collection } = options;
+  const { collection, databaseName, databaseType, databaseDriver } = options;
   let collectionNames;
   if (options.config) {
     const config = await loadConfigFile(options.config);
@@ -214,21 +237,40 @@ migrateCommand.command("sql").description("Generate SQL for database migration")
     );
     process.exit(1);
   }
-  const coreMigrations = _pg.migrationTableSchemaComponent.migrations({
-    connector: "PostgreSQL:pg"
+  const driverType = `${databaseType}:${databaseDriver}`;
+  const migrations = getMigrations({
+    driverType,
+    connectionString: void 0,
+    databaseName,
+    collectionNames
   });
-  const migrations = [
-    ...coreMigrations,
-    ...collectionNames.flatMap(
-      (collectionName) => _chunkAV4SHJQBcjs.pongoCollectionSchemaComponent.call(void 0, collectionName).migrations({
-        connector: "PostgreSQL:pg"
-        // TODO: Provide connector here
-      })
-    )
-  ];
   console.log("Printing SQL:");
   console.log(_dumbo.combineMigrations.call(void 0, ...migrations));
 });
+var getMigrations = ({
+  driverType,
+  connectionString,
+  databaseName,
+  collectionNames
+}) => {
+  const driver = _chunkTP73JUIXcjs.pongoDatabaseDriverRegistry.tryGet(driverType);
+  if (driver === null) {
+    console.error(
+      `Error: No database driver found for driver type "${driverType}". Make sure the driver is registered and the connection string is correct.`
+    );
+    process.exit(1);
+  }
+  const dbDefinition = _chunkTP73JUIXcjs.pongoSchema.db.from(databaseName, collectionNames);
+  const driverOptions = {
+    schema: { definition: dbDefinition }
+  };
+  const customOptions = {
+    connectionString: _nullishCoalesce(connectionString, () => ( driver.defaultConnectionString)),
+    databaseName
+  };
+  const db = driver.databaseFactory({ ...driverOptions, ...customOptions });
+  return db.schema.component.migrations;
+};
 
 // src/commandLine/shell.ts
 
@@ -239,6 +281,7 @@ migrateCommand.command("sql").description("Generate SQL for database migration")
 
 
 
+var _pg = require('@event-driven-io/dumbo/pg');
 var _clitable3 = require('cli-table3'); var _clitable32 = _interopRequireDefault(_clitable3);
 
 var _repl = require('repl'); var _repl2 = _interopRequireDefault(_repl);
@@ -310,7 +353,7 @@ var prettifyLogs = (logLevel) => {
 var startRepl = async (options) => {
   setLogLevel(_nullishCoalesce(process.env.DUMBO_LOG_LEVEL, () => ( options.logging.logLevel)));
   setLogStyle(_nullishCoalesce(process.env.DUMBO_LOG_STYLE, () => ( options.logging.logStyle)));
-  console.log(_dumbo.color.green("Starting Pongo Shell (version: 0.17.0-alpha.6)"));
+  console.log(_dumbo.color.green("Starting Pongo Shell (version: 0.17.0-beta.10)"));
   if (options.logging.printOptions) {
     console.log(_dumbo.color.green("With Options:"));
     console.log(_dumbo.prettyJson.call(void 0, options));
@@ -323,6 +366,15 @@ var startRepl = async (options) => {
       )
     );
   }
+  const { databaseType } = _dumbo.parseConnectionString.call(void 0, connectionString);
+  const driverType = `${databaseType}:${options.databaseDriver}`;
+  const driver = _chunkTP73JUIXcjs.pongoDatabaseDriverRegistry.tryGet(driverType);
+  if (driver === null) {
+    console.error(
+      `Error: No database driver found for driver type "${driverType}". Make sure the driver is installed and the connection string is correct.`
+    );
+    process.exit(1);
+  }
   const connectionCheck = await _pg.checkConnection.call(void 0, connectionString);
   if (!connectionCheck.successful) {
     if (connectionCheck.errorType === "ConnectionRefused") {
@@ -355,16 +407,24 @@ var startRepl = async (options) => {
   if (options.schema.collections.length > 0) {
     const collectionsSchema = {};
     for (const collectionName of options.schema.collections) {
-      collectionsSchema[collectionName] = _chunkAV4SHJQBcjs.pongoSchema.collection(collectionName);
+      collectionsSchema[collectionName] = _chunkTP73JUIXcjs.pongoSchema.collection(collectionName);
     }
-    const schema = _chunkAV4SHJQBcjs.pongoSchema.client({
-      database: _chunkAV4SHJQBcjs.pongoSchema.db(options.schema.database, collectionsSchema)
+    const schema = _chunkTP73JUIXcjs.pongoSchema.client({
+      database: _chunkTP73JUIXcjs.pongoSchema.db(options.schema.database, collectionsSchema)
     });
-    const typedClient = _chunkAV4SHJQBcjs.pongoClient.call(void 0, connectionString, {
+    const driverOptions = {
+      driver,
       schema: {
         definition: schema,
         autoMigration: options.schema.autoMigration
       }
+    };
+    const customOptions = {
+      connectionString
+    };
+    const typedClient = _chunkTP73JUIXcjs.pongoClient.call(void 0, {
+      ...driverOptions,
+      ...customOptions
     });
     db = typedClient.database;
     for (const collectionName of options.schema.collections) {
@@ -372,8 +432,18 @@ var startRepl = async (options) => {
     }
     pongo = typedClient;
   } else {
-    pongo = _chunkAV4SHJQBcjs.pongoClient.call(void 0, connectionString, {
-      schema: { autoMigration: options.schema.autoMigration }
+    const driverOptions = {
+      driver,
+      schema: {
+        autoMigration: options.schema.autoMigration
+      }
+    };
+    const customOptions = {
+      connectionString
+    };
+    pongo = _chunkTP73JUIXcjs.pongoClient.call(void 0, {
+      ...driverOptions,
+      ...customOptions
     });
     db = pongo.db(options.schema.database);
   }
@@ -402,6 +472,10 @@ var teardown = async () => {
 process.on("uncaughtException", teardown);
 process.on("SIGINT", teardown);
 var shellCommand = new (0, _commander.Command)("shell").description("Start an interactive Pongo shell").option(
+  "-drv, --database-driver <string>",
+  'Database driver that should be used for connection (e.g., "pg" for PostgreSQL, "sqlite3" for SQLite)',
+  "pg"
+).option(
   "-cs, --connectionString <string>",
   "Connection string for the database"
 ).option("-db, --database <string>", "Database name to connect", "postgres").option(
@@ -432,7 +506,8 @@ var shellCommand = new (0, _commander.Command)("shell").description("Start an in
       database,
       autoMigration: options.disableAutoMigrations ? "None" : "CreateOrUpdate"
     },
-    connectionString
+    connectionString,
+    databaseDriver: options.databaseDriver
   });
 });
 

package/dist/cli.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/home/runner/work/Pongo/Pongo/src/packages/pongo/dist/cli.cjs","../src/cli.ts","../src/commandLine/configFile.ts","../src/commandLine/migrate.ts","../src/commandLine/shell.ts"],"names":[],"mappings":"AAAA;AACA;AACE;AACA;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACA;ACRA,sCAAwB;ADUxB;AACA;AEZA;AACA,gEAAe;AAQf,IAAM,eAAA,EAAiB,CAAC,KAAA,EAAA,GAA0B;AAChD,EAAA,GAAA,CAAI,KAAA,CAAM,OAAA,IAAW,CAAA,EAAG;AACtB,IAAA,OAAO,KAAA;AAAA,EACT;AAEA,EAAA,IAAI,UAAA,EAAY,KAAA,CAAM,MAAA,CAAO,CAAC,CAAA,CAAE,WAAA,CAAY,EAAA,EAAI,KAAA,CAAM,KAAA,CAAM,CAAC,CAAA;AAE7D,EAAA,GAAA,CAAI,SAAA,CAAU,QAAA,CAAS,GAAG,CAAA,EAAG;AAC3B,IAAA,UAAA,EAAY,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,CAAA,CAAE,CAAA;AAAA,EACnC;AAEA,EAAA,OAAO,SAAA;AACT,CAAA;AAEA,IAAM,aAAA,EAAe,CAAC,gBAAA,EAA4B,CAAC,OAAO,CAAA,EAAA,GAAM;AAC9D,EAAA,MAAM,MAAA,EAAQ,eAAA,CACX,GAAA;AAAA,IACC,CAAC,IAAA,EAAA,GACC,CAAA,YAAA,EAAe,cAAA,CAAe,IAAI,CAAC,CAAA,oDAAA;AAAA,EACvC,CAAA,CACC,IAAA,CAAK,IAAI,CAAA;AAEZ,EAAA,MAAM,YAAA,EAAc,eAAA,CACjB,GAAA;AAAA,IACC,CAAC,IAAA,EAAA,GACC,CAAA,MAAA,EAAS,IAAI,CAAA,yBAAA,EAA4B,cAAA,CAAe,IAAI,CAAC,CAAA,GAAA,EAAM,IAAI,CAAA,GAAA;AAAA,EAC3E,CAAA,CACC,IAAA,CAAK,IAAI,CAAA;AAEZ,EAAA,OAAO,CAAA;AAAA;AAAA,EAEP,KAAK,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAKL,WAAW,CAAA;AAAA;AAAA;AAAA,EAAA,CAAA;AAIb,CAAA;AAEA,IAAM,qBAAA,EAAuB,CAAA;AAAA;AAAA,EAAwD,YAAA,CAAa,CAAC,CAAA,CAAA;AAC7F;AAAgB;AAAuE;AAC1E;AAAA;AAAuF;AACpG;AAAmB;AAA+H;AAClJ;AAAqB;AAAsF;AAEpG;AAGL,EAAA;AACF,EAAA;AAEI,IAAA;AAIA,IAAA;AAEK,IAAA;AACD,MAAA;AACA,MAAA;AACV,IAAA;AAEO,IAAA;AACD,EAAA;AACE,IAAA;AACK,IAAA;AACf,EAAA;AACF;AAEa;AAIP,EAAA;AACC,IAAA;AACS,IAAA;AACL,EAAA;AACC,IAAA;AACA,IAAA;AACK,IAAA;AACf,EAAA;AACF;AAEa;AAGG,EAAA;AACL,IAAA;AACT,EAAA;AAEc,EAAA;AACL,IAAA;AACT,EAAA;AAEc,EAAA;AACL,IAAA;AACT,EAAA;AAEY,EAAA;AAEN,EAAA;AAED,EAAA;AACI,IAAA;AACT,EAAA;AAEe,EAAA;AACN,IAAA;AACT,EAAA;AAEM,EAAA;AAEF,EAAA;AACK,IAAA;AACT,EAAA;AAEO,EAAA;AACT;AAaa;AACX,EAAA;AACF;AAGG;AAGC,EAAA;AACA,EAAA;AACgB,EAAA;AAEP,IAAA;AACT,EAAA;AACC,EAAA;AAEF;AACC,EAAA;AACA,EAAA;AAEM;AAGA,EAAA;AAGA,EAAA;AACI,IAAA;AACN,MAAA;AACF,IAAA;AACa,IAAA;AACf,EAAA;AAEe,EAAA;AACD,IAAA;AACH,EAAA;AACI,IAAA;AACH,MAAA;AACN,QAAA;AACF,MAAA;AACQ,MAAA;AACV,IAAA;AAEA,IAAA;AACF,EAAA;AACD;AF9Cc;AACA;AG9IR;AACT;AACE;AACA;AACA;AACK;AACE;AAkBI;AACX,EAAA;AACF;AAGG;AAGC,EAAA;AACA,EAAA;AAED;AACC,EAAA;AACA,EAAA;AACgB,EAAA;AAEP,IAAA;AACT,EAAA;AACC,EAAA;AAEK;AAGE,EAAA;AACF,EAAA;AAEF,EAAA;AAEC,EAAA;AACK,IAAA;AACN,MAAA;AAEF,IAAA;AACa,IAAA;AACf,EAAA;AAEY,EAAA;AACJ,IAAA;AAEN,IAAA;AACS,EAAA;AACT,IAAA;AACK,EAAA;AACG,IAAA;AACN,MAAA;AACF,IAAA;AACa,IAAA;AACf,EAAA;AAEa,EAAA;AAEP,EAAA;AAAsC,IAAA;AAE7B,MAAA;AAAA;AACZ,IAAA;AACH,EAAA;AAEM,EAAA;AACJ,IAAA;AACD,EAAA;AACF;AAGA;AAGC,EAAA;AACA,EAAA;AACgB,EAAA;AAEP,IAAA;AACT,EAAA;AACC,EAAA;AAEK;AAIE,EAAA;AAEJ,EAAA;AAEQ,EAAA;AACJ,IAAA;AAEN,IAAA;AACS,EAAA;AACT,IAAA;AACK,EAAA;AACG,IAAA;AACN,MAAA;AACF,IAAA;AACa,IAAA;AACf,EAAA;AAEM,EAAA;AACO,IAAA;AACZ,EAAA;AACK,EAAA;AACD,IAAA;AACA,IAAA;AAAyB,MAAA;AAExB,QAAA;AAAW;AACZ,MAAA;AACH,IAAA;AACF,EAAA;AAEY,EAAA;AACA,EAAA;AACb;AHkGc;AACA;AIxOjB;AACE;AACA;AACA;AACA;AACA;AAEK;AACE;AACF;AACE;AACQ;AASb;AAEE;AAKE,EAAA;AACE,IAAA;AACA,MAAA;AACO,MAAA;AAAK,QAAA;AAAA;AAEP,UAAA;AAAoC,QAAA;AAC7C,MAAA;AACF,IAAA;AACO,IAAA;AACR,EAAA;AACM,EAAA;AACT;AAEI;AAEE;AAIA;AAMA;AACQ,EAAA;AACG,IAAA;AACf,EAAA;AAEM,EAAA;AAEM,IAAA;AAAA;AAED,MAAA;AAAmD,IAAA;AAEnD,EAAA;AAEL,EAAA;AAEQ,EAAA;AACN,IAAA;AACK,IAAA;AACZ,EAAA;AAEO,EAAA;AACA,IAAA;AACJ,MAAA;AAAiB,QAAA;AAAA;AAER,UAAA;AAAS;AAEZ,YAAA;AAAyB;AAEvB,cAAA;AAAiC,YAAA;AAAA;AAEjC,cAAA;AAAsB,YAAA;AACxB,UAAA;AAII,QAAA;AACV,MAAA;AACF,IAAA;AACD,EAAA;AAEY,EAAA;AACf;AAEM;AACQ,EAAA;AACd;AAEM;AACQ,EAAA;AACd;AAEM;AACA,EAAA;AACQ,EAAA;AACd;AAEkB;AAeJ,EAAA;AACA,EAAA;AAEA,EAAA;AAEA,EAAA;AACE,IAAA;AACA,IAAA;AACd,EAAA;AAEM,EAAA;AAKQ,EAAA;AACJ,IAAA;AACA,MAAA;AACJ,QAAA;AACF,MAAA;AACF,IAAA;AACF,EAAA;AAEM,EAAA;AAED,EAAA;AACC,IAAA;AACM,MAAA;AACA,QAAA;AACJ,UAAA;AACF,QAAA;AACF,MAAA;AACS,IAAA;AACD,MAAA;AACA,QAAA;AACJ,UAAA;AACF,QAAA;AACF,MAAA;AACK,IAAA;AACG,MAAA;AACV,IAAA;AACY,IAAA;AACC,IAAA;AACf,EAAA;AAEY,EAAA;AACA,EAAA;AAEE,EAAA;AACJ,IAAA;AACG,IAAA;AACX,IAAA;AACQ,IAAA;AACT,EAAA;AAEG,EAAA;AAEQ,EAAA;AACJ,IAAA;AAEK,IAAA;AACT,MAAA;AAEF,IAAA;AAEM,IAAA;AACM,MAAA;AACX,IAAA;AAEK,IAAA;AACI,MAAA;AACN,QAAA;AACA,QAAA;AACF,MAAA;AACD,IAAA;AAEI,IAAA;AAEM,IAAA;AACH,MAAA;AACR,IAAA;AAEQ,IAAA;AACH,EAAA;AACG,IAAA;AACI,MAAA;AACX,IAAA;AAEU,IAAA;AACb,EAAA;AAEc,EAAA;AACA,EAAA;AAGA,EAAA;AACA,EAAA;AACA,EAAA;AACA,EAAA;AACA,EAAA;AACA,EAAA;AACA,EAAA;AAGL,EAAA;AACD,IAAA;AACO,IAAA;AACd,EAAA;AAEQ,EAAA;AACD,IAAA;AACO,IAAA;AACd,EAAA;AACH;AAEiB;AACH,EAAA;AACA,EAAA;AACd;AAEW;AACA;AAaL;AAGF,EAAA;AACA,EAAA;AAEM;AAEN,EAAA;AACA,EAAA;AACgB,EAAA;AAEP,IAAA;AACT,EAAA;AACC,EAAA;AAEF;AACC,EAAA;AACA,EAAA;AAEM;AAEN,EAAA;AACA,EAAA;AACA,EAAA;AAEM;AAGE,EAAA;AACF,EAAA;AAEA,EAAA;AACK,IAAA;AACP,MAAA;AACU,MAAA;AAGA,MAAA;AAKZ,IAAA;AACQ,IAAA;AACN,MAAA;AACA,MAAA;AACA,MAAA;AAGF,IAAA;AACA,IAAA;AACD,EAAA;AACF;AJ+Hc;AACA;AClbD;AAEH;AAEL;AACA;AACA;AAEM;AAEP;ADgbU;AACA;AACA","file":"/home/runner/work/Pongo/Pongo/src/packages/pongo/dist/cli.cjs","sourcesContent":[null,"#!/usr/bin/env node\nimport { Command } from 'commander';\nimport { configCommand, migrateCommand, shellCommand } from './commandLine';\n\nconst program = new Command();\n\nprogram.name('pongo').description('CLI tool for Pongo');\n\nprogram.addCommand(configCommand);\nprogram.addCommand(migrateCommand);\nprogram.addCommand(shellCommand);\n\nprogram.parse(process.argv);\n\nexport default program;\n","import { Command } from 'commander';\nimport fs from 'node:fs';\nimport {\n  objectEntries,\n  toDbSchemaMetadata,\n  type PongoDbSchemaMetadata,\n  type PongoSchemaConfig,\n} from '../core';\n\nconst formatTypeName = (input: string): string => {\n  if (input.length === 0) {\n    return input;\n  }\n\n  let formatted = input.charAt(0).toUpperCase() + input.slice(1);\n\n  if (formatted.endsWith('s')) {\n    formatted = formatted.slice(0, -1);\n  }\n\n  return formatted;\n};\n\nconst sampleConfig = (collectionNames: string[] = ['users']) => {\n  const types = collectionNames\n    .map(\n      (name) =>\n        `export type ${formatTypeName(name)} = { name: string; description: string; date: Date }`,\n    )\n    .join('\\n');\n\n  const collections = collectionNames\n    .map(\n      (name) =>\n        `      ${name}: pongoSchema.collection<${formatTypeName(name)}>('${name}'),`,\n    )\n    .join('\\n');\n\n  return `import { pongoSchema } from '@event-driven-io/pongo';\n\n${types}\n\nexport default {\n  schema: pongoSchema.client({\n    database: pongoSchema.db({\n${collections}\n    }),\n  }),\n};`;\n};\n\nconst missingDefaultExport = `Error: Config should contain default export, e.g.\\n\\n${sampleConfig()}`;\nconst missingSchema = `Error: Config should contain schema property, e.g.\\n\\n${sampleConfig()}`;\nconst missingDbs = `Error: Config should have at least a single database defined, e.g.\\n\\n${sampleConfig()}`;\nconst missingDefaultDb = `Error: Config should have a default database defined (without name or or with default database name), e.g.\\n\\n${sampleConfig()}`;\nconst missingCollections = `Error: Database should have defined at least one collection, e.g.\\n\\n${sampleConfig()}`;\n\nexport const loadConfigFile = async (\n  configPath: string,\n): Promise<PongoDbSchemaMetadata> => {\n  const configUrl = new URL(configPath, `file://${process.cwd()}/`);\n  try {\n    // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment\n    const imported: Partial<{ default: PongoSchemaConfig }> = await import(\n      configUrl.href\n    );\n\n    const parsed = parseDefaultDbSchema(imported);\n\n    if (typeof parsed === 'string') {\n      console.error(parsed);\n      process.exit(1);\n    }\n\n    return parsed;\n  } catch {\n    console.error(`Error: Couldn't load file: ${configUrl.href}`);\n    process.exit(1);\n  }\n};\n\nexport const generateConfigFile = (\n  configPath: string,\n  collectionNames: string[],\n): void => {\n  try {\n    fs.writeFileSync(configPath, sampleConfig(collectionNames), 'utf8');\n    console.log(`Configuration file stored at: ${configPath}`);\n  } catch (error) {\n    console.error(`Error: Couldn't store config file: ${configPath}!`);\n    console.error(error);\n    process.exit(1);\n  }\n};\n\nexport const parseDefaultDbSchema = (\n  imported: Partial<{ default: PongoSchemaConfig }>,\n): PongoDbSchemaMetadata | string => {\n  if (!imported.default) {\n    return missingDefaultExport;\n  }\n\n  if (!imported.default.schema) {\n    return missingSchema;\n  }\n\n  if (!imported.default.schema.dbs) {\n    return missingDbs;\n  }\n\n  const dbs = objectEntries(imported.default.schema.dbs).map((db) => db[1]);\n\n  const defaultDb = dbs.find((db) => db.name === undefined);\n\n  if (!defaultDb) {\n    return missingDefaultDb;\n  }\n\n  if (!defaultDb.collections) {\n    return missingCollections;\n  }\n\n  const collections = objectEntries(defaultDb.collections).map((col) => col[1]);\n\n  if (collections.length === 0) {\n    return missingCollections;\n  }\n\n  return toDbSchemaMetadata(defaultDb);\n};\n\ntype SampleConfigOptions =\n  | {\n      collection: string[];\n      print?: boolean;\n    }\n  | {\n      collection: string[];\n      generate?: boolean;\n      file?: string;\n    };\n\nexport const configCommand = new Command('config').description(\n  'Manage Pongo configuration',\n);\n\nconfigCommand\n  .command('sample')\n  .description('Generate or print sample configuration')\n  .option(\n    '-col, --collection <name>',\n    'Specify the collection name',\n    (value: string, previous: string[]) => {\n      // Accumulate collection names into an array (explicitly typing `previous` as `string[]`)\n      return previous.concat([value]);\n    },\n    [] as string[],\n  )\n  .option(\n    '-f, --file <path>',\n    'Path to configuration file with collection list',\n  )\n  .option('-g, --generate', 'Generate sample config file')\n  .option('-p, --print', 'Print sample config file')\n  .action((options: SampleConfigOptions) => {\n    const collectionNames =\n      options.collection.length > 0 ? options.collection : ['users'];\n\n    if (!('print' in options) && !('generate' in options)) {\n      console.error(\n        'Error: Please provide either:\\n--print param to print sample config or\\n--generate to generate sample config file',\n      );\n      process.exit(1);\n    }\n\n    if ('print' in options) {\n      console.log(`${sampleConfig(collectionNames)}`);\n    } else if ('generate' in options) {\n      if (!options.file) {\n        console.error(\n          'Error: You need to provide a config file through a --file',\n        );\n        process.exit(1);\n      }\n\n      generateConfigFile(options.file, collectionNames);\n    }\n  });\n","import { combineMigrations } from '@event-driven-io/dumbo';\nimport {\n  dumbo,\n  migrationTableSchemaComponent,\n  runPostgreSQLMigrations,\n} from '@event-driven-io/dumbo/pg';\nimport { Command } from 'commander';\nimport { pongoCollectionSchemaComponent } from '../core';\nimport { loadConfigFile } from './configFile';\n\ninterface MigrateRunOptions {\n  collection: string[];\n  connectionString: string;\n  config?: string;\n  dryRun?: boolean;\n}\n\ninterface MigrateSqlOptions {\n  print?: boolean;\n  write?: string;\n  config?: string;\n  collection: string[];\n}\n\nexport const migrateCommand = new Command('migrate').description(\n  'Manage database migrations',\n);\n\nmigrateCommand\n  .command('run')\n  .description('Run database migrations')\n  .option(\n    '-cs, --connection-string <string>',\n    'Connection string for the database',\n  )\n  .option(\n    '-col, --collection <name>',\n    'Specify the collection name',\n    (value: string, previous: string[]) => {\n      // Accumulate collection names into an array (explicitly typing `previous` as `string[]`)\n      return previous.concat([value]);\n    },\n    [] as string[],\n  )\n  .option('-f, --config <path>', 'Path to configuration file with Pongo config')\n  .option('-dr, --dryRun', 'Perform dry run without commiting changes', false)\n  .action(async (options: MigrateRunOptions) => {\n    const { collection, dryRun } = options;\n    const connectionString =\n      options.connectionString ?? process.env.DB_CONNECTION_STRING;\n    let collectionNames: string[];\n\n    if (!connectionString) {\n      console.error(\n        'Error: Connection string is required. Provide it either as a \"--connection-string\" parameter or through the DB_CONNECTION_STRING environment variable.' +\n          '\\nFor instance: --connection-string postgresql://postgres:postgres@localhost:5432/postgres',\n      );\n      process.exit(1);\n    }\n\n    if (options.config) {\n      const config = await loadConfigFile(options.config);\n\n      collectionNames = config.collections.map((c) => c.name);\n    } else if (collection) {\n      collectionNames = collection;\n    } else {\n      console.error(\n        'Error: You need to provide at least one collection name. Provide it either through \"--config\" file or as a \"--collection\" parameter.',\n      );\n      process.exit(1);\n    }\n\n    const pool = dumbo({ connectionString });\n\n    const migrations = collectionNames.flatMap((collectionsName) =>\n      pongoCollectionSchemaComponent(collectionsName).migrations({\n        connector: 'PostgreSQL:pg', // TODO: Provide connector here\n      }),\n    );\n\n    await runPostgreSQLMigrations(pool, migrations, {\n      dryRun,\n    });\n  });\n\nmigrateCommand\n  .command('sql')\n  .description('Generate SQL for database migration')\n  .option(\n    '-col, --collection <name>',\n    'Specify the collection name',\n    (value: string, previous: string[]) => {\n      // Accumulate collection names into an array (explicitly typing `previous` as `string[]`)\n      return previous.concat([value]);\n    },\n    [] as string[],\n  )\n  .option('-f, --config <path>', 'Path to configuration file with Pongo config')\n  .option('--print', 'Print the SQL to the console (default)', true)\n  //.option('--write <filename>', 'Write the SQL to a specified file')\n  .action(async (options: MigrateSqlOptions) => {\n    const { collection } = options;\n\n    let collectionNames: string[];\n\n    if (options.config) {\n      const config = await loadConfigFile(options.config);\n\n      collectionNames = config.collections.map((c) => c.name);\n    } else if (collection) {\n      collectionNames = collection;\n    } else {\n      console.error(\n        'Error: You need to provide at least one collection name. Provide it either through \"--config\" file or as a \"--collection\" parameter.',\n      );\n      process.exit(1);\n    }\n\n    const coreMigrations = migrationTableSchemaComponent.migrations({\n      connector: 'PostgreSQL:pg',\n    });\n    const migrations = [\n      ...coreMigrations,\n      ...collectionNames.flatMap((collectionName) =>\n        pongoCollectionSchemaComponent(collectionName).migrations({\n          connector: 'PostgreSQL:pg', // TODO: Provide connector here\n        }),\n      ),\n    ];\n\n    console.log('Printing SQL:');\n    console.log(combineMigrations(...migrations));\n  });\n","import {\n  color,\n  LogLevel,\n  LogStyle,\n  prettyJson,\n  SQL,\n  type MigrationStyle,\n} from '@event-driven-io/dumbo';\nimport { checkConnection } from '@event-driven-io/dumbo/pg';\nimport Table from 'cli-table3';\nimport { Command } from 'commander';\nimport repl from 'node:repl';\nimport {\n  pongoClient,\n  pongoSchema,\n  type PongoClient,\n  type PongoCollectionSchema,\n  type PongoDb,\n} from '../core';\n\nlet pongo: PongoClient;\n\nconst calculateColumnWidths = (\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  results: any[],\n  columnNames: string[],\n): number[] => {\n  const columnWidths = columnNames.map((col) => {\n    const maxWidth = Math.max(\n      col.length, // Header size\n      ...results.map((result) =>\n        // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n        result[col] ? String(result[col]).length : 0,\n      ),\n    );\n    return maxWidth + 2; // Add padding\n  });\n  return columnWidths;\n};\n\nlet shouldDisplayResultsAsTable = false;\n\nconst printResultsAsTable = (print?: boolean) =>\n  (shouldDisplayResultsAsTable = print === undefined || print === true);\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nconst printOutput = (obj: any): string =>\n  Array.isArray(obj) && shouldDisplayResultsAsTable\n    ? displayResultsAsTable(obj)\n    : prettyJson(obj);\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nconst displayResultsAsTable = (results: any[]): string => {\n  if (results.length === 0) {\n    return color.yellow('No documents found.');\n  }\n\n  const columnNames = results\n\n    .flatMap((result) =>\n      // eslint-disable-next-line @typescript-eslint/no-unsafe-argument\n      typeof result === 'object' ? Object.keys(result) : typeof result,\n    )\n    .filter((value, index, array) => array.indexOf(value) === index);\n\n  const columnWidths = calculateColumnWidths(results, columnNames);\n\n  const table = new Table({\n    head: columnNames.map((col) => color.cyan(col)),\n    colWidths: columnWidths,\n  });\n\n  results.forEach((result) => {\n    table.push(\n      columnNames.map((col) =>\n        // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n        result[col] !== undefined\n          ? // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n            Array.isArray(result[col])\n            ? // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n              displayResultsAsTable(result[col])\n            : // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n              prettyJson(result[col])\n          : typeof result === 'object'\n            ? ''\n            : result != undefined && result != undefined\n              ? prettyJson(result)\n              : '',\n      ),\n    );\n  });\n\n  return table.toString();\n};\n\nconst setLogLevel = (logLevel: string) => {\n  process.env.DUMBO_LOG_LEVEL = logLevel;\n};\n\nconst setLogStyle = (logLevel: string) => {\n  process.env.DUMBO_LOG_STYLE = logLevel;\n};\n\nconst prettifyLogs = (logLevel?: string) => {\n  if (logLevel !== undefined) setLogLevel(logLevel);\n  setLogStyle(LogStyle.PRETTY);\n};\n\nconst startRepl = async (options: {\n  logging: {\n    printOptions: boolean;\n    logLevel: LogLevel;\n    logStyle: LogStyle;\n  };\n  schema: {\n    database: string;\n    collections: string[];\n    autoMigration: MigrationStyle;\n  };\n  connectionString: string | undefined;\n}) => {\n  // TODO: This will change when we have proper tracing and logging config\n  // For now, that's enough\n  setLogLevel(process.env.DUMBO_LOG_LEVEL ?? options.logging.logLevel);\n  setLogStyle(process.env.DUMBO_LOG_STYLE ?? options.logging.logStyle);\n\n  console.log(color.green('Starting Pongo Shell (version: 0.17.0-alpha.6)'));\n\n  if (options.logging.printOptions) {\n    console.log(color.green('With Options:'));\n    console.log(prettyJson(options));\n  }\n\n  const connectionString =\n    options.connectionString ??\n    process.env.DB_CONNECTION_STRING ??\n    'postgresql://postgres:postgres@localhost:5432/postgres';\n\n  if (!(options.connectionString ?? process.env.DB_CONNECTION_STRING)) {\n    console.log(\n      color.yellow(\n        `No connection string provided, using: 'postgresql://postgres:postgres@localhost:5432/postgres'`,\n      ),\n    );\n  }\n\n  const connectionCheck = await checkConnection(connectionString);\n\n  if (!connectionCheck.successful) {\n    if (connectionCheck.errorType === 'ConnectionRefused') {\n      console.error(\n        color.red(\n          `Connection was refused. Check if the PostgreSQL server is running and accessible.`,\n        ),\n      );\n    } else if (connectionCheck.errorType === 'Authentication') {\n      console.error(\n        color.red(\n          `Authentication failed. Check the username and password in the connection string.`,\n        ),\n      );\n    } else {\n      console.error(color.red('Error connecting to PostgreSQL server'));\n    }\n    console.log(color.red('Exiting Pongo Shell...'));\n    process.exit();\n  }\n\n  console.log(color.green(`Successfully connected`));\n  console.log(color.green('Use db.<collection>.<method>() to query.'));\n\n  const shell = repl.start({\n    prompt: color.green('pongo> '),\n    useGlobal: true,\n    breakEvalOnSigint: true,\n    writer: printOutput,\n  });\n\n  let db: PongoDb;\n\n  if (options.schema.collections.length > 0) {\n    const collectionsSchema: Record<string, PongoCollectionSchema> = {};\n\n    for (const collectionName of options.schema.collections) {\n      collectionsSchema[collectionName] =\n        pongoSchema.collection(collectionName);\n    }\n\n    const schema = pongoSchema.client({\n      database: pongoSchema.db(options.schema.database, collectionsSchema),\n    });\n\n    const typedClient = pongoClient(connectionString, {\n      schema: {\n        definition: schema,\n        autoMigration: options.schema.autoMigration,\n      },\n    });\n\n    db = typedClient.database;\n\n    for (const collectionName of options.schema.collections) {\n      shell.context[collectionName] = typedClient.database[collectionName];\n    }\n\n    pongo = typedClient;\n  } else {\n    pongo = pongoClient(connectionString, {\n      schema: { autoMigration: options.schema.autoMigration },\n    });\n\n    db = pongo.db(options.schema.database);\n  }\n\n  shell.context.pongo = pongo;\n  shell.context.db = db;\n\n  // helpers\n  shell.context.SQL = SQL;\n  shell.context.setLogLevel = setLogLevel;\n  shell.context.setLogStyle = setLogStyle;\n  shell.context.prettifyLogs = prettifyLogs;\n  shell.context.printResultsAsTable = printResultsAsTable;\n  shell.context.LogStyle = LogStyle;\n  shell.context.LogLevel = LogLevel;\n\n  // Intercept REPL output to display results as a table if they are arrays\n  shell.on('exit', async () => {\n    await teardown();\n    process.exit();\n  });\n\n  shell.on('SIGINT', async () => {\n    await teardown();\n    process.exit();\n  });\n};\n\nconst teardown = async () => {\n  console.log(color.yellow('Exiting Pongo Shell...'));\n  await pongo.close();\n};\n\nprocess.on('uncaughtException', teardown);\nprocess.on('SIGINT', teardown);\n\ninterface ShellOptions {\n  database: string;\n  collection: string[];\n  connectionString?: string;\n  disableAutoMigrations: boolean;\n  logStyle?: string;\n  logLevel?: string;\n  prettyLog?: boolean;\n  printOptions?: boolean;\n}\n\nconst shellCommand = new Command('shell')\n  .description('Start an interactive Pongo shell')\n  .option(\n    '-cs, --connectionString <string>',\n    'Connection string for the database',\n  )\n  .option('-db, --database <string>', 'Database name to connect', 'postgres')\n  .option(\n    '-col, --collection <name>',\n    'Specify the collection name',\n    (value: string, previous: string[]) => {\n      // Accumulate collection names into an array (explicitly typing `previous` as `string[]`)\n      return previous.concat([value]);\n    },\n    [] as string[],\n  )\n  .option(\n    '-no-migrations, --disable-auto-migrations',\n    'Disable automatic migrations',\n  )\n  .option('-o, --print-options', 'Print shell options')\n  .option(\n    '-ll, --log-level <logLevel>',\n    'Log level: DISABLED, INFO, LOG, WARN, ERROR',\n    'DISABLED',\n  )\n  .option('-ls, --log-style', 'Log style: RAW, PRETTY', 'RAW')\n  .option('-p, --pretty-log', 'Turn on logging with prettified output')\n  .action(async (options: ShellOptions) => {\n    const { collection, database } = options;\n    const connectionString = options.connectionString;\n\n    await startRepl({\n      logging: {\n        printOptions: options.printOptions === true,\n        logStyle: options.prettyLog\n          ? LogStyle.PRETTY\n          : ((options.logStyle as LogStyle | undefined) ?? LogStyle.RAW),\n        logLevel: options.logLevel\n          ? (options.logLevel as LogLevel)\n          : options.prettyLog\n            ? LogLevel.INFO\n            : LogLevel.DISABLED,\n      },\n      schema: {\n        collections: collection,\n        database,\n        autoMigration: options.disableAutoMigrations\n          ? 'None'\n          : 'CreateOrUpdate',\n      },\n      connectionString,\n    });\n  });\n\nexport { shellCommand };\n"]}
+{"version":3,"sources":["/home/runner/work/Pongo/Pongo/src/packages/pongo/dist/cli.cjs","../src/cli.ts","../src/commandLine/configFile.ts","../src/commandLine/migrate.ts","../src/commandLine/shell.ts"],"names":[],"mappings":"AAAA;AACA;AACE;AACA;AACA;AACA;AACA;AACF,wDAA6B;AAC7B;AACA;ACRA,sCAAwB;ADUxB;AACA;AEZA;AACA,gEAAe;AAQf,IAAM,eAAA,EAAiB,CAAC,KAAA,EAAA,GAA0B;AAChD,EAAA,GAAA,CAAI,KAAA,CAAM,OAAA,IAAW,CAAA,EAAG;AACtB,IAAA,OAAO,KAAA;AAAA,EACT;AAEA,EAAA,IAAI,UAAA,EAAY,KAAA,CAAM,MAAA,CAAO,CAAC,CAAA,CAAE,WAAA,CAAY,EAAA,EAAI,KAAA,CAAM,KAAA,CAAM,CAAC,CAAA;AAE7D,EAAA,GAAA,CAAI,SAAA,CAAU,QAAA,CAAS,GAAG,CAAA,EAAG;AAC3B,IAAA,UAAA,EAAY,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,CAAA,CAAE,CAAA;AAAA,EACnC;AAEA,EAAA,OAAO,SAAA;AACT,CAAA;AAEA,IAAM,aAAA,EAAe,CAAC,gBAAA,EAA4B,CAAC,OAAO,CAAA,EAAA,GAAM;AAC9D,EAAA,MAAM,MAAA,EAAQ,eAAA,CACX,GAAA;AAAA,IACC,CAAC,IAAA,EAAA,GACC,CAAA,YAAA,EAAe,cAAA,CAAe,IAAI,CAAC,CAAA,oDAAA;AAAA,EACvC,CAAA,CACC,IAAA,CAAK,IAAI,CAAA;AAEZ,EAAA,MAAM,YAAA,EAAc,eAAA,CACjB,GAAA;AAAA,IACC,CAAC,IAAA,EAAA,GACC,CAAA,MAAA,EAAS,IAAI,CAAA,yBAAA,EAA4B,cAAA,CAAe,IAAI,CAAC,CAAA,GAAA,EAAM,IAAI,CAAA,GAAA;AAAA,EAC3E,CAAA,CACC,IAAA,CAAK,IAAI,CAAA;AAEZ,EAAA,OAAO,CAAA;AAAA;AAAA,EAEP,KAAK,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAKL,WAAW,CAAA;AAAA;AAAA;AAAA,EAAA,CAAA;AAIb,CAAA;AAEA,IAAM,qBAAA,EAAuB,CAAA;AAAA;AAAA,EAAwD,YAAA,CAAa,CAAC,CAAA,CAAA;AAC7F;AAAgB;AAAuE;AAC1E;AAAA;AAAuF;AACpG;AAAmB;AAA+H;AAClJ;AAAqB;AAAsF;AAEpG;AAGL,EAAA;AACF,EAAA;AAEI,IAAA;AAIA,IAAA;AAEK,IAAA;AACD,MAAA;AACA,MAAA;AACV,IAAA;AAEO,IAAA;AACD,EAAA;AACE,IAAA;AACK,IAAA;AACf,EAAA;AACF;AAEa;AAIP,EAAA;AACC,IAAA;AACS,IAAA;AACL,EAAA;AACC,IAAA;AACA,IAAA;AACK,IAAA;AACf,EAAA;AACF;AAEa;AAGG,EAAA;AACL,IAAA;AACT,EAAA;AAEc,EAAA;AACL,IAAA;AACT,EAAA;AAEc,EAAA;AACL,IAAA;AACT,EAAA;AAEY,EAAA;AAEN,EAAA;AAED,EAAA;AACI,IAAA;AACT,EAAA;AAEe,EAAA;AACN,IAAA;AACT,EAAA;AAEM,EAAA;AAEF,EAAA;AACK,IAAA;AACT,EAAA;AAEO,EAAA;AACT;AAaa;AACX,EAAA;AACF;AAGG;AAGC,EAAA;AACA,EAAA;AACgB,EAAA;AAEP,IAAA;AACT,EAAA;AACC,EAAA;AAEF;AACC,EAAA;AACA,EAAA;AAEM;AAGA,EAAA;AAGA,EAAA;AACI,IAAA;AACN,MAAA;AACF,IAAA;AACa,IAAA;AACf,EAAA;AAEe,EAAA;AACD,IAAA;AACH,EAAA;AACI,IAAA;AACH,MAAA;AACN,QAAA;AACF,MAAA;AACQ,MAAA;AACV,IAAA;AAEA,IAAA;AACF,EAAA;AACD;AF9Cc;AACA;AG9IjB;AACE;AACA;AACA;AACA;AAEK;AACE;AA+BI;AACX,EAAA;AACF;AAGG;AAGC,EAAA;AACA,EAAA;AACA,EAAA;AAED;AACC,EAAA;AACA,EAAA;AAED;AACC,EAAA;AACA,EAAA;AACA,EAAA;AAED;AACC,EAAA;AACA,EAAA;AAED;AACC,EAAA;AACA,EAAA;AACgB,EAAA;AAEP,IAAA;AACT,EAAA;AACC,EAAA;AAEK;AAGE,EAAA;AACF,EAAA;AAGA,EAAA;AAIF,EAAA;AAEC,EAAA;AACK,IAAA;AACN,MAAA;AAEF,IAAA;AACa,IAAA;AACf,EAAA;AAEY,EAAA;AACJ,IAAA;AAEN,IAAA;AACS,EAAA;AACT,IAAA;AACK,EAAA;AACG,IAAA;AACN,MAAA;AACF,IAAA;AACa,IAAA;AACf,EAAA;AAEM,EAAA;AAEA,EAAA;AACJ,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACD,EAAA;AAEY,EAAA;AAEP,EAAA;AACJ,IAAA;AACD,EAAA;AACF;AAGA;AAGC,EAAA;AACA,EAAA;AAED;AACC,EAAA;AACA,EAAA;AAED;AACC,EAAA;AACA,EAAA;AACA,EAAA;AAED;AACC,EAAA;AACA,EAAA;AACgB,EAAA;AAEP,IAAA;AACT,EAAA;AACC,EAAA;AAEK;AAIE,EAAA;AAEJ,EAAA;AAEQ,EAAA;AACJ,IAAA;AAEN,IAAA;AACS,EAAA;AACT,IAAA;AACK,EAAA;AACG,IAAA;AACN,MAAA;AACF,IAAA;AACa,IAAA;AACf,EAAA;AAEM,EAAA;AAEA,EAAA;AACJ,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACD,EAAA;AAEW,EAAA;AACA,EAAA;AACb;AAEG;AACJ,EAAA;AACA,EAAA;AACA,EAAA;AACA,EAAA;AAMI;AACW,EAAA;AAEA,EAAA;AACL,IAAA;AACN,MAAA;AACF,IAAA;AACa,IAAA;AACf,EAAA;AAEM,EAAA;AAEA,EAAA;AAIM,IAAA;AACZ,EAAA;AAEM,EAAA;AACJ,IAAA;AACA,IAAA;AACF,EAAA;AAEW,EAAA;AAED,EAAA;AACZ;AHuDiB;AACA;AIlRjB;AACE;AACA;AACA;AACA;AACA;AACA;AAEK;AACE;AACF;AACE;AACQ;AAWb;AAEE;AAKE,EAAA;AACE,IAAA;AACA,MAAA;AACO,MAAA;AAAK,QAAA;AAAA;AAEP,UAAA;AAAoC,QAAA;AAC7C,MAAA;AACF,IAAA;AACO,IAAA;AACR,EAAA;AACM,EAAA;AACT;AAEI;AAEE;AAIA;AAMA;AACQ,EAAA;AACG,IAAA;AACf,EAAA;AAEM,EAAA;AAEM,IAAA;AAAA;AAED,MAAA;AAAmD,IAAA;AAEnD,EAAA;AAEL,EAAA;AAEQ,EAAA;AACN,IAAA;AACK,IAAA;AACZ,EAAA;AAEO,EAAA;AACA,IAAA;AACJ,MAAA;AAAiB,QAAA;AAAA;AAER,UAAA;AAAS;AAEZ,YAAA;AAAyB;AAEvB,cAAA;AAAiC,YAAA;AAAA;AAEjC,cAAA;AAAsB,YAAA;AACxB,UAAA;AAII,QAAA;AACV,MAAA;AACF,IAAA;AACD,EAAA;AAEY,EAAA;AACf;AAEM;AACQ,EAAA;AACd;AAEM;AACQ,EAAA;AACd;AAEM;AACA,EAAA;AACQ,EAAA;AACd;AAEkB;AAgBJ,EAAA;AACA,EAAA;AAEA,EAAA;AAEA,EAAA;AACE,IAAA;AACA,IAAA;AACd,EAAA;AAEM,EAAA;AAKQ,EAAA;AACJ,IAAA;AACA,MAAA;AACJ,QAAA;AACF,MAAA;AACF,IAAA;AACF,EAAA;AAEQ,EAAA;AACF,EAAA;AAES,EAAA;AAEA,EAAA;AACL,IAAA;AACN,MAAA;AACF,IAAA;AACa,IAAA;AACf,EAAA;AAEM,EAAA;AAED,EAAA;AACC,IAAA;AACM,MAAA;AACA,QAAA;AACJ,UAAA;AACF,QAAA;AACF,MAAA;AACS,IAAA;AACD,MAAA;AACA,QAAA;AACJ,UAAA;AACF,QAAA;AACF,MAAA;AACK,IAAA;AACG,MAAA;AACV,IAAA;AACY,IAAA;AACC,IAAA;AACf,EAAA;AAEY,EAAA;AACA,EAAA;AAEE,EAAA;AACJ,IAAA;AACG,IAAA;AACX,IAAA;AACQ,IAAA;AACT,EAAA;AAEG,EAAA;AAEQ,EAAA;AACJ,IAAA;AAEK,IAAA;AACT,MAAA;AAEF,IAAA;AAEM,IAAA;AACM,MAAA;AACX,IAAA;AAEK,IAAA;AACJ,MAAA;AACQ,MAAA;AACN,QAAA;AACA,QAAA;AACF,MAAA;AACF,IAAA;AAGM,IAAA;AACJ,MAAA;AACF,IAAA;AAEM,IAAA;AACD,MAAA;AACA,MAAA;AACJ,IAAA;AAEI,IAAA;AAEM,IAAA;AACH,MAAA;AACR,IAAA;AAEQ,IAAA;AACH,EAAA;AACC,IAAA;AACJ,MAAA;AACQ,MAAA;AACN,QAAA;AACF,MAAA;AACF,IAAA;AAGM,IAAA;AACJ,MAAA;AACF,IAAA;AAEQ,IAAA;AACH,MAAA;AACA,MAAA;AACJ,IAAA;AAEU,IAAA;AACb,EAAA;AAEc,EAAA;AACA,EAAA;AAGA,EAAA;AACA,EAAA;AACA,EAAA;AACA,EAAA;AACA,EAAA;AACA,EAAA;AACA,EAAA;AAGL,EAAA;AACD,IAAA;AACO,IAAA;AACd,EAAA;AAEQ,EAAA;AACD,IAAA;AACO,IAAA;AACd,EAAA;AACH;AAEiB;AACH,EAAA;AACA,EAAA;AACd;AAEW;AACA;AAcL;AAGF,EAAA;AACA,EAAA;AACA,EAAA;AAED;AACC,EAAA;AACA,EAAA;AAEM;AAEN,EAAA;AACA,EAAA;AACgB,EAAA;AAEP,IAAA;AACT,EAAA;AACC,EAAA;AAEF;AACC,EAAA;AACA,EAAA;AAEM;AAEN,EAAA;AACA,EAAA;AACA,EAAA;AAEM;AAGE,EAAA;AACF,EAAA;AAEA,EAAA;AACK,IAAA;AACP,MAAA;AACU,MAAA;AAGA,MAAA;AAKZ,IAAA;AACQ,IAAA;AACN,MAAA;AACA,MAAA;AACA,MAAA;AAGF,IAAA;AACA,IAAA;AACA,IAAA;AACD,EAAA;AACF;AJ2Jc;AACA;AC7fD;AAEH;AAEL;AACA;AACA;AAEM;AAEP;AD2fU;AACA;AACA","file":"/home/runner/work/Pongo/Pongo/src/packages/pongo/dist/cli.cjs","sourcesContent":[null,"#!/usr/bin/env node\nimport { Command } from 'commander';\nimport { configCommand, migrateCommand, shellCommand } from './commandLine';\n\nconst program = new Command();\n\nprogram.name('pongo').description('CLI tool for Pongo');\n\nprogram.addCommand(configCommand);\nprogram.addCommand(migrateCommand);\nprogram.addCommand(shellCommand);\n\nprogram.parse(process.argv);\n\nexport default program;\n","import { Command } from 'commander';\nimport fs from 'node:fs';\nimport {\n  objectEntries,\n  toDbSchemaMetadata,\n  type PongoDbSchemaMetadata,\n  type PongoSchemaConfig,\n} from '../core';\n\nconst formatTypeName = (input: string): string => {\n  if (input.length === 0) {\n    return input;\n  }\n\n  let formatted = input.charAt(0).toUpperCase() + input.slice(1);\n\n  if (formatted.endsWith('s')) {\n    formatted = formatted.slice(0, -1);\n  }\n\n  return formatted;\n};\n\nconst sampleConfig = (collectionNames: string[] = ['users']) => {\n  const types = collectionNames\n    .map(\n      (name) =>\n        `export type ${formatTypeName(name)} = { name: string; description: string; date: Date }`,\n    )\n    .join('\\n');\n\n  const collections = collectionNames\n    .map(\n      (name) =>\n        `      ${name}: pongoSchema.collection<${formatTypeName(name)}>('${name}'),`,\n    )\n    .join('\\n');\n\n  return `import { pongoSchema } from '@event-driven-io/pongo';\n\n${types}\n\nexport default {\n  schema: pongoSchema.client({\n    database: pongoSchema.db({\n${collections}\n    }),\n  }),\n};`;\n};\n\nconst missingDefaultExport = `Error: Config should contain default export, e.g.\\n\\n${sampleConfig()}`;\nconst missingSchema = `Error: Config should contain schema property, e.g.\\n\\n${sampleConfig()}`;\nconst missingDbs = `Error: Config should have at least a single database defined, e.g.\\n\\n${sampleConfig()}`;\nconst missingDefaultDb = `Error: Config should have a default database defined (without name or or with default database name), e.g.\\n\\n${sampleConfig()}`;\nconst missingCollections = `Error: Database should have defined at least one collection, e.g.\\n\\n${sampleConfig()}`;\n\nexport const loadConfigFile = async (\n  configPath: string,\n): Promise<PongoDbSchemaMetadata> => {\n  const configUrl = new URL(configPath, `file://${process.cwd()}/`);\n  try {\n    // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment\n    const imported: Partial<{ default: PongoSchemaConfig }> = await import(\n      configUrl.href\n    );\n\n    const parsed = parseDefaultDbSchema(imported);\n\n    if (typeof parsed === 'string') {\n      console.error(parsed);\n      process.exit(1);\n    }\n\n    return parsed;\n  } catch {\n    console.error(`Error: Couldn't load file: ${configUrl.href}`);\n    process.exit(1);\n  }\n};\n\nexport const generateConfigFile = (\n  configPath: string,\n  collectionNames: string[],\n): void => {\n  try {\n    fs.writeFileSync(configPath, sampleConfig(collectionNames), 'utf8');\n    console.log(`Configuration file stored at: ${configPath}`);\n  } catch (error) {\n    console.error(`Error: Couldn't store config file: ${configPath}!`);\n    console.error(error);\n    process.exit(1);\n  }\n};\n\nexport const parseDefaultDbSchema = (\n  imported: Partial<{ default: PongoSchemaConfig }>,\n): PongoDbSchemaMetadata | string => {\n  if (!imported.default) {\n    return missingDefaultExport;\n  }\n\n  if (!imported.default.schema) {\n    return missingSchema;\n  }\n\n  if (!imported.default.schema.dbs) {\n    return missingDbs;\n  }\n\n  const dbs = objectEntries(imported.default.schema.dbs).map((db) => db[1]);\n\n  const defaultDb = dbs.find((db) => db.name === undefined);\n\n  if (!defaultDb) {\n    return missingDefaultDb;\n  }\n\n  if (!defaultDb.collections) {\n    return missingCollections;\n  }\n\n  const collections = objectEntries(defaultDb.collections).map((col) => col[1]);\n\n  if (collections.length === 0) {\n    return missingCollections;\n  }\n\n  return toDbSchemaMetadata(defaultDb);\n};\n\ntype SampleConfigOptions =\n  | {\n      collection: string[];\n      print?: boolean;\n    }\n  | {\n      collection: string[];\n      generate?: boolean;\n      file?: string;\n    };\n\nexport const configCommand = new Command('config').description(\n  'Manage Pongo configuration',\n);\n\nconfigCommand\n  .command('sample')\n  .description('Generate or print sample configuration')\n  .option(\n    '-col, --collection <name>',\n    'Specify the collection name',\n    (value: string, previous: string[]) => {\n      // Accumulate collection names into an array (explicitly typing `previous` as `string[]`)\n      return previous.concat([value]);\n    },\n    [] as string[],\n  )\n  .option(\n    '-f, --file <path>',\n    'Path to configuration file with collection list',\n  )\n  .option('-g, --generate', 'Generate sample config file')\n  .option('-p, --print', 'Print sample config file')\n  .action((options: SampleConfigOptions) => {\n    const collectionNames =\n      options.collection.length > 0 ? options.collection : ['users'];\n\n    if (!('print' in options) && !('generate' in options)) {\n      console.error(\n        'Error: Please provide either:\\n--print param to print sample config or\\n--generate to generate sample config file',\n      );\n      process.exit(1);\n    }\n\n    if ('print' in options) {\n      console.log(`${sampleConfig(collectionNames)}`);\n    } else if ('generate' in options) {\n      if (!options.file) {\n        console.error(\n          'Error: You need to provide a config file through a --file',\n        );\n        process.exit(1);\n      }\n\n      generateConfigFile(options.file, collectionNames);\n    }\n  });\n","import {\n  combineMigrations,\n  dumbo,\n  parseConnectionString,\n  runSQLMigrations,\n  type DatabaseDriverType,\n} from '@event-driven-io/dumbo';\nimport { Command } from 'commander';\nimport {\n  pongoDatabaseDriverRegistry,\n  pongoSchema,\n  type AnyPongoDatabaseDriverOptions,\n  type PongoCollectionSchema,\n  type PongoDatabaseFactoryOptions,\n  type PongoDocument,\n} from '../core';\nimport { loadConfigFile } from './configFile';\n\ninterface MigrateRunOptions {\n  collection: string[];\n  connectionString: string;\n  databaseType?: string;\n  databaseName?: string | undefined;\n  databaseDriver: string;\n  config?: string;\n  dryRun?: boolean;\n}\n\ninterface MigrateSqlOptions {\n  print?: boolean;\n  write?: string;\n  databaseType: string;\n  databaseName?: string | undefined;\n  databaseDriver: string;\n  config?: string;\n  collection: string[];\n}\n\nexport const migrateCommand = new Command('migrate').description(\n  'Manage database migrations',\n);\n\nmigrateCommand\n  .command('run')\n  .description('Run database migrations')\n  .option(\n    '-dbt, --database-type <string>',\n    'Database type that should be used for connection (e.g., PostgreSQL or SQLite)',\n    undefined,\n  )\n  .option(\n    '-drv, --database-driver <string>',\n    'Database driver that should be used for connection (e.g., \"pg\" for PostgreSQL, \"sqlite3\" for SQLite)',\n  )\n  .option(\n    '-dbn, --database-name <string>',\n    'Database name to connect to',\n    undefined,\n  )\n  .option(\n    '-cs, --connection-string <string>',\n    'Connection string for the database',\n  )\n  .option(\n    '-col, --collection <name>',\n    'Specify the collection name',\n    (value: string, previous: string[]) => {\n      // Accumulate collection names into an array (explicitly typing `previous` as `string[]`)\n      return previous.concat([value]);\n    },\n    [] as string[],\n  )\n  .option('-f, --config <path>', 'Path to configuration file with Pongo config')\n  .option('-dr, --dryRun', 'Perform dry run without commiting changes', false)\n  .action(async (options: MigrateRunOptions) => {\n    const { collection, dryRun, databaseName, databaseDriver } = options;\n    const connectionString =\n      options.connectionString ?? process.env.DB_CONNECTION_STRING;\n\n    const databaseType =\n      options.databaseType ??\n      parseConnectionString(connectionString).databaseType;\n\n    let collectionNames: string[];\n\n    if (!connectionString) {\n      console.error(\n        'Error: Connection string is required. Provide it either as a \"--connection-string\" parameter or through the DB_CONNECTION_STRING environment variable.' +\n          '\\nFor instance: --connection-string postgresql://postgres:postgres@localhost:5432/postgres',\n      );\n      process.exit(1);\n    }\n\n    if (options.config) {\n      const config = await loadConfigFile(options.config);\n\n      collectionNames = config.collections.map((c) => c.name);\n    } else if (collection) {\n      collectionNames = collection;\n    } else {\n      console.error(\n        'Error: You need to provide at least one collection name. Provide it either through \"--config\" file or as a \"--collection\" parameter.',\n      );\n      process.exit(1);\n    }\n\n    const driverType = `${databaseType}:${databaseDriver}` as const;\n\n    const migrations = getMigrations({\n      driverType,\n      connectionString,\n      databaseName,\n      collectionNames,\n    });\n\n    const pool = dumbo({ connectionString, driverType });\n\n    await runSQLMigrations(pool, migrations, {\n      dryRun,\n    });\n  });\n\nmigrateCommand\n  .command('sql')\n  .description('Generate SQL for database migration')\n  .option(\n    '-dbt, --database-type <string>',\n    'Database type that should be used for connection (e.g., PostgreSQL or SQLite)',\n  )\n  .option(\n    '-drv, --database-driver <string>',\n    'Database driver that should be used for connection (e.g., \"pg\" for PostgreSQL, \"sqlite3\" for SQLite)',\n  )\n  .option(\n    '-dbn, --database-name <string>',\n    'Database name to connect to',\n    undefined,\n  )\n  .option(\n    '-col, --collection <name>',\n    'Specify the collection name',\n    (value: string, previous: string[]) => {\n      // Accumulate collection names into an array (explicitly typing `previous` as `string[]`)\n      return previous.concat([value]);\n    },\n    [] as string[],\n  )\n  .option('-f, --config <path>', 'Path to configuration file with Pongo config')\n  .option('--print', 'Print the SQL to the console (default)', true)\n  //.option('--write <filename>', 'Write the SQL to a specified file')\n  .action(async (options: MigrateSqlOptions) => {\n    const { collection, databaseName, databaseType, databaseDriver } = options;\n\n    let collectionNames: string[];\n\n    if (options.config) {\n      const config = await loadConfigFile(options.config);\n\n      collectionNames = config.collections.map((c) => c.name);\n    } else if (collection) {\n      collectionNames = collection;\n    } else {\n      console.error(\n        'Error: You need to provide at least one collection name. Provide it either through \"--config\" file or as a \"--collection\" parameter.',\n      );\n      process.exit(1);\n    }\n\n    const driverType = `${databaseType}:${databaseDriver}` as const;\n\n    const migrations = getMigrations({\n      driverType,\n      connectionString: undefined,\n      databaseName,\n      collectionNames,\n    });\n\n    console.log('Printing SQL:');\n    console.log(combineMigrations(...migrations));\n  });\n\nconst getMigrations = ({\n  driverType,\n  connectionString,\n  databaseName,\n  collectionNames,\n}: {\n  driverType: DatabaseDriverType;\n  connectionString: string | undefined;\n  databaseName: string | undefined;\n  collectionNames: string[];\n}) => {\n  const driver = pongoDatabaseDriverRegistry.tryGet(driverType);\n\n  if (driver === null) {\n    console.error(\n      `Error: No database driver found for driver type \"${driverType}\". Make sure the driver is registered and the connection string is correct.`,\n    );\n    process.exit(1);\n  }\n\n  const dbDefinition = pongoSchema.db.from(databaseName, collectionNames);\n\n  const driverOptions: PongoDatabaseFactoryOptions<\n    Record<string, PongoCollectionSchema<PongoDocument>>,\n    AnyPongoDatabaseDriverOptions\n  > = {\n    schema: { definition: dbDefinition },\n  };\n\n  const customOptions = {\n    connectionString: connectionString ?? driver.defaultConnectionString,\n    databaseName,\n  };\n\n  const db = driver.databaseFactory({ ...driverOptions, ...customOptions });\n\n  return db.schema.component.migrations;\n};\n","import {\n  color,\n  LogLevel,\n  LogStyle,\n  parseConnectionString,\n  prettyJson,\n  SQL,\n  type MigrationStyle,\n} from '@event-driven-io/dumbo';\nimport { checkConnection } from '@event-driven-io/dumbo/pg';\nimport Table from 'cli-table3';\nimport { Command } from 'commander';\nimport repl from 'node:repl';\nimport {\n  pongoClient,\n  pongoDatabaseDriverRegistry,\n  pongoSchema,\n  type PongoClient,\n  type PongoClientOptions,\n  type PongoCollectionSchema,\n  type PongoDb,\n} from '../core';\n\nlet pongo: PongoClient;\n\nconst calculateColumnWidths = (\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  results: any[],\n  columnNames: string[],\n): number[] => {\n  const columnWidths = columnNames.map((col) => {\n    const maxWidth = Math.max(\n      col.length, // Header size\n      ...results.map((result) =>\n        // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n        result[col] ? String(result[col]).length : 0,\n      ),\n    );\n    return maxWidth + 2; // Add padding\n  });\n  return columnWidths;\n};\n\nlet shouldDisplayResultsAsTable = false;\n\nconst printResultsAsTable = (print?: boolean) =>\n  (shouldDisplayResultsAsTable = print === undefined || print === true);\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nconst printOutput = (obj: any): string =>\n  Array.isArray(obj) && shouldDisplayResultsAsTable\n    ? displayResultsAsTable(obj)\n    : prettyJson(obj);\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nconst displayResultsAsTable = (results: any[]): string => {\n  if (results.length === 0) {\n    return color.yellow('No documents found.');\n  }\n\n  const columnNames = results\n\n    .flatMap((result) =>\n      // eslint-disable-next-line @typescript-eslint/no-unsafe-argument\n      typeof result === 'object' ? Object.keys(result) : typeof result,\n    )\n    .filter((value, index, array) => array.indexOf(value) === index);\n\n  const columnWidths = calculateColumnWidths(results, columnNames);\n\n  const table = new Table({\n    head: columnNames.map((col) => color.cyan(col)),\n    colWidths: columnWidths,\n  });\n\n  results.forEach((result) => {\n    table.push(\n      columnNames.map((col) =>\n        // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n        result[col] !== undefined\n          ? // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n            Array.isArray(result[col])\n            ? // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n              displayResultsAsTable(result[col])\n            : // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access\n              prettyJson(result[col])\n          : typeof result === 'object'\n            ? ''\n            : result != undefined && result != undefined\n              ? prettyJson(result)\n              : '',\n      ),\n    );\n  });\n\n  return table.toString();\n};\n\nconst setLogLevel = (logLevel: string) => {\n  process.env.DUMBO_LOG_LEVEL = logLevel;\n};\n\nconst setLogStyle = (logLevel: string) => {\n  process.env.DUMBO_LOG_STYLE = logLevel;\n};\n\nconst prettifyLogs = (logLevel?: string) => {\n  if (logLevel !== undefined) setLogLevel(logLevel);\n  setLogStyle(LogStyle.PRETTY);\n};\n\nconst startRepl = async (options: {\n  logging: {\n    printOptions: boolean;\n    logLevel: LogLevel;\n    logStyle: LogStyle;\n  };\n  schema: {\n    database: string;\n    collections: string[];\n    autoMigration: MigrationStyle;\n  };\n  connectionString: string | undefined;\n  databaseDriver: string;\n}) => {\n  // TODO: This will change when we have proper tracing and logging config\n  // For now, that's enough\n  setLogLevel(process.env.DUMBO_LOG_LEVEL ?? options.logging.logLevel);\n  setLogStyle(process.env.DUMBO_LOG_STYLE ?? options.logging.logStyle);\n\n  console.log(color.green('Starting Pongo Shell (version: 0.17.0-beta.10)'));\n\n  if (options.logging.printOptions) {\n    console.log(color.green('With Options:'));\n    console.log(prettyJson(options));\n  }\n\n  const connectionString =\n    options.connectionString ??\n    process.env.DB_CONNECTION_STRING ??\n    'postgresql://postgres:postgres@localhost:5432/postgres';\n\n  if (!(options.connectionString ?? process.env.DB_CONNECTION_STRING)) {\n    console.log(\n      color.yellow(\n        `No connection string provided, using: 'postgresql://postgres:postgres@localhost:5432/postgres'`,\n      ),\n    );\n  }\n\n  const { databaseType } = parseConnectionString(connectionString);\n  const driverType = `${databaseType}:${options.databaseDriver}` as const;\n\n  const driver = pongoDatabaseDriverRegistry.tryGet(driverType);\n\n  if (driver === null) {\n    console.error(\n      `Error: No database driver found for driver type \"${driverType}\". Make sure the driver is installed and the connection string is correct.`,\n    );\n    process.exit(1);\n  }\n\n  const connectionCheck = await checkConnection(connectionString);\n\n  if (!connectionCheck.successful) {\n    if (connectionCheck.errorType === 'ConnectionRefused') {\n      console.error(\n        color.red(\n          `Connection was refused. Check if the PostgreSQL server is running and accessible.`,\n        ),\n      );\n    } else if (connectionCheck.errorType === 'Authentication') {\n      console.error(\n        color.red(\n          `Authentication failed. Check the username and password in the connection string.`,\n        ),\n      );\n    } else {\n      console.error(color.red('Error connecting to PostgreSQL server'));\n    }\n    console.log(color.red('Exiting Pongo Shell...'));\n    process.exit();\n  }\n\n  console.log(color.green(`Successfully connected`));\n  console.log(color.green('Use db.<collection>.<method>() to query.'));\n\n  const shell = repl.start({\n    prompt: color.green('pongo> '),\n    useGlobal: true,\n    breakEvalOnSigint: true,\n    writer: printOutput,\n  });\n\n  let db: PongoDb;\n\n  if (options.schema.collections.length > 0) {\n    const collectionsSchema: Record<string, PongoCollectionSchema> = {};\n\n    for (const collectionName of options.schema.collections) {\n      collectionsSchema[collectionName] =\n        pongoSchema.collection(collectionName);\n    }\n\n    const schema = pongoSchema.client({\n      database: pongoSchema.db(options.schema.database, collectionsSchema),\n    });\n\n    const driverOptions: PongoClientOptions = {\n      driver,\n      schema: {\n        definition: schema,\n        autoMigration: options.schema.autoMigration,\n      },\n    };\n\n    // TODO: Find a better way to pass custom driver settings\n    const customOptions = {\n      connectionString,\n    };\n\n    const typedClient = pongoClient({\n      ...driverOptions,\n      ...customOptions,\n    });\n\n    db = typedClient.database!;\n\n    for (const collectionName of options.schema.collections) {\n      shell.context[collectionName] = typedClient.database![collectionName];\n    }\n\n    pongo = typedClient;\n  } else {\n    const driverOptions: PongoClientOptions = {\n      driver,\n      schema: {\n        autoMigration: options.schema.autoMigration,\n      },\n    };\n\n    // TODO: Find a better way to pass custom driver settings\n    const customOptions = {\n      connectionString,\n    };\n\n    pongo = pongoClient({\n      ...driverOptions,\n      ...customOptions,\n    });\n\n    db = pongo.db(options.schema.database);\n  }\n\n  shell.context.pongo = pongo;\n  shell.context.db = db;\n\n  // helpers\n  shell.context.SQL = SQL;\n  shell.context.setLogLevel = setLogLevel;\n  shell.context.setLogStyle = setLogStyle;\n  shell.context.prettifyLogs = prettifyLogs;\n  shell.context.printResultsAsTable = printResultsAsTable;\n  shell.context.LogStyle = LogStyle;\n  shell.context.LogLevel = LogLevel;\n\n  // Intercept REPL output to display results as a table if they are arrays\n  shell.on('exit', async () => {\n    await teardown();\n    process.exit();\n  });\n\n  shell.on('SIGINT', async () => {\n    await teardown();\n    process.exit();\n  });\n};\n\nconst teardown = async () => {\n  console.log(color.yellow('Exiting Pongo Shell...'));\n  await pongo.close();\n};\n\nprocess.on('uncaughtException', teardown);\nprocess.on('SIGINT', teardown);\n\ninterface ShellOptions {\n  database: string;\n  collection: string[];\n  databaseDriver: string;\n  connectionString?: string;\n  disableAutoMigrations: boolean;\n  logStyle?: string;\n  logLevel?: string;\n  prettyLog?: boolean;\n  printOptions?: boolean;\n}\n\nconst shellCommand = new Command('shell')\n  .description('Start an interactive Pongo shell')\n  .option(\n    '-drv, --database-driver <string>',\n    'Database driver that should be used for connection (e.g., \"pg\" for PostgreSQL, \"sqlite3\" for SQLite)',\n    'pg',\n  )\n  .option(\n    '-cs, --connectionString <string>',\n    'Connection string for the database',\n  )\n  .option('-db, --database <string>', 'Database name to connect', 'postgres')\n  .option(\n    '-col, --collection <name>',\n    'Specify the collection name',\n    (value: string, previous: string[]) => {\n      // Accumulate collection names into an array (explicitly typing `previous` as `string[]`)\n      return previous.concat([value]);\n    },\n    [] as string[],\n  )\n  .option(\n    '-no-migrations, --disable-auto-migrations',\n    'Disable automatic migrations',\n  )\n  .option('-o, --print-options', 'Print shell options')\n  .option(\n    '-ll, --log-level <logLevel>',\n    'Log level: DISABLED, INFO, LOG, WARN, ERROR',\n    'DISABLED',\n  )\n  .option('-ls, --log-style', 'Log style: RAW, PRETTY', 'RAW')\n  .option('-p, --pretty-log', 'Turn on logging with prettified output')\n  .action(async (options: ShellOptions) => {\n    const { collection, database } = options;\n    const connectionString = options.connectionString;\n\n    await startRepl({\n      logging: {\n        printOptions: options.printOptions === true,\n        logStyle: options.prettyLog\n          ? LogStyle.PRETTY\n          : ((options.logStyle as LogStyle | undefined) ?? LogStyle.RAW),\n        logLevel: options.logLevel\n          ? (options.logLevel as LogLevel)\n          : options.prettyLog\n            ? LogLevel.INFO\n            : LogLevel.DISABLED,\n      },\n      schema: {\n        collections: collection,\n        database,\n        autoMigration: options.disableAutoMigrations\n          ? 'None'\n          : 'CreateOrUpdate',\n      },\n      connectionString,\n      databaseDriver: options.databaseDriver,\n    });\n  });\n\nexport { shellCommand };\n"]}