@elysiumjs/mnemosyne-drizzle 0.2.0 → 0.4.0

package/dist/commands/migration_run.command.js

@@ -1,15 +0,0 @@
-var __decorate=function(decorators,target,key,desc){var c=arguments.length,r=c<3?target:desc===null?desc=Object.getOwnPropertyDescriptor(target,key):desc,d;if(typeof Reflect==="object"&&typeof Reflect.decorate==="function")r=Reflect.decorate(decorators,target,key,desc);else for(var i=decorators.length-1;i>=0;i--)if(d=decorators[i])r=(c<3?d(r):c>3?d(target,key,r):d(target,key))||r;return c>3&&r&&Object.defineProperty(target,key,r),r};var __metadata=function(k,v){if(typeof Reflect==="object"&&typeof Reflect.metadata==="function")return Reflect.metadata(k,v)};import{Command,CommandArgumentType}from"@elysiumjs/core";import prompts from"prompts";import{Database}from"../database";import{getMigrationName,planMigrations}from"../migrations";let MigrationRunCommand=class MigrationRunCommand extends Command{static command="migration:run";static description="Runs pending migrations.";tenant="";connection="default";fresh=false;async run(){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)=>{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=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}]=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()}};__decorate([Command.arg({description:"The name of the tenant in which the migrations will be executed.",type:CommandArgumentType.STRING,required:true}),__metadata("design:type",String)],MigrationRunCommand.prototype,"tenant",undefined);__decorate([Command.arg({description:"The name of the connection to use.",type:CommandArgumentType.STRING,default:"default"}),__metadata("design:type",String)],MigrationRunCommand.prototype,"connection",undefined);__decorate([Command.arg({description:"Drop the entire schema before running the migrations.",type:CommandArgumentType.BOOLEAN,default:false}),__metadata("design:type",Boolean)],MigrationRunCommand.prototype,"fresh",undefined);MigrationRunCommand=__decorate([Command.register()],MigrationRunCommand);export{MigrationRunCommand};

package/dist/database.js

@@ -1 +0,0 @@
-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";import{Cache as DrizzleCache}from"drizzle-orm/cache/core";export class DrizzleDatabaseCache extends DrizzleCache{cacheStrategy;globalTtl;#storageEngine;#cache;usedTablesPerKey={};constructor(cacheStrategy="explicit",globalTtl=1,storageEngine="redis"){super();this.cacheStrategy=cacheStrategy;this.globalTtl=globalTtl;this.#storageEngine=storageEngine}async#storage(){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}strategy(){return this.cacheStrategy}async get(key){return await(await this.#storage()).get(key)??undefined}async put(key,response,tables,_isTag,config){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){this.usedTablesPerKey[table]=[key]}else{keys.push(key)}}}async onMutate(params){const tagsArray=params.tags?Array.isArray(params.tags)?params.tags:[params.tags]:[];const tablesArray=params.tables?Array.isArray(params.tables)?params.tables:[params.tables]:[];const keysToDelete=new Set;for(const table of tablesArray){const tableName=is(table,Table)?getTableName(table):table;const keys=this.usedTablesPerKey[tableName]??[];for(const key of keys)keysToDelete.add(key)}if(keysToDelete.size>0||tagsArray.length>0){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):table;this.usedTablesPerKey[tableName]=[]}}}}class DrizzleDatabase extends AbstractDatabase{createConnection(props){return this.driver.createConnection(props)}driver={createConnection(config){const cache=config.cache===undefined?undefined:config.cache===true?new DrizzleDatabaseCache:config.cache.enabled!==false?new DrizzleDatabaseCache(config.cache.strategy,config.cache.ttl,config.cache.storage):undefined;return drizzle({...config,cache})},withTransaction(connection,callback){return connection.transaction((tx)=>callback(tx))},getRawClient(connection){return connection.$client}}}export const Database=new DrizzleDatabase;

package/dist/drizzle.module.js

