@event-driven-io/pongo 0.17.0-beta.4 → 0.17.0-beta.40

package/dist/cli.js

@@ -1,525 +1,317 @@
 #!/usr/bin/env node
-import {
-  objectEntries,
-  pongoClient,
-  pongoDatabaseDriverRegistry,
-  pongoSchema,
-  toDbSchemaMetadata
-} from "./chunk-NCNRRYVE.js";
-
-// src/cli.ts
-import { Command as Command4 } from "commander";
-
-// src/commandLine/configFile.ts
+import { _ as toDbSchemaMetadata, j as objectEntries, o as pongoClient, p as pongoSchema, u as pongoDriverRegistry } from "./core-BHdOCUrr.js";
+import { JSONSerializer, LogLevel, LogStyle, SQL, color, combineMigrations, dumbo, parseConnectionString, prettyJson, runSQLMigrations } from "@event-driven-io/dumbo";
 import { Command } from "commander";
-import fs from "fs";
-var formatTypeName = (input) => {
-  if (input.length === 0) {
-    return input;
-  }
-  let formatted = input.charAt(0).toUpperCase() + input.slice(1);
-  if (formatted.endsWith("s")) {
-    formatted = formatted.slice(0, -1);
-  }
-  return formatted;
+import fs from "node:fs";
+import { checkConnection } from "@event-driven-io/dumbo/pg";
+import Table from "cli-table3";
+import repl from "node:repl";
+
+//#region src/commandLine/configFile.ts
+const formatTypeName = (input) => {
+	if (input.length === 0) return input;
+	let formatted = input.charAt(0).toUpperCase() + input.slice(1);
+	if (formatted.endsWith("s")) formatted = formatted.slice(0, -1);
+	return formatted;
 };
-var sampleConfig = (collectionNames = ["users"]) => {
-  const types = collectionNames.map(
-    (name) => `export type ${formatTypeName(name)} = { name: string; description: string; date: Date }`
-  ).join("\n");
-  const collections = collectionNames.map(
-    (name) => `      ${name}: pongoSchema.collection<${formatTypeName(name)}>('${name}'),`
-  ).join("\n");
-  return `import { pongoSchema } from '@event-driven-io/pongo';
+const sampleConfig = (collectionNames = ["users"]) => {
+	return `import { pongoSchema } from '@event-driven-io/pongo';
 
-${types}
+${collectionNames.map((name) => `export type ${formatTypeName(name)} = { name: string; description: string; date: Date }`).join("\n")}
 
 export default {
   schema: pongoSchema.client({
     database: pongoSchema.db({
-${collections}
+${collectionNames.map((name) => `      ${name}: pongoSchema.collection<${formatTypeName(name)}>('${name}'),`).join("\n")}
     }),
   }),
 };`;
 };
-var missingDefaultExport = `Error: Config should contain default export, e.g.
-
-${sampleConfig()}`;
-var missingSchema = `Error: Config should contain schema property, e.g.
-
-${sampleConfig()}`;
-var missingDbs = `Error: Config should have at least a single database defined, e.g.
-
-${sampleConfig()}`;
-var missingDefaultDb = `Error: Config should have a default database defined (without name or or with default database name), e.g.
-
-${sampleConfig()}`;
-var missingCollections = `Error: Database should have defined at least one collection, e.g.
-
-${sampleConfig()}`;
-var loadConfigFile = async (configPath) => {
-  const configUrl = new URL(configPath, `file://${process.cwd()}/`);
-  try {
-    const imported = await import(configUrl.href);
-    const parsed = parseDefaultDbSchema(imported);
-    if (typeof parsed === "string") {
-      console.error(parsed);
-      process.exit(1);
-    }
-    return parsed;
-  } catch {
-    console.error(`Error: Couldn't load file: ${configUrl.href}`);
-    process.exit(1);
-  }
+const missingDefaultExport = `Error: Config should contain default export, e.g.\n\n${sampleConfig()}`;
+const missingSchema = `Error: Config should contain schema property, e.g.\n\n${sampleConfig()}`;
+const missingDbs = `Error: Config should have at least a single database defined, e.g.\n\n${sampleConfig()}`;
+const missingDefaultDb = `Error: Config should have a default database defined (without name or or with default database name), e.g.\n\n${sampleConfig()}`;
+const missingCollections = `Error: Database should have defined at least one collection, e.g.\n\n${sampleConfig()}`;
+const loadConfigFile = async (configPath) => {
+	const configUrl = new URL(configPath, `file://${process.cwd()}/`);
+	try {
+		const parsed = parseDefaultDbSchema(await import(configUrl.href));
+		if (typeof parsed === "string") {
+			console.error(parsed);
+			process.exit(1);
+		}
+		return parsed;
+	} catch {
+		console.error(`Error: Couldn't load file: ${configUrl.href}`);
+		process.exit(1);
+	}
 };
-var generateConfigFile = (configPath, collectionNames) => {
-  try {
-    fs.writeFileSync(configPath, sampleConfig(collectionNames), "utf8");
-    console.log(`Configuration file stored at: ${configPath}`);
-  } catch (error) {
-    console.error(`Error: Couldn't store config file: ${configPath}!`);
-    console.error(error);
-    process.exit(1);
-  }
+const generateConfigFile = (configPath, collectionNames) => {
+	try {
+		fs.writeFileSync(configPath, sampleConfig(collectionNames), "utf8");
+		console.log(`Configuration file stored at: ${configPath}`);
+	} catch (error) {
+		console.error(`Error: Couldn't store config file: ${configPath}!`);
+		console.error(error);
+		process.exit(1);
+	}
 };
-var parseDefaultDbSchema = (imported) => {
-  if (!imported.default) {
-    return missingDefaultExport;
-  }
-  if (!imported.default.schema) {
-    return missingSchema;
-  }
-  if (!imported.default.schema.dbs) {
-    return missingDbs;
-  }
-  const dbs = objectEntries(imported.default.schema.dbs).map((db) => db[1]);
-  const defaultDb = dbs.find((db) => db.name === void 0);
-  if (!defaultDb) {
-    return missingDefaultDb;
-  }
-  if (!defaultDb.collections) {
-    return missingCollections;
-  }
-  const collections = objectEntries(defaultDb.collections).map((col) => col[1]);
-  if (collections.length === 0) {
-    return missingCollections;
-  }
-  return toDbSchemaMetadata(defaultDb);
+const parseDefaultDbSchema = (imported) => {
+	if (!imported.default) return missingDefaultExport;
+	if (!imported.default.schema) return missingSchema;
+	if (!imported.default.schema.dbs) return missingDbs;
+	const defaultDb = objectEntries(imported.default.schema.dbs).map((db) => db[1]).find((db) => db.name === void 0);
+	if (!defaultDb) return missingDefaultDb;
+	if (!defaultDb.collections) return missingCollections;
+	if (objectEntries(defaultDb.collections).map((col) => col[1]).length === 0) return missingCollections;
+	return toDbSchemaMetadata(defaultDb);
 };
-var configCommand = new Command("config").description(
-  "Manage Pongo configuration"
-);
-configCommand.command("sample").description("Generate or print sample configuration").option(
-  "-col, --collection <name>",
-  "Specify the collection name",
-  (value, previous) => {
-    return previous.concat([value]);
-  },
-  []
-).option(
-  "-f, --file <path>",
-  "Path to configuration file with collection list"
-).option("-g, --generate", "Generate sample config file").option("-p, --print", "Print sample config file").action((options) => {
-  const collectionNames = options.collection.length > 0 ? options.collection : ["users"];
-  if (!("print" in options) && !("generate" in options)) {
-    console.error(
-      "Error: Please provide either:\n--print param to print sample config or\n--generate to generate sample config file"
-    );
-    process.exit(1);
-  }
-  if ("print" in options) {
-    console.log(`${sampleConfig(collectionNames)}`);
-  } else if ("generate" in options) {
-    if (!options.file) {
-      console.error(
-        "Error: You need to provide a config file through a --file"
-      );
-      process.exit(1);
-    }
-    generateConfigFile(options.file, collectionNames);
-  }
+const configCommand = new Command("config").description("Manage Pongo configuration");
+configCommand.command("sample").description("Generate or print sample configuration").option("--col, --collection <name>", "Specify the collection name", (value, previous) => {
+	return previous.concat([value]);
+}, []).option("-f, --file <path>", "Path to configuration file with collection list").option("-g, --generate", "Generate sample config file").option("-p, --print", "Print sample config file").action((options) => {
+	const collectionNames = options.collection.length > 0 ? options.collection : ["users"];
+	if (!("print" in options) && !("generate" in options)) {
+		console.error("Error: Please provide either:\n--print param to print sample config or\n--generate to generate sample config file");
+		process.exit(1);
+	}
+	if ("print" in options) console.log(`${sampleConfig(collectionNames)}`);
+	else if ("generate" in options) {
+		if (!options.file) {
+			console.error("Error: You need to provide a config file through a --file");
+			process.exit(1);
+		}
+		generateConfigFile(options.file, collectionNames);
+	}
 });
 
-// src/commandLine/migrate.ts
-import {
-  combineMigrations,
-  dumbo,
-  parseConnectionString,
-  runSQLMigrations
-} from "@event-driven-io/dumbo";
-import { Command as Command2 } from "commander";
-var migrateCommand = new Command2("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(
-  "-col, --collection <name>",
-  "Specify the collection name",
-  (value, previous) => {
-    return previous.concat([value]);
-  },
-  []
-).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, databaseName, databaseDriver } = options;
-  const connectionString = options.connectionString ?? process.env.DB_CONNECTION_STRING;
-  const databaseType = options.databaseType ?? parseConnectionString(connectionString).databaseType;
-  let collectionNames;
-  if (!connectionString) {
-    console.error(
-      'Error: Connection string is required. Provide it either as a "--connection-string" parameter or through the DB_CONNECTION_STRING environment variable.\nFor instance: --connection-string postgresql://postgres:postgres@localhost:5432/postgres'
-    );
-    process.exit(1);
-  }
-  if (options.config) {
-    const config = await loadConfigFile(options.config);
-    collectionNames = config.collections.map((c) => c.name);
-  } else if (collection) {
-    collectionNames = collection;
-  } else {
-    console.error(
-      'Error: You need to provide at least one collection name. Provide it either through "--config" file or as a "--collection" parameter.'
-    );
-    process.exit(1);
-  }
-  const driverType = `${databaseType}:${databaseDriver}`;
-  const migrations = getMigrations({
-    driverType,
-    connectionString,
-    databaseName,
-    collectionNames
-  });
-  const pool = dumbo({ connectionString, driverType });
-  await runSQLMigrations(pool, migrations, {
-    dryRun
-  });
+//#endregion
+//#region src/commandLine/migrate.ts
+const migrateCommand = new 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("--col, --collection <name>", "Specify the collection name", (value, previous) => {
+	return previous.concat([value]);
+}, []).option("-f, --config <path>", "Path to configuration file with Pongo config").option("--dr, --dryRun", "Perform dry run without commiting changes", false).option("--t, --timeout <ms>", "Set the migration timeout in milliseconds", parseInt).action(async (options) => {
+	const { collection, dryRun, databaseName, databaseDriver, timeoutMs } = options;
+	const connectionString = options.connectionString ?? process.env.DB_CONNECTION_STRING;
+	const databaseType = options.databaseType ?? parseConnectionString(connectionString).databaseType;
+	let collectionNames;
+	if (!connectionString) {
+		console.error("Error: Connection string is required. Provide it either as a \"--connection-string\" parameter or through the DB_CONNECTION_STRING environment variable.\nFor instance: --connection-string postgresql://postgres:postgres@localhost:5432/postgres");
+		process.exit(1);
+	}
+	if (options.config) collectionNames = (await loadConfigFile(options.config)).collections.map((c) => c.name);
+	else if (collection) collectionNames = collection;
+	else {
+		console.error("Error: You need to provide at least one collection name. Provide it either through \"--config\" file or as a \"--collection\" parameter.");
+		process.exit(1);
+	}
+	const driverType = `${databaseType}:${databaseDriver}`;
+	const migrations = getMigrations({
+		driverType,
+		connectionString,
+		databaseName,
+		collectionNames
+	});
+	await runSQLMigrations(dumbo({
+		connectionString,
+		driverType
+	}), migrations, {
+		dryRun,
+		migrationTimeoutMs: timeoutMs
+	});
 });
-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) => {
-    return previous.concat([value]);
-  },
-  []
-).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, databaseName, databaseType, databaseDriver } = options;
-  let collectionNames;
-  if (options.config) {
-    const config = await loadConfigFile(options.config);
-    collectionNames = config.collections.map((c) => c.name);
-  } else if (collection) {
-    collectionNames = collection;
-  } else {
-    console.error(
-      'Error: You need to provide at least one collection name. Provide it either through "--config" file or as a "--collection" parameter.'
-    );
-    process.exit(1);
-  }
-  const driverType = `${databaseType}:${databaseDriver}`;
-  const migrations = getMigrations({
-    driverType,
-    connectionString: void 0,
-    databaseName,
-    collectionNames
-  });
-  console.log("Printing SQL:");
-  console.log(combineMigrations(...migrations));
+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) => {
+	return previous.concat([value]);
+}, []).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, databaseName, databaseType, databaseDriver } = options;
+	let collectionNames;
+	if (options.config) collectionNames = (await loadConfigFile(options.config)).collections.map((c) => c.name);
+	else if (collection) collectionNames = collection;
+	else {
+		console.error("Error: You need to provide at least one collection name. Provide it either through \"--config\" file or as a \"--collection\" parameter.");
+		process.exit(1);
+	}
+	const migrations = getMigrations({
+		driverType: `${databaseType}:${databaseDriver}`,
+		connectionString: void 0,
+		databaseName,
+		collectionNames
+	});
+	console.log("Printing SQL:");
+	console.log(combineMigrations(...migrations));
 });
