alchemy 0.25.0 → 0.26.0

package/lib/planetscale/branch.js

@@ -0,0 +1,161 @@
+import { Resource } from "../resource.js";
+import { PlanetScaleApi } from "./api.js";
+import { fixClusterSize, waitForDatabaseReady, } from "./utils.js";
+/**
+ * Create or manage a PlanetScale database branch
+ *
+ * @example
+ * // Create a branch from 'main'
+ * const branch = await Branch("feature-123", {
+ *   name: "feature-123",
+ *   organizationId: "my-org",
+ *   databaseName: "my-database",
+ *   parentBranch: "main"
+ * });
+ *
+ * @example
+ * // Create a branch from another branch object
+ * const parentBranch = await Branch("staging", {
+ *   name: "staging",
+ *   organizationId: "my-org",
+ *   databaseName: "my-database",
+ *   parentBranch: "main"
+ * });
+ *
+ * const featureBranch = await Branch("feature-456", {
+ *   name: "feature-456",
+ *   organizationId: "my-org",
+ *   databaseName: "my-database",
+ *   parentBranch: parentBranch // Using Branch object instead of string
+ * });
+ *
+ * @example
+ * // Create a branch from a backup
+ * const branch = await Branch("restored-branch", {
+ *   name: "restored-branch",
+ *   organizationId: "my-org",
+ *   databaseName: "my-database",
+ *   parentBranch: "main",
+ *   backupId: "backup-123",
+ *   clusterSize: "PS_10"
+ * });
+ */
+export const Branch = Resource("planetscale::Branch", async function (_id, props) {
+    const apiKey = props.apiKey?.unencrypted || process.env.PLANETSCALE_API_TOKEN;
+    if (!apiKey) {
+        throw new Error("PLANETSCALE_API_TOKEN environment variable is required");
+    }
+    const api = new PlanetScaleApi({ apiKey });
+    if (this.phase === "delete") {
+        try {
+            if (this.output?.name) {
+                const response = await api.delete(`/organizations/${props.organizationId}/databases/${props.databaseName}/branches/${this.output.name}`);
+                if (!response.ok && response.status !== 404) {
+                    throw new Error(`Failed to delete branch: ${response.statusText} ${await response.text()}`);
+                }
+            }
+        }
+        catch (error) {
+            console.error("Error deleting branch:", error);
+            throw error;
+        }
+        return this.destroy();
+    }
+    try {
+        const parentBranchName = !props.parentBranch
+            ? "main"
+            : typeof props.parentBranch === "string"
+                ? props.parentBranch
+                : props.parentBranch.name;
+        if (typeof props.parentBranch !== "string" && props.parentBranch) {
+            await waitForDatabaseReady(api, props.organizationId, props.databaseName, props.parentBranch.name);
+        }
+        // Check if branch exists
+        const getResponse = await api.get(`/organizations/${props.organizationId}/databases/${props.databaseName}/branches/${props.name}`);
+        if (!getResponse.ok && getResponse.status !== 404) {
+            // Error getting branch
+            throw new Error(`Failed to get branch: ${getResponse.statusText} ${await getResponse.text()}`);
+        }
+        if (getResponse.ok) {
+            // Branch exists
+            if (!props.adopt) {
+                throw new Error(`Branch ${props.name} already exists and adopt is false`);
+            }
+            const data = await getResponse.json();
+            const currentParentBranch = data.parent_branch || "main";
+            // Check immutable properties
+            if (props.parentBranch && parentBranchName !== currentParentBranch) {
+                throw new Error(`Cannot change parent branch from ${currentParentBranch} to ${parentBranchName}`);
+            }
+            if (this.output?.backupId) {
+                console.warn("BackupID is set, but branch already exists, so it will be ignored");
+            }
+            if (this.output?.seedData) {
+                console.warn("SeedData is set, but branch already exists, so it will be ignored");
+            }
+            // Update mutable properties if they've changed
+            if (props.safeMigrations !== undefined) {
+                const safeMigrationsResponse = await api[props.safeMigrations ? "post" : "delete"](`/organizations/${props.organizationId}/databases/${props.databaseName}/branches/${props.name}/safe-migrations`);
+                if (!safeMigrationsResponse.ok) {
+                    throw new Error(`Failed to ${props.safeMigrations ? "enable" : "disable"} safe migrations: ${safeMigrationsResponse.statusText}`);
+                }
+            }
+            if (props.clusterSize && data.cluster_size !== props.clusterSize) {
+                const clusterResponse = await api.patch(`/organizations/${props.organizationId}/databases/${props.databaseName}/branches/${props.name}/cluster`, {
+                    cluster_size: props.clusterSize,
+                });
+                if (!clusterResponse.ok) {
+                    throw new Error(`Failed to update cluster size: ${clusterResponse.statusText}`);
+                }
+            }
+            return this({
+                ...props,
+                name: props.name,
+                parentBranch: currentParentBranch,
+                createdAt: data.created_at,
+                updatedAt: data.updated_at,
+                htmlUrl: data.html_url,
+            });
+        }
+        await waitForDatabaseReady(api, props.organizationId, props.databaseName);
+        // Branch doesn't exist, create it
+        const createResponse = await api.post(`/organizations/${props.organizationId}/databases/${props.databaseName}/branches`, {
+            name: props.name,
+            parent_branch: parentBranchName,
+            backup_id: props.backupId,
+            seed_data: props.seedData,
+            cluster_size: props.clusterSize,
+        });
+        if (!createResponse.ok) {
+            throw new Error(`Failed to create branch: ${createResponse.statusText} ${await createResponse.text()}`);
+        }
+        const data = await createResponse.json();
+        // Handle safe migrations if specified
+        if (props.safeMigrations !== undefined) {
+            // We can't change the migrations mode if the database is not ready
+            await waitForDatabaseReady(api, props.organizationId, props.databaseName, props.name);
+            const safeMigrationsResponse = await api[props.safeMigrations ? "post" : "delete"](`/organizations/${props.organizationId}/databases/${props.databaseName}/branches/${props.name}/safe-migrations`);
+            if (!safeMigrationsResponse.ok) {
+                throw new Error(`Failed to ${props.safeMigrations ? "enable" : "disable"} safe migrations: ${safeMigrationsResponse.statusText} : ${await safeMigrationsResponse.text()}`);
+            }
+        }
+        // Handle cluster size update if specified
+        if (props.clusterSize) {
+            // Wait for database to be ready before modifying cluster size
+            await fixClusterSize(api, props.organizationId, props.databaseName, props.name, props.clusterSize, true);
+        }
+        return this({
+            ...props,
+            name: props.name,
+            parentBranch: data.parent_branch,
+            createdAt: data.created_at,
+            updatedAt: data.updated_at,
+            htmlUrl: data.html_url,
+        });
+    }
+    catch (error) {
+        console.error("Error managing branch:", error);
+        throw error;
+    }
+});
+//# sourceMappingURL=branch.js.map