@@ -1 +0,0 @@
-var __decorate=function(decorators,target,key,desc){var c=arguments.length,r=c<3?target:desc===null?desc=Object.getOwnPropertyDescriptor(target,key):desc,d;if(typeof Reflect==="object"&&typeof Reflect.decorate==="function")r=Reflect.decorate(decorators,target,key,desc);else for(var i=decorators.length-1;i>=0;i--)if(d=decorators[i])r=(c<3?d(r):c>3?d(target,key,r):d(target,key))||r;return c>3&&r&&Object.defineProperty(target,key,r),r};import{Module,needsDefaultAlias,runAll,Service}from"@elysiumjs/core";import{AbstractDatabase,MnemosyneModule,TENANCY_CONFIG_SERVICE,TENANCY_STRATEGY_SERVICE}from"@elysiumjs/mnemosyne";import{MigrationGenerateCommand}from"./commands/migration_generate.command";import{MigrationRunCommand}from"./commands/migration_run.command";import{Database}from"./database";import{DrizzleRLSTenancy,DrizzleSchemaTenancy}from"./tenancy";let DrizzleModule=class DrizzleModule extends Module.configurable({defaults:{default:"default"},required:["connections"]}){async onDestroy(){const names=Object.keys(this.config.connections).map(AbstractDatabase.getConnectionName);await runAll(names.filter((name)=>Service.isConstructed(name)).map((name)=>()=>Service.get(name)?.$client.close()))}};DrizzleModule=__decorate([Module.register({imports:[MnemosyneModule],commands:[MigrationGenerateCommand,MigrationRunCommand],services:(config)=>{const alias=needsDefaultAlias("DrizzleModule",config);const tenancy=config.tenancy??{mode:"schema"};return[...Object.entries(config.connections).map(([name,props])=>({name:AbstractDatabase.getConnectionName(name),global:true,factory:()=>Database.createConnection(props)})),...alias?[{name:AbstractDatabase.getConnectionName("default"),global:true,factory:(_config,get)=>get(AbstractDatabase.getConnectionName(config.default))}]:[],{name:TENANCY_CONFIG_SERVICE,global:true,factory:()=>tenancy},{name:TENANCY_STRATEGY_SERVICE,global:true,factory:()=>tenancy.mode==="rls"?new DrizzleRLSTenancy:new DrizzleSchemaTenancy}]}})],DrizzleModule);export{DrizzleModule};

package/dist/migrations.js

@@ -1 +0,0 @@
-import{basename,dirname}from"node:path";const FOLDER_MIGRATION_FILE="migration.sql";export const getMigrationName=(path)=>basename(path)===FOLDER_MIGRATION_FILE?basename(dirname(path)):basename(path);const descriptiveName=(name)=>name.replace(/\.sql$/,"").split("_").slice(1).join("_");const folderTime=(name)=>{const match=/^(\d{4})(\d{2})(\d{2})(\d{2})(\d{2})(\d{2})/.exec(name);return match?Date.UTC(+match[1],+match[2]-1,+match[3],+match[4],+match[5],+match[6]):undefined};export const planMigrations=(files,applied)=>{const appliedNames=new Set(applied.map(({name})=>name));const fileNames=new Set(files.map(getMigrationName));const moved=new Map;for(const record of[...applied].sort((a,b)=>a.name.localeCompare(b.name))){if(!record.name.endsWith(".sql")||fileNames.has(record.name))continue;const key=descriptiveName(record.name);moved.set(key,[...moved.get(key)??[],record])}const plan={pending:[],renames:[]};for(const file of[...files].sort()){const name=getMigrationName(file);if(appliedNames.has(name))continue;const from=basename(file)===FOLDER_MIGRATION_FILE?takeMoved(moved,name):undefined;if(from)plan.renames.push({from,to:name});else plan.pending.push(file)}return plan};const takeMoved=(moved,name)=>{const records=moved.get(descriptiveName(name));const generatedAt=folderTime(name);const index=records?.findIndex(({appliedAt})=>generatedAt===undefined||generatedAt<=appliedAt.getTime())??-1;return index<0?undefined:records.splice(index,1)[0].name};

package/dist/model.js

@@ -1 +0,0 @@
-import{AbstractModel,createSchemaFromModel}from"@elysiumjs/mnemosyne";import{getTableConfig}from"drizzle-orm/pg-core";import{getTableBuilders}from"./table";import{wrapTenantSchema}from"./tenancy";const mapDrizzleDataType=(col)=>{if(col.columnType==="PgUUID")return"uuid";if(col.dimensions)return"array";const[kind,constraint]=col.dataType.split(" ");switch(kind){case"number":return"number";case"boolean":return"boolean";case"array":return"array";case"bigint":return"bigint";case"object":switch(constraint){case"json":return"json";case"date":return"date";case"buffer":return"buffer";default:return"string"}default:return"string"}};export const drizzleAdapter={getTableName(table){return getTableConfig(table).name},getColumns(table){const config=getTableConfig(table);return config.columns.map((col)=>({name:col.name,dataType:mapDrizzleDataType(col),nullable:!col.notNull,hasDefault:col.hasDefault,isPrimaryKey:col.primary||config.primaryKeys.some((pk)=>pk.columns.some((c)=>c.name===col.name))}))},createTenantTable(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)}};export const createSchemaFromDrizzle=(table,opts)=>{return createSchemaFromModel(drizzleAdapter.getColumns(table),opts)};export const DrizzleModel=(table,options)=>{const config=getTableConfig(table);const builders=getTableBuilders(table);const Base=AbstractModel(config.name,builders??{},drizzleAdapter,table);if(options?.supportTenancy!==undefined){Base.supportTenancy=options.supportTenancy}return Base};

