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

package/dist/core/objects/foreign-data-wrapper/foreign-table/foreign-table.model.js

@@ -2,6 +2,7 @@ import { sql } from "@ts-safeql/sql-tag";
 import z from "zod";
 import { BasePgModel, columnPropsSchema, } from "../../base.model.js";
 import { privilegePropsSchema, } from "../../base.privilege-diff.js";
+import { normalizeSecurityLabels, securityLabelPropsSchema, } from "../../security-label.types.js";
 /**
  * All properties exposed by CREATE FOREIGN TABLE statement are included in diff output.
  * https://www.postgresql.org/docs/17/sql-createforeigntable.html
@@ -21,6 +22,7 @@ const foreignTablePropsSchema = z.object({
     comment: z.string().nullable(),
     columns: z.array(columnPropsSchema),
     privileges: z.array(privilegePropsSchema),
+    security_labels: z.array(securityLabelPropsSchema).default([]).optional(),
 });
 export class ForeignTable extends BasePgModel {
     schema;
@@ -31,6 +33,7 @@ export class ForeignTable extends BasePgModel {
     comment;
     columns;
     privileges;
+    security_labels;
     constructor(props) {
         super();
         // Identity fields
@@ -43,6 +46,7 @@ export class ForeignTable extends BasePgModel {
         this.comment = props.comment;
         this.columns = props.columns;
         this.privileges = props.privileges;
+        this.security_labels = props.security_labels ?? [];
     }
     get stableId() {
         return `foreignTable:${this.schema}.${this.name}`;
@@ -61,6 +65,7 @@ export class ForeignTable extends BasePgModel {
             comment: this.comment,
             columns: this.columns,
             privileges: this.privileges,
+            security_labels: this.security_labels,
         };
     }
     stableSnapshot() {
@@ -79,6 +84,7 @@ export class ForeignTable extends BasePgModel {
             data: {
                 ...this.dataFields,
                 columns: normalizeColumns(),
+                security_labels: normalizeSecurityLabels(this.security_labels),
             },
         };
     }
@@ -181,7 +187,20 @@ export async function extractForeignTables(pool) {
             join lateral aclexplode(src.acl) as x(grantor, grantee, privilege_type, is_grantable) on true
             group by x.grantee, x.privilege_type
           ) as grp
-        ), '[]') as privileges
+        ), '[]') as privileges,
+        coalesce(
+          (
+            select json_agg(
+              json_build_object('provider', sl.provider, 'label', sl.label)
+              order by sl.provider
+            )
+            from pg_catalog.pg_seclabel sl
+            where sl.objoid = ft.oid
+              and sl.classoid = 'pg_class'::regclass
+              and sl.objsubid = 0
+          ),
+          '[]'::json
+        ) as security_labels
       from
         foreign_tables ft
         left join pg_attribute a on a.attrelid = ft.oid and a.attnum > 0 and not a.attisdropped

package/dist/core/objects/index/index.diff.js

@@ -74,7 +74,6 @@ export function diffIndexes(main, branch, branchIndexableObjects) {
             "nulls_not_distinct",
             "immediate",
             "is_clustered",
-            "is_replica_identity",
             "column_collations",
             "operator_classes",
             "column_options",

package/dist/core/objects/index/index.model.d.ts

@@ -1,6 +1,7 @@
 import type { Pool } from "pg";
 import z from "zod";
 import { BasePgModel } from "../base.model.ts";
+import { type ExtractRetryOptions } from "../extract-with-retry.ts";
 declare const indexPropsSchema: z.ZodObject<{
     schema: z.ZodString;
     table_name: z.ZodString;
@@ -97,7 +98,6 @@ export declare class Index extends BasePgModel {
         nulls_not_distinct: boolean;
         immediate: boolean;
         is_clustered: boolean;
-        is_replica_identity: boolean;
         column_collations: (string | null)[];
         operator_classes: string[];
         column_options: number[];
@@ -132,7 +132,6 @@ export declare class Index extends BasePgModel {
             nulls_not_distinct: boolean;
             immediate: boolean;
             is_clustered: boolean;
-            is_replica_identity: boolean;
             index_expressions: string | null;
             partial_predicate: string | null;
             table_relkind: "r" | "m" | "p";
@@ -146,5 +145,5 @@ export declare class Index extends BasePgModel {
         };
     };
 }
-export declare function extractIndexes(pool: Pool): Promise<Index[]>;
+export declare function extractIndexes(pool: Pool, options?: ExtractRetryOptions): Promise<Index[]>;
 export {};

package/dist/core/objects/index/index.model.js

@@ -1,6 +1,7 @@
 import { sql } from "@ts-safeql/sql-tag";
 import z from "zod";
 import { BasePgModel } from "../base.model.js";
+import { extractWithDefinitionRetry, } from "../extract-with-retry.js";
 const TableRelkindSchema = z.enum([
     "r", // table (regular relation)
     "m", // materialized view
@@ -136,7 +137,11 @@ export class Index extends BasePgModel {
             nulls_not_distinct: this.nulls_not_distinct,
             immediate: this.immediate,
             is_clustered: this.is_clustered,
-            is_replica_identity: this.is_replica_identity,
+            // is_replica_identity excluded: the table's `replica_identity` /
+            // `replica_identity_index` is the source of truth, set via
+            // ALTER TABLE ... REPLICA IDENTITY USING INDEX. Including this flag here
+            // would trigger spurious DROP+CREATE of the index whenever the table's
+            // replica identity changes.
             // key_columns excluded: contains attribute numbers that can differ between databases
             // even when indexes are logically identical. The definition field already captures
             // the logical structure using column names, so we compare by definition instead.
@@ -184,8 +189,13 @@ export class Index extends BasePgModel {
         };
     }
 }
-export async function extractIndexes(pool) {
-    const { rows: indexRows } = await pool.query(sql `
+export async function extractIndexes(pool, options) {
+    const indexRows = await extractWithDefinitionRetry({
+        label: "indexes",
+        options,
+        hasNullDefinition: (row) => row.definition === null,
+        query: async () => {
+            const result = await pool.query(sql `
       with extension_oids as (
         select objid
         from pg_depend d
@@ -341,8 +351,9 @@ export async function extractIndexes(pool) {
 
       order by 1, 2
   `);
-    const validatedRows = indexRows
-        .map((row) => indexRowSchema.parse(row))
-        .filter((row) => row.definition !== null);
+            return result.rows.map((row) => indexRowSchema.parse(row));
+        },
+    });
+    const validatedRows = indexRows.filter((row) => row.definition !== null);
     return validatedRows.map((row) => new Index(row));
 }

package/dist/core/objects/materialized-view/changes/materialized-view.base.d.ts

@@ -2,7 +2,7 @@ import { BaseChange } from "../../base.change.ts";
 import type { MaterializedView } from "../materialized-view.model.ts";
 declare abstract class BaseMaterializedViewChange extends BaseChange {
     abstract readonly materializedView: MaterializedView;
-    abstract readonly scope: "object" | "comment" | "privilege";
+    abstract readonly scope: "object" | "comment" | "privilege" | "security_label";
     readonly objectType: "materialized_view";
 }
 export declare abstract class CreateMaterializedViewChange extends BaseMaterializedViewChange {

package/dist/core/objects/materialized-view/changes/materialized-view.security-label.d.ts

@@ -0,0 +1,28 @@
+import type { SecurityLabelProps } from "../../security-label.types.ts";
+import type { MaterializedView } from "../materialized-view.model.ts";
+import { CreateMaterializedViewChange, DropMaterializedViewChange } from "./materialized-view.base.ts";
+export type SecurityLabelMaterializedView = CreateSecurityLabelOnMaterializedView | DropSecurityLabelOnMaterializedView;
+export declare class CreateSecurityLabelOnMaterializedView extends CreateMaterializedViewChange {
+    readonly materializedView: MaterializedView;
+    readonly securityLabel: SecurityLabelProps;
+    readonly scope: "security_label";
+    constructor(props: {
+        materializedView: MaterializedView;
+        securityLabel: SecurityLabelProps;
+    });
+    get creates(): `securityLabel:${string}::provider:${string}`[];
+    get requires(): `materializedView:${string}`[];
+    serialize(): string;
+}
+export declare class DropSecurityLabelOnMaterializedView extends DropMaterializedViewChange {
+    readonly materializedView: MaterializedView;
+    readonly securityLabel: SecurityLabelProps;
+    readonly scope: "security_label";
+    constructor(props: {
+        materializedView: MaterializedView;
+        securityLabel: SecurityLabelProps;
+    });
+    get drops(): `securityLabel:${string}::provider:${string}`[];
+    get requires(): (`securityLabel:${string}::provider:${string}` | `materializedView:${string}`)[];
+    serialize(): string;
+}

package/dist/core/objects/materialized-view/changes/materialized-view.security-label.js

@@ -0,0 +1,61 @@
+import { quoteLiteral } from "../../base.change.js";
+import { stableId } from "../../utils.js";
+import { CreateMaterializedViewChange, DropMaterializedViewChange, } from "./materialized-view.base.js";
+export class CreateSecurityLabelOnMaterializedView extends CreateMaterializedViewChange {
+    materializedView;
+    securityLabel;
+    scope = "security_label";
+    constructor(props) {
+        super();
+        this.materializedView = props.materializedView;
+        this.securityLabel = props.securityLabel;
+    }
+    get creates() {
+        return [
+            stableId.securityLabel(this.materializedView.stableId, this.securityLabel.provider),
+        ];
+    }
+    get requires() {
+        return [this.materializedView.stableId];
+    }
+    serialize() {
+        return [
+            "SECURITY LABEL FOR",
+            this.securityLabel.provider,
+            "ON MATERIALIZED VIEW",
+            `${this.materializedView.schema}.${this.materializedView.name}`,
+            "IS",
+            quoteLiteral(this.securityLabel.label),
+        ].join(" ");
+    }
+}
+export class DropSecurityLabelOnMaterializedView extends DropMaterializedViewChange {
+    materializedView;
+    securityLabel;
+    scope = "security_label";
+    constructor(props) {
+        super();
+        this.materializedView = props.materializedView;
+        this.securityLabel = props.securityLabel;
+    }
+    get drops() {
+        return [
+            stableId.securityLabel(this.materializedView.stableId, this.securityLabel.provider),
+        ];
+    }
+    get requires() {
+        return [
+            stableId.securityLabel(this.materializedView.stableId, this.securityLabel.provider),
+            this.materializedView.stableId,
+        ];
+    }
+    serialize() {
+        return [
+            "SECURITY LABEL FOR",
+            this.securityLabel.provider,
+            "ON MATERIALIZED VIEW",
+            `${this.materializedView.schema}.${this.materializedView.name}`,
+            "IS NULL",
+        ].join(" ");
+    }
+}

package/dist/core/objects/materialized-view/changes/materialized-view.types.d.ts

@@ -3,5 +3,6 @@ import type { CommentMaterializedView } from "./materialized-view.comment.ts";
 import type { CreateMaterializedView } from "./materialized-view.create.ts";
 import type { DropMaterializedView } from "./materialized-view.drop.ts";
 import type { MaterializedViewPrivilege } from "./materialized-view.privilege.ts";
+import type { SecurityLabelMaterializedView } from "./materialized-view.security-label.ts";
 /** Union of all materialized-view-related change variants (`objectType: "materialized_view"`). @category Change Types */
