@elysiumjs/mnemosyne-drizzle 0.2.0 → 0.3.1

package/dist/tenancy.d.ts

@@ -6,6 +6,7 @@ import type { DrizzleConnection } from './database';
  * @author Axel Nana <axel.nana@workbud.com>
  * @param tenant The tenant identifier.
  * @returns The tenant's PostgreSQL schema.
+ * @throws {Error} If the tenant identifier is malformed.
  */
 export declare const getTenantSchema: (tenant: string) => PgSchema;
 /**
@@ -13,6 +14,7 @@ export declare const getTenantSchema: (tenant: string) => PgSchema;
  * @author Axel Nana <axel.nana@workbud.com>
  * @param tenant The tenant identifier.
  * @returns The newly registered (or existing) `PgSchema`.
+ * @throws {Error} If the tenant identifier is malformed.
  */
 export declare const registerTenantSchema: (tenant: string) => PgSchema;
 /**
@@ -26,26 +28,27 @@ export declare const registerTenantSchema: (tenant: string) => PgSchema;
  * @param tableName The table name.
  * @param columns The column definitions.
  * @returns A tenant-scoped `PgTable`.
+ * @throws {Error} If the tenant identifier is malformed.
  */
 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.
+ *
+ * The transaction is registered in the async context, so repositories and
+ * `Database.getConnection()` use it inside the callback. When a transaction is already open on the
+ * connection, the callback joins it if it is bound to the same tenant. Work for several tenants
+ * (such as an admin fan-out) must use one `withRLS()` per tenant, outside any transaction.
  * @author Axel Nana <axel.nana@workbud.com>
  * @param tenant The tenant identifier (or `null` for 'public').
  * @param callback The function to execute within the transaction.
  * @param connectionName The database connection name.
  * @returns The return value of the callback.
+ * @throws {TenantMismatchError} If the open transaction is bound to another tenant.
  */
 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>
- * @param tenant The tenant identifier (or `null` for 'public').
- * @param connectionName The database connection name.
- */
-export declare const setConnectionTenant: (tenant: string | null, connectionName?: string) => Promise<void>;
-/**
- * Reads the current tenant from the PostgreSQL session variable.
+ * Reads the current tenant from the PostgreSQL tenant variable, on the open transaction when
+ * there is one.
  * @author Axel Nana <axel.nana@workbud.com>
  * @param connectionName The database connection name.
  * @returns The current tenant identifier, or `null`.
@@ -84,5 +87,19 @@ export declare class DrizzleSchemaTenancy implements TenancyStrategy<PgTable, Dr
 export declare class DrizzleRLSTenancy implements TenancyStrategy<PgTable, DrizzleConnection> {
     readonly mode = "rls";
     resolveTable(table: PgTable, _tenant: string): PgTable;
+    /**
+     * Runs a callback in a transaction scoped to the given tenant.
+     *
+     * The transaction is bound to the tenant for its whole life, registered in the async context
+     * for the connection, and the tenant set as the current tenant, for the duration of the
+     * callback. When a transaction is already open on the connection, the callback joins it, which
+     * is only allowed when that transaction is bound to the same tenant.
+     * @author Axel Nana <axel.nana@workbud.com>
+     * @param connection The base connection (see `Database.getBaseConnection`).
+     * @param tenant The tenant identifier.
+     * @param callback The function to execute within the isolated context.
+     * @returns The return value of the callback.
+     * @throws {TenantMismatchError} If the open transaction is bound to another tenant.
+     */
     withIsolation<T>(connection: DrizzleConnection, tenant: string, callback: () => Promise<T>): Promise<T>;
 }

package/dist/tenancy.js

