@elysiumjs/mnemosyne-drizzle 0.1.4 → 0.2.0

package/dist/table.d.ts

@@ -1,10 +1,10 @@
-import type { PgColumnBuilderBase, PgTableWithColumns } from 'drizzle-orm/pg-core';
+import type { AnyPgColumnBuilder, PgTableWithColumns } from 'drizzle-orm/pg-core';
 /**
  * Retrieves the original column builders for a table.
  * @param table The table created with enhanced pgTable.
  * @returns The original column builder definitions, or `undefined`.
  */
-export declare const getTableBuilders: (table: PgTableWithColumns<any>) => Record<string, PgColumnBuilderBase> | undefined;
+export declare const getTableBuilders: (table: PgTableWithColumns<any>) => Record<string, AnyPgColumnBuilder> | undefined;
 /**
  * Creates a PostgreSQL table with automatic column builder registration.
  *
@@ -17,129 +17,53 @@ export declare const getTableBuilders: (table: PgTableWithColumns<any>) => Recor
  * @param extraConfig Optional extra Drizzle table configuration.
  * @returns A Drizzle PostgreSQL table with registered column builders.
  */
-export declare const pgTable: <TTableName extends string, TColumns extends Record<string, PgColumnBuilderBase>>(tableName: TTableName, columns: TColumns, extraConfig?: (table: any) => any) => PgTableWithColumns<{
+export declare const pgTable: <TTableName extends string, TColumns extends Record<string, AnyPgColumnBuilder>>(tableName: TTableName, columns: TColumns, extraConfig?: (table: any) => any) => PgTableWithColumns<{
     name: TTableName;
     schema: undefined;
-    columns: { [Key in keyof TColumns]: import("drizzle-orm/pg-core").PgColumn<{
-        name: (Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        })["name"];
+    columns: { [Key in keyof TColumns]: import("drizzle-orm/pg-core").PgBuildColumn<TTableName, TColumns[Key], {
+        name: string;
         tableName: TTableName;
-        dataType: (Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        })["dataType"];
-        columnType: (Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        })["columnType"];
-        data: Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } extends infer T ? T extends Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } ? T extends {
+        dataType: TColumns[Key]["_"]["dataType"];
+        data: TColumns[Key]["_"]["dimensions"] extends 1 | 2 | 3 | 4 | 5 ? TColumns[Key]["_"]["dimensions"] extends infer T ? T extends TColumns[Key]["_"]["dimensions"] ? T extends 1 ? (TColumns[Key]["_"] extends infer T_1 ? T_1 extends TColumns[Key]["_"] ? T_1 extends {
             $type: infer U;
-        } ? U : T["data"] : never : never;
-        driverParam: (Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        })["driverParam"];
-        notNull: Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } extends infer T_1 ? T_1 extends Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } ? T_1 extends {
-            notNull: true;
-        } ? true : false : never : never;
-        hasDefault: Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } extends infer T_2 ? T_2 extends Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } ? T_2 extends {
-            hasDefault: true;
-        } ? true : false : never : never;
-        isPrimaryKey: Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } extends infer T_3 ? T_3 extends Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } ? T_3 extends {
-            isPrimaryKey: true;
-        } ? true : false : never : never;
-        isAutoincrement: Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } extends infer T_4 ? T_4 extends Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } ? T_4 extends {
-            isAutoincrement: true;
-        } ? true : false : never : never;
-        hasRuntimeDefault: Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } extends infer T_5 ? T_5 extends Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } ? T_5 extends {
-            hasRuntimeDefault: true;
-        } ? true : false : never : never;
-        enumValues: (Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        })["enumValues"];
-        baseColumn: Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } extends infer T_6 ? T_6 extends Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } ? T_6 extends {
-            baseBuilder: infer U_1 extends import("drizzle-orm").ColumnBuilderBase;
-        } ? import("drizzle-orm").Column<{
-            name: U_1["_"]["name"];
-            tableName: TTableName;
-            dataType: U_1["_"]["dataType"];
-            columnType: U_1["_"]["columnType"];
-            data: U_1["_"] extends infer T_7 ? T_7 extends U_1["_"] ? T_7 extends {
-                $type: infer U;
-            } ? U : T_7["data"] : never : never;
-            driverParam: U_1["_"]["driverParam"];
-            notNull: U_1["_"] extends infer T_8 ? T_8 extends U_1["_"] ? T_8 extends {
-                notNull: true;
-            } ? true : false : never : never;
-            hasDefault: U_1["_"] extends infer T_9 ? T_9 extends U_1["_"] ? T_9 extends {
-                hasDefault: true;
-            } ? true : false : never : never;
-            isPrimaryKey: U_1["_"] extends infer T_10 ? T_10 extends U_1["_"] ? T_10 extends {
-                isPrimaryKey: true;
-            } ? true : false : never : never;
-            isAutoincrement: U_1["_"] extends infer T_11 ? T_11 extends U_1["_"] ? T_11 extends {
-                isAutoincrement: true;
-            } ? true : false : never : never;
-            hasRuntimeDefault: U_1["_"] extends infer T_12 ? T_12 extends U_1["_"] ? T_12 extends {
-                hasRuntimeDefault: true;
-            } ? true : false : never : never;
-            enumValues: U_1["_"]["enumValues"];
-            baseColumn: U_1["_"] extends infer T_13 ? T_13 extends U_1["_"] ? T_13 extends {
-                baseBuilder: infer U_1 extends import("drizzle-orm").ColumnBuilderBase;
-            } ? import("drizzle-orm").Column</*elided*/ any, {}, Omit<U_1["_"], "name" | "data" | "driverParam" | "tableName" | "notNull" | "hasDefault" | "isPrimaryKey" | "isAutoincrement" | "hasRuntimeDefault" | "dataType" | "columnType" | "enumValues" | "identity" | "dialect" | "baseColumn" | "generated" | "brand"> extends infer T_14 ? { [K in keyof T_14]: T_14[K]; } : never> : never : never : never;
-            identity: U_1["_"] extends infer T_15 ? T_15 extends U_1["_"] ? T_15 extends {
-                identity: "always";
-            } ? "always" : T_15 extends {
-                identity: "byDefault";
-            } ? "byDefault" : undefined : never : never;
-            generated: U_1["_"] extends infer T_16 ? T_16 extends U_1["_"] ? T_16 extends {
-                generated: infer G;
-            } ? unknown extends G ? undefined : G extends undefined ? undefined : G : undefined : never : never;
-        }, {}, Omit<U_1["_"], "name" | "data" | "driverParam" | "tableName" | "notNull" | "hasDefault" | "isPrimaryKey" | "isAutoincrement" | "hasRuntimeDefault" | "dataType" | "columnType" | "enumValues" | "identity" | "dialect" | "baseColumn" | "generated" | "brand"> extends infer T_17 ? { [K in keyof T_17]: T_17[K]; } : never> : never : never : never;
-        identity: Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } extends infer T_18 ? T_18 extends Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } ? T_18 extends {
-            identity: "always";
-        } ? "always" : T_18 extends {
-            identity: "byDefault";
-        } ? "byDefault" : undefined : never : never;
-        generated: Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } extends infer T_19 ? T_19 extends Omit<TColumns[Key]["_"], "name"> & {
-            name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-        } ? T_19 extends {
-            generated: infer G;
-        } ? unknown extends G ? undefined : G extends undefined ? undefined : G : undefined : never : never;
-    }, {}, Omit<Omit<TColumns[Key]["_"], "name"> & {
-        name: TColumns[Key]["_"]["name"] extends "" ? import("drizzle-orm").Assume<Key, string> : TColumns[Key]["_"]["name"];
-    }, "name" | "data" | "driverParam" | "tableName" | "notNull" | "hasDefault" | "isPrimaryKey" | "isAutoincrement" | "hasRuntimeDefault" | "dataType" | "columnType" | "enumValues" | "identity" | "dialect" | "baseColumn" | "generated" | "brand"> extends infer T_20 ? { [K_1 in keyof T_20]: T_20[K_1]; } : never>; };
-    dialect: "pg";
+        } ? U : T_1 extends {
+            data: infer D;
+        } ? D : unknown : never : never)[] : T extends 2 ? (TColumns[Key]["_"] extends infer T_1 ? T_1 extends TColumns[Key]["_"] ? T_1 extends {
+            $type: infer U;
+        } ? U : T_1 extends {
+            data: infer D;
+        } ? D : unknown : never : never)[][] : T extends 3 ? (TColumns[Key]["_"] extends infer T_1 ? T_1 extends TColumns[Key]["_"] ? T_1 extends {
+            $type: infer U;
+        } ? U : T_1 extends {
+            data: infer D;
+        } ? D : unknown : never : never)[][][] : T extends 4 ? (TColumns[Key]["_"] extends infer T_1 ? T_1 extends TColumns[Key]["_"] ? T_1 extends {
+            $type: infer U;
+        } ? U : T_1 extends {
+            data: infer D;
+        } ? D : unknown : never : never)[][][][] : T extends 5 ? (TColumns[Key]["_"] extends infer T_1 ? T_1 extends TColumns[Key]["_"] ? T_1 extends {
+            $type: infer U;
+        } ? U : T_1 extends {
+            data: infer D;
+        } ? D : unknown : never : never)[][][][][] : TColumns[Key]["_"] extends infer T_1 ? T_1 extends TColumns[Key]["_"] ? T_1 extends {
+            $type: infer U;
+        } ? U : T_1 extends {
+            data: infer D;
+        } ? D : unknown : never : never : never : never : TColumns[Key]["_"] extends infer T_1 ? T_1 extends TColumns[Key]["_"] ? T_1 extends {
+            $type: infer U;
+        } ? U : T_1 extends {
+            data: infer D;
+        } ? D : unknown : never : never;
+        driverParam: TColumns[Key]["_"]["dimensions"] extends 1 | 2 | 3 | 4 | 5 ? string | (TColumns[Key]["_"]["dimensions"] extends infer T_2 ? T_2 extends TColumns[Key]["_"]["dimensions"] ? T_2 extends 1 ? TColumns[Key]["_"]["driverParam"][] : T_2 extends 2 ? TColumns[Key]["_"]["driverParam"][][] : T_2 extends 3 ? TColumns[Key]["_"]["driverParam"][][][] : T_2 extends 4 ? TColumns[Key]["_"]["driverParam"][][][][] : T_2 extends 5 ? TColumns[Key]["_"]["driverParam"][][][][][] : TColumns[Key]["_"]["driverParam"] : never : never) : TColumns[Key]["_"]["driverParam"];
+        notNull: TColumns[Key]["_"]["notNull"] extends true ? true : false;
+        hasDefault: TColumns[Key]["_"]["hasDefault"] extends true ? true : false;
+        isPrimaryKey: false;
+        isAutoincrement: false;
+        hasRuntimeDefault: false;
+        enumValues: TColumns[Key]["_"] extends infer T_3 ? T_3 extends TColumns[Key]["_"] ? T_3 extends {
+            enumValues: infer E extends string[];
+        } ? E : undefined : never : never;
+        identity: TColumns[Key]["_"]["identity"] extends "always" | "byDefault" ? TColumns[Key]["_"]["identity"] : undefined;
+        generated: TColumns[Key]["_"]["generated"] extends true ? true : undefined;
+    }>; };
+    dialect: 'pg';
 }>;

