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

package/src/core/objects/rls-policy/changes/rls-policy.alter.test.ts

@@ -0,0 +1,250 @@
+import { describe, expect, test } from "bun:test";
+import { diffRlsPolicies } from "../rls-policy.diff.ts";
+import { RlsPolicy, type RlsPolicyProps } from "../rls-policy.model.ts";
+import {
+  AlterRlsPolicySetRoles,
+  AlterRlsPolicySetUsingExpression,
+  AlterRlsPolicySetWithCheckExpression,
+} from "./rls-policy.alter.ts";
+import { CreateRlsPolicy } from "./rls-policy.create.ts";
+import { DropRlsPolicy } from "./rls-policy.drop.ts";
+
+describe.concurrent("rls-policy", () => {
+  describe("alter", () => {
+    test("change roles", () => {
+      const props: Omit<RlsPolicyProps, "roles"> = {
+        schema: "public",
+        name: "test_policy",
+        table_name: "test_table",
+        command: "r",
+        permissive: true,
+        using_expression: "user_id = current_user_id()",
+        with_check_expression: null,
+        owner: "owner",
+        comment: null,
+      };
+      const policy = new RlsPolicy({
+        ...props,
+        roles: ["public"],
+      });
+
+      const change = new AlterRlsPolicySetRoles({
+        policy,
+        roles: ["role1", "role2"],
+      });
+
+      expect(change.serialize()).toBe(
+        "ALTER POLICY public.test_policy ON public.test_table TO role1, role2",
+      );
+    });
+
+    test("change roles to PUBLIC (default)", () => {
+      const props: Omit<RlsPolicyProps, "roles"> = {
+        schema: "public",
+        name: "test_policy",
+        table_name: "test_table",
+        command: "r",
+        permissive: true,
+        using_expression: "expr",
+        with_check_expression: null,
+        owner: "owner",
+        comment: null,
+      };
+      const policy = new RlsPolicy({
+        ...props,
+        roles: ["role1"],
+      });
+
+      const change = new AlterRlsPolicySetRoles({
+        policy,
+        roles: ["public"],
+      });
+
+      expect(change.serialize()).toBe(
+        "ALTER POLICY public.test_policy ON public.test_table TO PUBLIC",
+      );
+    });
+
+    test("drop + create rls policy when command changes", () => {
+      const props: Omit<RlsPolicyProps, "command"> = {
+        schema: "public",
+        name: "test_policy",
+        table_name: "test_table",
+        permissive: true,
+        roles: ["public"],
+        using_expression: "user_id = current_user_id()",
+        with_check_expression: null,
+        owner: "owner",
+        comment: null,
+      };
+      const main = new RlsPolicy({
+        ...props,
+        command: "r", // SELECT
+      });
+      const branch = new RlsPolicy({
+        ...props,
+        command: "w", // UPDATE
+      });
+
+      const changes = diffRlsPolicies(
+        { [main.stableId]: main },
+        { [branch.stableId]: branch },
+      );
+
+      expect(changes).toHaveLength(2);
+      expect(changes[0]).toBeInstanceOf(DropRlsPolicy);
+      expect(changes[1]).toBeInstanceOf(CreateRlsPolicy);
+      expect(changes[0].serialize()).toBe(
+        "DROP POLICY test_policy ON public.test_table",
+      );
+      expect(changes[1].serialize()).toBe(
+        "CREATE POLICY test_policy ON public.test_table FOR UPDATE USING (user_id = current_user_id())",
+      );
+    });
+
+    test("drop + create rls policy when permissive changes", () => {
+      const props: Omit<RlsPolicyProps, "permissive"> = {
+        schema: "public",
+        name: "test_policy",
+        table_name: "test_table",
+        command: "r",
+        roles: ["public"],
+        using_expression: "user_id = current_user_id()",
+        with_check_expression: null,
+        owner: "owner",
+        comment: null,
+      };
+      const main = new RlsPolicy({
+        ...props,
+        permissive: true,
+      });
+      const branch = new RlsPolicy({
+        ...props,
+        permissive: false,
+      });
+
+      const changes = diffRlsPolicies(
+        { [main.stableId]: main },
+        { [branch.stableId]: branch },
+      );
+
+      expect(changes).toHaveLength(2);
+      expect(changes[0]).toBeInstanceOf(DropRlsPolicy);
+      expect(changes[1]).toBeInstanceOf(CreateRlsPolicy);
+      expect(changes[0].serialize()).toBe(
+        "DROP POLICY test_policy ON public.test_table",
+      );
+      expect(changes[1].serialize()).toBe(
+        "CREATE POLICY test_policy ON public.test_table AS RESTRICTIVE FOR SELECT USING (user_id = current_user_id())",
+      );
+    });
+
+    test("alter using expression", () => {
+      const props: Omit<RlsPolicyProps, "using_expression"> = {
+        schema: "public",
+        name: "test_policy",
+        table_name: "test_table",
+        command: "r",
+        permissive: true,
+        roles: ["public"],
+        with_check_expression: null,
+        owner: "test",
+        comment: null,
+      };
+      const policy = new RlsPolicy({
+        ...props,
+        using_expression: "old_expr",
+      });
+
+      const change = new AlterRlsPolicySetUsingExpression({
+        policy,
+        usingExpression: "new_expr",
+      });
+
+      expect(change.serialize()).toBe(
+        "ALTER POLICY public.test_policy ON public.test_table USING (new_expr)",
+      );
+    });
+
+    test("clear using expression -> USING (true)", () => {
+      const props: Omit<RlsPolicyProps, "using_expression"> = {
+        schema: "public",
+        name: "test_policy",
+        table_name: "test_table",
+        command: "r",
+        permissive: true,
+        roles: ["public"],
+        with_check_expression: null,
+        owner: "test",
+        comment: null,
+      };
+      const policy = new RlsPolicy({
+        ...props,
+        using_expression: "old_expr",
+      });
+
+      const change = new AlterRlsPolicySetUsingExpression({
+        policy,
+        usingExpression: null,
+      });
+
+      expect(change.serialize()).toBe(
+        "ALTER POLICY public.test_policy ON public.test_table USING (true)",
+      );
+    });
+
+    test("alter with check expression", () => {
+      const props: Omit<RlsPolicyProps, "with_check_expression"> = {
+        schema: "public",
+        name: "test_policy",
+        table_name: "test_table",
+        command: "r",
+        permissive: true,
+        roles: ["public"],
+        using_expression: "expr",
+        owner: "test",
+        comment: null,
+      };
+      const policy = new RlsPolicy({
+        ...props,
+        with_check_expression: "old_check",
+      });
+
+      const change = new AlterRlsPolicySetWithCheckExpression({
+        policy,
+        withCheckExpression: "new_check",
+      });
+
+      expect(change.serialize()).toBe(
+        "ALTER POLICY public.test_policy ON public.test_table WITH CHECK (new_check)",
+      );
+    });
+
+    test("clear with check expression -> WITH CHECK (true)", () => {
+      const props: Omit<RlsPolicyProps, "with_check_expression"> = {
+        schema: "public",
+        name: "test_policy",
+        table_name: "test_table",
+        command: "r",
+        permissive: true,
+        roles: ["public"],
+        using_expression: "expr",
+        owner: "test",
+        comment: null,
+      };
+      const policy = new RlsPolicy({
+        ...props,
+        with_check_expression: "old_check",
+      });
+
+      const change = new AlterRlsPolicySetWithCheckExpression({
+        policy,
+        withCheckExpression: null,
+      });
+
+      expect(change.serialize()).toBe(
+        "ALTER POLICY public.test_policy ON public.test_table WITH CHECK (true)",
+      );
+    });
+  });
+});

