alchemy 0.63.1 → 0.65.0

package/lib/planetscale/password.js.map

@@ -1 +1 @@
-{"version":3,"file":"password.js","sourceRoot":"","sources":["../../src/planetscale/password.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAExC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAsG1C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAuJG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,QAAQ,CAC9B,uBAAuB,EACvB,KAAK,WAEH,EAAU,EACV,KAAoB;IAEpB,MAAM,MAAM,GACV,KAAK,CAAC,MAAM,EAAE,WAAW,IAAI,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;IACjE,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;IAC5E,CAAC;IACD,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa;QACjC,CAAC,CAAC,WAAW,EAAE;QACf,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,IAAI,WAAW,EAAE,CAAC,CAAC;IAC7C,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,QAAQ,EAAE,CAAC;IAEnH,MAAM,GAAG,GAAG,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IAC3C,MAAM,UAAU,GACd,KAAK,CAAC,MAAM,IAAI,IAAI;QAClB,CAAC,CAAC,MAAM;QACR,CAAC,CAAC,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ;YAChC,CAAC,CAAC,KAAK,CAAC,MAAM;YACd,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC;IAC1B,MAAM,YAAY,GAChB,OAAO,KAAK,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC;IAE5E,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC;gBACtB,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,MAAM,CAC/B,kBAAkB,KAAK,CAAC,cAAc,cAAc,YAAY,aAAa,UAAU,cAAc,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CACtH,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBAC5C,MAAM,IAAI,KAAK,CACb,4BAA4B,QAAQ,CAAC,UAAU,IAAI,MAAM,QAAQ,CAAC,IAAI,EAAE,EAAE,CAC3E,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;YAChD,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC;IACD,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5B,IACE,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,IAAI;YAC1B,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC;gBAC9D,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC;oBAChC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC;oBAC1B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,KAAK,CAAC,KAAK,CAAC,MAAM;oBAC/C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,KAAK,CAAC,KAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACpE,CAAC;YACD,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;QACxB,CAAC;QACD,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,KAAK,CACpC,kBAAkB,KAAK,CAAC,cAAc,cAAc,YAAY,aAAa,UAAU,cAAc,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EACrH;YACE,IAAI;YACJ,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB,CACF,CAAC;QAEF,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CACb,8BAA8B,cAAc,CAAC,UAAU,IAAI,MAAM,cAAc,CAAC,IAAI,EAAE,EAAE,CACzF,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC;YACV,GAAG,IAAI,CAAC,MAAM;YACd,GAAG,KAAK;SACT,CAAC,CAAC;IACL,CAAC;IAED,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,IAAI,CACnC,kBAAkB,KAAK,CAAC,cAAc,cAAc,YAAY,aAAa,UAAU,YAAY,EACnG;YACE,IAAI;YACJ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB,CACF,CAAC;QAEF,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CACb,8BAA8B,cAAc,CAAC,UAAU,IAAI,MAAM,cAAc,CAAC,IAAI,EAAE,EAAE,CACzF,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,IAAI,EAAO,CAAC;QAE9C,OAAO,IAAI,CAAC;YACV,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,IAAI,EAAE,IAAI,CAAC,eAAe;YAC1B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC;YACzC,QAAQ;YACR,GAAG,KAAK;YACR,IAAI,EAAE,GAAG,KAAK,CAAC,IAAI,IAAI,QAAQ,EAAE;SAClC,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;QAChD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC,CACF,CAAC"}
+{"version":3,"file":"password.js","sourceRoot":"","sources":["../../src/planetscale/password.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAExC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAyB,MAAM,qBAAqB,CAAC;AAiG/E;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyJG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,QAAQ,CAC9B,uBAAuB,EACvB,KAAK,WAEH,EAAU,EACV,KAAoB;IAEpB,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa;QACjC,CAAC,CAAC,WAAW,EAAE;QACf,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,IAAI,WAAW,EAAE,CAAC,CAAC;IAC7C,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,QAAQ,EAAE,CAAC;IAEnH,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAAC,KAAK,CAAC,CAAC;IACzC,MAAM,QAAQ,GACZ,OAAO,KAAK,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC;IAC5E,MAAM,MAAM,GACV,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ;QAC9B,CAAC,CAAC,KAAK,CAAC,MAAM;QACd,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,IAAI,MAAM,CAAC,CAAC;IACrC,MAAM,YAAY,GAChB,KAAK,CAAC,cAAc;QACpB,CAAC,CAAC,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,EAAE,cAAc,CAAC;YACjE,CAAC,OAAO,KAAK,CAAC,QAAQ,KAAK,QAAQ,IAAI,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC;IAE3E,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACjD,CAAC;IAED,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5B,IAAI,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC;YACpB,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CACrE;gBACE,IAAI,EAAE;oBACJ,YAAY;oBACZ,QAAQ;oBACR,MAAM;oBACN,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE;iBACnB;gBACD,MAAM,EAAE,MAAM;aACf,CACF,CAAC;YAEF,IAAI,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC1C,MAAM,IAAI,KAAK,CAAC,4BAA4B,MAAM,GAAG,EAAE;oBACrD,KAAK,EAAE,GAAG,CAAC,KAAK;iBACjB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC;IACD,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5B,yFAAyF;QACzF,IACE,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,IAAI,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CACxC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,KAAK,OAAO,CAC9C,EACD,CAAC;YACD,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;QACxB,CAAC;QACD,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC;YACzD,IAAI,EAAE;gBACJ,YAAY;gBACZ,QAAQ;gBACR,MAAM;gBACN,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE;aACnB;YACD,IAAI,EAAE;gBACJ,IAAI;gBACJ,KAAK,EAAE,KAAK,CAAC,KAAK;aACnB;SACF,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC;YACV,GAAG,IAAI,CAAC,MAAM;YACd,GAAG,KAAK;YACR,IAAI;SACL,CAAC,CAAC;IACL,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC;QACrE,IAAI,EAAE;YACJ,YAAY;YACZ,QAAQ;YACR,MAAM;SACP;QACD,IAAI,EAAE;YACJ,IAAI;YACJ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB;KACF,CAAC,CAAC;IAEH,OAAO,IAAI,CAAC;QACV,EAAE,EAAE,IAAI,CAAC,EAAE;QACX,SAAS,EAAE,IAAI,CAAC,UAAU;QAC1B,IAAI,EAAE,IAAI,CAAC,eAAe;QAC1B,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC;QACzC,QAAQ;QACR,GAAG,KAAK;QACR,IAAI,EAAE,GAAG,KAAK,CAAC,IAAI,IAAI,QAAQ,EAAE;KAClC,CAAC,CAAC;AACL,CAAC,CACF,CAAC;AAEF;;GAEG;AACH,MAAM,IAAI,GAAG,CAAI,CAAI,EAAE,CAAa,EAAE,EAAE;IACtC,MAAM,IAAI,GAAgB,EAAE,CAAC;IAC7B,KAAK,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;YACvC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC,CAAC"}

package/lib/planetscale/role.d.ts

@@ -0,0 +1,126 @@
+import type { Context } from "../context.ts";
+import { Resource } from "../resource.ts";
+import type { Secret } from "../secret.ts";
+import { type PlanetScaleProps } from "./api/client.gen.ts";
+import type { Branch } from "./branch.ts";
+import type { Database } from "./database.ts";
+/**
+ * Properties for creating or updating a PlanetScale PostgreSQL Role
+ */
+export interface RoleProps extends PlanetScaleProps {
+    /**
+     * The organization ID where the role will be created
+     * Required when using string database name, optional when using Database resource
+     */
+    organizationId?: string;
+    /**
+     * The database where the role will be created
+     * Can be either a database name (string) or Database resource
+     */
+    database: string | Database;
+    /**
+     * The branch where the role will be created
+     * Can be either a branch name (string) or Branch resource
+     * @default "main"
+     */
+    branch?: string | Branch;
+    /**
+     * Time to live in seconds
+     */
+    ttl?: number;
+    /**
+     * Roles to inherit from.
+     * The `"postgres"` role provides full administrator access to the database.
+     * You can also inherit from another Role resource.
+     */
+    inheritedRoles: InheritedRole[] | Role;
+}
+/**
+ * Roles that can be inherited from.
+ */
+export type InheritedRole = "postgres" | "pg_checkpoint" | "pg_create_subscription" | "pg_maintain" | "pg_monitor" | "pg_read_all_data" | "pg_read_all_settings" | "pg_read_all_stats" | "pg_signal_backend" | "pg_stat_scan_tables" | "pg_use_reserved_connections" | "pg_write_all_data" | (string & {});
+export interface Role extends Resource<"planetscale::Role">, Omit<RoleProps, "inheritedRoles"> {
+    /**
+     * The unique identifier for the role
+     */
+    id: string;
+    /**
+     * The name of the role
+     */
+    name: string;
+    /**
+     * The timestamp when the role expires (ISO 8601 format)
+     */
+    expiresAt: string;
+    /**
+     * The host URL for database connection
+     */
+    host: string;
+    /**
+     * The username for database authentication
+     */
+    username: string;
+    /**
+     * The encrypted password for database authentication
+     */
+    password: Secret<string>;
+    /**
+     * The database name
+     */
+    databaseName: string;
+    /**
+     * The direct connection URL for the database.
+     */
+    connectionUrl: Secret<string>;
+    /**
+     * The pooled connection URL for the database.
+     * Uses PSBouncer on port 6432. Recommended for production.
+     * @see https://planetscale.com/docs/postgres/connecting/psbouncer
+     */
+    connectionUrlPooled: Secret<string>;
+    /**
+     * The roles that this role inherits from.
+     */
+    inheritedRoles: InheritedRole[];
+}
+/**
+ * Create and manage database roles for PlanetScale PostgreSQL branches. Database roles provide secure access to your database with specific roles and permissions.
+ *
+ * For MySQL, use [Passwords](./password.ts) instead.
+ *
+ * @example
+ * ## Basic Role
+ *
+ * Create a default role with all permissions:
+ *
+ * ```ts
+ * const role = await Role("my-role", {
+ *   database: "my-database",
+ *   inheritedRoles: ["postgres"],
+ * });
+ * ```
+ *
+ * ## Role with TTL
+ *
+ * Create a role with a TTL of 1 hour:
+ *
+ * ```ts
+ * const role = await Role("my-role", {
+ *   database: "my-database",
+ *   ttl: 3600,
+ * });
+ * ```
+ *
+ * ## Role with Inherited Permissions
+ *
+ * Create a role with read-only access to all data and settings:
+ *
+ * ```ts
+ * const role = await Role("my-role", {
+ *   database: "my-database",
+ *   inheritedRoles: ["pg_read_all_data", "pg_read_all_settings"],
+ * });
+ * ```
+ */
+export declare const Role: (((this: any, id: string, props?: {}) => never) & (new (_: never) => never)) | ((this: Context<Role, RoleProps>, id: string, props: RoleProps) => Promise<Role>);
+//# sourceMappingURL=role.d.ts.map

package/lib/planetscale/role.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"role.d.ts","sourceRoot":"","sources":["../../src/planetscale/role.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAE3C,OAAO,EAAqB,KAAK,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAE/E,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAG9C;;GAEG;AACH,MAAM,WAAW,SAAU,SAAQ,gBAAgB;IACjD;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB;;;OAGG;IACH,QAAQ,EAAE,MAAM,GAAG,QAAQ,CAAC;IAE5B;;;;OAIG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAEzB;;OAEG;IACH,GAAG,CAAC,EAAE,MAAM,CAAC;IAEb;;;;OAIG;IACH,cAAc,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC;CACxC;AAED;;GAEG;AACH,MAAM,MAAM,aAAa,GACrB,UAAU,GACV,eAAe,GACf,wBAAwB,GACxB,aAAa,GACb,YAAY,GACZ,kBAAkB,GAClB,sBAAsB,GACtB,mBAAmB,GACnB,mBAAmB,GACnB,qBAAqB,GACrB,6BAA6B,GAC7B,mBAAmB,GACnB,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;AAElB,MAAM,WAAW,IACf,SAAQ,QAAQ,CAAC,mBAAmB,CAAC,EACnC,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC;IACnC;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IAEX;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAElB;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IAEjB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAEzB;;OAEG;IACH,YAAY,EAAE,MAAM,CAAC;IAErB;;OAEG;IACH,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAE9B;;;;OAIG;IACH,mBAAmB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAEpC;;OAEG;IACH,cAAc,EAAE,aAAa,EAAE,CAAC;CACjC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,eAAO,MAAM,IAAI,yFAGP,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC,MAC1B,MAAM,SACH,SAAS,KACf,OAAO,CAAC,IAAI,CAAC,CA+GjB,CAAC"}

package/lib/planetscale/role.js

@@ -0,0 +1,144 @@
+import { alchemy } from "../alchemy.js";
+import { Resource } from "../resource.js";
+import { logger } from "../util/logger.js";
+import { PlanetScaleClient } from "./api/client.gen.js";
+import { waitForBranchReady } from "./utils.js";
+/**
+ * Create and manage database roles for PlanetScale PostgreSQL branches. Database roles provide secure access to your database with specific roles and permissions.
+ *
+ * For MySQL, use [Passwords](./password.ts) instead.
+ *
+ * @example
+ * ## Basic Role
+ *
+ * Create a default role with all permissions:
+ *
+ * ```ts
+ * const role = await Role("my-role", {
+ *   database: "my-database",
+ *   inheritedRoles: ["postgres"],
+ * });
+ * ```
+ *
+ * ## Role with TTL
+ *
+ * Create a role with a TTL of 1 hour:
+ *
+ * ```ts
+ * const role = await Role("my-role", {
+ *   database: "my-database",
+ *   ttl: 3600,
+ * });
+ * ```
+ *
+ * ## Role with Inherited Permissions
+ *
+ * Create a role with read-only access to all data and settings:
+ *
+ * ```ts
+ * const role = await Role("my-role", {
+ *   database: "my-database",
+ *   inheritedRoles: ["pg_read_all_data", "pg_read_all_settings"],
+ * });
+ * ```
+ */
+export const Role = Resource("planetscale::Role", async function (id, props) {
+    const api = new PlanetScaleClient(props);
+    const organization = typeof props.branch === "object"
+        ? props.branch.organizationId
+        : typeof props.database === "object"
+            ? props.database.organizationId
+            : props.organizationId;
+    const database = typeof props.database === "string" ? props.database : props.database.name;
+    const branch = typeof props.branch === "string"
+        ? props.branch
+        : (props.branch?.name ?? "main");
+    const inheritedRoles = Array.isArray(props.inheritedRoles)
+        ? props.inheritedRoles
+        : props.inheritedRoles.inheritedRoles;
+    if (!organization) {
+        throw new Error("Organization ID is required");
+    }
+    switch (this.phase) {
+        case "delete": {
+            if (this.output?.id) {
+                const res = await api.organizations.databases.branches.roles.delete({
+                    path: {
+                        organization,
+                        database,
+                        branch,
+                        id: this.output.id,
+                    },
+                    result: "full",
+                });
+                if (res.error) {
+                    switch (res.error.status) {
+                        case 404:
+                            break;
+                        case 422:
+                            // This is a workaround for 422 unprocessable: Role is still referenced and cannot be dropped.
+                            // Essentially, the role is still linked to a branch, so we can't delete it here, but it'll be deleted when the branch is deleted.
+                            logger.warn([
+                                `Failed to delete role "${id}" (name: "${this.output.name}").`,
+                                `Error: ${res.error.message}`,
+                                `If you are deleting database "${database}" or branch "${branch}", the role will be deleted automatically.`,
+                                `Otherwise, consider manually deleting the role at: https://app.planetscale.com/${organization}/${database}/settings/roles`,
+                            ].join("\n"));
+                            break;
+                        default:
+                            throw new Error(`Failed to delete role "${id}"`, {
+                                cause: res.error,
+                            });
+                    }
+                }
+            }
+            return this.destroy();
+        }
+        case "create": {
+            const { kind, ready } = await api.organizations.databases.branches.get({
+                path: {
+                    organization,
+                    database,
+                    name: branch,
+                },
+            });
+            if (kind !== "postgresql") {
+                throw new Error(`Cannot create a role on MySQL database "${database}". Roles are only supported on PostgreSQL databases. For MySQL databases, please use the Password resource instead.`);
+            }
+            // Cannot create role until branch is ready
+            if (!ready) {
+                await waitForBranchReady(api, organization, database, branch);
+            }
+            const role = await api.organizations.databases.branches.roles.post({
+                path: {
+                    organization,
+                    database,
+                    branch,
+                },
+                body: {
+                    ttl: props.ttl,
+                    inherited_roles: inheritedRoles,
+                },
+            });
+            return this({
+                ...props,
+                id: role.id,
+                name: role.name,
+                host: role.access_host_url,
+                username: role.username,
+                password: alchemy.secret(role.password),
+                expiresAt: role.expires_at,
+                databaseName: role.database_name,
+                inheritedRoles,
+                connectionUrl: alchemy.secret(`postgresql://${role.username}:${role.password}@${role.access_host_url}:5432/${role.database_name}?sslmode=verify-full`),
+                connectionUrlPooled: alchemy.secret(`postgresql://${role.username}:${role.password}@${role.access_host_url}:6432/${role.database_name}?sslmode=verify-full`),
+            });
+        }
+        case "update": {
+            // According to the types, the only property that can be updated is the name.
+            // However, I was getting 500 errors when trying to update the name, so we'll just replace.
+            return this.replace();
+        }
+    }
+});
+//# sourceMappingURL=role.js.map

package/lib/planetscale/role.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"role.js","sourceRoot":"","sources":["../../src/planetscale/role.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAExC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,iBAAiB,EAAyB,MAAM,qBAAqB,CAAC;AAI/E,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAgHhD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,MAAM,CAAC,MAAM,IAAI,GAAG,QAAQ,CAC1B,mBAAmB,EACnB,KAAK,WAEH,EAAU,EACV,KAAgB;IAEhB,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAAC,KAAK,CAAC,CAAC;IACzC,MAAM,YAAY,GAChB,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ;QAC9B,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc;QAC7B,CAAC,CAAC,OAAO,KAAK,CAAC,QAAQ,KAAK,QAAQ;YAClC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,cAAc;YAC/B,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC;IAC7B,MAAM,QAAQ,GACZ,OAAO,KAAK,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC;IAC5E,MAAM,MAAM,GACV,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ;QAC9B,CAAC,CAAC,KAAK,CAAC,MAAM;QACd,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,IAAI,MAAM,CAAC,CAAC;IACrC,MAAM,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,cAAc,CAAC;QACxD,CAAC,CAAC,KAAK,CAAC,cAAc;QACtB,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,cAAc,CAAC;IACxC,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACjD,CAAC;IAED,QAAQ,IAAI,CAAC,KAAK,EAAE,CAAC;QACnB,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,IAAI,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC;gBACpB,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC;oBAClE,IAAI,EAAE;wBACJ,YAAY;wBACZ,QAAQ;wBACR,MAAM;wBACN,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE;qBACnB;oBACD,MAAM,EAAE,MAAM;iBACf,CAAC,CAAC;gBACH,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;oBACd,QAAQ,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;wBACzB,KAAK,GAAG;4BACN,MAAM;wBACR,KAAK,GAAG;4BACN,8FAA8F;4BAC9F,kIAAkI;4BAClI,MAAM,CAAC,IAAI,CACT;gCACE,0BAA0B,EAAE,aAAa,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK;gCAC9D,UAAU,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE;gCAC7B,iCAAiC,QAAQ,gBAAgB,MAAM,4CAA4C;gCAC3G,kFAAkF,YAAY,IAAI,QAAQ,iBAAiB;6BAC5H,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;4BACF,MAAM;wBACR;4BACE,MAAM,IAAI,KAAK,CAAC,0BAA0B,EAAE,GAAG,EAAE;gCAC/C,KAAK,EAAE,GAAG,CAAC,KAAK;6BACjB,CAAC,CAAC;oBACP,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;QACxB,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC;gBACrE,IAAI,EAAE;oBACJ,YAAY;oBACZ,QAAQ;oBACR,IAAI,EAAE,MAAM;iBACb;aACF,CAAC,CAAC;YACH,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CACb,2CAA2C,QAAQ,qHAAqH,CACzK,CAAC;YACJ,CAAC;YACD,2CAA2C;YAC3C,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,kBAAkB,CAAC,GAAG,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;YAChE,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC;gBACjE,IAAI,EAAE;oBACJ,YAAY;oBACZ,QAAQ;oBACR,MAAM;iBACP;gBACD,IAAI,EAAE;oBACJ,GAAG,EAAE,KAAK,CAAC,GAAG;oBACd,eAAe,EAAE,cAAc;iBACN;aAC5B,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;gBACV,GAAG,KAAK;gBACR,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,IAAI,CAAC,eAAe;gBAC1B,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC;gBACvC,SAAS,EAAE,IAAI,CAAC,UAAU;gBAC1B,YAAY,EAAE,IAAI,CAAC,aAAa;gBAChC,cAAc;gBACd,aAAa,EAAE,OAAO,CAAC,MAAM,CAC3B,gBAAgB,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,eAAe,SAAS,IAAI,CAAC,aAAa,sBAAsB,CACxH;gBACD,mBAAmB,EAAE,OAAO,CAAC,MAAM,CACjC,gBAAgB,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,eAAe,SAAS,IAAI,CAAC,aAAa,sBAAsB,CACxH;aACF,CAAC,CAAC;QACL,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,6EAA6E;YAC7E,2FAA2F;YAC3F,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;QACxB,CAAC;IACH,CAAC;AACH,CAAC,CACF,CAAC"}

package/lib/planetscale/utils.d.ts

@@ -1,15 +1,31 @@
-import type { PlanetScaleApi } from "./api.ts";
+import type { PlanetScaleClient } from "./api/client.gen.ts";
+import type { DatabaseBranch } from "./api/types.gen.ts";
 export type PlanetScaleClusterSize = "PS_DEV" | "PS_10" | "PS_20" | "PS_40" | "PS_80" | "PS_160" | "PS_320" | "PS_400" | "PS_640" | "PS_700" | "PS_900" | "PS_1280" | "PS_1400" | "PS_1800" | "PS_2100" | "PS_2560" | "PS_2700" | "PS_2800" | (string & {});
 /**
- * Wait for a database to be ready with exponential backoff
+ * Ensures that the given cluster size is in the correct format.
  */
-export declare function waitForDatabaseReady(api: PlanetScaleApi, organizationId: string, databaseName: string, branchName?: string): Promise<void>;
+export declare function sanitizeClusterSize(input: {
+    size: PlanetScaleClusterSize;
+    kind?: "mysql" | "postgresql";
+    arch?: "x86" | "arm";
+    region?: string;
+}): string;
 /**
- * Wait for a keyspace to be ready
+ * Polls a branch until it is ready.
  */
-export declare function waitForKeyspaceReady(api: PlanetScaleApi, org: string, db: string, branch: string, keyspace: string): Promise<void>;
+export declare function waitForBranchReady(api: PlanetScaleClient, organization: string, database: string, branch: string): Promise<DatabaseBranch>;
 /**
- * Fix cluster size for a branch
+ * Polls a database until it is ready.
  */
-export declare function fixClusterSize(api: PlanetScaleApi, organizationId: string, databaseName: string, branchName: string, expectedClusterSize: PlanetScaleClusterSize, isDBReady: boolean): Promise<void>;
+export declare function waitForDatabaseReady(api: PlanetScaleClient, organization: string, database: string): Promise<void>;
+/**
+ * Polls a keyspace until it is finished resizing.
+ */
+export declare function waitForKeyspaceReady(api: PlanetScaleClient, organization: string, database: string, branch: string, keyspace: string): Promise<void>;
+/**
+ * Ensure a branch is production and has the correct cluster size.
+ * If a branch is not production, it will be promoted to production because
+ * cluster sizes can only be configured for production branches.
+ */
+export declare function ensureProductionBranchClusterSize(api: PlanetScaleClient, organization: string, database: string, branch: string, kind: "mysql" | "postgresql", expectedClusterSize: PlanetScaleClusterSize): Promise<void>;
 //# sourceMappingURL=utils.d.ts.map

package/lib/planetscale/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/planetscale/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE/C,MAAM,MAAM,sBAAsB,GAC9B,QAAQ,GACR,OAAO,GACP,OAAO,GACP,OAAO,GACP,OAAO,GACP,QAAQ,GACR,QAAQ,GACR,QAAQ,GACR,QAAQ,GACR,QAAQ,GACR,QAAQ,GACR,SAAS,GACT,SAAS,GACT,SAAS,GACT,SAAS,GACT,SAAS,GACT,SAAS,GACT,SAAS,GACT,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;AAElB;;GAEG;AACH,wBAAsB,oBAAoB,CACxC,GAAG,EAAE,cAAc,EACnB,cAAc,EAAE,MAAM,EACtB,YAAY,EAAE,MAAM,EACpB,UAAU,CAAC,EAAE,MAAM,GAClB,OAAO,CAAC,IAAI,CAAC,CA2Bf;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CACxC,GAAG,EAAE,cAAc,EACnB,GAAG,EAAE,MAAM,EACX,EAAE,EAAE,MAAM,EACV,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC,CAiCf;AAED;;GAEG;AACH,wBAAsB,cAAc,CAClC,GAAG,EAAE,cAAc,EACnB,cAAc,EAAE,MAAM,EACtB,YAAY,EAAE,MAAM,EACpB,UAAU,EAAE,MAAM,EAClB,mBAAmB,EAAE,sBAAsB,EAC3C,SAAS,EAAE,OAAO,GACjB,OAAO,CAAC,IAAI,CAAC,CA0Ef"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/planetscale/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAEzD,MAAM,MAAM,sBAAsB,GAC9B,QAAQ,GACR,OAAO,GACP,OAAO,GACP,OAAO,GACP,OAAO,GACP,QAAQ,GACR,QAAQ,GACR,QAAQ,GACR,QAAQ,GACR,QAAQ,GACR,QAAQ,GACR,SAAS,GACT,SAAS,GACT,SAAS,GACT,SAAS,GACT,SAAS,GACT,SAAS,GACT,SAAS,GACT,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;AAElB;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,KAAK,EAAE;IACzC,IAAI,EAAE,sBAAsB,CAAC;IAC7B,IAAI,CAAC,EAAE,OAAO,GAAG,YAAY,CAAC;IAC9B,IAAI,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,GAAG,MAAM,CAaT;AAED;;GAEG;AACH,wBAAsB,kBAAkB,CACtC,GAAG,EAAE,iBAAiB,EACtB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,cAAc,CAAC,CASzB;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CACxC,GAAG,EAAE,iBAAiB,EACtB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC,CASf;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CACxC,GAAG,EAAE,iBAAiB,EACtB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC,CAYf;AAED;;;;GAIG;AACH,wBAAsB,iCAAiC,CACrD,GAAG,EAAE,iBAAiB,EACtB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,OAAO,GAAG,YAAY,EAC5B,mBAAmB,EAAE,sBAAsB,GAC1C,OAAO,CAAC,IAAI,CAAC,CA2Bf"}

package/lib/planetscale/utils.js

@@ -1,95 +1,189 @@
 /**
- * Wait for a database to be ready with exponential backoff
+ * Ensures that the given cluster size is in the correct format.
  */
-export async function waitForDatabaseReady(api, organizationId, databaseName, branchName) {
-    const startTime = Date.now();
-    let waitMs = 100;
-    const branchSuffix = branchName ? `/branches/${branchName}` : "";
-    while (true) {
-        const response = await api.get(`/organizations/${organizationId}/databases/${databaseName}${branchSuffix}`);
-        if (!response.ok) {
-            throw new Error(`Failed to check database state: ${response.statusText}`);
-        }
-        const data = await response.json();
-        if (data.ready === true) {
-            return;
-        }
-        if (Date.now() - startTime >= 60000) {
-            throw new Error(`Timeout waiting for database ${databaseName} to be ready`);
-        }
-        await new Promise((resolve) => setTimeout(resolve, waitMs));
-        waitMs = Math.min(waitMs * 2, 1000); // Cap at 1s intervals
+export function sanitizeClusterSize(input) {
+    // NAS-backed Postgres cluster sizes are formatted as `PS_<size>_<provider>_<arch>`,
+    // where <provider> is either "AWS" or "GCP", and <arch> is either "ARM" or "X86".
+    // Postgres clusters backed by PlanetScale Metal are more complex (e.g. "M6_160_AWS_INTEL_D_METAL_118"),
+    // so to avoid messing with those, we just check for AWS or GCP in the size.
+    if (input.kind === "postgresql" && !input.size.match(/(AWS|GCP)/)) {
+        // Infer the provider from the region.
+        // Not all AWS regions start with "aws-", but all GCP regions start with "gcp-".
+        const provider = input.region?.startsWith("gcp") ? "GCP" : "AWS";
+        const arch = (input.arch ?? "x86").toUpperCase();
+        return `${input.size}_${provider}_${arch}`;
     }
+    return input.size;
 }
 /**
- * Wait for a keyspace to be ready
+ * Polls a branch until it is ready.
  */
-export async function waitForKeyspaceReady(api, org, db, branch, keyspace) {
-    const start = Date.now();
-    let delay = 100;
-    while (true) {
-        const res = await api.get(`/organizations/${org}/databases/${db}/branches/${branch}/keyspaces/${keyspace}/resizes`);
-        if (!res.ok) {
-            throw new Error(`Error fetching keyspace "${keyspace}": ${res.statusText}`);
-        }
-        const ks = await res.json();
-        // once it's fully ready, we can proceed
-        if (ks.data.every((item) => item.state !== "resizing")) {
-            return;
+export async function waitForBranchReady(api, organization, database, branch) {
+    return await poll({
+        description: `branch "${branch}" ready`,
+        fn: () => api.organizations.databases.branches.get({
+            path: { organization, database, name: branch },
+        }),
+        predicate: (branch) => branch.ready,
+    });
+}
+/**
+ * Polls a database until it is ready.
+ */
+export async function waitForDatabaseReady(api, organization, database) {
+    await poll({
+        description: `database "${database}" ready`,
+        fn: () => api.organizations.databases.get({
+            path: { organization, name: database },
+        }),
+        predicate: (database) => database.ready,
+    });
+}
+/**
+ * Polls a keyspace until it is finished resizing.
+ */
+export async function waitForKeyspaceReady(api, organization, database, branch, keyspace) {
+    await poll({
+        description: `keyspace "${keyspace}" ready`,
+        fn: () => api.organizations.databases.branches.keyspaces.resizes.list({
+            path: { organization, database, branch, name: keyspace },
+        }),
+        predicate: (response) => response.data.every((item) => item.state !== "resizing"),
+        initialDelay: 100,
+        maxDelay: 1000,
+    });
+}
+/**
+ * Ensure a branch is production and has the correct cluster size.
+ * If a branch is not production, it will be promoted to production because
+ * cluster sizes can only be configured for production branches.
+ */
+export async function ensureProductionBranchClusterSize(api, organization, database, branch, kind, expectedClusterSize) {
+    switch (kind) {
+        case "mysql": {
+            // Vitess databases must be promoted before resizing
+            await ensureProductionBranch(api, organization, database, branch);
+            await ensureMySQLClusterSize(api, organization, database, branch, expectedClusterSize);
+            break;
         }
-        if (Date.now() - start > 600_000) {
-            throw new Error(`Timeout waiting for keyspace "${keyspace}" to be ready`);
+        case "postgresql": {
+            // Postgres databases must be resized first before promoting, otherwise 500 error
+            await ensurePostgresClusterSize(api, organization, database, branch, expectedClusterSize);
+            await ensureProductionBranch(api, organization, database, branch);
+            break;
         }
-        await new Promise((r) => setTimeout(r, delay));
-        delay = Math.min(delay * 2, 1_000);
     }
 }
 /**
- * Fix cluster size for a branch
+ * Checks if a branch is production and promotes it if it is not.
  */
-export async function fixClusterSize(api, organizationId, databaseName, branchName, expectedClusterSize, isDBReady) {
-    if (!isDBReady) {
-        await waitForDatabaseReady(api, organizationId, databaseName);
-    }
-    // 1. Ensure branch is production
-    let branchRes = await api.get(`/organizations/${organizationId}/databases/${databaseName}/branches/${branchName}`);
-    if (!branchRes.ok) {
-        throw new Error(`Unable to get branch data: ${branchRes.statusText}`);
-    }
-    let branchData = await branchRes.json();
-    if (!branchData.production) {
-        if (!branchData.ready) {
-            await waitForDatabaseReady(api, organizationId, databaseName, branchName);
+async function ensureProductionBranch(api, organization, database, name) {
+    const data = await api.organizations.databases.branches.get({
+        path: {
+            organization,
+            database,
+            name,
+        },
+    });
+    if (!data.production) {
+        if (!data.ready) {
+            await waitForBranchReady(api, organization, database, name);
         }
-        const promoteRes = await api.post(`/organizations/${organizationId}/databases/${databaseName}/branches/${branchName}/promote`);
-        if (!promoteRes.ok) {
-            throw new Error(`Unable to promote branch: ${promoteRes.statusText}`);
-        }
-    }
-    // 2. Load default keyspace
-    const ksListRes = await api.get(`/organizations/${organizationId}/databases/${databaseName}/branches/${branchName}/keyspaces`);
-    if (!ksListRes.ok) {
-        throw new Error(`Failed to list keyspaces: ${ksListRes.statusText}`);
+        await api.organizations.databases.branches.promote.post({
+            path: { organization, database, name },
+        });
     }
-    const ksList = (await ksListRes.json()).data;
-    const defaultKsData = ksList.find((x) => x.name === databaseName); // Default keypsace is always the same name as the database
-    if (ksList.length === 0 || !defaultKsData) {
-        throw new Error(`No default keyspace found for branch ${branchName}`);
+}
+/**
+ * Ensures that a MySQL branch has the correct cluster size.
+ */
+async function ensureMySQLClusterSize(api, organization, database, branch, expectedClusterSize) {
+    // 1. Load default keyspace
+    const keyspaces = await api.organizations.databases.branches.keyspaces.list({
+        path: {
+            organization,
+            database,
+            branch,
+        },
+    });
+    const defaultKeyspace = keyspaces.data.find((x) => x.name === database); // Default keyspace is always the same name as the database
+    if (!defaultKeyspace) {
+        throw new Error(`No default keyspace found for branch ${branch}`);
     }
-    const defaultKs = defaultKsData.name;
-    let currentSize = defaultKsData.cluster_name;
-    // 3. Wait until any in-flight resize is done
-    await waitForKeyspaceReady(api, organizationId, databaseName, branchName, defaultKs);
-    // 4. If size mismatch, trigger resize and wait again
+    // 2. Wait until any in-flight resize is done
+    await waitForKeyspaceReady(api, organization, database, branch, defaultKeyspace.name);
+    // 3. If size mismatch, trigger resize and wait again
     // Ideally this would use the undocumented Keyspaces API, but there seems to be a missing oauth scope that we cannot add via the console yet
-    if (currentSize !== expectedClusterSize) {
-        const resizeRes = await api.patch(`/organizations/${organizationId}/databases/${databaseName}/branches/${branchName}/cluster`, { cluster_size: expectedClusterSize });
-        if (!resizeRes.ok) {
-            const text = await resizeRes.text();
-            throw new Error(`Failed to start resize: ${resizeRes.statusText} ${text}`);
-        }
+    if (defaultKeyspace.cluster_name !== expectedClusterSize) {
+        await api.organizations.databases.branches.cluster.patch({
+            path: {
+                organization,
+                database,
+                name: branch,
+            },
+            body: { cluster_size: expectedClusterSize },
+        });
         // Poll until the resize completes
-        await waitForKeyspaceReady(api, organizationId, databaseName, branchName, defaultKs);
+        await waitForKeyspaceReady(api, organization, database, branch, defaultKeyspace.name);
+    }
+}
+/**
+ * Ensures that a Postgres branch has the correct cluster size.
+ */
+async function ensurePostgresClusterSize(api, organization, database, branch, expectedClusterSize) {
+    const data = await api.organizations.databases.branches.get({
+        path: {
+            organization,
+            database,
+            name: branch,
+        },
+    });
+    if (data.cluster_name === expectedClusterSize) {
+        return;
+    }
+    await waitForPendingPostgresChanges(api, organization, database, branch);
+    const change = await api.organizations.databases.branches.changes.patch({
+        path: {
+            organization,
+            database,
+            branch,
+        },
+        body: {
+            cluster_size: expectedClusterSize,
+        },
+    });
+    await waitForPendingPostgresChanges(api, organization, database, branch, change.id);
+}
+/**
+ * Polls for a pending Postgres change to be completed.
+ */
+async function waitForPendingPostgresChanges(api, organization, database, branch, changeId) {
+    await poll({
+        description: `changes for branch "${branch}"`,
+        fn: () => api.organizations.databases.branches.changes.list({
+            path: { organization, database, branch },
+        }),
+        predicate: (response) => response.data.every((change) => change.state === "completed" ||
+            change.state === "canceled" ||
+            (changeId && change.id === changeId)),
+    });
+}
+/**
+ * Polls a function until it returns a result that satisfies the predicate.
+ */
+async function poll(input) {
+    const start = Date.now();
+    let delay = input.initialDelay ?? 250;
+    while (true) {
+        const result = await input.fn();
+        if (input.predicate(result)) {
+            return result;
+        }
+        if (Date.now() - start >= (input.timeout ?? 1_000_000)) {
+            throw new Error(`Timed out waiting for ${input.description} after ${Math.round((input.timeout ?? 1_000_000) / 1000)}s`);
+        }
+        await new Promise((resolve) => setTimeout(resolve, delay));
+        delay = Math.min(delay * 2, input.maxDelay ?? 5_000);
     }
 }
 //# sourceMappingURL=utils.js.map