package/dist/table.js

@@ -0,0 +1 @@
+import{pgTable as drizzlePgTable}from"drizzle-orm/pg-core";const tableRegistry=new WeakMap;export const getTableBuilders=(table)=>{return tableRegistry.get(table)};export const pgTable=(tableName,columns,extraConfig)=>{const table=drizzlePgTable(tableName,columns,extraConfig);tableRegistry.set(table,columns);return table};

package/dist/tenancy.d.ts

@@ -1,5 +1,5 @@
 import type { TenancyStrategy } from '@elysiumjs/mnemosyne';
-import type { PgColumnBuilderBase, PgSchema, PgTable, PgTransaction } from 'drizzle-orm/pg-core';
+import type { AnyPgColumnBuilder, PgAsyncTransaction, PgSchema, PgTable } from 'drizzle-orm/pg-core';
 import type { DrizzleConnection } from './database';
 /**
  * Returns the Drizzle `PgSchema` for the given tenant, creating it if needed.
@@ -27,7 +27,7 @@ export declare const registerTenantSchema: (tenant: string) => PgSchema;
  * @param columns The column definitions.
  * @returns A tenant-scoped `PgTable`.
  */
-export declare const wrapTenantSchema: <TTableName extends string, TColumnsMap extends Record<string, PgColumnBuilderBase>>(tenant: string, tableName: TTableName, columns: TColumnsMap) => PgTable;
+export declare const wrapTenantSchema: <TTableName extends string, TColumnsMap extends Record<string, AnyPgColumnBuilder>>(tenant: string, tableName: TTableName, columns: TColumnsMap) => PgTable;
 /**
  * Executes a callback within an RLS-scoped transaction for the given tenant.
  * @author Axel Nana <axel.nana@workbud.com>
@@ -36,7 +36,7 @@ export declare const wrapTenantSchema: <TTableName extends string, TColumnsMap e
  * @param connectionName The database connection name.
  * @returns The return value of the callback.
  */
