arkormx 2.0.0-next.2 → 2.0.0-next.3

package/dist/cli.mjs

@@ -1371,6 +1371,28 @@ const getMigrationPlan = async (migration, direction = "up") => {
 	else await instance.down(schema);
 	return schema.getOperations();
 };
+const supportsDatabaseMigrationExecution = (adapter) => {
+	return typeof adapter?.executeSchemaOperations === "function";
+};
+const supportsDatabaseReset = (adapter) => {
+	return typeof adapter?.resetDatabase === "function";
+};
+const stripPrismaSchemaModelsAndEnums = (schema) => {
+	const stripped = schema.replace(PRISMA_MODEL_REGEX, "").replace(PRISMA_ENUM_REGEX, "").replace(/\n{3,}/g, "\n\n").trimEnd();
+	return stripped.length > 0 ? `${stripped}\n` : "";
+};
+const applyMigrationToDatabase = async (adapter, migration) => {
+	if (!supportsDatabaseMigrationExecution(adapter)) throw new ArkormException("The configured adapter does not support database-backed migration execution.");
+	const operations = await getMigrationPlan(migration, "up");
+	await adapter.executeSchemaOperations(operations);
+	return { operations };
+};
+const applyMigrationRollbackToDatabase = async (adapter, migration) => {
+	if (!supportsDatabaseMigrationExecution(adapter)) throw new ArkormException("The configured adapter does not support database-backed migration execution.");
+	const operations = await getMigrationPlan(migration, "down");
+	await adapter.executeSchemaOperations(operations);
+	return { operations };
+};
 /**
 * Apply the schema operations defined in a migration to a Prisma schema 
 * file, updating the file on disk if specified, and return the updated 
@@ -2509,10 +2531,13 @@ var MakeSeederCommand = class extends Command {
 
 //#endregion
 //#region src/helpers/migration-history.ts
-const DEFAULT_STATE = {
+const createEmptyAppliedMigrationsState = () => ({
 	version: 1,
 	migrations: [],
 	runs: []
+});
+const supportsDatabaseMigrationState = (adapter) => {
+	return typeof adapter?.readAppliedMigrationsState === "function" && typeof adapter?.writeAppliedMigrationsState === "function";
 };
 const resolveMigrationStateFilePath = (cwd, configuredPath) => {
 	if (configuredPath && configuredPath.trim().length > 0) return resolve(configuredPath);
@@ -2527,10 +2552,10 @@ const computeMigrationChecksum = (filePath) => {
 	return createHash("sha256").update(source).digest("hex");
 };
 const readAppliedMigrationsState = (stateFilePath) => {
-	if (!existsSync(stateFilePath)) return { ...DEFAULT_STATE };
+	if (!existsSync(stateFilePath)) return createEmptyAppliedMigrationsState();
 	try {
 		const parsed = JSON.parse(readFileSync(stateFilePath, "utf-8"));
-		if (!Array.isArray(parsed.migrations)) return { ...DEFAULT_STATE };
+		if (!Array.isArray(parsed.migrations)) return createEmptyAppliedMigrationsState();
 		return {
 			version: 1,
 			migrations: parsed.migrations.filter((migration) => {
@@ -2541,14 +2566,25 @@ const readAppliedMigrationsState = (stateFilePath) => {
 			}) : []
 		};
 	} catch {
-		return { ...DEFAULT_STATE };
+		return createEmptyAppliedMigrationsState();
 	}
 };
+const readAppliedMigrationsStateFromStore = async (adapter, stateFilePath) => {
+	if (supportsDatabaseMigrationState(adapter)) return await adapter.readAppliedMigrationsState();
+	return readAppliedMigrationsState(stateFilePath);
+};
 const writeAppliedMigrationsState = (stateFilePath, state) => {
 	const directory = dirname(stateFilePath);
 	if (!existsSync(directory)) mkdirSync(directory, { recursive: true });
 	writeFileSync(stateFilePath, JSON.stringify(state, null, 2));
 };
+const writeAppliedMigrationsStateToStore = async (adapter, stateFilePath, state) => {
+	if (supportsDatabaseMigrationState(adapter)) {
+		await adapter.writeAppliedMigrationsState(state);
+		return;
+	}
+	writeAppliedMigrationsState(stateFilePath, state);
+};
 const isMigrationApplied = (state, identity, checksum) => {
 	const matched = state.migrations.find((migration) => migration.id === identity);
 	if (!matched) return false;
@@ -2667,7 +2703,9 @@ var MigrateCommand = class extends Command {
 		const classes = this.option("all") || !this.argument("name") ? await this.loadAllMigrations(migrationsDir) : (await this.loadNamedMigration(migrationsDir, this.argument("name"))).filter(([cls]) => cls !== void 0);
 		if (classes.length === 0) return void this.error("Error: No migration classes found to run.");
 		const stateFilePath = resolveMigrationStateFilePath(process.cwd(), this.option("state-file") ? String(this.option("state-file")) : void 0);
-		let appliedState = readAppliedMigrationsState(stateFilePath);
+		let appliedState = await readAppliedMigrationsStateFromStore(this.app.getConfig("adapter"), stateFilePath);
+		const adapter = this.app.getConfig("adapter");
+		const useDatabaseMigrations = supportsDatabaseMigrationExecution(adapter);
 		const skipped = [];
 		const changed = [];
 		const pending = classes.filter(([migrationClass, file]) => {
@@ -2689,10 +2727,16 @@ var MigrateCommand = class extends Command {
 			this.success("No pending migration classes to apply.");
 			return;
 		}
-		for (const [MigrationClassItem] of pending) await applyMigrationToPrismaSchema(MigrationClassItem, {
-			schemaPath,
-			write: true
-		});
+		for (const [MigrationClassItem] of pending) {
+			if (useDatabaseMigrations) {
+				await applyMigrationToDatabase(adapter, MigrationClassItem);
+				continue;
+			}
+			await applyMigrationToPrismaSchema(MigrationClassItem, {
+				schemaPath,
+				write: true
+			});
+		}
 		if (appliedState) {
 			const runAppliedIds = [];
 			for (const [migrationClass, file] of pending) {
@@ -2711,10 +2755,10 @@ var MigrateCommand = class extends Command {
 				appliedAt: (/* @__PURE__ */ new Date()).toISOString(),
 				migrationIds: runAppliedIds
 			});