-export type MaterializedViewChange = AlterMaterializedView | CommentMaterializedView | CreateMaterializedView | DropMaterializedView | MaterializedViewPrivilege;
+export type MaterializedViewChange = AlterMaterializedView | CommentMaterializedView | CreateMaterializedView | DropMaterializedView | MaterializedViewPrivilege | SecurityLabelMaterializedView;

package/dist/core/objects/materialized-view/materialized-view.diff.js

@@ -1,11 +1,13 @@
 import { diffObjects } from "../base.diff.js";
 import { diffPrivileges, emitColumnPrivilegeChanges, } from "../base.privilege-diff.js";
+import { diffSecurityLabels } from "../security-label.types.js";
 import { deepEqual, hasNonAlterableChanges } from "../utils.js";
 import { AlterMaterializedViewChangeOwner, AlterMaterializedViewSetStorageParams, } from "./changes/materialized-view.alter.js";
 import { CreateCommentOnMaterializedView, CreateCommentOnMaterializedViewColumn, DropCommentOnMaterializedView, DropCommentOnMaterializedViewColumn, } from "./changes/materialized-view.comment.js";
 import { CreateMaterializedView } from "./changes/materialized-view.create.js";
 import { DropMaterializedView } from "./changes/materialized-view.drop.js";
 import { GrantMaterializedViewPrivileges, RevokeGrantOptionMaterializedViewPrivileges, RevokeMaterializedViewPrivileges, } from "./changes/materialized-view.privilege.js";