-export declare const withRLS: <T>(tenant: string | null, callback: (tx: PgTransaction<any, any, any>) => Promise<T>, connectionName?: string) => Promise<T>;
+export declare const withRLS: <T>(tenant: string | null, callback: (tx: PgAsyncTransaction<any, any>) => Promise<T>, connectionName?: string) => Promise<T>;
 /**
  * Sets the current tenant on a persistent database connection (session-level).
  * @author Axel Nana <axel.nana@workbud.com>

package/dist/tenancy.js

@@ -0,0 +1 @@
+import{getTenancyConfig}from"@elysiumjs/mnemosyne";import{sql}from"drizzle-orm";import{getTableConfig,pgPolicy,pgSchema,pgTable}from"drizzle-orm/pg-core";import{Database}from"./database";import{getTableBuilders}from"./table";const tenantSchemas=new Map;const schemaRegistry=new Map;export const getTenantSchema=(tenant)=>{return schemaRegistry.get(tenant)??registerTenantSchema(tenant)};export const registerTenantSchema=(tenant)=>{if(schemaRegistry.has(tenant)){return schemaRegistry.get(tenant)}const tenantSchema=pgSchema(tenant);schemaRegistry.set(tenant,tenantSchema);return tenantSchema};export const wrapTenantSchema=(tenant,tableName,columns)=>{if(tenant==="public"){return pgTable(tableName,columns)}const fullTableName=`${tenant}.${tableName}`;if(tenantSchemas.has(fullTableName)){return tenantSchemas.get(fullTableName)}const tenantSchema=getTenantSchema(tenant);const schemaTable=tenantSchema.table(tableName,columns);tenantSchemas.set(fullTableName,schemaTable);return schemaTable};export const withRLS=async(tenant,callback,connectionName="default")=>{const db=Database.getConnection(connectionName);const config=getTenancyConfig();const varName=config.rls?.sessionVariable??"app.current_tenant";const value=tenant??"public";return db.transaction(async(tx)=>{await tx.execute(sql`SELECT set_config(${varName}, ${value}, true)`);return callback(tx)})};export const setConnectionTenant=async(tenant,connectionName="default")=>{const db=Database.getConnection(connectionName);const config=getTenancyConfig();const varName=config.rls?.sessionVariable??"app.current_tenant";const value=tenant??"public";await db.execute(sql`SELECT set_config(${varName}, ${value}, false)`)};export const getSessionTenant=async(connectionName="default")=>{const db=Database.getConnection(connectionName);const config=getTenancyConfig();const varName=config.rls?.sessionVariable??"app.current_tenant";const result=await db.execute(sql`SELECT current_setting(${varName}, true) as tenant`);return result?.rows?.[0]?.tenant??null};export const createRLSPolicy=(tableName,columnName,policyName)=>{const config=getTenancyConfig();const varName=config.rls?.sessionVariable??"app.current_tenant";const column=columnName??config.rls?.defaultColumn??"tenant_id";const name=policyName??`${tableName}_tenant_isolation`;return pgPolicy(name,{for:"all",to:"public",using:sql`${sql.identifier(column)} = current_setting(${varName}, true)::text`,withCheck:sql`${sql.identifier(column)} = current_setting(${varName}, true)::text`})};export class DrizzleSchemaTenancy{mode="schema";resolveTable(table,tenant){const builders=getTableBuilders(table);if(!builders){throw new Error(`No column builders registered for table. `+`Use pgTable() from '@elysiumjs/mnemosyne-drizzle' for tenant support.`)}const config=getTableConfig(table);return wrapTenantSchema(tenant,config.name,builders)}async withIsolation(_connection,_tenant,callback){return callback()}}export class DrizzleRLSTenancy{mode="rls";resolveTable(table,_tenant){return table}async withIsolation(connection,tenant,callback){const config=getTenancyConfig();const varName=config.rls?.sessionVariable??"app.current_tenant";return connection.transaction(async(tx)=>{await tx.execute(sql`SELECT set_config(${varName}, ${tenant}, true)`);return callback()})}}

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@elysiumjs/mnemosyne-drizzle",
-	"version": "0.1.4",
+	"version": "0.2.0",
 	"license": "Apache-2.0",
 	"description": "Drizzle ORM driver for @elysiumjs/mnemosyne",
 	"private": false,
@@ -13,9 +13,8 @@
 	"scripts": {
 		"format": "prettier --write \"./src/**/*.ts\"",
 		"clean": "rimraf dist",