package/lib/planetscale/branch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"branch.js","sourceRoot":"","sources":["../../src/planetscale/branch.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC1C,OAAO,EACL,cAAc,EACd,oBAAoB,GAErB,MAAM,YAAY,CAAC;AAuGpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,MAAM,CAAC,MAAM,MAAM,GAAG,QAAQ,CAC5B,qBAAqB,EACrB,KAAK,WAEH,GAAW,EACX,KAAkB;IAElB,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;IAED,MAAM,GAAG,GAAG,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IAE3C,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,KAAK,CAAC,YAAY,aAAa,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CACtG,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,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YAC/C,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,YAAY;YAC1C,CAAC,CAAC,MAAM;YACR,CAAC,CAAC,OAAO,KAAK,CAAC,YAAY,KAAK,QAAQ;gBACtC,CAAC,CAAC,KAAK,CAAC,YAAY;gBACpB,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC;QAE9B,IAAI,OAAO,KAAK,CAAC,YAAY,KAAK,QAAQ,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;YACjE,MAAM,oBAAoB,CACxB,GAAG,EACH,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,YAAY,CAAC,IAAI,CACxB,CAAC;QACJ,CAAC;QAED,yBAAyB;QACzB,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,GAAG,CAC/B,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,YAAY,aAAa,KAAK,CAAC,IAAI,EAAE,CAChG,CAAC;QAEF,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,WAAW,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YAClD,uBAAuB;YACvB,MAAM,IAAI,KAAK,CACb,yBAAyB,WAAW,CAAC,UAAU,IAAI,MAAM,WAAW,CAAC,IAAI,EAAE,EAAE,CAC9E,CAAC;QACJ,CAAC;QAED,IAAI,WAAW,CAAC,EAAE,EAAE,CAAC;YACnB,gBAAgB;YAChB,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CACb,UAAU,KAAK,CAAC,IAAI,oCAAoC,CACzD,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,IAAI,EAAO,CAAC;YAC3C,MAAM,mBAAmB,GAAG,IAAI,CAAC,aAAa,IAAI,MAAM,CAAC;YAEzD,6BAA6B;YAC7B,IAAI,KAAK,CAAC,YAAY,IAAI,gBAAgB,KAAK,mBAAmB,EAAE,CAAC;gBACnE,MAAM,IAAI,KAAK,CACb,oCAAoC,mBAAmB,OAAO,gBAAgB,EAAE,CACjF,CAAC;YACJ,CAAC;YACD,IAAI,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,CAAC;gBAC1B,OAAO,CAAC,IAAI,CACV,mEAAmE,CACpE,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,CAAC;gBAC1B,OAAO,CAAC,IAAI,CACV,mEAAmE,CACpE,CAAC;YACJ,CAAC;YAED,+CAA+C;YAC/C,IAAI,KAAK,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;gBACvC,MAAM,sBAAsB,GAAG,MAAM,GAAG,CACtC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CACzC,CACC,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,YAAY,aAAa,KAAK,CAAC,IAAI,kBAAkB,CAChH,CAAC;gBAEF,IAAI,CAAC,sBAAsB,CAAC,EAAE,EAAE,CAAC;oBAC/B,MAAM,IAAI,KAAK,CACb,aAAa,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,qBAAqB,sBAAsB,CAAC,UAAU,EAAE,CACjH,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,YAAY,KAAK,KAAK,CAAC,WAAW,EAAE,CAAC;gBACjE,MAAM,eAAe,GAAG,MAAM,GAAG,CAAC,KAAK,CACrC,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,YAAY,aAAa,KAAK,CAAC,IAAI,UAAU,EACvG;oBACE,YAAY,EAAE,KAAK,CAAC,WAAW;iBAChC,CACF,CAAC;gBAEF,IAAI,CAAC,eAAe,CAAC,EAAE,EAAE,CAAC;oBACxB,MAAM,IAAI,KAAK,CACb,kCAAkC,eAAe,CAAC,UAAU,EAAE,CAC/D,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,OAAO,IAAI,CAAC;gBACV,GAAG,KAAK;gBACR,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,YAAY,EAAE,mBAAmB;gBACjC,SAAS,EAAE,IAAI,CAAC,UAAU;gBAC1B,SAAS,EAAE,IAAI,CAAC,UAAU;gBAC1B,OAAO,EAAE,IAAI,CAAC,QAAQ;aACvB,CAAC,CAAC;QACL,CAAC;QACD,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;QAE1E,kCAAkC;QAClC,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,IAAI,CACnC,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,YAAY,WAAW,EACjF;YACE,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,aAAa,EAAE,gBAAgB;YAC/B,SAAS,EAAE,KAAK,CAAC,QAAQ;YACzB,SAAS,EAAE,KAAK,CAAC,QAAQ;YACzB,YAAY,EAAE,KAAK,CAAC,WAAW;SAChC,CACF,CAAC;QAEF,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CACb,4BAA4B,cAAc,CAAC,UAAU,IAAI,MAAM,cAAc,CAAC,IAAI,EAAE,EAAE,CACvF,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,IAAI,EAAO,CAAC;QAE9C,sCAAsC;QACtC,IAAI,KAAK,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;YACvC,mEAAmE;YACnE,MAAM,oBAAoB,CACxB,GAAG,EACH,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,IAAI,CACX,CAAC;YACF,MAAM,sBAAsB,GAAG,MAAM,GAAG,CACtC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CACzC,CACC,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,YAAY,aAAa,KAAK,CAAC,IAAI,kBAAkB,CAChH,CAAC;YAEF,IAAI,CAAC,sBAAsB,CAAC,EAAE,EAAE,CAAC;gBAC/B,MAAM,IAAI,KAAK,CACb,aAAa,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,qBAAqB,sBAAsB,CAAC,UAAU,MAAM,MAAM,sBAAsB,CAAC,IAAI,EAAE,EAAE,CAC1J,CAAC;YACJ,CAAC;QACH,CAAC;QAED,0CAA0C;QAC1C,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;YACtB,8DAA8D;YAC9D,MAAM,cAAc,CAClB,GAAG,EACH,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,WAAW,EACjB,IAAI,CACL,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC;YACV,GAAG,KAAK;YACR,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,YAAY,EAAE,IAAI,CAAC,aAAa;YAChC,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,OAAO,EAAE,IAAI,CAAC,QAAQ;SACvB,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;QAC/C,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC,CACF,CAAC"}