+import { CreateSecurityLabelOnMaterializedView, DropSecurityLabelOnMaterializedView, } from "./changes/materialized-view.security-label.js";
 /**
  * Diff two sets of materialized views from main and branch catalogs.
  *
@@ -48,6 +50,14 @@ export function diffMaterializedViews(ctx, main, branch) {
                 }));
             }
         }
+        // Security labels on the matview itself (columns of matviews are not
+        // supported targets of SECURITY LABEL, so we only label the relation).
+        for (const label of mv.security_labels) {
+            changes.push(new CreateSecurityLabelOnMaterializedView({
+                materializedView: mv,
+                securityLabel: label,
+            }));
+        }
         // PRIVILEGES: For created objects, compare against default privileges state
         // The migration script will run ALTER DEFAULT PRIVILEGES before CREATE (via constraint spec),
         // so objects are created with the default privileges state in effect.
@@ -156,6 +166,14 @@ export function diffMaterializedViews(ctx, main, branch) {
                     }));
                 }
             }
+            // SECURITY LABELS
+            changes.push(...diffSecurityLabels(mainMaterializedView.security_labels, branchMaterializedView.security_labels, (securityLabel) => new CreateSecurityLabelOnMaterializedView({
+                materializedView: branchMaterializedView,
+                securityLabel,
+            }), (securityLabel) => new DropSecurityLabelOnMaterializedView({
+                materializedView: mainMaterializedView,
+                securityLabel,
+            })));
             // COMMENT changes on columns
             const mainCols = new Map(mainMaterializedView.columns.map((c) => [c.name, c]));
             const branchCols = new Map(branchMaterializedView.columns.map((c) => [c.name, c]));

package/dist/core/objects/materialized-view/materialized-view.model.d.ts

@@ -2,6 +2,8 @@ import type { Pool } from "pg";
 import z from "zod";
 import { BasePgModel, type TableLikeObject } from "../base.model.ts";
 import { type PrivilegeProps } from "../base.privilege-diff.ts";
+import { type ExtractRetryOptions } from "../extract-with-retry.ts";
+import { type SecurityLabelProps } from "../security-label.types.ts";
 declare const materializedViewPropsSchema: z.ZodObject<{
     schema: z.ZodString;
     name: z.ZodString;
@@ -41,6 +43,10 @@ declare const materializedViewPropsSchema: z.ZodObject<{
         collation: z.ZodNullable<z.ZodString>;
         default: z.ZodNullable<z.ZodString>;
         comment: z.ZodNullable<z.ZodString>;
+        security_labels: z.ZodOptional<z.ZodArray<z.ZodObject<{
+            provider: z.ZodString;
+            label: z.ZodString;
+        }, z.z.core.$strip>>>;
     }, z.z.core.$strip>>;
     privileges: z.ZodArray<z.ZodObject<{
         grantee: z.ZodString;
@@ -48,6 +54,10 @@ declare const materializedViewPropsSchema: z.ZodObject<{
         grantable: z.ZodBoolean;
         columns: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString>>>;
     }, z.z.core.$strip>>;
+    security_labels: z.ZodOptional<z.ZodDefault<z.ZodArray<z.ZodObject<{
+        provider: z.ZodString;
+        label: z.ZodString;
+    }, z.z.core.$strip>>>>;
 }, z.z.core.$strip>;
 type MaterializedViewPrivilegeProps = PrivilegeProps;
 export type MaterializedViewProps = z.infer<typeof materializedViewPropsSchema>;
@@ -70,6 +80,7 @@ export declare class MaterializedView extends BasePgModel implements TableLikeOb
     readonly comment: MaterializedViewProps["comment"];
     readonly columns: MaterializedViewProps["columns"];
     readonly privileges: MaterializedViewPrivilegeProps[];
+    readonly security_labels: SecurityLabelProps[];
     constructor(props: MaterializedViewProps);
     get stableId(): `materializedView:${string}`;
     get identityFields(): {
@@ -108,6 +119,10 @@ export declare class MaterializedView extends BasePgModel implements TableLikeOb
             collation: string | null;
             default: string | null;
             comment: string | null;
+            security_labels?: {
+                provider: string;
+                label: string;
+            }[] | undefined;
         }[];
         privileges: {
             grantee: string;
@@ -115,6 +130,10 @@ export declare class MaterializedView extends BasePgModel implements TableLikeOb
             grantable: boolean;
             columns?: string[] | null | undefined;
         }[];
+        security_labels: {
+            provider: string;
+            label: string;
+        }[];
     };
     stableSnapshot(): {
         identity: {
@@ -125,6 +144,10 @@ export declare class MaterializedView extends BasePgModel implements TableLikeOb
             columns: {
                 [x: string]: unknown;
             }[];
+            security_labels: {
+                provider: string;
+                label: string;
+            }[];
             definition: string;
             row_security: boolean;
             force_row_security: boolean;
@@ -148,5 +171,5 @@ export declare class MaterializedView extends BasePgModel implements TableLikeOb
         };
     };
 }
-export declare function extractMaterializedViews(pool: Pool): Promise<MaterializedView[]>;
+export declare function extractMaterializedViews(pool: Pool, options?: ExtractRetryOptions): Promise<MaterializedView[]>;
 export {};

package/dist/core/objects/materialized-view/materialized-view.model.js

@@ -2,6 +2,8 @@ import { sql } from "@ts-safeql/sql-tag";
 import z from "zod";
 import { BasePgModel, columnPropsSchema, } from "../base.model.js";
 import { privilegePropsSchema, } from "../base.privilege-diff.js";
+import { extractWithDefinitionRetry, } from "../extract-with-retry.js";
+import { normalizeSecurityLabels, securityLabelPropsSchema, } from "../security-label.types.js";
 import { ReplicaIdentitySchema } from "../table/table.model.js";
 const materializedViewPropsSchema = z.object({
     schema: z.string(),
@@ -22,6 +24,15 @@ const materializedViewPropsSchema = z.object({
     comment: z.string().nullable(),
     columns: z.array(columnPropsSchema),
     privileges: z.array(privilegePropsSchema),
+    security_labels: z.array(securityLabelPropsSchema).default([]).optional(),
+});
+// pg_get_viewdef(oid) can return NULL when the underlying matview (or its
+// pg_rewrite row) is dropped between catalog scan and resolution, or under
+// transient catalog state during recovery. An unreadable matview cannot be
+// diffed, so we accept NULL here and filter the row out at extraction time
+// rather than crashing the whole catalog parse with a ZodError.
+const materializedViewRowSchema = materializedViewPropsSchema.extend({
+    definition: z.string().nullable(),
 });
 export class MaterializedView extends BasePgModel {
     schema;
@@ -42,6 +53,7 @@ export class MaterializedView extends BasePgModel {
     comment;
     columns;
     privileges;
+    security_labels;
     constructor(props) {
         super();
         // Identity fields
@@ -64,6 +76,7 @@ export class MaterializedView extends BasePgModel {
         this.comment = props.comment;
         this.columns = props.columns;
         this.privileges = props.privileges;
+        this.security_labels = props.security_labels ?? [];
     }
     get stableId() {
         return `materializedView:${this.schema}.${this.name}`;
@@ -92,6 +105,7 @@ export class MaterializedView extends BasePgModel {
             comment: this.comment,
             columns: this.columns,
             privileges: this.privileges,
+            security_labels: this.security_labels,
         };
     }
     stableSnapshot() {
@@ -110,12 +124,18 @@ export class MaterializedView extends BasePgModel {
             data: {
                 ...this.dataFields,
                 columns: normalizeColumns(),
+                security_labels: normalizeSecurityLabels(this.security_labels),
             },
         };
     }
 }
-export async function extractMaterializedViews(pool) {
-    const { rows: mvRows } = await pool.query(sql `
+export async function extractMaterializedViews(pool, options) {
+    const mvRows = await extractWithDefinitionRetry({
+        label: "materialized views",
+        options,
+        hasNullDefinition: (row) => row.definition === null,
+        query: async () => {
+            const result = await pool.query(sql `
 with extension_oids as (
   select
     objid
@@ -205,7 +225,20 @@ select
       join lateral aclexplode(src.acl) as x(grantor, grantee, privilege_type, is_grantable) on true
       group by x.grantee, x.privilege_type
     ) as grp
-  ), '[]') as privileges
+  ), '[]') as privileges,
+  coalesce(
+    (
+      select json_agg(
+        json_build_object('provider', sl.provider, 'label', sl.label)
+        order by sl.provider
+      )
+      from pg_catalog.pg_seclabel sl
+      where sl.objoid = c.oid
+        and sl.classoid = 'pg_class'::regclass
+        and sl.objsubid = 0
+    ),
+    '[]'::json
+  ) as security_labels
 from
   pg_catalog.pg_class c
   left outer join extension_oids e on c.oid = e.objid
@@ -220,7 +253,9 @@ group by
 order by
   c.relnamespace::regnamespace, c.relname
   `);
-    // Validate and parse each row using the Zod schema
-    const validatedRows = mvRows.map((row) => materializedViewPropsSchema.parse(row));
+            return result.rows.map((row) => materializedViewRowSchema.parse(row));
+        },
+    });
+    const validatedRows = mvRows.filter((row) => row.definition !== null);
     return validatedRows.map((row) => new MaterializedView(row));
 }

package/dist/core/objects/procedure/changes/procedure.base.d.ts

@@ -2,7 +2,7 @@ import { BaseChange } from "../../base.change.ts";
 import type { Procedure } from "../procedure.model.ts";
 declare abstract class BaseProcedureChange extends BaseChange {
     abstract readonly procedure: Procedure;
-    abstract readonly scope: "object" | "comment" | "privilege";
+    abstract readonly scope: "object" | "comment" | "privilege" | "security_label";
     readonly objectType: "procedure";
 }
 export declare abstract class CreateProcedureChange extends BaseProcedureChange {

package/dist/core/objects/procedure/changes/procedure.security-label.d.ts

@@ -0,0 +1,28 @@
+import type { SecurityLabelProps } from "../../security-label.types.ts";
+import type { Procedure } from "../procedure.model.ts";
+import { CreateProcedureChange, DropProcedureChange } from "./procedure.base.ts";
+export type SecurityLabelProcedure = CreateSecurityLabelOnProcedure | DropSecurityLabelOnProcedure;
+export declare class CreateSecurityLabelOnProcedure extends CreateProcedureChange {
+    readonly procedure: Procedure;
+    readonly securityLabel: SecurityLabelProps;
+    readonly scope: "security_label";
+    constructor(props: {
+        procedure: Procedure;
+        securityLabel: SecurityLabelProps;
+    });
+    get creates(): `securityLabel:${string}::provider:${string}`[];
+    get requires(): `procedure:${string}`[];
+    serialize(): string;
+}
+export declare class DropSecurityLabelOnProcedure extends DropProcedureChange {
+    readonly procedure: Procedure;
+    readonly securityLabel: SecurityLabelProps;
+    readonly scope: "security_label";
+    constructor(props: {
+        procedure: Procedure;
+        securityLabel: SecurityLabelProps;
+    });
+    get drops(): `securityLabel:${string}::provider:${string}`[];
+    get requires(): (`securityLabel:${string}::provider:${string}` | `procedure:${string}`)[];
+    serialize(): string;
+}

package/dist/core/objects/procedure/changes/procedure.security-label.js

@@ -0,0 +1,69 @@
+import { quoteLiteral } from "../../base.change.js";
+import { stableId } from "../../utils.js";
+import { CreateProcedureChange, DropProcedureChange, } from "./procedure.base.js";
+function targetKeyword(p) {
+    return p.kind === "p" ? "PROCEDURE" : "FUNCTION";
+}
+function procedureIdentity(p) {
+    return `${p.schema}.${p.name}(${(p.argument_types ?? []).join(",")})`;
+}
+export class CreateSecurityLabelOnProcedure extends CreateProcedureChange {
+    procedure;
+    securityLabel;
+    scope = "security_label";
+    constructor(props) {
+        super();
+        this.procedure = props.procedure;
+        this.securityLabel = props.securityLabel;
+    }
+    get creates() {
+        return [
+            stableId.securityLabel(this.procedure.stableId, this.securityLabel.provider),
+        ];
+    }
+    get requires() {
+        return [this.procedure.stableId];
+    }
+    serialize() {
+        return [
+            "SECURITY LABEL FOR",
+            this.securityLabel.provider,
+            "ON",
+            targetKeyword(this.procedure),
+            procedureIdentity(this.procedure),
+            "IS",
+            quoteLiteral(this.securityLabel.label),
+        ].join(" ");
+    }
+}
+export class DropSecurityLabelOnProcedure extends DropProcedureChange {
+    procedure;
+    securityLabel;
+    scope = "security_label";
+    constructor(props) {
+        super();
+        this.procedure = props.procedure;
+        this.securityLabel = props.securityLabel;
+    }
+    get drops() {
+        return [
+            stableId.securityLabel(this.procedure.stableId, this.securityLabel.provider),
+        ];
+    }
+    get requires() {
+        return [
+            stableId.securityLabel(this.procedure.stableId, this.securityLabel.provider),
+            this.procedure.stableId,
+        ];
+    }
+    serialize() {
+        return [
+            "SECURITY LABEL FOR",
+            this.securityLabel.provider,
+            "ON",
+            targetKeyword(this.procedure),
+            procedureIdentity(this.procedure),
+            "IS NULL",
+        ].join(" ");
+    }
+}

package/dist/core/objects/procedure/changes/procedure.types.d.ts

@@ -3,5 +3,6 @@ import type { CommentProcedure } from "./procedure.comment.ts";
 import type { CreateProcedure } from "./procedure.create.ts";
 import type { DropProcedure } from "./procedure.drop.ts";
 import type { ProcedurePrivilege } from "./procedure.privilege.ts";
+import type { SecurityLabelProcedure } from "./procedure.security-label.ts";
 /** Union of all procedure-related change variants (`objectType: "procedure"`). @category Change Types */