-		"build:js": "bun build ./src/index.ts --outdir ./dist --target bun --packages external --root ./src --minify",
-		"build:ts": "tsc",
-		"build": "bun run clean && bun run build:js && bun run build:ts",
+		"build": "bun run clean && tsc && bun run build:minify",
+		"build:minify": "bun scripts/minify.ts",
 		"typecheck": "tsc --noEmit",
 		"prepublishOnly": "bun run build"
 	},
@@ -31,14 +30,17 @@
 	"types": "dist/index.d.ts",
 	"sideEffects": false,
 	"peerDependencies": {
-		"@elysiumjs/core": "^0.7.0",
-		"@elysiumjs/mnemosyne": "^0.1.1"
+		"@elysiumjs/core": "^0.8.0",
+		"@elysiumjs/mnemosyne": "^0.2.0"
 	},
 	"dependencies": {
-		"drizzle-orm": "^0.45.1"
+		"drizzle-orm": "^1.0.0-rc.5",
+		"prompts": "^2.4.2",
+		"radash": "^12.1.1"
 	},
 	"devDependencies": {
-		"@elysiumjs/core": "0.7.0",
-		"@elysiumjs/mnemosyne": "0.1.1"
+		"@types/prompts": "^2.4.9",
+		"@elysiumjs/core": "0.8.0",
+		"@elysiumjs/mnemosyne": "0.2.0"
 	}
 }

