@zeno-lib/db 0.6.0 → 0.7.0-beta.0

package/dist/config.mjs

@@ -1,4 +1,41 @@
 import { defineConfig } from "drizzle-kit";
+import { existsSync, readdirSync, renameSync } from "node:fs";
+import { join } from "node:path";
+import { styleText } from "node:util";
+//#region src/move-generated-sql.ts
+const GENERATED_SQL = "migration.sql";
+/**
+* Move each `<name>/migration.sql` to `<name>.sql`,
+* because the Supabase CLI only reads `.sql` files at the top of the folder.
+* Throws if a `<name>.sql` already exists.
+*/
+function moveGeneratedSql(options) {
+	const { migrationsDir } = options;
+	const names = existsSync(migrationsDir) ? readdirSync(migrationsDir).filter((name) => existsSync(join(migrationsDir, name, GENERATED_SQL))).sort() : [];
+	const taken = names.map((name) => join(migrationsDir, `${name}.sql`)).filter((target) => existsSync(target));
+	if (taken.length > 0) throw new Error(`Cannot move the generated SQL, because these files already exist: ${taken.join(", ")}`);
+	return { moved: names.map((name) => {
+		const target = join(migrationsDir, `${name}.sql`);
+		renameSync(join(migrationsDir, name, GENERATED_SQL), target);
+		return target;
+	}) };
+}
+/** Run `moveGeneratedSql` when `drizzle-kit generate` exits. */
+function moveGeneratedSqlAfterGenerate(options) {
+	if (process.argv[2] !== "generate") return;
+	process.once("exit", (code) => {
+		if (code !== 0) return;
+		try {
+			const check = styleText("green", "✓", { stream: process.stderr });
+			for (const path of moveGeneratedSql(options).moved) console.error(`${check} Moved the generated SQL to ${path}`);
+		} catch (error) {
+			const message = error instanceof Error ? error.message : String(error);
+			console.error(styleText("red", `✗ ${message}`, { stream: process.stderr }));
+			process.exitCode = 1;
+		}
+	});
+}
+//#endregion
 //#region src/config.ts
 /**
 * Roles a Supabase project ships that drizzle-kit's `provider: "supabase"` does
@@ -43,7 +80,7 @@ const DEFAULT_SCHEMA_FILTER = ["public"];
 function defineDrizzleConfig(overrides = {}) {
 	const { entities, ...configOverrides } = overrides;
 	const roleOverrides = typeof entities?.roles === "object" ? entities.roles : {};
-	return defineConfig({
+	const config = defineConfig({
 		dbCredentials: { url: process.env.SUPABASE_DATABASE_URL ?? "" },
 		dialect: "postgresql",
 		entities: {
@@ -59,6 +96,8 @@ function defineDrizzleConfig(overrides = {}) {
 		schemaFilter: DEFAULT_SCHEMA_FILTER,
 		...configOverrides
 	});
+	if (config.out) moveGeneratedSqlAfterGenerate({ migrationsDir: config.out });
+	return config;
 }
 //#endregion
 export { defineDrizzleConfig, supabaseManagedRoles };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zeno-lib/db",
-  "version": "0.6.0",
+  "version": "0.7.0-beta.0",
   "type": "module",
   "exports": {
     ".": {
@@ -92,7 +92,6 @@
     "stop": "pnpm exec supabase stop",
     "reset": "pnpm exec supabase db reset",
     "db:generate": "drizzle-kit generate",
-    "db:migrate": "drizzle-kit migrate",
     "db:push": "drizzle-kit push",
     "db:studio": "drizzle-kit studio",
     "test": "vitest --run",

package/src/config.ts

@@ -1,6 +1,7 @@
 // https://orm.drizzle.team/docs/drizzle-config-file
 // https://orm.drizzle.team/docs/rls#migrations  (entities.roles.provider: "supabase")
 import { type Config, defineConfig } from "drizzle-kit"
+import { moveGeneratedSqlAfterGenerate } from "./move-generated-sql.ts"
 
 /**
  * Roles a Supabase project ships that drizzle-kit's `provider: "supabase"` does
@@ -54,7 +55,7 @@ export function defineDrizzleConfig(
   const roleOverrides =
     typeof entities?.roles === "object" ? entities.roles : {}
 
-  return defineConfig({
+  const config = defineConfig({
     dbCredentials: { url: process.env.SUPABASE_DATABASE_URL ?? "" },
     dialect: "postgresql",
     // Tells drizzle-kit that Supabase's built-in roles (anon, authenticated,
@@ -74,4 +75,10 @@ export function defineDrizzleConfig(
     schemaFilter: DEFAULT_SCHEMA_FILTER,
     ...configOverrides,
   } as Config)
+
+  if (config.out) {
+    moveGeneratedSqlAfterGenerate({ migrationsDir: config.out })
+  }
+
+  return config
 }

package/src/move-generated-sql.ts

@@ -0,0 +1,80 @@
+import { existsSync, readdirSync, renameSync } from "node:fs"
+import { join } from "node:path"
+import { styleText } from "node:util"
+
+export type MoveGeneratedSqlOptions = {
+  /** drizzle-kit writes migrations to this folder. */
+  migrationsDir: string
+}
+
+export type MoveGeneratedSqlResult = {
+  /** The new paths of the SQL files, sorted by name. */
+  moved: string[]
+}
+
+const GENERATED_SQL = "migration.sql"
+
+/**
+ * Move each `<name>/migration.sql` to `<name>.sql`,
+ * because the Supabase CLI only reads `.sql` files at the top of the folder.
+ * Throws if a `<name>.sql` already exists.
+ */
+export function moveGeneratedSql(
+  options: MoveGeneratedSqlOptions
+): MoveGeneratedSqlResult {
+  const { migrationsDir } = options
+
+  const names = existsSync(migrationsDir)
+    ? readdirSync(migrationsDir)
+        .filter((name) => existsSync(join(migrationsDir, name, GENERATED_SQL)))
+        .sort()
+    : []
+
+  const taken = names
+    .map((name) => join(migrationsDir, `${name}.sql`))
+    .filter((target) => existsSync(target))
+  if (taken.length > 0) {
+    throw new Error(
+      `Cannot move the generated SQL, because these files already exist: ${taken.join(", ")}`
+    )
+  }
+
+  const moved = names.map((name) => {
+    const target = join(migrationsDir, `${name}.sql`)
+    renameSync(join(migrationsDir, name, GENERATED_SQL), target)
+    return target
+  })
+
+  return { moved }
+}
+
+/** Run `moveGeneratedSql` when `drizzle-kit generate` exits. */
+export function moveGeneratedSqlAfterGenerate(
+  options: MoveGeneratedSqlOptions
+): void {
+  // drizzle-kit loads the config for every command.
+  // Only `generate` writes a migration.
+  if (process.argv[2] !== "generate") {
+    return
+  }
+
+  process.once("exit", (code) => {
+    if (code !== 0) {
+      return
+    }
+
+    // Print to stderr, so the JSON from `--output json` on stdout stays valid.
+    try {
+      const check = styleText("green", "✓", { stream: process.stderr })
+      for (const path of moveGeneratedSql(options).moved) {
+        console.error(`${check} Moved the generated SQL to ${path}`)
+      }
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error)
+      console.error(
+        styleText("red", `✗ ${message}`, { stream: process.stderr })
+      )
+      process.exitCode = 1
+    }
+  })
+}