@elysiumjs/mnemosyne-drizzle 0.2.0 → 0.4.0

package/src/index.ts

@@ -1,47 +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.
-
-// Re-export drizzle-orm/pg-core with enhanced pgTable
-export * from 'drizzle-orm';
-export * from 'drizzle-orm/pg-core';
-export type {
-	SelectedFields,
-	SelectedFieldsFlat,
-	SelectedFieldsOrdered,
-	TableConfig
-} from 'drizzle-orm/pg-core';
-export { pgTable, getTableBuilders } from './table';
-
-// Re-exports
-export { Database, DrizzleDatabaseCache } from './database';
-export type { DrizzleConnectionProps, DrizzleConnection } from './database';
-export {
-	DrizzleModel,
-	createSchemaFromDrizzle,
-	drizzleAdapter,
-	type DrizzleModelOptions
-} from './model';
-export { DrizzleModule, type DrizzleModuleConfig } from './drizzle.module';
-export { DrizzleRepository } from './repository';
-export {
-	DrizzleSchemaTenancy,
-	DrizzleRLSTenancy,
-	wrapTenantSchema,
-	registerTenantSchema,
-	getTenantSchema,
-	withRLS,
-	setConnectionTenant,
-	getSessionTenant,
-	createRLSPolicy
-} from './tenancy';

package/src/migrations.ts