package/scripts/minify.ts

@@ -0,0 +1,10 @@
+import { Glob } from 'bun';
+
+const glob = new Glob('dist/**/*.js');
+
+for await (const path of glob.scan('.')) {
+	const file = Bun.file(path);
+	const code = await file.text();
+	const result = new Bun.Transpiler({ minifyWhitespace: true, trimUnusedImports: true }).transformSync(code);
+	await Bun.write(file, result);
+}

package/src/commands/migration_generate.command.ts

@@ -0,0 +1,101 @@
+// Copyright (c) 2025-present Workbud Technologies Inc. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import { existsSync } from 'node:fs';
+import { exists } from 'node:fs/promises';
+import { dirname, join } from 'node:path';
+
+import { Command, CommandArgumentType } from '@elysiumjs/core';
+import prompts from 'prompts';
+
+/**
+ * Gets the root path of the Elysium.js project: the closest parent holding an `.elysiumrc`.
+ * @author Axel Nana <axel.nana@workbud.com>
+ */
+const getProjectPath = () => {
+	const startPath = process.cwd();
+	let dir = startPath;
+
+	while (true) {
+		if (existsSync(join(dir, '.elysiumrc'))) return dir;
+
+		const parentDir = dirname(dir);
+		if (dir === parentDir) break;
+
+		dir = parentDir;
+	}
+
+	return startPath;
+};
+
+/**
+ * Generates migration files.
+ * @author Axel Nana <axel.nana@workbud.com>
+ */
+@Command.register()
+export class MigrationGenerateCommand extends Command {
+	public static readonly command: string = 'migration:generate';
+	public static readonly description: string = 'Generates migration files.';
+
+	@Command.arg({
+		description: 'The name of the migration to create',
+		type: CommandArgumentType.STRING
+	})
+	private name?: string;
+
+	public async run(): Promise<void> {
+		if (!this.name) {
+			const answers = await prompts(
+				[
+					{
+						type: 'text',
+						name: 'name',
+						message: 'Migration Name:',
+						initial: 'create_users_table',
+						validate(value: string) {
+							if (value.length < 1) {
+								return 'Migration name cannot be empty';
+							}
+
+							return true;
+						}
+					}
+				],
+				{
+					onCancel: () => {
+						this.error('Operation cancelled.');
+						process.exit(0);
+					}
+				}
+			);
+
+			this.name = answers.name;
+		}
+
+		const projectRoot = getProjectPath();
+		let schemaPath = `${projectRoot}/src/database/schemas`;
+		if (await exists(`${projectRoot}/src/database/schema.ts`)) {
+			schemaPath = `${projectRoot}/src/database/schema.ts`;
+		}
+
+		await Bun.$`bunx --bun drizzle-kit generate --dialect postgresql --schema ${schemaPath} --out ${projectRoot}/src/database/migrations --name ${this.name}`;
+
+		const sqlFiles = new Bun.Glob(`${projectRoot}/src/database/migrations/**/*.sql`).scan();
+		for await (const path of sqlFiles) {
+			const file = Bun.file(path);
+			const content = await file.text();
+			await file.write(content.replace(/"public"\./gm, ''));
+		}
+	}
+}