package/src/core/objects/rls-policy/changes/rls-policy.alter.ts

@@ -0,0 +1,128 @@
+import type { RlsPolicy } from "../rls-policy.model.ts";
+import { AlterRlsPolicyChange } from "./rls-policy.base.ts";
+
+/**
+ * Alter an RLS policy.
+ *
+ * @see https://www.postgresql.org/docs/17/sql-alterpolicy.html
+ *
+ * Synopsis
+ * ```sql
+ * ALTER POLICY name ON table_name
+ *     [ TO { role_name | PUBLIC | CURRENT_ROLE | CURRENT_USER | SESSION_USER } [, ...] ]
+ *     [ USING ( using_expression ) ]
+ *     [ WITH CHECK ( with_check_expression ) ]
+ * ```
+ */
+
+export type AlterRlsPolicy =
+  | AlterRlsPolicySetRoles
+  | AlterRlsPolicySetUsingExpression
+  | AlterRlsPolicySetWithCheckExpression;
+
+/**
+ * ALTER POLICY ... TO roles ...
+ */
+export class AlterRlsPolicySetRoles extends AlterRlsPolicyChange {
+  public readonly policy: RlsPolicy;
+  public readonly roles: string[];
+  public readonly scope = "object" as const;
+
+  constructor(props: { policy: RlsPolicy; roles: string[] }) {
+    super();
+    this.policy = props.policy;
+    this.roles = props.roles;
+  }
+
+  get requires() {
+    return [this.policy.stableId];
+  }
+
+  serialize(): string {
+    const targetRoles = this.roles;
+    const toPublic =
+      targetRoles.length === 0 ||
+      (targetRoles.length === 1 && targetRoles[0].toLowerCase() === "public");
+    const rolesSql = toPublic ? "PUBLIC" : targetRoles.join(", ");
+
+    return [
+      "ALTER POLICY",
+      `${this.policy.schema}.${this.policy.name}`,
+      "ON",
+      `${this.policy.schema}.${this.policy.table_name}`,
+      "TO",
+      rolesSql,
+    ].join(" ");
+  }
+}
+
+/**
+ * ALTER POLICY ... USING (...)
+ */
+export class AlterRlsPolicySetUsingExpression extends AlterRlsPolicyChange {
+  public readonly policy: RlsPolicy;
+  public readonly usingExpression: string | null;
+  public readonly scope = "object" as const;
+
+  constructor(props: { policy: RlsPolicy; usingExpression: string | null }) {
+    super();
+    this.policy = props.policy;
+    this.usingExpression = props.usingExpression;
+  }
+
+  get requires() {
+    return [this.policy.stableId];
+  }
+
+  serialize(): string {
+    const expr = this.usingExpression ?? "true";
+    return [
+      "ALTER POLICY",
+      `${this.policy.schema}.${this.policy.name}`,
+      "ON",
+      `${this.policy.schema}.${this.policy.table_name}`,
+      "USING",
+      `(${expr})`,
+    ].join(" ");
+  }
+}
+
+/**
+ * ALTER POLICY ... WITH CHECK (...)
+ */
+export class AlterRlsPolicySetWithCheckExpression extends AlterRlsPolicyChange {
+  public readonly policy: RlsPolicy;
+  public readonly withCheckExpression: string | null;
+  public readonly scope = "object" as const;
+
+  constructor(props: {
+    policy: RlsPolicy;
+    withCheckExpression: string | null;
+  }) {
+    super();
+    this.policy = props.policy;
+    this.withCheckExpression = props.withCheckExpression;
+  }
+
+  get requires() {
+    return [this.policy.stableId];
+  }
+
+  serialize(): string {
+    const expr = this.withCheckExpression ?? "true";
+    return [
+      "ALTER POLICY",
+      `${this.policy.schema}.${this.policy.name}`,
+      "ON",
+      `${this.policy.schema}.${this.policy.table_name}`,
+      "WITH CHECK",
+      `(${expr})`,
+    ].join(" ");
+  }
+}
+
+/**
+ * Replace an RLS policy by dropping and recreating it.
+ * This is used when properties that cannot be altered via ALTER POLICY change.
+ */
+// NOTE: ReplaceRlsPolicy removed. Non-alterable changes are emitted as Drop + Create in rls-policy.diff.ts.