-export type ProcedureChange = AlterProcedure | CommentProcedure | CreateProcedure | DropProcedure | ProcedurePrivilege;
+export type ProcedureChange = AlterProcedure | CommentProcedure | CreateProcedure | DropProcedure | ProcedurePrivilege | SecurityLabelProcedure;

package/dist/core/objects/procedure/procedure.diff.js

@@ -1,11 +1,13 @@
 import { diffObjects } from "../base.diff.js";
 import { diffPrivileges, emitObjectPrivilegeChanges, filterPublicBuiltInDefaults, } from "../base.privilege-diff.js";
+import { diffSecurityLabels } from "../security-label.types.js";
 import { deepEqual, hasNonAlterableChanges } from "../utils.js";
 import { AlterProcedureChangeOwner, AlterProcedureSetConfig, AlterProcedureSetLeakproof, AlterProcedureSetParallel, AlterProcedureSetSecurity, AlterProcedureSetStrictness, AlterProcedureSetVolatility, } from "./changes/procedure.alter.js";
 import { CreateCommentOnProcedure, DropCommentOnProcedure, } from "./changes/procedure.comment.js";
 import { CreateProcedure } from "./changes/procedure.create.js";
 import { DropProcedure } from "./changes/procedure.drop.js";
 import { GrantProcedurePrivileges, RevokeGrantOptionProcedurePrivileges, RevokeProcedurePrivileges, } from "./changes/procedure.privilege.js";