package/lib/planetscale/database.d.ts

@@ -0,0 +1,143 @@
+import type { Context } from "../context.ts";
+import { Resource } from "../resource.ts";
+import type { Secret } from "../secret.ts";
+import { type PlanetScaleClusterSize } from "./utils.ts";
+/**
+ * Properties for creating or updating a PlanetScale Database
+ */
+export interface DatabaseProps {
+    /**
+     * The name of the database
+     */
+    name: string;
+    /**
+     * The organization ID where the database will be created
+     */
+    organizationId: string;
+    /**
+     * PlanetScale API token (overrides environment variable)
+     */
+    apiKey?: Secret;
+    /**
+     * Whether to adopt the database if it already exists in Planetscale
+     */
+    adopt?: boolean;
+    /**
+     * The region where the database will be created (create only)
+     */
+    region?: {
+        /**
+         * The slug identifier of the region
+         */
+        slug: string;
+    };
+    /**
+     * Whether to require approval for deployments
+     */
+    requireApprovalForDeploy?: boolean;
+    /**
+     * Whether to allow data branching
+     */
+    allowDataBranching?: boolean;
+    /**
+     * Whether to enable automatic migrations
+     */
+    automaticMigrations?: boolean;
+    /**
+     * Whether to restrict branch creation to the same region as database
+     */
+    restrictBranchRegion?: boolean;
+    /**
+     * Whether to collect full queries from the database
+     */
+    insightsRawQueries?: boolean;
+    /**
+     * Whether web console can be used on production branch
+     */
+    productionBranchWebConsole?: boolean;
+    /**
+     * The default branch of the database
+     */
+    defaultBranch?: string;
+    /**
+     * Migration framework to use on the database
+     */
+    migrationFramework?: string;
+    /**
+     * Name of table to use as migration table
+     */
+    migrationTableName?: string;
+    /**
+     * The database cluster size (required)
+     */
+    clusterSize: PlanetScaleClusterSize;
+}
+/**
+ * Represents a PlanetScale Database
+ */
+export interface Database extends Resource<"planetscale::Database">, DatabaseProps {
+    /**
+     * The unique identifier of the database
+     */
+    id: string;
+    /**
+     * The current state of the database
+     */
+    state: string;
+    /**
+     * The default branch name
+     */
+    defaultBranch: string;
+    /**
+     * The plan type
+     */
+    plan: string;
+    /**
+     * Time at which the database was created
+     */
+    createdAt: string;
+    /**
+     * Time at which the database was last updated
+     */
+    updatedAt: string;
+    /**
+     * HTML URL to access the database
+     */
+    htmlUrl: string;
+}
+/**
+ * Create, manage and delete PlanetScale databases
+ *
+ * @example
+ * // Create a basic database in a specific organization
+ * const db = await Database("my-app-db", {
+ *   name: "my-app-db",
+ *   organizationId: "my-org",
+ *   clusterSize: "PS_10"
+ * });
+ *
+ * @example
+ * // Create a database with specific region and settings
+ * const db = await Database("my-app-db", {
+ *   name: "my-app-db",
+ *   organizationId: "my-org",
+ *   region: {
+ *     slug: "us-east"
+ *   },
+ *   clusterSize: "PS_10",
+ *   requireApprovalForDeploy: true,
+ *   allowDataBranching: true,
+ *   automaticMigrations: true
+ * });
+ *
+ * @example
+ * // Create a database with custom API key
+ * const db = await Database("my-app-db", {
+ *   name: "my-app-db",
+ *   organizationId: "my-org",
+ *   apiKey: alchemy.secret(process.env.CUSTOM_PLANETSCALE_TOKEN),
+ *   clusterSize: "PS_10"
+ * });
+ */
+export declare const Database: (((this: any, id: string, props?: {}) => never) & (new (_: never) => never)) | ((this: Context<Database>, _id: string, props: DatabaseProps) => Promise<Database>);
+//# sourceMappingURL=database.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"database.d.ts","sourceRoot":"","sources":["../../src/planetscale/database.ts"],"names":[],"mappings":"AAAA,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,EAEL,KAAK,sBAAsB,EAE5B,MAAM,YAAY,CAAC;AAEpB;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,cAAc,EAAE,MAAM,CAAC;IAEvB;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;OAEG;IACH,KAAK,CAAC,EAAE,OAAO,CAAC;IAEhB;;OAEG;IACH,MAAM,CAAC,EAAE;QACP;;WAEG;QACH,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;IAEF;;OAEG;IACH,wBAAwB,CAAC,EAAE,OAAO,CAAC;IAEnC;;OAEG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAE7B;;OAEG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAE9B;;OAEG;IACH,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAE/B;;OAEG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAE7B;;OAEG;IACH,0BAA0B,CAAC,EAAE,OAAO,CAAC;IAErC;;OAEG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IAEvB;;OAEG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAE5B;;OAEG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAE5B;;OAEG;IACH,WAAW,EAAE,sBAAsB,CAAC;CACrC;AAED;;GAEG;AACH,MAAM,WAAW,QACf,SAAQ,QAAQ,CAAC,uBAAuB,CAAC,EACvC,aAAa;IACf;;OAEG;IACH,EAAE,EAAE,MAAM,CAAC;IAEX;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,aAAa,EAAE,MAAM,CAAC;IAEtB;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAElB;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAElB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AACH,eAAO,MAAM,QAAQ,yFAGX,OAAO,CAAC,QAAQ,CAAC,OAClB,MAAM,SACJ,aAAa,KACnB,OAAO,CAAC,QAAQ,CAAC,CAwNrB,CAAC"}

