@supabase/pg-delta 1.0.0-alpha.3 → 1.0.0-alpha.30

package/dist/core/declarative-apply/extract-catalog-providers.js

@@ -0,0 +1,159 @@
+/**
+ * Extract functions, types, and other catalog objects from the target database
+ * so pg-topo can treat them as external providers and suppress false
+ * UNRESOLVED_DEPENDENCY diagnostics (e.g. now(), gen_random_uuid(), nextval(),
+ * auth.users, extensions.uuid_generate_v4, etc.).
+ */
+function addProvider(providers, ref, alsoUnderPublic = false) {
+    providers.push(ref);
+    if (alsoUnderPublic &&
+        (ref.schema === "pg_catalog" || ref.schema === "information_schema")) {
+        providers.push({ ...ref, schema: "public" });
+    }
+}
+/**
+ * Query the target database for all catalog objects that can be dependencies.
+ * Returns ObjectRefs that pg-topo can use as external providers so it does
+ * not flag false UNRESOLVED_DEPENDENCY diagnostics (e.g. now(), text, public).
+ *
+ * This is intentionally separate from {@link extractCatalog} in catalog.model.ts:
+ * - extractCatalog is for schema diffing and excludes system objects (filters
+ *   out pg_catalog, information_schema, extension-owned). It also does not
+ *   extract languages. We need the opposite here: all objects that might be
+ *   referenced by SQL, including built-in functions, types, and schemas.
+ * - Objects in pg_catalog/information_schema are registered under both their
+ *   real schema and "public" (via addProvider's alsoUnderPublic) so
+ *   unqualified references in SQL resolve the same way the parser does.
+ */
+export async function extractCatalogProviders(pool) {
+    const providers = [];
+    const [functionsResult, typesResult, schemasResult, relationsResult, extensionsResult, rolesResult, languagesResult, collationsResult, fdwsResult, serversResult, eventTriggersResult, publicationsResult, subscriptionsResult,] = await Promise.all([
+        pool.query(`
+      SELECT
+        p.proname AS name,
+        n.nspname AS schema,
+        p.prokind AS kind,
+        COALESCE((
+          SELECT string_agg(
+            CASE WHEN p.proargmodes IS NOT NULL AND p.proargmodes[ord] = 'v'
+                 THEN 'VARIADIC ' || format_type(t.oid, NULL)
+                 ELSE format_type(t.oid, NULL)
+            END, ',' ORDER BY ord)
+          FROM unnest(p.proargtypes) WITH ORDINALITY AS t(oid, ord)
+        ), '') AS signature
+      FROM pg_proc p
+      JOIN pg_namespace n ON n.oid = p.pronamespace
+    `),
+        pool.query(`
+      SELECT t.typname AS name, n.nspname AS schema, t.typtype AS typetype
+      FROM pg_type t
+      JOIN pg_namespace n ON n.oid = t.typnamespace
+      WHERE n.nspname NOT LIKE 'pg_toast%'
+        AND t.typtype IN ('b', 'c', 'd', 'e', 'r')
+    `),
+        pool.query(`
+      SELECT nspname AS name FROM pg_namespace
+      WHERE nspname NOT LIKE 'pg_toast%'
+    `),
+        pool.query(`
+      SELECT c.relname AS name, n.nspname AS schema, c.relkind AS relkind
+      FROM pg_class c
+      JOIN pg_namespace n ON n.oid = c.relnamespace
+      WHERE c.relkind IN ('r', 'p', 'v', 'm', 'S', 'i')
+        AND n.nspname NOT LIKE 'pg_toast%'
+    `),
+        pool.query(`
+      SELECT e.extname AS name, n.nspname AS schema
+      FROM pg_extension e
+      LEFT JOIN pg_namespace n ON n.oid = e.extnamespace
+    `),
+        pool.query(`SELECT rolname AS name FROM pg_roles`),
+        pool.query(`SELECT lanname AS name FROM pg_language`),
+        pool.query(`
+      SELECT c.collname AS name, n.nspname AS schema
+      FROM pg_collation c
+      JOIN pg_namespace n ON n.oid = c.collnamespace
+      WHERE n.nspname NOT LIKE 'pg_toast%'
+    `),
+        pool.query(`SELECT fdwname AS name FROM pg_foreign_data_wrapper`),
+        pool.query(`SELECT srvname AS name FROM pg_foreign_server`),
+        pool.query(`SELECT evtname AS name FROM pg_event_trigger`),
+        pool.query(`SELECT pubname AS name FROM pg_publication`),
+        pool.query(`SELECT subname AS name FROM pg_subscription`),
+    ]);
+    for (const fn of functionsResult.rows) {
+        const kind = fn.kind === "a"
+            ? "aggregate"
+            : fn.kind === "p"
+                ? "procedure"
+                : "function";
+        const sig = fn.signature.trim() ? `(${fn.signature})` : "()";
+        const ref = {
+            kind,
+            name: fn.name,
+            schema: fn.schema,
+            signature: sig,
+        };
+        addProvider(providers, ref, true);
+    }
+    for (const t of typesResult.rows) {
+        const kind = t.typetype === "d" ? "domain" : "type";
+        const ref = { kind, name: t.name, schema: t.schema };
+        addProvider(providers, ref, true);
+    }
+    for (const row of schemasResult.rows) {
+        providers.push({ kind: "schema", name: row.name });
+    }
+    const relkindToKind = {
+        r: "table",
+        p: "table",
+        v: "view",
+        m: "materialized_view",
+        S: "sequence",
+        i: "index",
+    };
+    for (const row of relationsResult.rows) {
+        const kind = relkindToKind[row.relkind];
+        if (!kind)
+            continue;
+        const ref = { kind, name: row.name, schema: row.schema };
+        addProvider(providers, ref, true);
+    }
+    for (const row of extensionsResult.rows) {
+        providers.push({
+            kind: "extension",
+            name: row.name,
+            schema: row.schema ?? undefined,
+        });
+    }
+    for (const row of rolesResult.rows) {
+        providers.push({ kind: "role", name: row.name });
+    }
+    for (const row of languagesResult.rows) {
+        providers.push({ kind: "language", name: row.name });
+    }
+    for (const row of collationsResult.rows) {
+        const ref = {
+            kind: "collation",
+            name: row.name,
+            schema: row.schema,
+        };
+        addProvider(providers, ref, true);
+    }
+    for (const row of fdwsResult.rows) {
+        providers.push({ kind: "foreign_data_wrapper", name: row.name });
+    }
+    for (const row of serversResult.rows) {
+        providers.push({ kind: "foreign_server", name: row.name });
+    }
+    for (const row of eventTriggersResult.rows) {
+        providers.push({ kind: "event_trigger", name: row.name });
+    }
+    for (const row of publicationsResult.rows) {
+        providers.push({ kind: "publication", name: row.name });
+    }
+    for (const row of subscriptionsResult.rows) {
+        providers.push({ kind: "subscription", name: row.name });
+    }
+    return providers;
+}