package/src/core/objects/rls-policy/changes/rls-policy.base.ts

@@ -0,0 +1,20 @@
+import { BaseChange } from "../../base.change.ts";
+import type { RlsPolicy } from "../rls-policy.model.ts";
+
+abstract class BaseRlsPolicyChange extends BaseChange {
+  abstract readonly policy: RlsPolicy;
+  abstract readonly scope: "object" | "comment";
+  readonly objectType: "rls_policy" = "rls_policy";
+}
+
+export abstract class CreateRlsPolicyChange extends BaseRlsPolicyChange {
+  readonly operation = "create" as const;
+}
+
+export abstract class AlterRlsPolicyChange extends BaseRlsPolicyChange {
+  readonly operation = "alter" as const;
+}
+
+export abstract class DropRlsPolicyChange extends BaseRlsPolicyChange {
+  readonly operation = "drop" as const;
+}

package/src/core/objects/rls-policy/changes/rls-policy.comment.ts

@@ -0,0 +1,69 @@
+import { quoteLiteral } from "../../base.change.ts";
+import { stableId } from "../../utils.ts";
+import type { RlsPolicy } from "../rls-policy.model.ts";
+import {
+  CreateRlsPolicyChange,
+  DropRlsPolicyChange,
+} from "./rls-policy.base.ts";
+
+export type CommentRlsPolicy =
+  | CreateCommentOnRlsPolicy
+  | DropCommentOnRlsPolicy;
+
+export class CreateCommentOnRlsPolicy extends CreateRlsPolicyChange {
+  public readonly policy: RlsPolicy;
+  public readonly scope = "comment" as const;
+
+  constructor(props: { policy: RlsPolicy }) {
+    super();
+    this.policy = props.policy;
+  }
+
+  get creates() {
+    return [stableId.comment(this.policy.stableId)];
+  }
+
+  get requires() {
+    return [this.policy.stableId];
+  }
+
+  serialize(): string {
+    return [
+      "COMMENT ON POLICY",
+      this.policy.name,
+      "ON",
+      `${this.policy.schema}.${this.policy.table_name}`,
+      "IS",
+      // biome-ignore lint/style/noNonNullAssertion: rls policy comment is not nullable in this case
+      quoteLiteral(this.policy.comment!),
+    ].join(" ");
+  }
+}
+
+export class DropCommentOnRlsPolicy extends DropRlsPolicyChange {
+  public readonly policy: RlsPolicy;
+  public readonly scope = "comment" as const;
+
+  constructor(props: { policy: RlsPolicy }) {
+    super();
+    this.policy = props.policy;
+  }
+
+  get drops() {
+    return [stableId.comment(this.policy.stableId)];
+  }
+
+  get requires() {
+    return [stableId.comment(this.policy.stableId), this.policy.stableId];
+  }
+
+  serialize(): string {
+    return [
+      "COMMENT ON POLICY",
+      this.policy.name,
+      "ON",
+      `${this.policy.schema}.${this.policy.table_name}`,
+      "IS NULL",
+    ].join(" ");
+  }
+}

