@supabase/pg-delta 1.0.0-alpha.2 → 1.0.0-alpha.21

package/src/cli/exit-code.test.ts

@@ -0,0 +1,19 @@
+import { describe, expect, test } from "bun:test";
+import { getCommandExitCode, setCommandExitCode } from "./exit-code.ts";
+
+describe("exit-code", () => {
+  test("getCommandExitCode returns undefined when never set", () => {
+    expect(getCommandExitCode()).toBeUndefined();
+  });
+
+  test("setCommandExitCode then getCommandExitCode returns the value", () => {
+    setCommandExitCode(2);
+    expect(getCommandExitCode()).toBe(2);
+  });
+
+  test("setCommandExitCode overwrites previous value", () => {
+    setCommandExitCode(2);
+    setCommandExitCode(0);
+    expect(getCommandExitCode()).toBe(0);
+  });
+});

package/src/cli/exit-code.ts

@@ -0,0 +1,7 @@
+let _exitCode: number | undefined;
+export function setCommandExitCode(code: number) {
+  _exitCode = code;
+}
+export function getCommandExitCode() {
+  return _exitCode;
+}

package/src/cli/formatters/index.ts

@@ -0,0 +1,5 @@
+/**
+ * Formatters for displaying plans in various formats.
+ */
+
+export { formatTree } from "./tree/tree.ts";

package/src/cli/formatters/tree/tree-builder.ts