package/dist/core/declarative-apply/index.d.ts

@@ -0,0 +1,49 @@
+/**
+ * Declarative schema apply – orchestrator.
+ *
+ * Accepts pre-read SQL content (file path + sql string per file), uses pg-topo
+ * for static dependency analysis and topological ordering, then applies
+ * statements to a target database using iterative rounds to handle any
+ * remaining dependency gaps. File discovery and reading are done by the caller
+ * (e.g. CLI) so I/O errors can be handled there.
+ */
+import type { Diagnostic } from "@supabase/pg-topo";
+import type { Pool } from "pg";
+import { type ApplyResult, type RoundResult } from "./round-apply.ts";
+import type { SqlFileEntry } from "./discover-sql.ts";
+interface DeclarativeApplyOptions {
+    /** Pre-read SQL files: filePath (relative) and sql content. Caller does discovery and read. */
+    content: SqlFileEntry[];
+    /** Target database connection URL (required if pool is not provided) */
+    targetUrl?: string;
+    /** Existing pool to use (caller owns it; not closed). If provided, targetUrl is ignored. */
+    pool?: Pool;
+    /** Max rounds before giving up (default: 100) */
+    maxRounds?: number;
+    /** Run final function body validation (default: true) */
+    validateFunctionBodies?: boolean;
+    /** Disable function body checks during rounds (default: true) */
+    disableCheckFunctionBodies?: boolean;
+    /** Progress callback fired after each round */
+    onRoundComplete?: (round: RoundResult) => void;
+}
+export interface DeclarativeApplyResult {
+    /** Result from the round-based apply engine */
+    apply: ApplyResult;
+    /** Diagnostics from pg-topo's static analysis (warnings, not fatal) */
+    diagnostics: Diagnostic[];
+    /** Total number of statements discovered */
+    totalStatements: number;
+}
+/**
+ * Apply a declarative SQL schema to a target database.
+ *
+ * 1. Call pg-topo analyzeAndSort on the provided SQL strings
+ * 2. Remap synthetic statement IDs to caller-provided file paths
+ * 3. Apply statements round-by-round to the target database
+ * 4. Optionally validate function bodies in a final pass
+ */
+export declare function applyDeclarativeSchema(options: DeclarativeApplyOptions): Promise<DeclarativeApplyResult>;
+export type { SqlFileEntry } from "./discover-sql.ts";
+export { loadDeclarativeSchema } from "./discover-sql.ts";
+export type { ApplyResult, RoundResult } from "./round-apply.ts";