package/src/core/objects/rls-policy/changes/rls-policy.create.test.ts

@@ -0,0 +1,74 @@
+import { describe, expect, test } from "bun:test";
+import { RlsPolicy } from "../rls-policy.model.ts";
+import { CreateRlsPolicy } from "./rls-policy.create.ts";
+
+describe("rls-policy", () => {
+  test("create minimal", () => {
+    const policy = new RlsPolicy({
+      schema: "public",
+      name: "test_policy_min",
+      table_name: "test_table",
+      command: "*",
+      permissive: true,
+      roles: [],
+      using_expression: null,
+      with_check_expression: null,
+      owner: "test",
+      comment: null,
+    });
+
+    const change = new CreateRlsPolicy({
+      policy,
+    });
+
+    expect(change.serialize()).toBe(
+      "CREATE POLICY test_policy_min ON public.test_table",
+    );
+  });
+
+  test("create", () => {
+    const policy = new RlsPolicy({
+      schema: "public",
+      name: "test_policy",
+      table_name: "test_table",
+      command: "r",
+      permissive: true,
+      roles: ["public"],
+      using_expression: "user_id = current_user_id()",
+      with_check_expression: null,
+      owner: "test",
+      comment: null,
+    });
+
+    const change = new CreateRlsPolicy({
+      policy,
+    });
+
+    expect(change.serialize()).toBe(
+      "CREATE POLICY test_policy ON public.test_table FOR SELECT USING (user_id = current_user_id())",
+    );
+  });
+
+  test("create with all options", () => {
+    const policy = new RlsPolicy({
+      schema: "public",
+      name: "test_policy_all",
+      table_name: "test_table",
+      command: "w",
+      permissive: false,
+      roles: ["role1", "role2"],
+      using_expression: "expr1",
+      with_check_expression: "expr2",
+      owner: "test",
+      comment: null,
+    });
+
+    const change = new CreateRlsPolicy({
+      policy,
+    });
+
+    expect(change.serialize()).toBe(
+      "CREATE POLICY test_policy_all ON public.test_table AS RESTRICTIVE FOR UPDATE TO role1, role2 USING (expr1) WITH CHECK (expr2)",
+    );
+  });
+});

package/src/core/objects/rls-policy/changes/rls-policy.create.ts