@@ -1 +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()})}}
+import{getTenancyConfig,isValidTenantId,withTenant}from"@elysiumjs/mnemosyne";import{sql}from"drizzle-orm";import{getTableConfig,pgPolicy,pgSchema,pgTable}from"drizzle-orm/pg-core";import{assertTransactionTenant,bindTransactionTenant,Database}from"./database";import{getTableBuilders}from"./table";const tenantSchemas=new Map;const schemaRegistry=new Map;const assertTenantId=(tenant)=>{if(!isValidTenantId(tenant)){throw new Error(`Invalid tenant identifier: ${JSON.stringify(tenant.slice(0,64))}`)}};export const getTenantSchema=(tenant)=>{assertTenantId(tenant);return schemaRegistry.get(tenant)??registerTenantSchema(tenant)};export const registerTenantSchema=(tenant)=>{assertTenantId(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)=>{assertTenantId(tenant);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")=>{return new DrizzleRLSTenancy().withIsolation(Database.getBaseConnection(connectionName),tenant??"public",()=>callback(Database.getConnection(connectionName)))};export const getSessionTenant=async(connectionName="default")=>{const db=Database.getConnection(connectionName);const config=getTenancyConfig();const varName=config.rls?.sessionVariable??"app.current_tenant";const[row]=await db.execute(sql`SELECT current_setting(${varName}, true) as tenant`);return row?.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 active=Database.getActiveTransaction(connection);if(active!==undefined){assertTransactionTenant(active,tenant);return withTenant(tenant,callback)}return connection.transaction(async(tx)=>{await bindTransactionTenant(tx,tenant);return withTenant(tenant,()=>Database.runInTransaction(connection,tx,callback))})}}

package/package.json

@@ -1,22 +1,28 @@
 {
 	"name": "@elysiumjs/mnemosyne-drizzle",
-	"version": "0.2.0",
+	"version": "0.3.1",
 	"license": "Apache-2.0",
 	"description": "Drizzle ORM driver for @elysiumjs/mnemosyne",
 	"private": false,
 	"author": "Workbud Technologies Inc.",
 	"repository": {
 		"type": "git",
-		"url": "https://github.com/workbud/elysium"
+		"url": "git+https://github.com/na2axl/elysiumjs.git",
+		"directory": "@elysiumjs/mnemosyne-drizzle"
+	},
+	"homepage": "https://github.com/na2axl/elysiumjs/tree/main/@elysiumjs/mnemosyne-drizzle#readme",
+	"bugs": {
+		"url": "https://github.com/na2axl/elysiumjs/issues"
 	},
-	"homepage": "https://github.com/workbud/elysium",
 	"scripts": {
 		"format": "prettier --write \"./src/**/*.ts\"",
 		"clean": "rimraf dist",
 		"build": "bun run clean && tsc && bun run build:minify",
 		"build:minify": "bun scripts/minify.ts",
-		"typecheck": "tsc --noEmit",
-		"prepublishOnly": "bun run build"
+		"typecheck": "tsc --noEmit && tsc -p tsconfig.types.json",
+		"prepublishOnly": "bun run build",
+		"test": "bun test",
+		"test:coverage": "bun test --coverage"
 	},
 	"keywords": [
 		"elysium",
@@ -28,10 +34,14 @@
 	"module": "dist/index.js",
 	"type": "module",
 	"types": "dist/index.d.ts",
+	"files": [
+		"dist",
+		"CHANGELOG.md"
+	],
 	"sideEffects": false,
 	"peerDependencies": {
-		"@elysiumjs/core": "^0.8.0",
-		"@elysiumjs/mnemosyne": "^0.2.0"
+		"@elysiumjs/core": "^0.10.0",
+		"@elysiumjs/mnemosyne": "^0.3.1"
 	},
 	"dependencies": {
 		"drizzle-orm": "^1.0.0-rc.5",
@@ -40,7 +50,7 @@
 	},
 	"devDependencies": {
 		"@types/prompts": "^2.4.9",
-		"@elysiumjs/core": "0.8.0",
-		"@elysiumjs/mnemosyne": "0.2.0"
+		"@elysiumjs/core": "0.10.0",
+		"@elysiumjs/mnemosyne": "0.3.1"
 	}
 }

package/.omc/state/agent-replay-3920b579-6d8e-4f5f-924a-319202936685.jsonl

@@ -1,7 +0,0 @@
-{"t":0,"agent":"aac206b","agent_type":"executor","event":"agent_start","parent_mode":"none"}
-{"t":0,"agent":"a914068","agent_type":"executor","event":"agent_stop","success":true,"duration_ms":31478}
-{"t":0,"agent":"aa0cc16","agent_type":"untracked-native-fork","event":"agent_stop","success":true,"synthetic":true,"telemetry_status":"unmatched_stop","reason":"SubagentStop arrived without a matching SubagentStart; native Agent/Task start telemetry was not observed."}
-{"t":0,"agent":"a55f688","agent_type":"untracked-native-fork","event":"agent_stop","success":true,"synthetic":true,"telemetry_status":"unmatched_stop","reason":"SubagentStop arrived without a matching SubagentStart; native Agent/Task start telemetry was not observed."}
-{"t":0,"agent":"a0ba584","agent_type":"untracked-native-fork","event":"agent_stop","success":true,"synthetic":true,"telemetry_status":"unmatched_stop","reason":"SubagentStop arrived without a matching SubagentStart; native Agent/Task start telemetry was not observed."}
-{"t":0,"agent":"a6e8774","agent_type":"untracked-native-fork","event":"agent_stop","success":true,"synthetic":true,"telemetry_status":"unmatched_stop","reason":"SubagentStop arrived without a matching SubagentStart; native Agent/Task start telemetry was not observed."}
-{"t":0,"agent":"aac206b","agent_type":"executor","event":"agent_stop","success":true,"duration_ms":168112}

package/.omc/state/sessions/3920b579-6d8e-4f5f-924a-319202936685/last-tool-error-state.json

@@ -1,7 +0,0 @@
-{
-  "tool_name": "Bash",
-  "tool_input_preview": "{\"command\":\"perl -0pi -e 's/\\\\t\\\\t\\\\tconst migrations: \\\\{ name: string \\\\}\\\\[\\\\] = await sql`SELECT name FROM elysium_migrations;`;\\\\n\\\\t\\\\t\\\\tconst completedMigrations = migrations\\\\.map\\\\(\\\\(row\\\\)...",
-  "error": "Exit code 1\n\t\t\t}\n\n\t\t\tconst rows: { name: string; applied_at: Date }[] =\n\t\t\t\tawait sql`SELECT name, applied_at FROM elysium_migrations;`;\n\t\t\tconst completedMigrations = rows.map((row) => ({ name: row.name, appliedAt: row.applied_at }));\n\n\t\t\tconst filesIt = new Bun.Glob(`./src/database/migrations/**/*.sql`).scan({\n\t\t\t\tabsolute: true,\n\t\t\t\tonlyFiles: true\n\t\t\t});\n\n\t\t\tconst files = await Array.fromAsync(filesIt);\n\t\t\tconst { pending: sqlFiles, renames } = planMigrations(files, completedMigrations);\nChe...",
-  "timestamp": "2026-10-26T02:29:07.154Z",
-  "retry_count": 1
-}

package/.omc/state/sessions/3920b579-6d8e-4f5f-924a-319202936685/mission-state.json

@@ -1,93 +0,0 @@
-{
-  "updatedAt": "2026-10-24T23:32:53.900Z",
-  "missions": [
-    {
-      "id": "session:3920b579-6d8e-4f5f-924a-319202936685:none",
-      "source": "session",
-      "name": "none",
-      "objective": "Session mission",
-      "createdAt": "2026-10-24T23:30:05.788Z",
-      "updatedAt": "2026-10-24T23:32:53.900Z",
-      "status": "done",
-      "workerCount": 1,
-      "taskCounts": {
-        "total": 1,
-        "pending": 0,
-        "blocked": 0,
-        "inProgress": 0,
-        "completed": 1,
-        "failed": 0
-      },
-      "agents": [
-        {
-          "name": "executor:aac206b",
-          "role": "executor",
-          "ownership": "aac206b9ab7956eb4",
-          "status": "done",
-          "currentStep": null,
-          "latestUpdate": "completed",
-          "completedSummary": null,
-          "updatedAt": "2026-10-24T23:32:53.900Z"
-        }
-      ],
-      "timeline": [
-        {
-          "id": "session-start:aac206b9ab7956eb4:2026-10-24T23:30:05.788Z",
-          "at": "2026-10-24T23:30:05.788Z",
-          "kind": "update",
-          "agent": "executor:aac206b",
-          "detail": "started executor:aac206b",
-          "sourceKey": "session-start:aac206b9ab7956eb4"
-        },
-        {
-          "id": "session-stop:a914068b2d3b26e56:2026-10-24T23:30:37.266Z",
-          "at": "2026-10-24T23:30:37.266Z",
-          "kind": "completion",
-          "agent": "executor:aac206b",
-          "detail": "completed",
-          "sourceKey": "session-stop:a914068b2d3b26e56"
-        },
-        {
-          "id": "session-stop:aa0cc16535a1e4c83:2026-10-24T23:31:08.989Z",
-          "at": "2026-10-24T23:31:08.989Z",
-          "kind": "completion",
-          "agent": "executor:aac206b",
-          "detail": "completed",
-          "sourceKey": "session-stop:aa0cc16535a1e4c83"
-        },
-        {
-          "id": "session-stop:a55f6887513d75db7:2026-10-24T23:31:40.515Z",
-          "at": "2026-10-24T23:31:40.515Z",
-          "kind": "completion",
-          "agent": "executor:aac206b",
-          "detail": "completed",
-          "sourceKey": "session-stop:a55f6887513d75db7"
-        },
-        {
-          "id": "session-stop:a0ba584b3d851e834:2026-10-24T23:32:12.704Z",
-          "at": "2026-10-24T23:32:12.704Z",
-          "kind": "completion",
-          "agent": "executor:aac206b",
-          "detail": "completed",
-          "sourceKey": "session-stop:a0ba584b3d851e834"
-        },
-        {
-          "id": "session-stop:a6e8774450f70f52b:2026-10-24T23:32:44.262Z",
-          "at": "2026-10-24T23:32:44.262Z",
-          "kind": "completion",
-          "agent": "executor:aac206b",
-          "detail": "completed",
-          "sourceKey": "session-stop:a6e8774450f70f52b"
-        },
-        {
-          "id": "session-stop:aac206b9ab7956eb4:2026-10-24T23:32:53.900Z",
-          "at": "2026-10-24T23:32:53.900Z",
-          "kind": "completion",
-          "agent": "executor:aac206b",
-          "detail": "completed",
-          "sourceKey": "session-stop:aac206b9ab7956eb4"
-        }
-      ]
-    }
-  ]
-}

package/.omc/state/sessions/3920b579-6d8e-4f5f-924a-319202936685/pre-tool-advisory-throttle.json

@@ -1,18 +0,0 @@
-{
-  "version": 1,
-  "entries": {
-    "445ed27a3872b681d98190bae61ccb84954e1bc4e140df5370be958dee776b3a": {
-      "last_emitted_at_ms": 1792981759903,
-      "message": "Verify changes work after editing. Test functionality before marking complete."
-    },
-    "79a93d4a2f8f50b95f852280616242fee1855dc99a3c75211917f55e72e95fae": {
-      "last_emitted_at_ms": 1792981745167,
-      "message": "Use parallel execution for independent tasks. Use run_in_background for long operations (npm install, builds, tests)."
-    },
-    "466399dafa2d20f60587180bad0a07358eb7f2bce724df0ff682f038ad33f5ad": {
-      "last_emitted_at_ms": 1792981448165,
-      "message": "Read multiple files in parallel when possible for faster analysis."
-    }
-  },
-  "updated_at": "2026-10-26T02:29:19.903Z"
-}

package/.omc/state/sessions/3920b579-6d8e-4f5f-924a-319202936685/subagent-tracking-state.json

@@ -1,61 +0,0 @@
-{
-  "agents": [
-    {
-      "agent_id": "aac206b9ab7956eb4",
-      "agent_type": "oh-my-claudecode:executor",
-      "started_at": "2026-10-24T23:30:05.788Z",
-      "parent_mode": "none",
-      "status": "completed",
-      "completed_at": "2026-10-24T23:32:53.900Z",
-      "duration_ms": 168112
-    },
-    {
-      "agent_id": "aa0cc16535a1e4c83",
-      "agent_type": "untracked-native-fork",
-      "started_at": "2026-10-24T23:31:08.989Z",
-      "parent_mode": "none",
-      "status": "completed",
-      "completed_at": "2026-10-24T23:31:08.989Z",
-      "synthetic": true,
-      "telemetry_status": "unmatched_stop",
-      "telemetry_note": "SubagentStop arrived without a matching SubagentStart; native Agent/Task start telemetry was not observed."
-    },
-    {
-      "agent_id": "a55f6887513d75db7",
-      "agent_type": "untracked-native-fork",
-      "started_at": "2026-10-24T23:31:40.515Z",
-      "parent_mode": "none",
-      "status": "completed",
-      "completed_at": "2026-10-24T23:31:40.515Z",
-      "synthetic": true,
-      "telemetry_status": "unmatched_stop",
-      "telemetry_note": "SubagentStop arrived without a matching SubagentStart; native Agent/Task start telemetry was not observed."
-    },
-    {
-      "agent_id": "a0ba584b3d851e834",
-      "agent_type": "untracked-native-fork",
-      "started_at": "2026-10-24T23:32:12.704Z",
-      "parent_mode": "none",
-      "status": "completed",
-      "completed_at": "2026-10-24T23:32:12.704Z",
-      "synthetic": true,
-      "telemetry_status": "unmatched_stop",
-      "telemetry_note": "SubagentStop arrived without a matching SubagentStart; native Agent/Task start telemetry was not observed."
-    },
-    {
-      "agent_id": "a6e8774450f70f52b",
-      "agent_type": "untracked-native-fork",
-      "started_at": "2026-10-24T23:32:44.262Z",
-      "parent_mode": "none",
-      "status": "completed",
-      "completed_at": "2026-10-24T23:32:44.262Z",
-      "synthetic": true,
-      "telemetry_status": "unmatched_stop",
-      "telemetry_note": "SubagentStop arrived without a matching SubagentStart; native Agent/Task start telemetry was not observed."
-    }
-  ],
-  "total_spawned": 1,
-  "total_completed": 6,
-  "total_failed": 0,
-  "last_updated": "2026-10-24T23:32:53.907Z"
-}

package/.turbo/turbo-build.log

@@ -1,3 +0,0 @@
-$ bun run clean && tsc && bun run build:minify
-$ rimraf dist
-$ bun scripts/minify.ts

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();
-	}
-}