package/dist/core/declarative-apply/index.js

@@ -0,0 +1,134 @@
+/**
+ * Declarative schema apply – orchestrator.
+ *
+ * Accepts pre-read SQL content (file path + sql string per file), uses pg-topo
+ * for static dependency analysis and topological ordering, then applies
+ * statements to a target database using iterative rounds to handle any
+ * remaining dependency gaps. File discovery and reading are done by the caller
+ * (e.g. CLI) so I/O errors can be handled there.
+ */
+import { analyzeAndSort } from "@supabase/pg-topo";
+import { createManagedPool } from "../postgres-config.js";
+import { extractCatalogProviders } from "./extract-catalog-providers.js";
+import { roundApply, } from "./round-apply.js";
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+/**
+ * Convert pg-topo StatementNodes into StatementEntries for the apply engine.
+ */
+function toStatementEntries(nodes) {
+    return nodes.map((node) => ({
+        id: `${node.id.filePath}:${node.id.statementIndex}`,
+        sql: node.sql,
+        statementClass: node.statementClass,
+    }));
+}
+function remapStatementId(statementId, filePathMap) {
+    if (!statementId)
+        return undefined;
+    return {
+        ...statementId,
+        filePath: filePathMap.get(statementId.filePath) ?? statementId.filePath,
+    };
+}
+// ---------------------------------------------------------------------------
+// Main entry point
+// ---------------------------------------------------------------------------
+/**
+ * Apply a declarative SQL schema to a target database.
+ *
+ * 1. Call pg-topo analyzeAndSort on the provided SQL strings
+ * 2. Remap synthetic statement IDs to caller-provided file paths
+ * 3. Apply statements round-by-round to the target database
+ * 4. Optionally validate function bodies in a final pass
+ */
+export async function applyDeclarativeSchema(options) {
+    const { content, targetUrl, pool: providedPool, maxRounds = 100, validateFunctionBodies = true, disableCheckFunctionBodies = true, onRoundComplete, } = options;
+    if (content.length === 0) {
+        return {
+            apply: {
+                status: "success",
+                totalRounds: 0,
+                totalApplied: 0,
+                totalSkipped: 0,
+                rounds: [],
+            },
+            diagnostics: [],
+            totalStatements: 0,
+        };
+    }
+    let pool;
+    let closePool;
+    if (providedPool != null) {
+        pool = providedPool;
+    }
+    else if (targetUrl != null) {
+        const managed = await createManagedPool(targetUrl, { label: "target" });
+        pool = managed.pool;
+        closePool = managed.close;
+    }
+    else {
+        throw new Error("Either targetUrl or pool must be provided");
+    }
+    try {
+        const externalProviders = await extractCatalogProviders(pool);
+        // Step 1: pg-topo analyze and sort (no file I/O; uses synthetic <input:i> paths)
+        const sqlContents = content.map((entry) => entry.sql);
+        const analyzeResult = await analyzeAndSort(sqlContents, {
+            externalProviders,
+        });
+        const { ordered, diagnostics } = analyzeResult;
+        // Step 2: Remap <input:i> to real file paths
+        const filePathMap = new Map();
+        for (let i = 0; i < content.length; i += 1) {
+            filePathMap.set(`<input:${i}>`, content[i].filePath);
+        }
+        const remappedOrdered = ordered.map((node) => ({
+            ...node,
+            id: {
+                ...node.id,
+                filePath: filePathMap.get(node.id.filePath) ?? node.id.filePath,
+            },
+        }));
+        const remappedDiagnostics = diagnostics.map((d) => ({
+            ...d,
+            statementId: remapStatementId(d.statementId, filePathMap),
+        }));
+        if (ordered.length === 0) {
+            return {
+                apply: {
+                    status: "success",
+                    totalRounds: 0,
+                    totalApplied: 0,
+                    totalSkipped: 0,
+                    rounds: [],
+                },
+                diagnostics: remappedDiagnostics,
+                totalStatements: 0,
+            };
+        }
+        // Step 3: Convert to statement entries and apply
+        const statements = toStatementEntries(remappedOrdered);
+        const applyResult = await roundApply({
+            pool,
+            statements,
+            maxRounds,
+            disableCheckFunctionBodies,
+            finalValidation: validateFunctionBodies,
+            onRoundComplete,
+        });
+        return {
+            apply: applyResult,
+            diagnostics: remappedDiagnostics,
+            totalStatements: remappedOrdered.length,
+        };
+    }
+    finally {
+        if (closePool) {
+            await closePool();
+        }
+    }
+}
+// Re-export file discovery for programmatic callers (e.g. Supabase CLI edge-runtime templates)
+export { loadDeclarativeSchema } from "./discover-sql.js";