package/src/commands/migration_run.command.ts

@@ -0,0 +1,160 @@
+// Copyright (c) 2025-present Workbud Technologies Inc. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import { Command, CommandArgumentType } from '@elysiumjs/core';
+import prompts from 'prompts';
+
+import { Database } from '../database';
+import { getMigrationName, planMigrations } from '../migrations';
+
+/**
+ * Execute migration files.
+ * @author Axel Nana <axel.nana@workbud.com>
+ */
+@Command.register()
+export class MigrationRunCommand extends Command {
+	public static readonly command: string = 'migration:run';
+	public static readonly description: string = 'Runs pending migrations.';
+
+	@Command.arg({
+		description: 'The name of the tenant in which the migrations will be executed.',
+		type: CommandArgumentType.STRING,
+		required: true
+	})
+	private tenant: string = '';
+
+	@Command.arg({
+		description: 'The name of the connection to use.',
+		type: CommandArgumentType.STRING,
+		default: 'default'
+	})
+	private connection: string = 'default';
+
+	@Command.arg({
+		description: 'Drop the entire schema before running the migrations.',
+		type: CommandArgumentType.BOOLEAN,
+		default: false
+	})
+	private fresh: boolean = false;
+
+	public async run(): Promise<void> {
+		if (!Database.connectionExists(this.connection)) {
+			this.error(`Connection ${this.bold(this.connection)} does not exist.`);
+			return process.exit(1);
+		}
+
+		const sql = Database.getConnection(this.connection).$client;
+		await sql.connect();
+
+		await sql.begin(async (sql: Bun.SQL) => {
+			if (this.fresh) {
+				await sql`DROP SCHEMA IF EXISTS ${sql(this.tenant)} CASCADE;`;
+				await sql`CREATE SCHEMA ${sql(this.tenant)};`;
+			}
+
+			const [{ schema_exists }] = await sql`SELECT EXISTS (
+			SELECT schema_name
+			  FROM information_schema.schemata
+			 WHERE schema_name = ${this.tenant}
+		) AS schema_exists;`;
+
+			if (!schema_exists) {
+				const answers = await prompts(
+					[
+						{
+							type: 'confirm',
+							name: 'create',
+							message: `Schema ${this.tenant} does not exist. Do you want to create it?`,
+							initial: true
+						}
+					],
+					{
+						onCancel: () => {
+							this.error('Operation cancelled.');
+							process.exit(0);
+						}
+					}
+				);
+
+				if (answers.create) {
+					await sql`CREATE SCHEMA ${sql(this.tenant)};`;
+				} else {
+					this.error(`Schema ${this.bold(this.tenant)} creation cancelled.`);
+					return process.exit(1);
+				}
+			}
+
+			await sql`SET search_path TO ${sql(this.tenant)};`;
+
+			const [{ table_exists }] = await sql`SELECT EXISTS (
+			SELECT *
+			  FROM information_schema.tables
+			 WHERE table_name = 'elysium_migrations'
+			   AND table_schema = ${this.tenant}
+		) AS table_exists;`;
+
+			if (!table_exists) {
+				await sql`CREATE TABLE elysium_migrations (
+				id SERIAL PRIMARY KEY,
+				name TEXT NOT NULL,
+				batch INTEGER NOT NULL,
+				applied_at TIMESTAMP NOT NULL
+			);`;
+			}
+
+			const rows: { name: string; applied_at: Date }[] =
+				await sql`SELECT name, applied_at FROM elysium_migrations;`;
+			const completedMigrations = rows.map((row) => ({
+				name: row.name,
+				appliedAt: row.applied_at
+			}));
+
+			const filesIt = new Bun.Glob(`./src/database/migrations/**/*.sql`).scan({
+				absolute: true,
+				onlyFiles: true
+			});
+
+			const files = await Array.fromAsync(filesIt);
+			const { pending: sqlFiles, renames } = planMigrations(files, completedMigrations);
+
+			for (const { from, to } of renames) {
+				await sql`UPDATE elysium_migrations SET name = ${to} WHERE name = ${from};`;
+				this.info(`Migration ${this.bold(from)} is now tracked as ${this.bold(to)}.`);
+			}
+
+			if (sqlFiles.length === 0) {
+				this.success('No new migrations to apply.');
+				return;
+			}
+
+			const s = this.spinner(`Running migrations for tenant ${this.bold(this.tenant)}...`);
+
+			const [{ max }]: [{ max?: number }] =
+				await sql`SELECT MAX(batch) + 1 AS max FROM elysium_migrations;`;
+
+			const batch = max ?? 0;
+
+			for (const path of sqlFiles) {
+				await sql.file(path);
+				const [{ name }] =
+					await sql`INSERT INTO elysium_migrations (name, batch, applied_at) VALUES (${getMigrationName(path)}, ${batch}, ${new Date()}) RETURNING name;`;
+				s.pause(`Migration ${this.bold(name)} applied successfully.`);
+			}
+
+			s.complete(`Migrations for tenant ${this.bold(this.tenant)} applied successfully.`);
+		});
+
+		await sql.close();
+	}
+}