package/lib/planetscale/database.js

@@ -0,0 +1,184 @@
+import { Resource } from "../resource.js";
+import { PlanetScaleApi } from "./api.js";
+import { fixClusterSize, waitForDatabaseReady, } from "./utils.js";
+/**
+ * Create, manage and delete PlanetScale databases
+ *
+ * @example
+ * // Create a basic database in a specific organization
+ * const db = await Database("my-app-db", {
+ *   name: "my-app-db",
+ *   organizationId: "my-org",
+ *   clusterSize: "PS_10"
+ * });
+ *
+ * @example
+ * // Create a database with specific region and settings
+ * const db = await Database("my-app-db", {
+ *   name: "my-app-db",
+ *   organizationId: "my-org",
+ *   region: {
+ *     slug: "us-east"
+ *   },
+ *   clusterSize: "PS_10",
+ *   requireApprovalForDeploy: true,
+ *   allowDataBranching: true,
+ *   automaticMigrations: true
+ * });
+ *
+ * @example
+ * // Create a database with custom API key
+ * const db = await Database("my-app-db", {
+ *   name: "my-app-db",
+ *   organizationId: "my-org",
+ *   apiKey: alchemy.secret(process.env.CUSTOM_PLANETSCALE_TOKEN),
+ *   clusterSize: "PS_10"
+ * });
+ */
+export const Database = Resource("planetscale::Database", async function (_id, props) {
+    const apiKey = props.apiKey?.unencrypted || process.env.PLANETSCALE_API_TOKEN;
+    if (!apiKey) {
+        throw new Error("PLANETSCALE_API_TOKEN environment variable is required");
+    }
+    const api = new PlanetScaleApi({ apiKey });
+    if (this.phase === "delete") {
+        try {
+            if (this.output?.name) {
+                const response = await api.delete(`/organizations/${props.organizationId}/databases/${this.output.name}`);
+                if (!response.ok && response.status !== 404) {
+                    throw new Error(`Failed to delete database: ${response.statusText} ${await response.text()}`);
+                }
+            }
+        }
+        catch (error) {
+            console.error("Error deleting database:", error);
+            throw error;
+        }
+        return this.destroy();
+    }
+    try {
+        // Check if database exists
+        const getResponse = await api.get(`/organizations/${props.organizationId}/databases/${props.name}`);
+        const getData = await getResponse.json();
+        if (this.phase === "update" || (props.adopt && getResponse.ok)) {
+            if (!getResponse.ok) {
+                throw new Error(`Database ${props.name} not found`);
+            }
+            // Update database settings
+            // If updating to a non-'main' default branch, create it first
+            if (props.defaultBranch && props.defaultBranch !== "main") {
+                const branchResponse = await api.get(`/organizations/${props.organizationId}/databases/${props.name}/branches/${props.defaultBranch}`);
+                if (!getData.ready) {
+                    await waitForDatabaseReady(api, props.organizationId, props.name);
+                }
+                if (!branchResponse.ok && branchResponse.status === 404) {
+                    // Create the branch
+                    const createBranchResponse = await api.post(`/organizations/${props.organizationId}/databases/${props.name}/branches`, {
+                        name: props.defaultBranch,
+                        parent_branch: "main",
+                    });
+                    if (!createBranchResponse.ok) {
+                        throw new Error(`Failed to create default branch: ${createBranchResponse.statusText} ${await createBranchResponse.text()}`);
+                    }
+                }
+            }
+            const updateResponse = await api.patch(`/organizations/${props.organizationId}/databases/${props.name}`, {
+                automatic_migrations: props.automaticMigrations,
+                migration_framework: props.migrationFramework,
+                migration_table_name: props.migrationTableName,
+                require_approval_for_deploy: props.requireApprovalForDeploy,
+                restrict_branch_region: props.restrictBranchRegion,
+                allow_data_branching: props.allowDataBranching,
+                insights_raw_queries: props.insightsRawQueries,
+                production_branch_web_console: props.productionBranchWebConsole,
+                default_branch: props.defaultBranch,
+            });
+            if (!updateResponse.ok) {
+                throw new Error(`Failed to update database: ${updateResponse.statusText} ${await updateResponse.text()}`);
+            }
+            await fixClusterSize(api, props.organizationId, props.name, props.defaultBranch || "main", props.clusterSize, getData.ready);
+            const data = await updateResponse.json();
+            return this({
+                ...props,
+                id: data.id,
+                state: data.state,
+                defaultBranch: data.default_branch,
+                plan: data.plan,
+                createdAt: data.created_at,
+                updatedAt: data.updated_at,
+                htmlUrl: data.html_url,
+            });
+        }
+        if (getResponse.ok) {
+            throw new Error(`Database with name ${props.name} already exists`);
+        }
+        // Create new database
+        const createResponse = await api.post(`/organizations/${props.organizationId}/databases`, {
+            name: props.name,
+            region_slug: props.region?.slug,
+            require_approval_for_deploy: props.requireApprovalForDeploy,
+            allow_data_branching: props.allowDataBranching,
+            automatic_migrations: props.automaticMigrations,
+            restrict_branch_region: props.restrictBranchRegion,
+            insights_raw_queries: props.insightsRawQueries,
+            production_branch_web_console: props.productionBranchWebConsole,
+            migration_framework: props.migrationFramework,
+            migration_table_name: props.migrationTableName,
+            cluster_size: props.clusterSize,
+        });
+        if (!createResponse.ok) {
+            throw new Error(`Failed to create database: ${createResponse.statusText} ${await createResponse.text()}`);
+        }
+        const data = await createResponse.json();
+        // If a non-'main' default branch is specified, create it
+        if (props.defaultBranch && props.defaultBranch !== "main") {
+            await waitForDatabaseReady(api, props.organizationId, props.name);
+            // Check if branch exists
+            const branchResponse = await api.get(`/organizations/${props.organizationId}/databases/${props.name}/branches/${props.defaultBranch}`);
+            if (!branchResponse.ok && branchResponse.status === 404) {
+                // Create the branch
+                const createBranchResponse = await api.post(`/organizations/${props.organizationId}/databases/${props.name}/branches`, {
+                    name: props.defaultBranch,
+                    parent_branch: "main",
+                });
+                if (!createBranchResponse.ok) {
+                    throw new Error(`Failed to create default branch: ${createBranchResponse.statusText} ${await createBranchResponse.text()}`);
+                }
+                await fixClusterSize(api, props.organizationId, props.name, props.defaultBranch || "main", props.clusterSize, false);
+                // Update database to use new branch as default
+                const updateResponse = await api.patch(`/organizations/${props.organizationId}/databases/${props.name}`, {
+                    default_branch: props.defaultBranch,
+                });
+                if (!updateResponse.ok) {
+                    throw new Error(`Failed to set default branch: ${updateResponse.statusText} ${await updateResponse.text()}`);
+                }
+                const updatedData = await updateResponse.json();
+                return this({
+                    ...props,
+                    id: data.id,
+                    state: updatedData.state,
+                    defaultBranch: updatedData.default_branch,
+                    plan: updatedData.plan,
+                    createdAt: updatedData.created_at,
+                    updatedAt: updatedData.updated_at,
+                    htmlUrl: updatedData.html_url,
+                });
+            }
+        }
+        return this({
+            ...props,
+            id: data.id,
+            state: data.state,
+            defaultBranch: data.default_branch || "main",
+            plan: data.plan,
+            createdAt: data.created_at,
+            updatedAt: data.updated_at,
+            htmlUrl: data.html_url,
+        });
+    }
+    catch (error) {
+        console.error("Error managing database:", error);
+        throw error;
+    }
+});
+//# sourceMappingURL=database.js.map