package/dist/core/declarative-apply/round-apply.d.ts

@@ -0,0 +1,100 @@
+/**
+ * Round-based declarative schema apply engine.
+ *
+ * Applies SQL statements to a database using iterative rounds:
+ * 1. Try each pending statement
+ * 2. On dependency errors, defer to next round
+ * 3. Repeat until all applied or no progress (stuck)
+ * 4. Optional final validation pass for function bodies
+ */
+import type { Pool } from "pg";
+export interface StatementEntry {
+    /** Unique identifier for the statement (e.g. "file:index") */
+    id: string;
+    /** The SQL to execute */
+    sql: string;
+    /** Optional statement classification (e.g. CREATE_FUNCTION) */
+    statementClass?: string;
+}
+export interface StatementError {
+    /** Statement that failed */
+    statement: StatementEntry;
+    /** PostgreSQL error code (SQLSTATE) */
+    code: string;
+    /** Human-readable error message */
+    message: string;
+    /** Whether this was classified as a dependency error */
+    isDependencyError: boolean;
+    /** 1-based character offset in the statement SQL where the error occurred */
+    position?: number;
+    /** PostgreSQL error detail (e.g. token invalid) */
+    detail?: string;
+    /** PostgreSQL hint */
+    hint?: string;
+}
+export interface RoundResult {
+    /** Round number (1-based) */
+    round: number;
+    /** Number of statements successfully applied this round */
+    applied: number;
+    /** Number of statements deferred to next round */
+    deferred: number;
+    /** Number of statements that failed with non-dependency errors */
+    failed: number;
+    /** Errors encountered this round */
+    errors: StatementError[];
+}
+export interface ApplyResult {
+    /** Overall status */
+    status: "success" | "stuck" | "error";
+    /** Total number of rounds executed */
+    totalRounds: number;
+    /** Total number of statements successfully applied */
+    totalApplied: number;
+    /** Total number of statements skipped (environment/capability errors) */
+    totalSkipped: number;
+    /** Statements that could not be applied (stuck) */
+    stuckStatements?: StatementError[];
+    /** Non-dependency errors that caused hard failures */
+    errors?: StatementError[];
+    /** Errors from the final function body validation pass */
+    validationErrors?: StatementError[];
+    /** Per-round results */
+    rounds: RoundResult[];
+}
+interface RoundApplyOptions {
+    /** Target database pool */
+    pool: Pool;
+    /** Ordered SQL statements to apply */
+    statements: StatementEntry[];
+    /** Max rounds before giving up (default: 100) */
+    maxRounds?: number;
+    /** Disable function body checks during application (default: true) */
+    disableCheckFunctionBodies?: boolean;
+    /** Run final validation with check_function_bodies=on (default: true) */
+    finalValidation?: boolean;
+    /** Progress callback fired after each round */
+    onRoundComplete?: (round: RoundResult) => void;
+}
+/**
+ * Apply SQL statements to a database using iterative rounds.
+ *
+ * Algorithm:
+ * 1. Optionally set check_function_bodies = off
+ * 2. For each round, iterate over pending statements:
+ *    - On success: mark as applied
+ *    - On dependency error: defer to next round
+ *    - On environment error: skip permanently with warning
+ *    - On other error: mark as failed
+ * 3. If a round makes no progress (0 applied), stop (stuck)
+ * 4. If finalValidation is true, re-run CREATE FUNCTION/PROCEDURE
+ *    with check_function_bodies = on
+ */
+export declare function roundApply(options: RoundApplyOptions): Promise<ApplyResult>;
+/**
+ * Rewrite a CREATE FUNCTION/PROCEDURE statement to use OR REPLACE for
+ * idempotent re-execution during validation. Handles leading line-comments,
+ * block comments (pg-topo annotations), and avoids double-adding OR REPLACE.
+ */
+export declare function rewriteAsOrReplace(sql: string): string;
+export {};