okmodel 0.3.0 → 0.5.0

package/dist/shared/wxn4b7pp.js

@@ -0,0 +1,252 @@
+import {
+  cellText,
+  VIEW_DEPENDENCIES,
+  viewDependencyEdges,
+  PUBLIC,
+  introspectSchema
+} from "./fancm5pt.js";
+import {
+  planMigration
+} from "./ayz64vzj.js";
+import {
+  quoteIdent,
+  createObjectSql
+} from "./8xsbagad.js";
+import {
+  normaliseViewQuery,
+  viewObject,
+  materializedViewObject
+} from "./y005b761.js";
+import {
+  catalog
+} from "./g7ez901n.js";
+import {
+  OkmError2
+} from "./52szzakm.js";
+
+// src/tooling/migrate/drift.ts
+import postgres from "postgres";
+
+// src/dialects/pg/view/scratch.ts
+var SCRATCH = "okm_view_seal";
+var VIEW_SHAPE = `
+  select c.relname as name, c.relkind as kind, pg_get_viewdef(c.oid, true) as query,
+    (
+      select coalesce(string_agg(
+        a.attname || '|' || format_type(a.atttypid, a.atttypmod),
+        E'\\n' order by a.attnum
+      ), '')
+      from pg_attribute a
+      where a.attrelid = c.oid and a.attnum > 0 and not a.attisdropped
+    ) as columns
+  from pg_class c
+  join pg_namespace n on n.oid = c.relnamespace
+  where n.nspname = $1 and c.relkind in ('v', 'm')
+`;
+async function sealViews(runner, source) {
+  const views = source.objects.filter((object) => object.kind === "view" || object.kind === "materializedView");
+  if (views.length === 0)
+    return source;
+  await dropScratch(runner);
+  await runner.query(`create schema ${quoteIdent(SCRATCH)}`);
+  try {
+    await runner.query(`set search_path to ${quoteIdent(SCRATCH)}`);
+    await createObjects(runner, source, "type");
+    for (const object of source.objects) {
+      if (object.kind !== "table")
+        continue;
+      await runner.query(stubTable(object, source));
+    }
+    await createObjects(runner, source, "function");
+    await createViews(runner, views);
+    const shapes = await runner.query(VIEW_SHAPE, [SCRATCH]);
+    const deps = await runner.query(VIEW_DEPENDENCIES, [SCRATCH]);
+    const edges = viewDependencyEdges(deps, PUBLIC, source.objects);
+    const byName = new Map(shapes.map((row) => [cellText(row.name), row]));
+    const next = source.objects.map((object) => {
+      if (object.kind !== "view" && object.kind !== "materializedView")
+        return object;
+      const row = byName.get(object.identity.name);
+      if (row === undefined) {
+        throw new OkmError2("OKM1020", `View ${object.identity.name} was not created in the scratch schema.`, {
+          fix: { summary: "The query has to run against the tables in this schema." }
+        });
+      }
+      const columns = readColumns(object, cellText(row.columns));
+      const query = normaliseViewQuery(cellText(row.query));
+      const dependencies = edges.get(object.identity.name) ?? [];
+      if (object.kind === "view") {
+        return viewObject({
+          namespace: object.identity.namespace,
+          name: object.identity.name,
+          columns,
+          query,
+          owner: object.owner,
+          provenance: object.provenance,
+          dependencies
+        });
+      }
+      return materializedViewObject({
+        namespace: object.identity.namespace,
+        name: object.identity.name,
+        columns,
+        query,
+        owner: object.owner,
+        provenance: object.provenance,
+        dependencies,
+        ...object.definition.refresh !== undefined ? { refresh: object.definition.refresh } : {}
+      });
+    });
+    return catalog(next);
+  } finally {
+    await runner.query("set search_path to public");
+    await dropScratch(runner);
+  }
+}
+async function createObjects(runner, source, kind) {
+  for (const object of source.objects) {
+    if (object.kind !== kind)
+      continue;
+    const statement = createObjectSql(object, SCRATCH);
+    if (statement !== undefined)
+      await runner.query(statement);
+  }
+}
+async function createViews(runner, views) {
+  const pending = [...views];
+  let last = "";
+  for (let pass = 0;pass < views.length && pending.length > 0; pass += 1) {
+    const failed = [];
+    let created = 0;
+    for (const object of pending) {
+      const sql = object.kind === "materializedView" ? `create materialized view ${quoteIdent(object.identity.name)} as ${object.definition.query} with no data` : `create view ${quoteIdent(object.identity.name)} as ${object.definition.query}`;
+      try {
+        await runner.query(sql);
+        created += 1;
+      } catch (error) {
+        last = error instanceof Error ? error.message : String(error);
+        failed.push(object);
+      }
+    }
+    if (created === 0) {
+      throw new OkmError2("OKM1020", `View ${failed[0]?.identity.name ?? ""} could not be created. ${last}`, {
+        fix: { summary: "The query has to run against the tables in this schema." }
+      });
+    }
+    pending.length = 0;
+    pending.push(...failed);
+  }
+}
+function stubTable(object, source) {
+  const columns = source.objects.filter((item) => item.kind === "column" && item.identity.parent.name === object.identity.name);
+  const lines = columns.map((column) => `${quoteIdent(column.identity.name)} ${column.definition.dataType}`);
+  return `create table ${quoteIdent(object.identity.name)} (${lines.join(", ")})`;
+}
+function readColumns(object, printed) {
+  const live = printed.length === 0 ? [] : printed.split(`
+`);
+  const declared = object.definition.columns;
+  if (live.length !== declared.length) {
+    throw mismatch(object.identity.name, declared.map((column) => column.name).join(", "), printed);
+  }
+  return declared.map((column, index) => {
+    const line = live[index] ?? "";
+    const splitAt = line.indexOf("|");
+    const name = splitAt < 0 ? line : line.slice(0, splitAt);
+    const type = splitAt < 0 ? "" : line.slice(splitAt + 1);
+    if (name !== column.name || type !== column.dataType) {
+      throw mismatch(object.identity.name, `${column.name}:${column.dataType}`, live[index] ?? "");
+    }
+    return column;
+  });
+}
+function mismatch(view, declared, live) {
+  return new OkmError2("OKM1020", `View ${view} declares ${declared} and the database has ${live}.`, { fix: { summary: "Declare the columns the query returns, in that order, with those types." } });
+}
+async function dropScratch(runner) {
+  const present = await runner.query(`select 1 as ok from pg_namespace where nspname = $1`, [
+    SCRATCH
+  ]);
+  if (present.length === 0)
+    return;
+  const rows = await runner.query(`
+    select c.relname as name, c.relkind as kind
+    from pg_class c
+    join pg_namespace n on n.oid = c.relnamespace
+    where n.nspname = $1 and c.relkind in ('v', 'm', 'r')
+  `, [SCRATCH]);
+  const views = rows.filter((row) => cellText(row.kind) === "v" || cellText(row.kind) === "m");
+  for (let pass = 0;pass < Math.max(views.length, 1); pass += 1) {
+    for (const row of views) {
+      const kind = cellText(row.kind) === "m" ? "materialized view" : "view";
+      try {
+        await runner.query(`drop ${kind} if exists ${quoteIdent(SCRATCH)}.${quoteIdent(cellText(row.name))}`);
+      } catch {}
+    }
+  }
+  for (const row of rows) {
+    if (cellText(row.kind) !== "r")
+      continue;
+    await runner.query(`drop table if exists ${quoteIdent(SCRATCH)}.${quoteIdent(cellText(row.name))}`);
+  }
+  await runner.query(`drop schema if exists ${quoteIdent(SCRATCH)} cascade`);
+}
+
+// src/tooling/migrate/drift.ts
+async function assertAuthorDrift2(url, author, roles) {
+  const sql = postgres(url, {
+    max: 1,
+    connect_timeout: 5,
+    idle_timeout: 1,
+    onnotice: () => {}
+  });
+  try {
+    const present = await sql`
+      select to_regclass('public.okm_meta')::text as reg
+    `;
+    if (present[0]?.reg == null)
+      return;
+    const runner = catalogQuery2(sql);
+    const authorOnServer = await sealViews(runner, author);
+    const live = await introspectSchema(runner, "public", "public", managedRoleOptions2(roles));
+    const plan = planMigration({ before: live, after: authorOnServer, name: "check" });
+    if (plan.steps.length === 0)
+      return;
+    throw new OkmError2("OKM1520", `The database differs from the schema.
+${plan.steps.map((step) => step.sql).join(`
+`)}`, {
+      fix: {
+        summary: "Change the schema or the database so okm check reports no difference."
+      }
+    });
+  } finally {
+    await sql.end({ timeout: 5 });
+  }
+}
+function catalogQuery2(sql) {
+  return {
+    async query(text, params) {
+      const rows = await sql.unsafe(text, params === undefined ? undefined : [...params]);
+      return rows.map((row) => {
+        const copy = {};
+        for (const [key, value] of Object.entries(row))
+          copy[key] = value;
+        return copy;
+      });
+    }
+  };
+}
+function managedRoleOptions2(roles) {
+  const names = managedRoleNames(roles);
+  return names === undefined ? undefined : { managedRoles: names };
+}
+function managedRoleNames(roles) {
+  if (roles === undefined)
+    return;
+  const names = new Set([roles.migration, roles.app]);
+  for (const role of roles.managed ?? [])
+    names.add(role.name);
+  return [...names];
+}
+
+export { sealViews, assertAuthorDrift2, catalogQuery2, managedRoleOptions2 };