package/dist/repository.js

@@ -1 +0,0 @@
-import{AbstractRepository}from"@elysiumjs/mnemosyne";import{eq}from"drizzle-orm";import{Database}from"./database";export const DrizzleRepository=(model)=>{class R extends AbstractRepository(model,Database){async all(){return await this.db.select().from(model.table)}async paginate(count,page=1){const total=await this.db.$count(model.table);const offset=Math.max(page-1,0)*count;return{page,total,data:await this.db.select().from(model.table).offset(offset).limit(count)}}async find(id){const[row]=await this.db.select().from(model.table).where(eq(model.table.id,id));return row??null}async findBy(column,value){const[row]=await this.db.select().from(model.table).where(eq(model.table[column],value));return row??null}async insert(data){const[row]=await this.db.insert(model.table).values(data).returning();return row}async update(id,data){const[row]=await this.db.update(model.table).set(data).where(eq(model.table.id,id)).returning();return row}async updateAll(data){return await this.db.update(model.table).set(data).returning()}async delete(id){const[row]=await this.db.delete(model.table).where(eq(model.table.id,id)).returning();return row}async deleteAll(){return await this.db.delete(model.table).returning()}async exists(id){const count=await this.db.$count(model.table,eq(model.table.id,id));return count>0}}return R};

package/dist/table.js

@@ -1 +0,0 @@
-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.js

@@ -1 +0,0 @@
-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/scripts/minify.ts

@@ -1,10 +0,0 @@
-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

@@ -1,101 +0,0 @@
-// 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

@@ -1,160 +0,0 @@
-// 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

@@ -1,205 +0,0 @@
-// Copyright (c) 2026-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 type { CacheInterface, DatabaseCacheConfig, DatabaseDriver } from '@elysiumjs/mnemosyne';
-import type { DrizzleConfig } from 'drizzle-orm';
-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';
-import { Cache as DrizzleCache } from 'drizzle-orm/cache/core';
-
-// ============================================================================
-// Types
-// ============================================================================
-
-/**
- * Connection configuration for Drizzle ORM.
- *
- * Extends `DrizzleConfig` with connection details and optional caching.
- *
- * @author Axel Nana <axel.nana@workbud.com>
- * @template TSchema The Drizzle schema type.
- */
-export type DrizzleConnectionProps<
-	TSchema extends Record<string, unknown> = Record<string, never>
-> = Omit<DrizzleConfig<TSchema>, 'cache'> &
-	({ connection: string | ({ url?: string } & Bun.SQLOptions) } | { client: Bun.SQL }) & {
-		cache?: true | DatabaseCacheConfig;
-	};
-
-/**
- * A Drizzle database connection backed by Bun's SQL driver.
- * @author Axel Nana <axel.nana@workbud.com>
- */
-export type DrizzleConnection = BunSQLDatabase<Record<string, never>> & { $client: Bun.SQL };
-
-// ============================================================================
-// DrizzleDatabaseCache
-// ============================================================================
-
-/**
- * Drizzle query cache backed by the framework's cache storage.
- *
- * Extends the Drizzle `Cache` base class and implements the
- * framework's `DatabaseCacheStrategy` interface so that query
- * results can be transparently cached and invalidated.
- *
- * @author Axel Nana <axel.nana@workbud.com>
- */
-export class DrizzleDatabaseCache extends DrizzleCache {
-	#storageEngine: 'redis' | 'memory';
-	#cache?: Omit<CacheInterface, 'tags'>;
-	private usedTablesPerKey: Record<string, string[]> = {};
-
-	constructor(
-		private readonly cacheStrategy: 'explicit' | 'all' = 'explicit',
-		private readonly globalTtl: number = 1,
-		storageEngine: 'redis' | 'memory' = 'redis'
-	) {
-		super();
-		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' {
-		return this.cacheStrategy;
-	}
-
-	public override async get(key: string): Promise<any[] | undefined> {
-		return (await (await this.#storage()).get<any[]>(key)) ?? undefined;
-	}
-
-	public override async put(
-		key: string,
-		response: any,
-		tables: string[],
-		_isTag: boolean,
-		config?: CacheConfig
-	): Promise<void> {
-		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) {
-				this.usedTablesPerKey[table] = [key];
-			} else {
-				keys.push(key);
-			}
-		}
-	}
-
-	public override async onMutate(params: {
-		tags?: string | string[];
-		tables?: string | string[] | Table<any> | Table<any>[];
-	}): Promise<void> {
-		const tagsArray = params.tags ? (Array.isArray(params.tags) ? params.tags : [params.tags]) : [];
-		const tablesArray = params.tables
-			? Array.isArray(params.tables)
-				? params.tables
-				: [params.tables]
-			: [];
-		const keysToDelete = new Set<string>();
-
-		for (const table of tablesArray) {
-			const tableName = is(table, Table) ? getTableName(table as Table) : (table as string);
-			const keys = this.usedTablesPerKey[tableName] ?? [];
-			for (const key of keys) keysToDelete.add(key);
-		}
-
-		if (keysToDelete.size > 0 || tagsArray.length > 0) {
-			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] = [];
-			}
-		}
-	}
-}
-
-// ============================================================================
-// DrizzleDatabase
-// ============================================================================
-
-/**
- * Drizzle-specific database manager.
- *
- * Implements the abstract database class with a driver that creates
- * Drizzle connections using Bun's built-in SQL client.
- *
- * @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 =
-				config.cache === undefined
-					? undefined
-					: config.cache === true
-						? new DrizzleDatabaseCache()
-						: config.cache.enabled !== false
-							? new DrizzleDatabaseCache(
-									config.cache.strategy,
-									config.cache.ttl,
-									config.cache.storage
-								)
-							: undefined;
-			return drizzle({ ...config, cache }) as DrizzleConnection;
-		},
-
-		withTransaction(connection, callback) {
-			return (connection as any).transaction((tx: any) => callback(tx));
-		},
-
-		getRawClient(connection) {
-			return connection.$client;
-		}
-	};
-}
-
-/**
- * Singleton Drizzle database manager.
- * @author Axel Nana <axel.nana@workbud.com>
- */
-export const Database = new DrizzleDatabase();