@@ -0,0 +1,100 @@
+import { stableId } from "../../utils.ts";
+import type { RlsPolicy } from "../rls-policy.model.ts";
+import { CreateRlsPolicyChange } from "./rls-policy.base.ts";
+
+/**
+ * Create an RLS policy.
+ *
+ * @see https://www.postgresql.org/docs/17/sql-createpolicy.html
+ *
+ * Synopsis
+ * ```sql
+ * CREATE POLICY name ON table_name
+ *     [ AS { PERMISSIVE | RESTRICTIVE } ]
+ *     [ FOR { ALL | SELECT | INSERT | UPDATE | DELETE } ]
+ *     [ TO { role_name | PUBLIC | CURRENT_ROLE | CURRENT_USER | SESSION_USER } [, ...] ]
+ *     [ USING ( using_expression ) ]
+ *     [ WITH CHECK ( with_check_expression ) ]
+ * ```
+ */
+export class CreateRlsPolicy extends CreateRlsPolicyChange {
+  public readonly policy: RlsPolicy;
+  public readonly scope = "object" as const;
+
+  constructor(props: { policy: RlsPolicy }) {
+    super();
+    this.policy = props.policy;
+  }
+
+  get creates() {
+    return [this.policy.stableId];
+  }
+
+  get requires() {
+    const dependencies = new Set<string>();
+
+    // Schema dependency
+    dependencies.add(stableId.schema(this.policy.schema));
+
+    // Table dependency
+    dependencies.add(
+      stableId.table(this.policy.schema, this.policy.table_name),
+    );
+
+    // Owner dependency
+    dependencies.add(stableId.role(this.policy.owner));
+
+    return Array.from(dependencies);
+  }
+
+  serialize(): string {
+    const parts: string[] = ["CREATE POLICY"];
+
+    // Add policy name
+    parts.push(this.policy.name);
+
+    // Add ON table
+    parts.push("ON", `${this.policy.schema}.${this.policy.table_name}`);
+
+    // Add AS RESTRICTIVE only if false (default is PERMISSIVE)
+    if (!this.policy.permissive) {
+      parts.push("AS RESTRICTIVE");
+    }
+
+    // Add FOR command
+    const commandMap: Record<string, string> = {
+      r: "FOR SELECT",
+      a: "FOR INSERT",
+      w: "FOR UPDATE",
+      d: "FOR DELETE",
+      "*": "FOR ALL",
+    };
+    // Default is FOR ALL; only print when not default
+    if (this.policy.command && this.policy.command !== "*") {
+      parts.push(commandMap[this.policy.command]);
+    }
+
+    // Add TO roles
+    // Default is TO PUBLIC; avoid printing explicit PUBLIC in CREATE
+    if (this.policy.roles && this.policy.roles.length > 0) {
+      const onlyPublic =
+        this.policy.roles.length === 1 &&
+        this.policy.roles[0].toLowerCase() === "public";
+      if (!onlyPublic) {
+        parts.push("TO", this.policy.roles.join(", "));
+      }
+    }
+
+    // Add USING expression
+    if (this.policy.using_expression) {
+      parts.push("USING", `(${this.policy.using_expression})`);
+    }
+
+    // Add WITH CHECK expression
+    if (this.policy.with_check_expression) {
+      parts.push("WITH CHECK", `(${this.policy.with_check_expression})`);
+    }
+
+    return parts.join(" ");
+  }
+}

package/src/core/objects/rls-policy/changes/rls-policy.drop.test.ts

@@ -0,0 +1,28 @@
+import { describe, expect, test } from "bun:test";
+import { RlsPolicy } from "../rls-policy.model.ts";
+import { DropRlsPolicy } from "./rls-policy.drop.ts";
+
+describe("rls-policy", () => {
+  test("drop", () => {
+    const policy = new RlsPolicy({
+      schema: "public",
+      name: "test_policy",
+      table_name: "test_table",
+      command: "r",
+      permissive: true,
+      roles: ["public"],
+      using_expression: "user_id = current_user_id()",
+      with_check_expression: null,
+      owner: "test",
+      comment: null,
+    });
+
+    const change = new DropRlsPolicy({
+      policy,
+    });
+
+    expect(change.serialize()).toBe(
+      "DROP POLICY test_policy ON public.test_table",
+    );
+  });
+});