-var getMigrations = ({
-  driverType,
-  connectionString,
-  databaseName,
-  collectionNames
-}) => {
-  const driver = 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 = pongoSchema.db.from(databaseName, collectionNames);
-  const driverOptions = {
-    schema: { definition: dbDefinition }
-  };
-  const customOptions = {
-    connectionString: connectionString ?? driver.defaultConnectionString,
-    databaseName
-  };
-  const db = driver.databaseFactory({ ...driverOptions, ...customOptions });
-  return db.schema.component.migrations;
+const getMigrations = ({ driverType, connectionString, databaseName, collectionNames }) => {
+	const driver = pongoDriverRegistry.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 driverOptions = {
+		schema: { definition: pongoSchema.db.from(databaseName, collectionNames) },
+		serializer: JSONSerializer
+	};
+	const customOptions = {
+		connectionString,
+		databaseName
+	};
+	return driver.databaseFactory({
+		...driverOptions,
+		...customOptions
+	}).schema.component.migrations;
 };
 
-// src/commandLine/shell.ts
-import {
-  color,
-  LogLevel,
-  LogStyle,
-  parseConnectionString as parseConnectionString2,
-  prettyJson,
-  SQL
-} from "@event-driven-io/dumbo";
-import { checkConnection } from "@event-driven-io/dumbo/pg";
-import Table from "cli-table3";
-import { Command as Command3 } from "commander";
-import repl from "repl";
-var pongo;
-var calculateColumnWidths = (results, columnNames) => {
-  const columnWidths = columnNames.map((col) => {
-    const maxWidth = Math.max(
-      col.length,
-      ...results.map(
-        (result) => (
-          // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-          result[col] ? String(result[col]).length : 0
-        )
-      )
-    );
-    return maxWidth + 2;
-  });
-  return columnWidths;
+//#endregion
+//#region src/commandLine/shell.ts
+let pongo;
+const calculateColumnWidths = (results, columnNames) => {
+	return columnNames.map((col) => {
+		return Math.max(col.length, ...results.map((result) => result[col] ? String(result[col]).length : 0)) + 2;
+	});
 };
-var shouldDisplayResultsAsTable = false;
-var printResultsAsTable = (print) => shouldDisplayResultsAsTable = print === void 0 || print === true;
-var printOutput = (obj) => Array.isArray(obj) && shouldDisplayResultsAsTable ? displayResultsAsTable(obj) : prettyJson(obj);
-var displayResultsAsTable = (results) => {
-  if (results.length === 0) {
-    return color.yellow("No documents found.");
-  }
-  const columnNames = results.flatMap(
-    (result) => (
-      // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
-      typeof result === "object" ? Object.keys(result) : typeof result
-    )
-  ).filter((value, index, array) => array.indexOf(value) === index);
-  const columnWidths = calculateColumnWidths(results, columnNames);
-  const table = new Table({
-    head: columnNames.map((col) => color.cyan(col)),
-    colWidths: columnWidths
-  });
-  results.forEach((result) => {
-    table.push(
-      columnNames.map(
-        (col) => (
-          // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-          result[col] !== void 0 ? (
-            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-            Array.isArray(result[col]) ? (
-              // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-              displayResultsAsTable(result[col])
-            ) : (
-              // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-              prettyJson(result[col])
-            )
-          ) : typeof result === "object" ? "" : result != void 0 && result != void 0 ? prettyJson(result) : ""
-        )
-      )
-    );
-  });
-  return table.toString();
+let shouldDisplayResultsAsTable = false;
+const printResultsAsTable = (print) => shouldDisplayResultsAsTable = print === void 0 || print === true;
+const printOutput = (obj) => Array.isArray(obj) && shouldDisplayResultsAsTable ? displayResultsAsTable(obj) : prettyJson(obj);
+const displayResultsAsTable = (results) => {
+	if (results.length === 0) return color.yellow("No documents found.");
+	const columnNames = results.flatMap((result) => typeof result === "object" ? Object.keys(result) : typeof result).filter((value, index, array) => array.indexOf(value) === index);
+	const columnWidths = calculateColumnWidths(results, columnNames);
+	const table = new Table({
+		head: columnNames.map((col) => color.cyan(col)),
+		colWidths: columnWidths
+	});
+	results.forEach((result) => {
+		table.push(columnNames.map((col) => result[col] !== void 0 ? Array.isArray(result[col]) ? displayResultsAsTable(result[col]) : prettyJson(result[col]) : typeof result === "object" ? "" : result != void 0 && result != void 0 ? prettyJson(result) : ""));
+	});
+	return table.toString();
 };
-var setLogLevel = (logLevel) => {
-  process.env.DUMBO_LOG_LEVEL = logLevel;
+const setLogLevel = (logLevel) => {
+	process.env.DUMBO_LOG_LEVEL = logLevel;
 };
-var setLogStyle = (logLevel) => {
-  process.env.DUMBO_LOG_STYLE = logLevel;
+const setLogStyle = (logLevel) => {
+	process.env.DUMBO_LOG_STYLE = logLevel;
 };
-var prettifyLogs = (logLevel) => {
-  if (logLevel !== void 0) setLogLevel(logLevel);
-  setLogStyle(LogStyle.PRETTY);
+const prettifyLogs = (logLevel) => {
+	if (logLevel !== void 0) setLogLevel(logLevel);
+	setLogStyle(LogStyle.PRETTY);
 };
-var startRepl = async (options) => {
-  setLogLevel(process.env.DUMBO_LOG_LEVEL ?? options.logging.logLevel);
-  setLogStyle(process.env.DUMBO_LOG_STYLE ?? options.logging.logStyle);
-  console.log(color.green("Starting Pongo Shell (version: 0.17.0-beta.4)"));
-  if (options.logging.printOptions) {
-    console.log(color.green("With Options:"));
-    console.log(prettyJson(options));
-  }
-  const connectionString = options.connectionString ?? process.env.DB_CONNECTION_STRING ?? "postgresql://postgres:postgres@localhost:5432/postgres";
-  if (!(options.connectionString ?? process.env.DB_CONNECTION_STRING)) {
-    console.log(
-      color.yellow(
-        `No connection string provided, using: 'postgresql://postgres:postgres@localhost:5432/postgres'`
-      )
-    );
-  }
-  const { databaseType } = parseConnectionString2(connectionString);
-  const driverType = `${databaseType}:${options.databaseDriver}`;
-  const driver = 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 checkConnection(connectionString);
-  if (!connectionCheck.successful) {
-    if (connectionCheck.errorType === "ConnectionRefused") {
-      console.error(
-        color.red(
-          `Connection was refused. Check if the PostgreSQL server is running and accessible.`
-        )
-      );
-    } else if (connectionCheck.errorType === "Authentication") {
-      console.error(
-        color.red(
-          `Authentication failed. Check the username and password in the connection string.`
-        )
-      );
-    } else {
-      console.error(color.red("Error connecting to PostgreSQL server"));
-    }
-    console.log(color.red("Exiting Pongo Shell..."));
-    process.exit();
-  }
-  console.log(color.green(`Successfully connected`));
-  console.log(color.green("Use db.<collection>.<method>() to query."));
-  const shell = repl.start({
-    prompt: color.green("pongo> "),
-    useGlobal: true,
-    breakEvalOnSigint: true,
-    writer: printOutput
-  });
-  let db;
-  if (options.schema.collections.length > 0) {
-    const collectionsSchema = {};
-    for (const collectionName of options.schema.collections) {
-      collectionsSchema[collectionName] = pongoSchema.collection(collectionName);
-    }
-    const schema = pongoSchema.client({
-      database: pongoSchema.db(options.schema.database, collectionsSchema)
-    });
-    const driverOptions = {
-      driver,
-      schema: {
-        definition: schema,
-        autoMigration: options.schema.autoMigration
-      }
-    };
-    const customOptions = {
-      connectionString
-    };
-    const typedClient = pongoClient({
-      ...driverOptions,
-      ...customOptions
-    });
-    db = typedClient.database;
-    for (const collectionName of options.schema.collections) {
-      shell.context[collectionName] = typedClient.database[collectionName];
-    }
-    pongo = typedClient;
-  } else {
-    const driverOptions = {
-      driver,
-      schema: {
-        autoMigration: options.schema.autoMigration
-      }
-    };
-    const customOptions = {
-      connectionString
-    };
-    pongo = pongoClient({
-      ...driverOptions,
-      ...customOptions
-    });
-    db = pongo.db(options.schema.database);
-  }
-  shell.context.pongo = pongo;
-  shell.context.db = db;
-  shell.context.SQL = SQL;
-  shell.context.setLogLevel = setLogLevel;
-  shell.context.setLogStyle = setLogStyle;
-  shell.context.prettifyLogs = prettifyLogs;
-  shell.context.printResultsAsTable = printResultsAsTable;
-  shell.context.LogStyle = LogStyle;
-  shell.context.LogLevel = LogLevel;
-  shell.on("exit", async () => {
-    await teardown();
-    process.exit();
-  });
-  shell.on("SIGINT", async () => {
-    await teardown();
-    process.exit();
-  });
+const startRepl = async (options) => {
+	setLogLevel(process.env.DUMBO_LOG_LEVEL ?? options.logging.logLevel);
+	setLogStyle(process.env.DUMBO_LOG_STYLE ?? options.logging.logStyle);
+	console.log(color.green("Starting Pongo Shell (version: 0.17.0-beta.40)"));
+	if (options.logging.printOptions) {
+		console.log(color.green("With Options:"));
+		console.log(prettyJson(options));
+	}
+	const connectionString = options.connectionString ?? process.env.DB_CONNECTION_STRING ?? "postgresql://postgres:postgres@localhost:5432/postgres";
+	if (!(options.connectionString ?? process.env.DB_CONNECTION_STRING)) console.log(color.yellow(`No connection string provided, using: 'postgresql://postgres:postgres@localhost:5432/postgres'`));
+	const { databaseType } = parseConnectionString(connectionString);
+	const driverType = `${databaseType}:${options.databaseDriver}`;
+	const driver = pongoDriverRegistry.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 checkConnection(connectionString);
+	if (!connectionCheck.successful) {
+		if (connectionCheck.errorType === "ConnectionRefused") console.error(color.red(`Connection was refused. Check if the PostgreSQL server is running and accessible.`));
+		else if (connectionCheck.errorType === "Authentication") console.error(color.red(`Authentication failed. Check the username and password in the connection string.`));
+		else console.error(color.red("Error connecting to PostgreSQL server"));
+		console.log(color.red("Exiting Pongo Shell..."));
+		process.exit();
+	}
+	console.log(color.green(`Successfully connected`));
+	console.log(color.green("Use db.<collection>.<method>() to query."));
+	const shell = repl.start({
+		prompt: color.green("pongo> "),
+		useGlobal: true,
+		breakEvalOnSigint: true,
+		writer: printOutput
+	});
+	let db;
+	if (options.schema.collections.length > 0) {
+		const collectionsSchema = {};
+		for (const collectionName of options.schema.collections) collectionsSchema[collectionName] = pongoSchema.collection(collectionName);
+		const driverOptions = {
+			driver,
+			schema: {
+				definition: pongoSchema.client({ database: pongoSchema.db(options.schema.database, collectionsSchema) }),
+				autoMigration: options.schema.autoMigration
+			}
+		};
+		const customOptions = { connectionString };
+		const typedClient = pongoClient({
+			...driverOptions,
+			...customOptions
+		});
+		db = typedClient.database;
+		for (const collectionName of options.schema.collections) shell.context[collectionName] = typedClient.database[collectionName];
+		pongo = typedClient;
+	} else {
+		const driverOptions = {
+			driver,
+			schema: { autoMigration: options.schema.autoMigration }
+		};
+		const customOptions = { connectionString };
+		pongo = pongoClient({
+			...driverOptions,
+			...customOptions
+		});
+		db = pongo.db(options.schema.database);
+	}
+	shell.context.pongo = pongo;
+	shell.context.db = db;
+	shell.context.SQL = SQL;
+	shell.context.setLogLevel = setLogLevel;
+	shell.context.setLogStyle = setLogStyle;
+	shell.context.prettifyLogs = prettifyLogs;
+	shell.context.printResultsAsTable = printResultsAsTable;
+	shell.context.LogStyle = LogStyle;
+	shell.context.LogLevel = LogLevel;
+	shell.on("exit", async () => {
+		await teardown();
+		process.exit();
+	});
+	shell.on("SIGINT", async () => {
+		await teardown();
+		process.exit();
+	});
 };
-var teardown = async () => {
-  console.log(color.yellow("Exiting Pongo Shell..."));
-  await pongo.close();
+const teardown = async () => {
+	console.log(color.yellow("Exiting Pongo Shell..."));
+	await pongo.close();
 };
 process.on("uncaughtException", teardown);
 process.on("SIGINT", teardown);
-var shellCommand = new Command3("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(
-  "-col, --collection <name>",
-  "Specify the collection name",
-  (value, previous) => {
-    return previous.concat([value]);
-  },
-  []
-).option(
-  "-no-migrations, --disable-auto-migrations",
-  "Disable automatic migrations"
-).option("-o, --print-options", "Print shell options").option(
-  "-ll, --log-level <logLevel>",
-  "Log level: DISABLED, INFO, LOG, WARN, ERROR",
-  "DISABLED"
-).option("-ls, --log-style", "Log style: RAW, PRETTY", "RAW").option("-p, --pretty-log", "Turn on logging with prettified output").action(async (options) => {
-  const { collection, database } = options;
-  const connectionString = options.connectionString;
-  await startRepl({
-    logging: {
-      printOptions: options.printOptions === true,
-      logStyle: options.prettyLog ? LogStyle.PRETTY : options.logStyle ?? LogStyle.RAW,
-      logLevel: options.logLevel ? options.logLevel : options.prettyLog ? LogLevel.INFO : LogLevel.DISABLED
-    },
-    schema: {
-      collections: collection,
-      database,
-      autoMigration: options.disableAutoMigrations ? "None" : "CreateOrUpdate"
-    },
-    connectionString,
-    databaseDriver: options.databaseDriver
-  });
+const shellCommand = new 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("--col, --collection <name>", "Specify the collection name", (value, previous) => {
+	return previous.concat([value]);
+}, []).option("--no-migrations, --disable-auto-migrations", "Disable automatic migrations").option("-o, --print-options", "Print shell options").option("--ll, --log-level <logLevel>", "Log level: DISABLED, INFO, LOG, WARN, ERROR", "DISABLED").option("--ls, --log-style", "Log style: RAW, PRETTY", "RAW").option("-p, --pretty-log", "Turn on logging with prettified output").action(async (options) => {
+	const { collection, database } = options;
+	const connectionString = options.connectionString;
+	await startRepl({
+		logging: {
+			printOptions: options.printOptions === true,
+			logStyle: options.prettyLog ? LogStyle.PRETTY : options.logStyle ?? LogStyle.RAW,
+			logLevel: options.logLevel ? options.logLevel : options.prettyLog ? LogLevel.INFO : LogLevel.DISABLED
+		},
+		schema: {
+			collections: collection,
+			database,
+			autoMigration: options.disableAutoMigrations ? "None" : "CreateOrUpdate"
+		},
+		connectionString,
+		databaseDriver: options.databaseDriver
+	});
 });
 
-// src/cli.ts
-var program = new Command4();
+//#endregion
+//#region src/cli.ts
+const program = new Command();
 program.name("pongo").description("CLI tool for Pongo");
 program.addCommand(configCommand);
 program.addCommand(migrateCommand);
 program.addCommand(shellCommand);
 program.parse(process.argv);
-var cli_default = program;
-export {
-  cli_default as default
-};
+
+//#endregion
+export { program as default };
 //# sourceMappingURL=cli.js.map