package/src/drizzle.module.ts

@@ -1,97 +0,0 @@
-// Copyright (c) 2026-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 type { TenancyConfig } from '@elysiumjs/mnemosyne';
-import type { DrizzleConnection, DrizzleConnectionProps } from './database';
-
-import { Module, needsDefaultAlias, runAll, Service } from '@elysiumjs/core';
-import {
-	AbstractDatabase,
-	MnemosyneModule,
-	TENANCY_CONFIG_SERVICE,
-	TENANCY_STRATEGY_SERVICE
-} from '@elysiumjs/mnemosyne';
-
-import { MigrationGenerateCommand } from './commands/migration_generate.command';
-import { MigrationRunCommand } from './commands/migration_run.command';
-import { Database } from './database';
-import { DrizzleRLSTenancy, DrizzleSchemaTenancy } from './tenancy';
-
-/**
- * Configuration of the {@link DrizzleModule}.
- * @author Axel Nana <axel.nana@workbud.com>
- */
-export type DrizzleModuleConfig = {
-	/** The name of the connection that `default` resolves to. */
-	default: string;
-
-	/** The connections to declare, by name. */
-	connections: Record<string, DrizzleConnectionProps>;
-
-	/** The tenancy configuration. Defaults to schema-based tenancy. */
-	tenancy?: TenancyConfig;
-};
-
-/**
- * Declares Drizzle connections as global services named `db.connection.{name}`, plus
- * `db.connection.default` aliasing the configured default, and registers the tenancy config
- * and strategy. Connections are created on first use. Also provides the migration commands.
- * @author Axel Nana <axel.nana@workbud.com>
- */
-@Module.register<DrizzleModuleConfig>({
-	imports: [MnemosyneModule],
-	commands: [MigrationGenerateCommand, MigrationRunCommand],
-	services: (config) => {
-		const alias = needsDefaultAlias('DrizzleModule', config);
-		const tenancy = config.tenancy ?? { mode: 'schema' };
-		return [
-			...Object.entries(config.connections).map(([name, props]) => ({
-				name: AbstractDatabase.getConnectionName(name),
-				global: true,
-				factory: () => Database.createConnection(props)
-			})),
-			...(alias
-				? [
-						{
-							name: AbstractDatabase.getConnectionName('default'),
-							global: true,
-							factory: (_config: unknown, get: <T>(name: string) => T) =>
-								get<DrizzleConnection>(AbstractDatabase.getConnectionName(config.default))
-						}
-					]
-				: []),
-			{ name: TENANCY_CONFIG_SERVICE, global: true, factory: () => tenancy },
-			{
-				name: TENANCY_STRATEGY_SERVICE,
-				global: true,
-				factory: () =>
-					tenancy.mode === 'rls' ? new DrizzleRLSTenancy() : new DrizzleSchemaTenancy()
-			}
-		];
-	}
-})
-export class DrizzleModule extends Module.configurable<DrizzleModuleConfig>({
-	defaults: { default: 'default' },
-	required: ['connections']
-}) {
-	/** Closes the clients of the connections that were created. */
-	public override async onDestroy(): Promise<void> {
-		const names = Object.keys(this.config.connections).map(AbstractDatabase.getConnectionName);
-		await runAll(
-			names
-				.filter((name) => Service.isConstructed(name))
-				.map((name) => () => Service.get<DrizzleConnection>(name)?.$client.close())
-		);
-	}
-}