package/lib/planetscale/database.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"database.js","sourceRoot":"","sources":["../../src/planetscale/database.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC1C,OAAO,EACL,cAAc,EAEd,oBAAoB,GACrB,MAAM,YAAY,CAAC;AAiIpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,QAAQ,CAC9B,uBAAuB,EACvB,KAAK,WAEH,GAAW,EACX,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;IAED,MAAM,GAAG,GAAG,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IAE3C,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,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CACvE,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBAC5C,MAAM,IAAI,KAAK,CACb,8BAA8B,QAAQ,CAAC,UAAU,IAAI,MAAM,QAAQ,CAAC,IAAI,EAAE,EAAE,CAC7E,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;YACjD,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC;IAED,IAAI,CAAC;QACH,2BAA2B;QAC3B,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,GAAG,CAC/B,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,IAAI,EAAE,CACjE,CAAC;QACF,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,IAAI,EAAO,CAAC;QAC9C,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YAC/D,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CAAC,YAAY,KAAK,CAAC,IAAI,YAAY,CAAC,CAAC;YACtD,CAAC;YACD,2BAA2B;YAC3B,8DAA8D;YAC9D,IAAI,KAAK,CAAC,aAAa,IAAI,KAAK,CAAC,aAAa,KAAK,MAAM,EAAE,CAAC;gBAC1D,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,GAAG,CAClC,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,IAAI,aAAa,KAAK,CAAC,aAAa,EAAE,CACjG,CAAC;gBACF,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;oBACnB,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gBACpE,CAAC;gBACD,IAAI,CAAC,cAAc,CAAC,EAAE,IAAI,cAAc,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBACxD,oBAAoB;oBACpB,MAAM,oBAAoB,GAAG,MAAM,GAAG,CAAC,IAAI,CACzC,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,IAAI,WAAW,EACzE;wBACE,IAAI,EAAE,KAAK,CAAC,aAAa;wBACzB,aAAa,EAAE,MAAM;qBACtB,CACF,CAAC;oBAEF,IAAI,CAAC,oBAAoB,CAAC,EAAE,EAAE,CAAC;wBAC7B,MAAM,IAAI,KAAK,CACb,oCAAoC,oBAAoB,CAAC,UAAU,IAAI,MAAM,oBAAoB,CAAC,IAAI,EAAE,EAAE,CAC3G,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;YAED,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,KAAK,CACpC,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,IAAI,EAAE,EAChE;gBACE,oBAAoB,EAAE,KAAK,CAAC,mBAAmB;gBAC/C,mBAAmB,EAAE,KAAK,CAAC,kBAAkB;gBAC7C,oBAAoB,EAAE,KAAK,CAAC,kBAAkB;gBAC9C,2BAA2B,EAAE,KAAK,CAAC,wBAAwB;gBAC3D,sBAAsB,EAAE,KAAK,CAAC,oBAAoB;gBAClD,oBAAoB,EAAE,KAAK,CAAC,kBAAkB;gBAC9C,oBAAoB,EAAE,KAAK,CAAC,kBAAkB;gBAC9C,6BAA6B,EAAE,KAAK,CAAC,0BAA0B;gBAC/D,cAAc,EAAE,KAAK,CAAC,aAAa;aACpC,CACF,CAAC;YAEF,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC;gBACvB,MAAM,IAAI,KAAK,CACb,8BAA8B,cAAc,CAAC,UAAU,IAAI,MAAM,cAAc,CAAC,IAAI,EAAE,EAAE,CACzF,CAAC;YACJ,CAAC;YAED,MAAM,cAAc,CAClB,GAAG,EACH,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,aAAa,IAAI,MAAM,EAC7B,KAAK,CAAC,WAAW,EACjB,OAAO,CAAC,KAAK,CACd,CAAC;YAEF,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,IAAI,EAAO,CAAC;YAC9C,OAAO,IAAI,CAAC;gBACV,GAAG,KAAK;gBACR,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,aAAa,EAAE,IAAI,CAAC,cAAc;gBAClC,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,SAAS,EAAE,IAAI,CAAC,UAAU;gBAC1B,SAAS,EAAE,IAAI,CAAC,UAAU;gBAC1B,OAAO,EAAE,IAAI,CAAC,QAAQ;aACvB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,WAAW,CAAC,EAAE,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,sBAAsB,KAAK,CAAC,IAAI,iBAAiB,CAAC,CAAC;QACrE,CAAC;QAED,sBAAsB;QACtB,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,IAAI,CACnC,kBAAkB,KAAK,CAAC,cAAc,YAAY,EAClD;YACE,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,WAAW,EAAE,KAAK,CAAC,MAAM,EAAE,IAAI;YAC/B,2BAA2B,EAAE,KAAK,CAAC,wBAAwB;YAC3D,oBAAoB,EAAE,KAAK,CAAC,kBAAkB;YAC9C,oBAAoB,EAAE,KAAK,CAAC,mBAAmB;YAC/C,sBAAsB,EAAE,KAAK,CAAC,oBAAoB;YAClD,oBAAoB,EAAE,KAAK,CAAC,kBAAkB;YAC9C,6BAA6B,EAAE,KAAK,CAAC,0BAA0B;YAC/D,mBAAmB,EAAE,KAAK,CAAC,kBAAkB;YAC7C,oBAAoB,EAAE,KAAK,CAAC,kBAAkB;YAC9C,YAAY,EAAE,KAAK,CAAC,WAAW;SAChC,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,yDAAyD;QACzD,IAAI,KAAK,CAAC,aAAa,IAAI,KAAK,CAAC,aAAa,KAAK,MAAM,EAAE,CAAC;YAC1D,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YAElE,yBAAyB;YACzB,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,GAAG,CAClC,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,IAAI,aAAa,KAAK,CAAC,aAAa,EAAE,CACjG,CAAC;YAEF,IAAI,CAAC,cAAc,CAAC,EAAE,IAAI,cAAc,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACxD,oBAAoB;gBACpB,MAAM,oBAAoB,GAAG,MAAM,GAAG,CAAC,IAAI,CACzC,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,IAAI,WAAW,EACzE;oBACE,IAAI,EAAE,KAAK,CAAC,aAAa;oBACzB,aAAa,EAAE,MAAM;iBACtB,CACF,CAAC;gBAEF,IAAI,CAAC,oBAAoB,CAAC,EAAE,EAAE,CAAC;oBAC7B,MAAM,IAAI,KAAK,CACb,oCAAoC,oBAAoB,CAAC,UAAU,IAAI,MAAM,oBAAoB,CAAC,IAAI,EAAE,EAAE,CAC3G,CAAC;gBACJ,CAAC;gBAED,MAAM,cAAc,CAClB,GAAG,EACH,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,aAAa,IAAI,MAAM,EAC7B,KAAK,CAAC,WAAW,EACjB,KAAK,CACN,CAAC;gBAEF,+CAA+C;gBAC/C,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,KAAK,CACpC,kBAAkB,KAAK,CAAC,cAAc,cAAc,KAAK,CAAC,IAAI,EAAE,EAChE;oBACE,cAAc,EAAE,KAAK,CAAC,aAAa;iBACpC,CACF,CAAC;gBAEF,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC;oBACvB,MAAM,IAAI,KAAK,CACb,iCAAiC,cAAc,CAAC,UAAU,IAAI,MAAM,cAAc,CAAC,IAAI,EAAE,EAAE,CAC5F,CAAC;gBACJ,CAAC;gBAED,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,IAAI,EAAO,CAAC;gBACrD,OAAO,IAAI,CAAC;oBACV,GAAG,KAAK;oBACR,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,KAAK,EAAE,WAAW,CAAC,KAAK;oBACxB,aAAa,EAAE,WAAW,CAAC,cAAc;oBACzC,IAAI,EAAE,WAAW,CAAC,IAAI;oBACtB,SAAS,EAAE,WAAW,CAAC,UAAU;oBACjC,SAAS,EAAE,WAAW,CAAC,UAAU;oBACjC,OAAO,EAAE,WAAW,CAAC,QAAQ;iBAC9B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;YACV,GAAG,KAAK;YACR,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,aAAa,EAAE,IAAI,CAAC,cAAc,IAAI,MAAM;YAC5C,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,OAAO,EAAE,IAAI,CAAC,QAAQ;SACvB,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC,CACF,CAAC"}