package/src/database.ts

@@ -14,9 +14,10 @@
 
 import type { CacheInterface, DatabaseCacheConfig, DatabaseDriver } from '@elysiumjs/mnemosyne';
 import type { DrizzleConfig } from 'drizzle-orm';
-import type { BunSQLDatabase } from 'drizzle-orm/bun-sql';
+import type { BunSQLDatabase } from 'drizzle-orm/bun-sql/postgres';
 import type { CacheConfig } from 'drizzle-orm/cache/core/types';
 
+import { Service, ServiceNotReadyError } from '@elysiumjs/core';
 import { AbstractDatabase, createDatabaseCacheStorage } from '@elysiumjs/mnemosyne';
 import { getTableName, is, Table } from 'drizzle-orm';
 import { drizzle } from 'drizzle-orm/bun-sql';
@@ -61,7 +62,8 @@ export type DrizzleConnection = BunSQLDatabase<Record<string, never>> & { $clien
  * @author Axel Nana <axel.nana@workbud.com>
  */
 export class DrizzleDatabaseCache extends DrizzleCache {
-	#cache: Omit<CacheInterface, 'tags'>;
+	#storageEngine: 'redis' | 'memory';
+	#cache?: Omit<CacheInterface, 'tags'>;
 	private usedTablesPerKey: Record<string, string[]> = {};
 
 	constructor(
@@ -70,7 +72,23 @@ export class DrizzleDatabaseCache extends DrizzleCache {
 		storageEngine: 'redis' | 'memory' = 'redis'
 	) {
 		super();
-		this.#cache = createDatabaseCacheStorage({ storage: storageEngine });
+		this.#storageEngine = storageEngine;
+	}
+
+	/**
+	 * Resolves the cache storage on first use rather than at construction, so that a connection
+	 * can be built before the module providing the cache has finished initialising.
+	 */
+	async #storage(): Promise<Omit<CacheInterface, 'tags'>> {
+		if (this.#cache) return this.#cache;
+		try {
+			this.#cache = createDatabaseCacheStorage({ storage: this.#storageEngine });
+		} catch (error) {
+			if (!(error instanceof ServiceNotReadyError)) throw error;
+			await Service.resolve(error.service);
+			this.#cache = createDatabaseCacheStorage({ storage: this.#storageEngine });
+		}
+		return this.#cache;
 	}
 
 	public override strategy(): 'explicit' | 'all' {
@@ -78,7 +96,7 @@ export class DrizzleDatabaseCache extends DrizzleCache {
 	}
 
 	public override async get(key: string): Promise<any[] | undefined> {
-		return (await this.#cache.get<any[]>(key)) ?? undefined;
+		return (await (await this.#storage()).get<any[]>(key)) ?? undefined;
 	}
 
 	public override async put(
@@ -88,7 +106,7 @@ export class DrizzleDatabaseCache extends DrizzleCache {
 		_isTag: boolean,
 		config?: CacheConfig
 	): Promise<void> {
-		await this.#cache.set(key, response, (config?.ex ?? this.globalTtl) * 1000);
+		await (await this.#storage()).set(key, response, (config?.ex ?? this.globalTtl) * 1000);
 		for (const table of tables) {
 			const keys = this.usedTablesPerKey[table];
 			if (keys === undefined) {
@@ -118,8 +136,9 @@ export class DrizzleDatabaseCache extends DrizzleCache {
 		}
 
 		if (keysToDelete.size > 0 || tagsArray.length > 0) {
-			await this.#cache.mdel(tagsArray);
-			await this.#cache.mdel([...keysToDelete]);
+			const cache = await this.#storage();
+			await cache.mdel(tagsArray);
+			await cache.mdel([...keysToDelete]);
 			for (const table of tablesArray) {
 				const tableName = is(table, Table) ? getTableName(table as Table) : (table as string);
 				this.usedTablesPerKey[tableName] = [];
@@ -141,6 +160,17 @@ export class DrizzleDatabaseCache extends DrizzleCache {
  * @author Axel Nana <axel.nana@workbud.com>
  */
 class DrizzleDatabase extends AbstractDatabase<DrizzleConnection, DrizzleConnectionProps> {
+	/**
+	 * Creates a Drizzle connection from its properties. Used by the `DrizzleModule` to build
+	 * the `db.connection.{name}` services.
+	 * @author Axel Nana <axel.nana@workbud.com>
+	 * @param props The connection properties.
+	 * @returns A new connection.
+	 */
+	public createConnection(props: DrizzleConnectionProps): DrizzleConnection {
+		return this.driver.createConnection(props);
+	}
+
 	protected driver: DatabaseDriver<DrizzleConnection, DrizzleConnectionProps> = {
 		createConnection(config) {
 			const cache =