@@ -0,0 +1,380 @@
+/**
+ * Builds a generic tree structure from a HierarchicalPlan.
+ * Shows only structural changes (scope === "object"), grouped for readability.
+ */
+
+import {
+  AlterTableAddColumn,
+  AlterTableAlterColumnDropDefault,
+  AlterTableAlterColumnDropNotNull,
+  AlterTableAlterColumnSetDefault,
+  AlterTableAlterColumnSetNotNull,
+  AlterTableAlterColumnType,
+  AlterTableDropColumn,
+} from "../../../core/objects/table/changes/table.alter.ts";
+import type {
+  ChangeEntry,
+  ChangeGroup,
+  HierarchicalPlan,
+} from "../../../core/plan/index.ts";
+import { getObjectName } from "../../../core/plan/serialize.ts";
+import type { TreeGroup, TreeItem } from "./tree-renderer.ts";
+
+function withCount(name: string, count: number): string {
+  return count > 0 ? `${name} ${count}` : name;
+}
+
+function structural(group: ChangeGroup): ChangeGroup {
+  const onlyStructural = (entry: ChangeEntry) =>
+    entry.original.scope === "object";
+  return {
+    create: group.create.filter(onlyStructural),
+    alter: group.alter.filter(onlyStructural),
+    drop: group.drop.filter(onlyStructural),
+  };
+}
+
+function hasStructural(group: ChangeGroup): boolean {
+  const structuralGroup = structural(group);
+  return (
+    structuralGroup.create.length +
+      structuralGroup.alter.length +
+      structuralGroup.drop.length >
+    0
+  );
+}
+
+function symbol(group: ChangeGroup): string {
+  const g = structural(group);
+  if (g.create.length > 0) return "+";
+  if (g.alter.length > 0) return "~";
+  if (g.drop.length > 0) return "-";
+  return "";
+}
+
+function displayName(entry: ChangeEntry): string {
+  const { original } = entry;
+  if (
+    original instanceof AlterTableAddColumn ||
+    original instanceof AlterTableDropColumn ||
+    original instanceof AlterTableAlterColumnType ||
+    original instanceof AlterTableAlterColumnSetDefault ||
+    original instanceof AlterTableAlterColumnDropDefault ||
+    original instanceof AlterTableAlterColumnSetNotNull ||
+    original instanceof AlterTableAlterColumnDropNotNull
+  ) {
+    return original.column.name;
+  }
+  return getObjectName(original);
+}
+
+function toItems(group: ChangeGroup): TreeItem[] {
+  const items: TreeItem[] = [];
+  const s = structural(group);
+  for (const entry of s.create) {
+    items.push({ name: `+ ${displayName(entry)}` });
+  }
+  for (const entry of s.alter) {
+    items.push({ name: `~ ${displayName(entry)}` });
+  }
+  for (const entry of s.drop) {
+    items.push({ name: `- ${displayName(entry)}` });
+  }
+  return items;
+}
+
+function tableChildren(
+  table: HierarchicalPlan["schemas"][string]["tables"][string],
+): TreeGroup[] {
+  const groups: TreeGroup[] = [];
+
+  const pushGroup = (name: string, changeGroup: ChangeGroup) => {
+    if (hasStructural(changeGroup)) {
+      const items = toItems(changeGroup);
+      const label = items.length > 0 ? `${name} ${items.length}` : name;
+      groups.push({ name: label, items });
+    }
+  };
+
+  pushGroup("columns", table.columns);
+  pushGroup("indexes", table.indexes);
+  pushGroup("triggers", table.triggers);
+  pushGroup("rules", table.rules);
+  pushGroup("policies", table.policies);
+
+  const partitionNames = Object.keys(table.partitions).sort();
+  if (partitionNames.length > 0) {
+    const partitionGroups = partitionNames
+      .map((name) => {
+        const part = table.partitions[name];
+        const sym = tableSymbol(part);
+        const childGroups = tableChildren(part);
+        if (!hasStructural(part.changes) && childGroups.length === 0) {
+          return null;
+        }
+        return {
+          name: sym ? `${sym} ${name}` : name,
+          groups: childGroups,
+        };
+      })
+      .filter(Boolean) as TreeGroup[];
+    if (partitionGroups.length > 0) {
+      groups.push({
+        name: withCount("partitions", partitionGroups.length),
+        groups: partitionGroups,
+      });
+    }
+  }
+
+  return groups;
+}
+
+function tableSymbol(
+  table: HierarchicalPlan["schemas"][string]["tables"][string],
+): string {
+  const own = symbol(table.changes);
+  if (own) return own;
+  const childSymbols = [
+    symbol(table.columns),
+    symbol(table.indexes),
+    symbol(table.triggers),
+    symbol(table.rules),
+    symbol(table.policies),
+  ];
+  if (childSymbols.includes("+")) return "+";
+  if (childSymbols.includes("~")) return "~";
+  if (childSymbols.includes("-")) return "-";
+  for (const part of Object.values(table.partitions)) {
+    const s = tableSymbol(part);
+    if (s) return s;
+  }
+  return "";
+}
+
+function matviewSymbol(
+  mv: HierarchicalPlan["schemas"][string]["materializedViews"][string],
+): string {
+  const own = symbol(mv.changes);
+  if (own) return own;
+  const child = symbol(mv.indexes);
+  return child;
+}
+
+function matviewChildren(
+  mv: HierarchicalPlan["schemas"][string]["materializedViews"][string],
+): TreeGroup[] {
+  const groups: TreeGroup[] = [];
+  if (hasStructural(mv.indexes)) {
+    const items = toItems(mv.indexes);
+    groups.push({
+      name: items.length > 0 ? `indexes (${items.length})` : "indexes",
+      items,
+    });
+  }
+  return groups;
+}
+
+function buildCluster(cluster: HierarchicalPlan["cluster"]): TreeGroup[] {
+  const groups: Array<[string, ChangeGroup]> = [
+    ["roles", cluster.roles],
+    ["extensions", cluster.extensions],
+    ["event triggers", cluster.eventTriggers],
+    ["publications", cluster.publications],
+    ["subscriptions", cluster.subscriptions],
+    ["foreign-data-wrappers", cluster.foreignDataWrappers],
+    ["servers", cluster.servers],
+    ["user-mappings", cluster.userMappings],
+  ];
+
+  return groups
+    .filter(([, grp]) => hasStructural(grp))
+    .map(([name, grp]) => {
+      const items = toItems(grp);
+      return { name: withCount(name, items.length), items };
+    });
+}
+
+function buildSchema(schema: HierarchicalPlan["schemas"][string]): TreeGroup[] {
+  const groups: TreeGroup[] = [];
+
+  const pushItems = (name: string, changeGroup: ChangeGroup) => {
+    if (hasStructural(changeGroup)) {
+      const items = toItems(changeGroup);
+      const label = items.length > 0 ? `${name} ${items.length}` : name;
+      groups.push({ name: label, items });
+    }
+  };
+
+  const tableNames = Object.keys(schema.tables).sort();
+  if (tableNames.length > 0) {
+    const tableGroups = tableNames
+      .map((name) => {
+        const table = schema.tables[name];
+        const changeSymbol = tableSymbol(table);
+        const children = tableChildren(table);
+        if (!hasStructural(table.changes) && children.length === 0) return null;
+        return {
+          name: changeSymbol ? `${changeSymbol} ${name}` : name,
+          groups: children,
+        };
+      })
+      .filter(Boolean) as TreeGroup[];
+    if (tableGroups.length > 0) {
+      groups.push({
+        name: withCount("tables", tableGroups.length),
+        groups: tableGroups,
+      });
+    }
+  }
+
+  const viewNames = Object.keys(schema.views).sort();
+  if (viewNames.length > 0) {
+    const viewGroups = viewNames
+      .map((name) => {
+        const view = schema.views[name];
+        const changeSymbol = tableSymbol(view);
+        const children = tableChildren(view);
+        if (!hasStructural(view.changes) && children.length === 0) return null;
+        return {
+          name: changeSymbol ? `${changeSymbol} ${name}` : name,
+          groups: children,
+        };
+      })
+      .filter(Boolean) as TreeGroup[];
+    if (viewGroups.length > 0) {
+      groups.push({
+        name: withCount("views", viewGroups.length),
+        groups: viewGroups,
+      });
+    }
+  }
+
+  const mvNames = Object.keys(schema.materializedViews).sort();
+  if (mvNames.length > 0) {
+    const mvGroups = mvNames
+      .map((name) => {
+        const mv = schema.materializedViews[name];
+        const changeSymbol = matviewSymbol(mv);
+        const children = matviewChildren(mv);
+        if (!hasStructural(mv.changes) && children.length === 0) return null;
+        return {
+          name: changeSymbol ? `${changeSymbol} ${name}` : name,
+          groups: children,
+        };
+      })
+      .filter(Boolean) as TreeGroup[];
+    if (mvGroups.length > 0) {
+      groups.push({
+        name: withCount("materialized views", mvGroups.length),
+        groups: mvGroups,
+      });
+    }
+  }
+
+  pushItems("functions", schema.functions);
+  pushItems("procedures", schema.procedures);
+  pushItems("aggregates", schema.aggregates);
+  pushItems("sequences", schema.sequences);
+
+  const typeGroups: TreeGroup[] = [];
+  if (hasStructural(schema.types.enums)) {
+    const items = toItems(schema.types.enums);
+    typeGroups.push({
+      name: items.length > 0 ? `enums ${items.length}` : "enums",
+      items,
+    });
+  }
+  if (hasStructural(schema.types.composites)) {
+    const items = toItems(schema.types.composites);
+    typeGroups.push({
+      name:
+        items.length > 0
+          ? `composite types ${items.length}`
+          : "composite types",
+      items,
+    });
+  }
+  if (hasStructural(schema.types.ranges)) {
+    const items = toItems(schema.types.ranges);
+    typeGroups.push({
+      name: items.length > 0 ? `ranges ${items.length}` : "ranges",
+      items,
+    });
+  }
+  if (hasStructural(schema.types.domains)) {
+    const items = toItems(schema.types.domains);
+    typeGroups.push({
+      name: items.length > 0 ? `domains ${items.length}` : "domains",
+      items,
+    });
+  }
+  if (typeGroups.length > 0) {
+    groups.push({ name: `types ${typeGroups.length}`, groups: typeGroups });
+  }
+
+  pushItems("collations", schema.collations);
+
+  const ftNames = Object.keys(schema.foreignTables).sort();
+  if (ftNames.length > 0) {
+    const ftGroups = ftNames
+      .map((name) => {
+        const ft = schema.foreignTables[name];
+        const changeSymbol = tableSymbol(ft);
+        const children = tableChildren(ft);
+        if (!hasStructural(ft.changes) && children.length === 0) return null;
+        return {
+          name: changeSymbol ? `${changeSymbol} ${name}` : name,
+          groups: children,
+        };
+      })
+      .filter(Boolean) as TreeGroup[];
+    if (ftGroups.length > 0) {
+      groups.push({
+        name: withCount("foreign tables", ftGroups.length),
+        groups: ftGroups,
+      });
+    }
+  }
+
+  return groups;
+}
+
+/**
+ * Build a generic tree structure from a HierarchicalPlan.
+ */
+export function buildPlanTree(plan: HierarchicalPlan): TreeGroup {
+  const rootGroups: TreeGroup[] = [];
+
+  const clusterGroups = buildCluster(plan.cluster);
+  if (clusterGroups.length > 0) {
+    rootGroups.push({
+      name: withCount("cluster", clusterGroups.length),
+      groups: clusterGroups,
+    });
+  }
+
+  const schemaNames = Object.keys(plan.schemas).sort();
+  if (schemaNames.length > 0) {
+    const schemaGroups = schemaNames
+      .map((schemaName) => {
+        const schema = plan.schemas[schemaName];
+        const sym = symbol(schema.changes);
+        const children = buildSchema(schema);
+        if (!hasStructural(schema.changes) && children.length === 0) {
+          return null;
+        }
+        const label = sym ? `${sym} ${schemaName}` : schemaName;
+        return { name: label, groups: children };
+      })
+      .filter(Boolean) as TreeGroup[];
+
+    if (schemaGroups.length > 0) {
+      rootGroups.push({
+        name: withCount("schemas", schemaGroups.length),
+        groups: schemaGroups,
+      });
+    }
+  }
+
+  return { name: "Plan", groups: rootGroups };
+}