@@ -1,100 +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 { basename, dirname } from 'node:path';
-
-/** A migration recorded as applied in `elysium_migrations`. */
-export type AppliedMigration = {
-	name: string;
-	appliedAt: Date;
-};
-
-/** The migrations to run, and the applied records to rename to their new names. */
-export type MigrationPlan = {
-	pending: string[];
-	renames: { from: string; to: string }[];
-};
-
-const FOLDER_MIGRATION_FILE = 'migration.sql';
-
-/**
- * Gets the name under which a migration file is tracked. Drizzle Kit 1.0 stores each migration
- * as `<folder>/migration.sql`, in which case the folder name identifies the migration.
- * @author Axel Nana <axel.nana@workbud.com>
- * @param path The path of the migration file.
- */
-export const getMigrationName = (path: string): string =>
-	basename(path) === FOLDER_MIGRATION_FILE ? basename(dirname(path)) : basename(path);
-
-/** Drops the `.sql` extension and the leading index or timestamp: `0001_users.sql` → `users`. */
-const descriptiveName = (name: string): string =>
-	name
-		.replace(/\.sql$/, '')
-		.split('_')
-		.slice(1)
-		.join('_');
-
-/** Reads the UTC `YYYYMMDDHHmmss` timestamp that starts a Drizzle Kit 1.0 folder name. */
-const folderTime = (name: string): number | undefined => {
-	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;
-};
-
-/**
- * Decides which migration files to run.
- *
- * `drizzle-kit up` moves each flat `<prefix>_<name>.sql` file into a
- * `<timestamp>_<name>/migration.sql` folder, where the timestamp is when the migration was
- * generated. An applied record whose flat file is gone is matched, in order, to a folder with
- * the same `<name>` generated before the record was applied; that folder is not run again and
- * the record is renamed. Folders generated later are new migrations and run.
- * @author Axel Nana <axel.nana@workbud.com>
- * @param files The paths of the migration files.
- * @param applied The applied migrations.
- */
-export const planMigrations = (files: string[], applied: AppliedMigration[]): MigrationPlan => {
-	const appliedNames = new Set(applied.map(({ name }) => name));
-	const fileNames = new Set(files.map(getMigrationName));
-
-	const moved = new Map<string, AppliedMigration[]>();
-	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: MigrationPlan = { 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;
-};
-
-/** Removes and returns the first moved record that the folder `name` can be the new home of. */
-const takeMoved = (moved: Map<string, AppliedMigration[]>, name: string): string | undefined => {
-	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/src/model.ts

@@ -1,178 +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 { ColumnMetadata, ModelAdapter } from '@elysiumjs/mnemosyne';
-import type { PgColumn, PgTableWithColumns, TableConfig } from 'drizzle-orm/pg-core';
-
-import { AbstractModel, createSchemaFromModel } from '@elysiumjs/mnemosyne';
-import { getTableConfig } from 'drizzle-orm/pg-core';
-
-import { getTableBuilders } from './table';
-import { wrapTenantSchema } from './tenancy';
-
-// ============================================================================
-// Column Type Mapping
-// ============================================================================
-
-/**
- * Maps a Drizzle column to a normalized data type.
- * @param col The Drizzle column to inspect.
- * @returns A normalized data type string.
- */
-const mapDrizzleDataType = (col: PgColumn): ColumnMetadata['dataType'] => {
-	if (col.columnType === 'PgUUID') return 'uuid';
-	if ((col as { dimensions?: number }).dimensions) return 'array';
-
-	// Drizzle 1.0 column data types are `<kind>` or `<kind> <constraint>` (e.g. `object json`).
-	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';
-	}
-};
-
-// ============================================================================
-// Drizzle Model Adapter
-// ============================================================================
-
-/**
- * Drizzle-specific model adapter.
- *
- * Bridges Drizzle's `PgTableWithColumns` to the framework's
- * normalized model metadata.
- *
- * @author Axel Nana <axel.nana@workbud.com>
- */
-export const drizzleAdapter: ModelAdapter<PgTableWithColumns<any>> = {
-	getTableName(table) {
-		return getTableConfig(table).name;
-	},
-
-	getColumns(table): ColumnMetadata[] {
-		const config = getTableConfig(table);
-		return config.columns.map((col: PgColumn) => ({
-			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) as PgTableWithColumns<any>;
-	}
-};
-
-// ============================================================================
-// Schema Helper
-// ============================================================================
-
-/**
- * Creates Elysia validation schemas from a Drizzle table.
- *
- * This is a convenience wrapper around `createSchemaFromModel` that
- * automatically extracts column metadata via the `drizzleAdapter`.
- *
- * @author Axel Nana <axel.nana@workbud.com>
- * @param table The Drizzle table to generate schemas for.
- * @param opts Schema generation options.
- * @returns An Elysia validation schema.
- */
-export const createSchemaFromDrizzle = (
-	table: PgTableWithColumns<any>,
-	opts?: { mode?: 'create' | 'update' | 'select' }
-) => {
-	return createSchemaFromModel(drizzleAdapter.getColumns(table), opts);
-};
-
-// ============================================================================
-// DrizzleModel Options
-// ============================================================================
-
-/**
- * Options for creating a DrizzleModel.
- */
-export interface DrizzleModelOptions {
-	/**
-	 * Enable tenant schema support for this model.
-	 * @default false
-	 */
-	supportTenancy?: boolean;
-}
-
-// ============================================================================
-// DrizzleModel Function
-// ============================================================================
-
-/**
- * Creates a model class backed by a Drizzle pgTable.
- *
- * @author Axel Nana <axel.nana@workbud.com>
- * @param table The Drizzle table (created with enhanced pgTable).
- * @param options Model configuration options.
- * @returns A model class with Drizzle-backed static metadata.
- */
-export const DrizzleModel = <TTable extends PgTableWithColumns<TableConfig>>(
-	table: TTable,
-	options?: DrizzleModelOptions
-) => {
-	const config = getTableConfig(table);
-	const builders = getTableBuilders(table);
-
-	type TSelect = TTable['$inferSelect'];
-	type TInsert = TTable['$inferInsert'];
-
-	const Base = AbstractModel(config.name, builders ?? {}, drizzleAdapter, table);
-
-	// Override supportTenancy if specified
-	if (options?.supportTenancy !== undefined) {
-		(Base as any).supportTenancy = options.supportTenancy;
-	}
-
-	return Base as typeof Base & {
-		readonly $inferSelect: TSelect;
-		readonly $inferInsert: TInsert;
-		readonly $inferUpdate: Partial<TInsert>;
-		readonly table: TTable;
-	};
-};

package/src/repository.ts

@@ -1,113 +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 { IdType, ModelClass } from '@elysiumjs/mnemosyne';
-import type { PgTableWithColumns } from 'drizzle-orm/pg-core';
-import type { DrizzleConnection } from './database';
-
-import { AbstractRepository } from '@elysiumjs/mnemosyne';
-import { eq } from 'drizzle-orm';
-
-import { Database } from './database';
-
-// ============================================================================
-// DrizzleRepository Mixin
-// ============================================================================
-
-/**
- * Drizzle-specific repository mixin with concrete CRUD operations.
- *
- * Extends the abstract repository with Drizzle query builder calls
- * for all standard CRUD methods.
- *
- * @author Axel Nana <axel.nana@workbud.com>
- * @template TModel The model class type.
- * @param model The model class to create the repository for.
- * @returns A concrete repository class with CRUD methods.
- */
-export const DrizzleRepository = <
-	TModel extends ModelClass<any, any, any, PgTableWithColumns<any>>
->(
-	model: TModel
-) => {
-	type TSelect = TModel['$inferSelect'];
-	type TInsert = TModel['$inferInsert'];
-	type TUpdate = TModel['$inferUpdate'];
-
-	class R extends AbstractRepository<TModel, DrizzleConnection>(model, Database) {
-		public async all(): Promise<TSelect[]> {
-			return await this.db.select().from(model.table);
-		}
-
-		public async paginate(
-			count: number,
-			page: number = 1
-		): Promise<{ page: number; data: TSelect[]; total: number }> {
-			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)
-			};
-		}
-
-		public async find(id: IdType): Promise<TSelect | null> {
-			const [row] = await this.db.select().from(model.table).where(eq(model.table.id, id));
-			return (row ?? null) as TSelect | null;
-		}
-
-		public async findBy<TKey extends keyof TSelect>(
-			column: TKey,
-			value: TSelect[TKey]
-		): Promise<TSelect | null> {
-			const [row] = await this.db.select().from(model.table).where(eq(model.table[column], value));
-			return (row ?? null) as TSelect | null;
-		}
-
-		public async insert(data: TInsert): Promise<TSelect> {
-			const [row] = await this.db.insert(model.table).values(data).returning();
-			return row as TSelect;
-		}
-
-		public async update(id: IdType, data: TUpdate): Promise<TSelect> {
-			const [row] = await this.db
-				.update(model.table)
-				.set(data)
-				.where(eq(model.table.id, id))
-				.returning();
-			return row as TSelect;
-		}
-
-		public async updateAll(data: TUpdate): Promise<TSelect[]> {
-			return await this.db.update(model.table).set(data).returning();
-		}
-
-		public async delete(id: IdType): Promise<TSelect> {
-			const [row] = await this.db.delete(model.table).where(eq(model.table.id, id)).returning();
-			return row as TSelect;
-		}
-
-		public async deleteAll(): Promise<TSelect[]> {
-			return await this.db.delete(model.table).returning();
-		}
-
-		public async exists(id: IdType): Promise<boolean> {
-			const count = await this.db.$count(model.table, eq(model.table.id, id));
-			return count > 0;
-		}
-	}
-
-	return R;
-};

package/src/table.ts

@@ -1,70 +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 { AnyPgColumnBuilder, PgTableWithColumns } from 'drizzle-orm/pg-core';
-
-import { pgTable as drizzlePgTable } from 'drizzle-orm/pg-core';
-
-// ============================================================================
-// Column Builder Registry
-// ============================================================================
-
-/**
- * Maps tables to their original column builder definitions.
- *
- * When `pgTable()` is called, column builders are consumed and turned into
- * built `PgColumn` instances. This registry preserves the original builder
- * objects so that tenant tables can be re-created in different schemas.
- */
-const tableRegistry = new WeakMap<PgTableWithColumns<any>, Record<string, AnyPgColumnBuilder>>();
-
-/**
- * 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 const getTableBuilders = (
-	table: PgTableWithColumns<any>
-): Record<string, AnyPgColumnBuilder> | undefined => {
-	return tableRegistry.get(table);
-};
-
-// ============================================================================
-// Enhanced pgTable Function
-// ============================================================================
-
-/**
- * Creates a PostgreSQL table with automatic column builder registration.
- *
- * This is an enhanced version of Drizzle's `pgTable` that automatically
- * registers column builders for tenancy support. It has the exact same
- * signature and behavior as the original.
- *
- * @param tableName The name of the database table.
- * @param columns The column definitions.
- * @param extraConfig Optional extra Drizzle table configuration.
- * @returns A Drizzle PostgreSQL table with registered column builders.
- */
-export const pgTable = <
-	TTableName extends string,
-	TColumns extends Record<string, AnyPgColumnBuilder>
->(
-	tableName: TTableName,
-	columns: TColumns,
-	extraConfig?: (table: any) => any
-) => {
-	const table = drizzlePgTable(tableName, columns, extraConfig);
-	tableRegistry.set(table, columns);
-	return table;
-};