@voltro/database 0.86.0 → 0.87.0

package/dist/migrate.js

@@ -403,9 +403,9 @@ $$ LANGUAGE plpgsql;
 	for (let n of e.stale) r.push(`DROP TRIGGER IF EXISTS ${re(n, t)} ON "${n}"`), r.push(`ALTER TABLE "${n}" REPLICA IDENTITY DEFAULT`);
 	return r;
 }, W = (e, t) => {
-	if (t !== "postgres") return;
+	if (t !== "postgres") return [];
 	let n = [];
-	return pe(e) && n.push("CREATE EXTENSION IF NOT EXISTS vector;"), me(e) && n.push("CREATE EXTENSION IF NOT EXISTS postgis;"), e.some((e) => e.appliedIndexes?.some((e) => e.kind === "gin" && r(e.kindOptions)?.split(".").at(-1) === "gin_trgm_ops")) && n.push("CREATE EXTENSION IF NOT EXISTS pg_trgm;"), n.length === 0 ? void 0 : n.join("\n");
+	return pe(e) && n.push("CREATE EXTENSION IF NOT EXISTS vector;"), me(e) && n.push("CREATE EXTENSION IF NOT EXISTS postgis;"), e.some((e) => e.appliedIndexes?.some((e) => e.kind === "gin" && r(e.kindOptions)?.split(".").at(-1) === "gin_trgm_ops")) && n.push("CREATE EXTENSION IF NOT EXISTS pg_trgm;"), n;
 }, pe = (e) => e.some((e) => Object.values(e.fields).some((e) => e.type === "vector")), me = (e) => e.some((e) => Object.values(e.fields).some((e) => e.spatial !== void 0)), G = (e) => {
 	let t = new Map(e.map((e) => [e.tableName, e])), n = v(e.map((e) => e.tableName), (e) => {
 		let n = t.get(e);
@@ -468,7 +468,7 @@ $$ LANGUAGE plpgsql;
 	}
 	return n;
 }, J = (e, t) => {
-	if (t !== "postgres") return "";
+	if (t !== "postgres") return [];
 	let n = /* @__PURE__ */ new Map();
 	for (let t of e) for (let e of Object.values(t.fields)) {
 		if (e.type !== "enum" || !e.enumName || !e.enumValues) continue;
@@ -476,15 +476,15 @@ $$ LANGUAGE plpgsql;
 		if (t && t.length !== e.enumValues.length) throw Error(`dbEnum '${e.enumName}' declared with conflicting value sets across tables. Define it once and re-use the handle.`);
 		n.set(e.enumName, e.enumValues);
 	}
-	if (n.size === 0) return "";
+	if (n.size === 0) return [];
 	let r = [];
 	for (let [e, t] of n) {
 		let n = t.map((e) => `'${e.replace(/'/g, "''")}'`).join(", ");
 		r.push(`DO $$ BEGIN\n  IF NOT EXISTS (SELECT 1 FROM pg_type WHERE typname = '${e.replace(/'/g, "''")}') THEN\n    CREATE TYPE "${e}" AS ENUM (${n});\n  END IF;\nEND $$;`);
 	}
-	return r.join("\n\n");
+	return r;
 }, Y = (e, t) => {
-	if (t !== "postgres") return "";
+	if (t !== "postgres") return [];
 	let r = /* @__PURE__ */ new Set();
 	for (let t of e) for (let e of Object.values(t.fields)) e.type === "enum" && e.enumName && r.add(e.enumName);
 	let i = [];
@@ -496,7 +496,7 @@ DECLARE
 BEGIN
   SELECT oid INTO type_oid FROM pg_type WHERE typname = '${n}';\n  IF type_oid IS NOT NULL\n     AND EXISTS (SELECT 1 FROM pg_enum WHERE enumtypid = type_oid AND enumlabel = '${a}')\n     AND NOT EXISTS (SELECT 1 FROM pg_enum WHERE enumtypid = type_oid AND enumlabel = '${o}') THEN\n    IF current_setting('server_version_num')::int >= 140000 THEN\n      EXECUTE format('ALTER TYPE %I RENAME VALUE %L TO %L', '${n}', '${a}', '${o}');\n    ELSE\n      UPDATE pg_enum SET enumlabel = '${o}' WHERE enumtypid = type_oid AND enumlabel = '${a}';\n    END IF;\n  END IF;\nEND $$;`);
 	}
-	return i.join("\n\n");
+	return i;
 }, he = (e) => {
 	let t = [], n = [];
 	for (let r of e) s(r) ? n.push(r) : t.push(r);
@@ -505,12 +505,8 @@ BEGIN
 		views: n
 	};
 }, X = (e, t, n = x, r = {}) => {
-	let i = I(t, r), { tables: o, views: s } = he(e), { sorted: l, cyclic: u } = G(o.map((e) => _(e, t))), d = q(u, t), f = [], p = W(l, t);
-	p && f.push(p);
-	let m = Y(l, t);
-	m && f.push(m);
-	let h = J(l, t);
-	h && f.push(h);
+	let i = I(t, r), { tables: o, views: s } = he(e), { sorted: l, cyclic: u } = G(o.map((e) => _(e, t))), d = q(u, t), f = [];
+	f.push(...W(l, t)), f.push(...Y(l, t)), f.push(...J(l, t));
 	for (let e of l) f.push(L(e, t, null, d.get(e.tableName), i));
 	if (!a(t)) for (let e of u) f.push(K(e, t));
 	for (let e of l) {
@@ -542,10 +538,8 @@ BEGIN
 	}
 	return p.join("\n\n") + "\n";
 }, ve = (e, t, n = x) => {
-	let { sorted: r, cyclic: i } = G(e), a = q(i, t), o = [], s = Y(r, t);
-	s && o.push(s);
-	let c = J(r, t);
-	c && o.push(c);
+	let { sorted: r, cyclic: i } = G(e), a = q(i, t), o = [];
+	o.push(...Y(r, t)), o.push(...J(r, t));
 	for (let e of r) o.push(L(e, t, null, a.get(e.tableName)));
 	for (let e of i) o.push(K(e, t));
 	return o.push(...V(r, t, n)), o.join("\n\n") + "\n";
@@ -589,21 +583,47 @@ BEGIN
 	let i = n === "turso" ? Te(t, n, r) : e.withTransaction(Te(t, n, r)), a = (e) => e instanceof Z ? e.error : e;
 	return (r !== void 0 && ee(n) === "plan-transactional" ? i.pipe(C.retry(we((e) => r(a(e)) === "retry"))) : i).pipe(C.catch((e) => C.die(e instanceof Z ? Error(g("migrate: statement failed", e.statement, e.error, 400).trim()) : e)));
 }, Ee = (e) => {
-	let t = [], n = "", r = !1;
-	for (let i = 0; i < e.length; i++) {
-		if (e.slice(i, i + 2) === "$$") {
-			r = !r, n += "$$", i += 1;
-			continue;
+	let t = [], n = 0, r = 0, i = (r) => {
+		let i = e.slice(n, r).trim();
+		i.length > 0 && t.push(i);
+	};
+	for (; r < e.length;) {
+		let t = e[r];
+		if (t === "'" || t === "\"" || t === "`") {
+			for (r++; r < e.length;) {
+				if (e[r] === t) {
+					if (e[r + 1] === t) {
+						r += 2;
+						continue;
+					}
+					break;
+				}
+				r++;
+			}
+			r++;
+		} else if (t === "-" && e[r + 1] === "-") {
+			let t = e.indexOf("\n", r);
+			r = t < 0 ? e.length : t + 1;
+		} else if (t === "/" && e[r + 1] === "*") {
+			let t = e.indexOf("*/", r + 2);
+			r = t < 0 ? e.length : t + 2;
+		} else {
+			if (t === "$") {
+				let t = /^\$[A-Za-z_]*\$/.exec(e.slice(r))?.[0];
+				if (t !== void 0) {
+					let n = e.indexOf(t, r + t.length);
+					r = n < 0 ? e.length : n + t.length;
+					continue;
+				}
+			}
+			if (t === ";") {
+				let t = e.slice(n, r);
+				(!/^\s*CREATE\s+(?:TEMP(?:ORARY)?\s+)?TRIGGER\b[\s\S]*\bBEGIN\b/i.test(t) || /\bEND\s*$/i.test(t)) && (i(r), n = r + 1);
+			}
+			r++;
 		}
-		let a = e[i], o = /^\s*CREATE\s+(?:TEMP(?:ORARY)?\s+)?TRIGGER\b[\s\S]*\bBEGIN\b/i.test(n) && !/\bEND\s*$/i.test(n);
-		if (a === ";" && !r && o) n += a;
-		else if (a === ";" && !r) {
-			let e = n.trim();
-			e.length > 0 && t.push(e), n = "";
-		} else n += a;
-	}
-	let i = n.trim();
-	return i.length > 0 && t.push(i), t;
+	}
+	return i(e.length), t;
 }, De = async (e, t, n = "postgres", r = {}) => {
 	await C.runPromise(ye(e, n, void 0, r).pipe(C.provide(t)));
 }, $ = (e, t, n = "postgres", r = {}) => C.gen(function* () {