package/lib/planetscale/index.d.ts

@@ -0,0 +1,5 @@
+export * from "./api.ts";
+export * from "./utils.ts";
+export * from "./database.ts";
+export * from "./branch.ts";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/planetscale/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,YAAY,CAAC;AAC3B,cAAc,eAAe,CAAC;AAC9B,cAAc,aAAa,CAAC"}

package/lib/planetscale/index.js

@@ -0,0 +1,5 @@
+export * from "./api.js";
+export * from "./utils.js";
+export * from "./database.js";
+export * from "./branch.js";
+//# sourceMappingURL=index.js.map

package/lib/planetscale/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/planetscale/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,YAAY,CAAC;AAC3B,cAAc,eAAe,CAAC;AAC9B,cAAc,aAAa,CAAC"}

package/lib/planetscale/utils.d.ts

@@ -0,0 +1,15 @@
+import type { PlanetScaleApi } from "./api.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
+ */
+export declare function waitForDatabaseReady(api: PlanetScaleApi, organizationId: string, databaseName: string, branchName?: string): Promise<void>;
+/**
+ * Wait for a keyspace to be ready
+ */
+export declare function waitForKeyspaceReady(api: PlanetScaleApi, org: string, db: string, branch: string, keyspace: string): Promise<void>;
+/**
+ * Fix cluster size for a branch
+ */
+export declare function fixClusterSize(api: PlanetScaleApi, organizationId: string, databaseName: string, branchName: string, expectedClusterSize: PlanetScaleClusterSize, isDBReady: boolean): Promise<void>;
+//# sourceMappingURL=utils.d.ts.map

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

@@ -0,0 +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"}

package/lib/planetscale/utils.js

@@ -0,0 +1,95 @@
+/**
+ * Wait for a database to be ready with exponential backoff
+ */
+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
+    }
+}
+/**
+ * Wait for a keyspace to be 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;
+        }
+        if (Date.now() - start > 600_000) {
+            throw new Error(`Timeout waiting for keyspace "${keyspace}" to be ready`);
+        }
+        await new Promise((r) => setTimeout(r, delay));
+        delay = Math.min(delay * 2, 1_000);
+    }
+}
+/**
+ * Fix cluster size for a branch
+ */
+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);
+        }
+        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}`);
+    }
+    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}`);
+    }
+    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
+    // 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}`);
+        }
+        // Poll until the resize completes
+        await waitForKeyspaceReady(api, organizationId, databaseName, branchName, defaultKs);
+    }
+}
+//# sourceMappingURL=utils.js.map