-			writeAppliedMigrationsState(stateFilePath, appliedState);
+			await writeAppliedMigrationsStateToStore(adapter, stateFilePath, appliedState);
 		}
-		if (!this.option("skip-generate")) runPrismaCommand(["generate"], process.cwd());
-		if (!this.option("skip-migrate")) if (this.option("deploy")) runPrismaCommand(["migrate", "deploy"], process.cwd());
+		if (!useDatabaseMigrations && !this.option("skip-generate")) runPrismaCommand(["generate"], process.cwd());
+		if (!useDatabaseMigrations && !this.option("skip-migrate")) if (this.option("deploy")) runPrismaCommand(["migrate", "deploy"], process.cwd());
 		else runPrismaCommand([
 			"migrate",
 			"dev",
@@ -2774,6 +2818,85 @@ var MigrateCommand = class extends Command {
 	}
 };
 
+//#endregion
+//#region src/cli/commands/MigrateFreshCommand.ts
+var MigrateFreshCommand = class extends Command {
+	signature = `migrate:fresh
+        {--skip-generate : Skip prisma generate}
+        {--skip-migrate : Skip prisma database sync}
+        {--state-file= : Path to applied migration state file}
+        {--schema= : Explicit prisma schema path}
+    `;
+	description = "Reset the database and rerun all migration classes";
+	async handle() {
+		this.app.command = this;
+		const configuredMigrationsDir = this.app.getConfig("paths")?.migrations ?? join(process.cwd(), "database", "migrations");
+		const migrationsDir = this.app.resolveRuntimeDirectoryPath(configuredMigrationsDir);
+		if (!existsSync(migrationsDir)) return void this.error(`Error: Migrations directory not found: ${this.app.formatPathForLog(configuredMigrationsDir)}`);
+		const adapter = this.app.getConfig("adapter");
+		const useDatabaseMigrations = supportsDatabaseMigrationExecution(adapter);
+		const schemaPath = this.option("schema") ? resolve(String(this.option("schema"))) : join(process.cwd(), "prisma", "schema.prisma");
+		const stateFilePath = resolveMigrationStateFilePath(process.cwd(), this.option("state-file") ? String(this.option("state-file")) : void 0);
+		const migrations = await this.loadAllMigrations(migrationsDir);
+		if (migrations.length === 0) return void this.error("Error: No migration classes found to run.");
+		if (supportsDatabaseReset(adapter)) await adapter.resetDatabase();
+		else {
+			if (!existsSync(schemaPath)) return void this.error(`Error: Prisma schema file not found: ${this.app.formatPathForLog(schemaPath)}`);
+			writeFileSync(schemaPath, stripPrismaSchemaModelsAndEnums(readFileSync(schemaPath, "utf-8")));
+		}
+		let appliedState = createEmptyAppliedMigrationsState();
+		await writeAppliedMigrationsStateToStore(adapter, stateFilePath, appliedState);
+		for (const [MigrationClassItem] of migrations) {
+			if (useDatabaseMigrations) {
+				await applyMigrationToDatabase(adapter, MigrationClassItem);
+				continue;
+			}
+			await applyMigrationToPrismaSchema(MigrationClassItem, {
+				schemaPath,
+				write: true
+			});
+		}
+		for (const [migrationClass, file] of migrations) appliedState = markMigrationApplied(appliedState, {
+			id: buildMigrationIdentity(file, migrationClass.name),
+			file,
+			className: migrationClass.name,
+			appliedAt: (/* @__PURE__ */ new Date()).toISOString(),
+			checksum: computeMigrationChecksum(file)
+		});
+		appliedState = markMigrationRun(appliedState, {
+			id: buildMigrationRunId(),
+			appliedAt: (/* @__PURE__ */ new Date()).toISOString(),
+			migrationIds: appliedState.migrations.map((migration) => migration.id)
+		});
+		await writeAppliedMigrationsStateToStore(adapter, stateFilePath, appliedState);
+		if (!useDatabaseMigrations) {
+			const schemaArgs = this.option("schema") ? ["--schema", schemaPath] : [];
+			if (!this.option("skip-generate")) runPrismaCommand(["generate", ...schemaArgs], process.cwd());
+			if (!this.option("skip-migrate")) runPrismaCommand([
+				"db",
+				"push",
+				"--force-reset",
+				...schemaArgs
+			], process.cwd());
+		}
+		this.success(`Refreshed database with ${migrations.length} migration(s).`);
+		migrations.forEach(([_, file]) => this.success(this.app.splitLogger("Migrated", file)));
+	}
+	async loadAllMigrations(migrationsDir) {
+		const files = readdirSync(migrationsDir).filter((file) => /\.(ts|js|mjs|cjs)$/i.test(file)).sort((left, right) => left.localeCompare(right)).map((file) => this.app.resolveRuntimeScriptPath(join(migrationsDir, file)));
+		return (await Promise.all(files.map(async (file) => (await this.loadMigrationClassesFromFile(file)).map((cls) => [cls, file])))).flat();
+	}
+	async loadMigrationClassesFromFile(filePath) {
+		const imported = await RuntimeModuleLoader.load(filePath);
+		return Object.values(imported).filter((value) => {
+			if (typeof value !== "function") return false;
+			const candidate = value;
+			const prototype = candidate.prototype;
+			return candidate[MIGRATION_BRAND] === true || typeof prototype?.up === "function" && typeof prototype?.down === "function";
+		});
+	}
+};
+
 //#endregion
 //#region src/cli/commands/MigrateRollbackCommand.ts
 /**
@@ -2802,7 +2925,9 @@ var MigrateRollbackCommand = class extends Command {
 		if (!existsSync(migrationsDir)) return void this.error(`Error: Migrations directory not found: ${this.app.formatPathForLog(configuredMigrationsDir)}`);
 		const schemaPath = this.option("schema") ? resolve(String(this.option("schema"))) : join(process.cwd(), "prisma", "schema.prisma");
 		const stateFilePath = resolveMigrationStateFilePath(process.cwd(), this.option("state-file") ? String(this.option("state-file")) : void 0);
-		let appliedState = readAppliedMigrationsState(stateFilePath);
+		const adapter = this.app.getConfig("adapter");
+		const useDatabaseMigrations = supportsDatabaseMigrationExecution(adapter);
+		let appliedState = await readAppliedMigrationsStateFromStore(adapter, stateFilePath);
 		const stepOption = this.option("step");
 		const stepCount = stepOption == null ? void 0 : Number(stepOption);
 		if (stepCount != null && (!Number.isFinite(stepCount) || stepCount <= 0 || !Number.isInteger(stepCount))) return void this.error("Error: --step must be a positive integer.");
@@ -2824,17 +2949,23 @@ var MigrateRollbackCommand = class extends Command {
 			rollbackClasses.forEach(([_, file]) => this.success(this.app.splitLogger("WouldRollback", file)));
 			return;
 		}
-		for (const [MigrationClassItem] of rollbackClasses) await applyMigrationRollbackToPrismaSchema(MigrationClassItem, {
-			schemaPath,
-			write: true
-		});
+		for (const [MigrationClassItem] of rollbackClasses) {
+			if (useDatabaseMigrations) {
+				await applyMigrationRollbackToDatabase(adapter, MigrationClassItem);
+				continue;
+			}
+			await applyMigrationRollbackToPrismaSchema(MigrationClassItem, {
+				schemaPath,
+				write: true
+			});
+		}
 		for (const [migrationClass, file] of rollbackClasses) {
 			const identity = buildMigrationIdentity(file, migrationClass.name);
 			appliedState = removeAppliedMigration(appliedState, identity);
 		}
-		writeAppliedMigrationsState(stateFilePath, appliedState);
-		if (!this.option("skip-generate")) runPrismaCommand(["generate"], process.cwd());
-		if (!this.option("skip-migrate")) if (this.option("deploy")) runPrismaCommand(["migrate", "deploy"], process.cwd());
+		await writeAppliedMigrationsStateToStore(adapter, stateFilePath, appliedState);
+		if (!useDatabaseMigrations && !this.option("skip-generate")) runPrismaCommand(["generate"], process.cwd());
+		if (!useDatabaseMigrations && !this.option("skip-migrate")) if (this.option("deploy")) runPrismaCommand(["migrate", "deploy"], process.cwd());
 		else runPrismaCommand([
 			"migrate",
 			"dev",
@@ -2878,7 +3009,14 @@ var MigrationHistoryCommand = class extends Command {
 	async handle() {
 		this.app.command = this;
 		const stateFilePath = resolveMigrationStateFilePath(process.cwd(), this.option("state-file") ? String(this.option("state-file")) : void 0);
+		const adapter = this.app.getConfig("adapter");
+		const usesDatabaseState = supportsDatabaseMigrationState(adapter);
 		if (this.option("delete")) {
+			if (usesDatabaseState) {
+				await adapter.writeAppliedMigrationsState(createEmptyAppliedMigrationsState());
+				this.success("Deleted tracked migration state from database.");
+				return;
+			}
 			if (!existsSync(stateFilePath)) {
 				this.success(`No migration state file found at ${this.app.formatPathForLog(stateFilePath)}`);
 				return;
@@ -2888,22 +3026,19 @@ var MigrationHistoryCommand = class extends Command {
 			return;
 		}
 		if (this.option("reset")) {
-			writeAppliedMigrationsState(stateFilePath, {
-				version: 1,
-				migrations: []
-			});
-			this.success(`Reset migration state: ${this.app.formatPathForLog(stateFilePath)}`);
+			await writeAppliedMigrationsStateToStore(adapter, stateFilePath, createEmptyAppliedMigrationsState());
+			this.success(usesDatabaseState ? "Reset migration state in database." : `Reset migration state: ${this.app.formatPathForLog(stateFilePath)}`);
 			return;
 		}
-		const state = readAppliedMigrationsState(stateFilePath);
+		const state = await readAppliedMigrationsStateFromStore(adapter, stateFilePath);
 		if (this.option("json")) {
 			this.success(JSON.stringify({
-				path: stateFilePath,
+				path: usesDatabaseState ? "database" : stateFilePath,
 				...state
 			}, null, 2));
 			return;
 		}
-		this.success(this.app.splitLogger("State", stateFilePath));
+		this.success(this.app.splitLogger("State", usesDatabaseState ? "database" : stateFilePath));
 		this.success(this.app.splitLogger("Tracked", String(state.migrations.length)));
 		if (state.migrations.length === 0) {
 			this.success("No tracked migrations found.");
@@ -3096,6 +3231,7 @@ await Kernel.init(app, {
 		ModelsSyncCommand,
 		SeedCommand,
 		MigrateCommand,
+		MigrateFreshCommand,
 		MigrateRollbackCommand,
 		MigrationHistoryCommand
 	],