+import { CreateSecurityLabelOnProcedure, DropSecurityLabelOnProcedure, } from "./changes/procedure.security-label.js";
 /**
  * Diff two sets of procedures from main and branch catalogs.
  *
@@ -30,6 +32,12 @@ export function diffProcedures(ctx, main, branch) {
         if (proc.comment !== null) {
             changes.push(new CreateCommentOnProcedure({ procedure: proc }));
         }
+        for (const label of proc.security_labels) {
+            changes.push(new CreateSecurityLabelOnProcedure({
+                procedure: proc,
+                securityLabel: label,
+            }));
+        }
         // PRIVILEGES: For created objects, compare against default privileges state
         // The migration script will run ALTER DEFAULT PRIVILEGES before CREATE (via constraint spec),
         // so objects are created with the default privileges state in effect.
@@ -139,6 +147,14 @@ export function diffProcedures(ctx, main, branch) {
                     changes.push(new CreateCommentOnProcedure({ procedure: branchProcedure }));
                 }
             }
+            // SECURITY LABELS
+            changes.push(...diffSecurityLabels(mainProcedure.security_labels, branchProcedure.security_labels, (securityLabel) => new CreateSecurityLabelOnProcedure({
+                procedure: branchProcedure,
+                securityLabel,
+            }), (securityLabel) => new DropSecurityLabelOnProcedure({
+                procedure: mainProcedure,
+                securityLabel,
+            })));
             // SECURITY DEFINER/INVOKER
             if (mainProcedure.security_definer !== branchProcedure.security_definer) {
                 changes.push(new AlterProcedureSetSecurity({