alchemy 0.63.0 → 0.64.0

package/lib/planetscale/branch.js

@@ -1,6 +1,6 @@
 import { Resource } from "../resource.js";
-import { PlanetScaleApi } from "./api.js";
-import { fixClusterSize, waitForDatabaseReady, } from "./utils.js";
+import { PlanetScaleClient } from "./api/client.gen.js";
+import { ensureProductionBranchClusterSize, waitForDatabaseReady, } from "./utils.js";
 /**
  * Create or manage a PlanetScale database branch
  *
@@ -42,126 +42,140 @@ import { fixClusterSize, waitForDatabaseReady, } from "./utils.js";
  */
 export const Branch = Resource("planetscale::Branch", async function (id, props) {
     const adopt = props.adopt ?? this.scope.adopt;
-    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 });
+    const api = new PlanetScaleClient(props);
     const branchName = props.name ?? this.output?.name ?? this.scope.createPhysicalName(id);
     if (this.phase === "update" && this.output.name !== branchName) {
         // TODO(sam): maybe we don't need to replace? just branch again? or rename?
         this.replace();
     }
     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()}`);
-                }
+        if (this.output?.name) {
+            const response = await api.organizations.databases.branches.delete({
+                path: {
+                    organization: props.organizationId,
+                    database: props.databaseName,
+                    name: this.output.name,
+                },
+                result: "full",
+            });
+            if (response.error && response.error.status !== 404) {
+                throw new Error(`Failed to delete branch: ${response.error.message}`, {
+                    cause: response.error,
+                });
             }
         }
-        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);
+    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.organizations.databases.branches.get({
+        path: {
+            organization: props.organizationId,
+            database: props.databaseName,
+            name: branchName,
+        },
+        result: "full",
+    });
+    if (getResponse.error && getResponse.error.status !== 404) {
+        // Error getting branch
+        throw new Error(`Failed to get branch: ${getResponse.error.message}`, {
+            cause: getResponse.error,
+        });
+    }
+    if (getResponse.data) {
+        // Branch exists
+        if (!adopt) {
+            throw new Error(`Branch ${branchName} already exists and adopt is false`);
         }
-        // Check if branch exists
-        const getResponse = await api.get(`/organizations/${props.organizationId}/databases/${props.databaseName}/branches/${branchName}`);
-        if (!getResponse.ok && getResponse.status !== 404) {
-            // Error getting branch
-            throw new Error(`Failed to get branch: ${getResponse.statusText} ${await getResponse.text()}`);
+        const data = getResponse.data;
+        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 (getResponse.ok) {
-            // Branch exists
-            if (!adopt) {
-                throw new Error(`Branch ${branchName} 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/${branchName}/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/${branchName}/cluster`, {
-                    cluster_size: props.clusterSize,
-                });
-                if (!clusterResponse.ok) {
-                    throw new Error(`Failed to update cluster size: ${clusterResponse.statusText}`);
-                }
-            }
-            return this({
-                ...props,
-                name: branchName,
-                parentBranch: currentParentBranch,
-                createdAt: data.created_at,
-                updatedAt: data.updated_at,
-                htmlUrl: data.html_url,
-            });
+        if (this.output?.backupId) {
+            console.warn("BackupID is set, but branch already exists, so it will be ignored");
         }
-        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: branchName,
-            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()}`);
+        if (this.output?.seedData) {
+            console.warn("SeedData is set, but branch already exists, so it will be ignored");
         }
-        const data = await createResponse.json();
-        // Handle safe migrations if specified
+        // Update mutable properties if they've changed
         if (props.safeMigrations !== undefined) {
-            // We can't change the migrations mode if the database is not ready
-            await waitForDatabaseReady(api, props.organizationId, props.databaseName, branchName);
-            const safeMigrationsResponse = await api[props.safeMigrations ? "post" : "delete"](`/organizations/${props.organizationId}/databases/${props.databaseName}/branches/${branchName}/safe-migrations`);
-            if (!safeMigrationsResponse.ok) {
-                throw new Error(`Failed to ${props.safeMigrations ? "enable" : "disable"} safe migrations: ${safeMigrationsResponse.statusText} : ${await safeMigrationsResponse.text()}`);
-            }
+            await api.organizations.databases.branches.safeMigrations[props.safeMigrations ? "post" : "delete"]({
+                path: {
+                    organization: props.organizationId,
+                    database: props.databaseName,
+                    name: branchName,
+                },
+            });
         }
-        // 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, branchName, props.clusterSize, true);
+        if (props.clusterSize && data.cluster_name !== props.clusterSize) {
+            await api.organizations.databases.branches.cluster.patch({
+                path: {
+                    organization: props.organizationId,
+                    database: props.databaseName,
+                    name: branchName,
+                },
+                body: {
+                    cluster_size: props.clusterSize,
+                },
+            });
         }
         return this({
             ...props,
             name: branchName,
-            parentBranch: data.parent_branch,
+            parentBranch: currentParentBranch,
             createdAt: data.created_at,
             updatedAt: data.updated_at,
             htmlUrl: data.html_url,
         });
     }
-    catch (error) {
-        console.error("Error managing branch:", error);
-        throw error;
+    await waitForDatabaseReady(api, props.organizationId, props.databaseName);
+    // Branch doesn't exist, create it
+    const data = await api.organizations.databases.branches.post({
+        path: {
+            organization: props.organizationId,
+            database: props.databaseName,
+        },
+        body: {
+            name: branchName,
+            parent_branch: parentBranchName,
+            backup_id: props.backupId,
+            seed_data: props.seedData,
+            cluster_size: props.clusterSize,
+        },
+    });
+    // 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, branchName);
+        await api.organizations.databases.branches.safeMigrations[props.safeMigrations ? "post" : "delete"]({
+            path: {
+                organization: props.organizationId,
+                database: props.databaseName,
+                name: branchName,
+            },
+        });
+    }
+    // Handle cluster size update if specified
+    if (props.clusterSize) {
+        // Wait for database to be ready before modifying cluster size
+        await ensureProductionBranchClusterSize(api, props.organizationId, props.databaseName, branchName, props.clusterSize, true);
     }
+    return this({
+        ...props,
+        name: branchName,
+        parentBranch: data.parent_branch,
+        createdAt: data.created_at,
+        updatedAt: data.updated_at,
+        htmlUrl: data.html_url,
+    });
 });
 //# sourceMappingURL=branch.js.map

package/lib/planetscale/branch.js.map

@@ -1 +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;AAyGpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,MAAM,CAAC,MAAM,MAAM,GAAG,QAAQ,CAC5B,qBAAqB,EACrB,KAAK,WAEH,EAAU,EACV,KAAkB;IAElB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;IAC9C,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,MAAM,UAAU,GACd,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;IAEvE,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAC/D,2EAA2E;QAC3E,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAED,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,UAAU,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,EAAE,CAAC;gBACX,MAAM,IAAI,KAAK,CACb,UAAU,UAAU,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,UAAU,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,UAAU,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,UAAU;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,UAAU;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,UAAU,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,UAAU,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,UAAU,EACV,KAAK,CAAC,WAAW,EACjB,IAAI,CACL,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC;YACV,GAAG,KAAK;YACR,IAAI,EAAE,UAAU;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"}
+{"version":3,"file":"branch.js","sourceRoot":"","sources":["../../src/planetscale/branch.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EACL,iCAAiC,EACjC,oBAAoB,GAErB,MAAM,YAAY,CAAC;AAqGpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AACH,MAAM,CAAC,MAAM,MAAM,GAAG,QAAQ,CAC5B,qBAAqB,EACrB,KAAK,WAEH,EAAU,EACV,KAAkB;IAElB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;IAE9C,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAEzC,MAAM,UAAU,GACd,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;IAEvE,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAC/D,2EAA2E;QAC3E,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAED,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5B,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC;YACtB,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACjE,IAAI,EAAE;oBACJ,YAAY,EAAE,KAAK,CAAC,cAAc;oBAClC,QAAQ,EAAE,KAAK,CAAC,YAAY;oBAC5B,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;iBACvB;gBACD,MAAM,EAAE,MAAM;aACf,CAAC,CAAC;YAEH,IAAI,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACpD,MAAM,IAAI,KAAK,CACb,4BAA4B,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,EACpD;oBACE,KAAK,EAAE,QAAQ,CAAC,KAAK;iBACtB,CACF,CAAC;YACJ,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC;IAED,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,YAAY;QAC1C,CAAC,CAAC,MAAM;QACR,CAAC,CAAC,OAAO,KAAK,CAAC,YAAY,KAAK,QAAQ;YACtC,CAAC,CAAC,KAAK,CAAC,YAAY;YACpB,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC;IAE9B,IAAI,OAAO,KAAK,CAAC,YAAY,KAAK,QAAQ,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;QACjE,MAAM,oBAAoB,CACxB,GAAG,EACH,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,YAAY,CAAC,IAAI,CACxB,CAAC;IACJ,CAAC;IAED,yBAAyB;IACzB,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC;QACjE,IAAI,EAAE;YACJ,YAAY,EAAE,KAAK,CAAC,cAAc;YAClC,QAAQ,EAAE,KAAK,CAAC,YAAY;YAC5B,IAAI,EAAE,UAAU;SACjB;QACD,MAAM,EAAE,MAAM;KACf,CAAC,CAAC;IAEH,IAAI,WAAW,CAAC,KAAK,IAAI,WAAW,CAAC,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QAC1D,uBAAuB;QACvB,MAAM,IAAI,KAAK,CAAC,yBAAyB,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE;YACpE,KAAK,EAAE,WAAW,CAAC,KAAK;SACzB,CAAC,CAAC;IACL,CAAC;IAED,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC;QACrB,gBAAgB;QAChB,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CACb,UAAU,UAAU,oCAAoC,CACzD,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC;QAC9B,MAAM,mBAAmB,GAAG,IAAI,CAAC,aAAa,IAAI,MAAM,CAAC;QAEzD,6BAA6B;QAC7B,IAAI,KAAK,CAAC,YAAY,IAAI,gBAAgB,KAAK,mBAAmB,EAAE,CAAC;YACnE,MAAM,IAAI,KAAK,CACb,oCAAoC,mBAAmB,OAAO,gBAAgB,EAAE,CACjF,CAAC;QACJ,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,CAAC;YAC1B,OAAO,CAAC,IAAI,CACV,mEAAmE,CACpE,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,CAAC;YAC1B,OAAO,CAAC,IAAI,CACV,mEAAmE,CACpE,CAAC;QACJ,CAAC;QAED,+CAA+C;QAC/C,IAAI,KAAK,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;YACvC,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,cAAc,CACvD,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CACzC,CAAC;gBACA,IAAI,EAAE;oBACJ,YAAY,EAAE,KAAK,CAAC,cAAc;oBAClC,QAAQ,EAAE,KAAK,CAAC,YAAY;oBAC5B,IAAI,EAAE,UAAU;iBACjB;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,YAAY,KAAK,KAAK,CAAC,WAAW,EAAE,CAAC;YACjE,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC;gBACvD,IAAI,EAAE;oBACJ,YAAY,EAAE,KAAK,CAAC,cAAc;oBAClC,QAAQ,EAAE,KAAK,CAAC,YAAY;oBAC5B,IAAI,EAAE,UAAU;iBACjB;gBACD,IAAI,EAAE;oBACJ,YAAY,EAAE,KAAK,CAAC,WAAW;iBAChC;aACF,CAAC,CAAC;QACL,CAAC;QAED,OAAO,IAAI,CAAC;YACV,GAAG,KAAK;YACR,IAAI,EAAE,UAAU;YAChB,YAAY,EAAE,mBAAmB;YACjC,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,OAAO,EAAE,IAAI,CAAC,QAAQ;SACvB,CAAC,CAAC;IACL,CAAC;IACD,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;IAE1E,kCAAkC;IAClC,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC;QAC3D,IAAI,EAAE;YACJ,YAAY,EAAE,KAAK,CAAC,cAAc;YAClC,QAAQ,EAAE,KAAK,CAAC,YAAY;SAC7B;QACD,IAAI,EAAE;YACJ,IAAI,EAAE,UAAU;YAChB,aAAa,EAAE,gBAAgB;YAC/B,SAAS,EAAE,KAAK,CAAC,QAAQ;YACzB,SAAS,EAAE,KAAK,CAAC,QAAQ;YACzB,YAAY,EAAE,KAAK,CAAC,WAAW;SAChC;KACF,CAAC,CAAC;IAEH,sCAAsC;IACtC,IAAI,KAAK,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;QACvC,mEAAmE;QACnE,MAAM,oBAAoB,CACxB,GAAG,EACH,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,YAAY,EAClB,UAAU,CACX,CAAC;QACF,MAAM,GAAG,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,cAAc,CACvD,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CACzC,CAAC;YACA,IAAI,EAAE;gBACJ,YAAY,EAAE,KAAK,CAAC,cAAc;gBAClC,QAAQ,EAAE,KAAK,CAAC,YAAY;gBAC5B,IAAI,EAAE,UAAU;aACjB;SACF,CAAC,CAAC;IACL,CAAC;IAED,0CAA0C;IAC1C,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACtB,8DAA8D;QAC9D,MAAM,iCAAiC,CACrC,GAAG,EACH,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,YAAY,EAClB,UAAU,EACV,KAAK,CAAC,WAAW,EACjB,IAAI,CACL,CAAC;IACJ,CAAC;IAED,OAAO,IAAI,CAAC;QACV,GAAG,KAAK;QACR,IAAI,EAAE,UAAU;QAChB,YAAY,EAAE,IAAI,CAAC,aAAa;QAChC,SAAS,EAAE,IAAI,CAAC,UAAU;QAC1B,SAAS,EAAE,IAAI,CAAC,UAAU;QAC1B,OAAO,EAAE,IAAI,CAAC,QAAQ;KACvB,CAAC,CAAC;AACL,CAAC,CACF,CAAC"}

package/lib/planetscale/database.d.ts

@@ -1,11 +1,11 @@
 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 PlanetScaleClusterSize } from "./utils.ts";
 /**
  * Properties for creating or updating a PlanetScale Database
  */
-export interface DatabaseProps {
+export interface DatabaseProps extends PlanetScaleProps {
     /**
      * The name of the database
      *
@@ -16,10 +16,6 @@ export interface DatabaseProps {
      * 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
      */

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

@@ -1 +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;;;;OAIG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;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,IAAI,EAAE,MAAM,CAAC;IAEb;;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,MACnB,MAAM,SACH,aAAa,KACnB,OAAO,CAAC,QAAQ,CAAC,CAsOrB,CAAC"}
+{"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,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAE5D,OAAO,EAEL,KAAK,sBAAsB,EAE5B,MAAM,YAAY,CAAC;AAEpB;;GAEG;AACH,MAAM,WAAW,aAAc,SAAQ,gBAAgB;IACrD;;;;OAIG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,cAAc,EAAE,MAAM,CAAC;IAEvB;;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,IAAI,EAAE,MAAM,CAAC;IAEb;;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,MACnB,MAAM,SACH,aAAa,KACnB,OAAO,CAAC,QAAQ,CAAC,CAkOrB,CAAC"}

package/lib/planetscale/database.js

@@ -1,6 +1,6 @@
 import { Resource } from "../resource.js";
-import { PlanetScaleApi } from "./api.js";
-import { fixClusterSize, waitForDatabaseReady, } from "./utils.js";
+import { PlanetScaleClient } from "./api/client.gen.js";
+import { ensureProductionBranchClusterSize, waitForDatabaseReady, } from "./utils.js";
 /**
  * Create, manage and delete PlanetScale databases
  *
@@ -36,58 +36,82 @@ import { fixClusterSize, waitForDatabaseReady, } from "./utils.js";
  * });
  */
 export const Database = Resource("planetscale::Database", async function (id, props) {
-    const adopt = props.adopt ?? this.scope.adopt;
-    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 });
+    const api = new PlanetScaleClient(props);
     const databaseName = props.name ?? this.output?.name ?? this.scope.createPhysicalName(id);
     if (this.phase === "update" && this.output.name !== databaseName) {
-        await api.patch(`/organizations/${props.organizationId}/databases/${this.output.name}`, { new_name: databaseName });
+        await api.organizations.databases.patch({
+            path: {
+                organization: props.organizationId,
+                name: this.output.name,
+            },
+            body: { new_name: databaseName },
+        });
     }
     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()}`);
-                }
+        if (this.output?.name) {
+            const response = await api.organizations.databases.delete({
+                path: {
+                    organization: props.organizationId,
+                    name: this.output.name,
+                },
+                result: "full",
+            });
+            if (response.error && response.error.status !== 404) {
+                throw new Error(`Failed to delete database "${this.output.name}"`, {
+                    cause: response.error,
+                });
             }
         }
-        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/${databaseName}`);
-        const getData = await getResponse.json();
-        if (this.phase === "update" || (adopt && getResponse.ok)) {
-            if (!getResponse.ok) {
-                throw new Error(`Database ${databaseName} not found`);
+    // Check if database exists
+    const getResponse = await api.organizations.databases.get({
+        path: {
+            organization: props.organizationId,
+            name: databaseName,
+        },
+        result: "full",
+    });
+    if (this.phase === "update" || (props.adopt && getResponse.data)) {
+        if (!getResponse.data) {
+            throw new Error(`Database "${databaseName}" not found`, {
+                cause: getResponse.error,
+            });
+        }
+        // Update database settings
+        // If updating to a non-'main' default branch, create it first
+        if (props.defaultBranch && props.defaultBranch !== "main") {
+            const branchResponse = await api.organizations.databases.branches.get({
+                path: {
+                    organization: props.organizationId,
+                    database: databaseName,
+                    name: props.defaultBranch,
+                },
+                result: "full",
+            });
+            if (!branchResponse.data) {
+                await waitForDatabaseReady(api, props.organizationId, databaseName);
             }
-            // 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/${databaseName}/branches/${props.defaultBranch}`);
-                if (!getData.ready) {
-                    await waitForDatabaseReady(api, props.organizationId, databaseName);
-                }
-                if (!branchResponse.ok && branchResponse.status === 404) {
-                    // Create the branch
-                    const createBranchResponse = await api.post(`/organizations/${props.organizationId}/databases/${databaseName}/branches`, {
+            if (branchResponse.error && branchResponse.error.status === 404) {
+                // Create the branch
+                await api.organizations.databases.branches.post({
+                    path: {
+                        organization: props.organizationId,
+                        database: databaseName,
+                    },
+                    body: {
                         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/${databaseName}`, {
+        }
+        const updateResponse = await api.organizations.databases.patch({
+            path: {
+                organization: props.organizationId,
+                name: databaseName,
+            },
+            body: {
                 automatic_migrations: props.automaticMigrations,
                 migration_framework: props.migrationFramework,
                 migration_table_name: props.migrationTableName,
@@ -97,31 +121,43 @@ export const Database = Resource("planetscale::Database", async function (id, pr
                 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, databaseName, props.defaultBranch || "main", props.clusterSize, getData.ready);
-            const data = await updateResponse.json();
-            return this({
-                ...props,
-                id: data.id,
-                name: databaseName,
-                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 ${databaseName} already exists`);
-        }
-        // Create new database
-        const createResponse = await api.post(`/organizations/${props.organizationId}/databases`, {
+            },
+        });
+        await ensureProductionBranchClusterSize(api, props.organizationId, databaseName, props.defaultBranch || "main", props.clusterSize, updateResponse.ready);
+        return this({
+            ...props,
+            id: updateResponse.id,
             name: databaseName,
-            region_slug: props.region?.slug,
+            state: updateResponse.state,
+            defaultBranch: updateResponse.default_branch,
+            plan: updateResponse.plan,
+            createdAt: updateResponse.created_at,
+            updatedAt: updateResponse.updated_at,
+            htmlUrl: updateResponse.html_url,
+        });
+    }
+    if (getResponse.data) {
+        throw new Error(`Database with name "${databaseName}" already exists`);
+    }
+    // Create new database
+    let data = await api.organizations.databases.post({
+        path: {
+            organization: props.organizationId,
+        },
+        body: {
+            name: databaseName,
+            region: props.region?.slug,
+            kind: "mysql",
+            cluster_size: props.clusterSize,
+        },
+    });
+    // These settings can't be set on creation, so we need to patch them after creation.
+    data = await api.organizations.databases.patch({
+        path: {
+            organization: props.organizationId,
+            name: databaseName,
+        },
+        body: {
             require_approval_for_deploy: props.requireApprovalForDeploy,
             allow_data_branching: props.allowDataBranching,
             automatic_migrations: props.automaticMigrations,
@@ -130,63 +166,66 @@ export const Database = Resource("planetscale::Database", async function (id, pr
             production_branch_web_console: props.productionBranchWebConsole,
             migration_framework: props.migrationFramework,
             migration_table_name: props.migrationTableName,
-            cluster_size: props.clusterSize,
+        },
+    });
+    // If a non-'main' default branch is specified, create it
+    if (props.defaultBranch && props.defaultBranch !== "main") {
+        await waitForDatabaseReady(api, props.organizationId, databaseName);
+        // Check if branch exists
+        const branchResponse = await api.organizations.databases.branches.get({
+            path: {
+                organization: props.organizationId,
+                database: databaseName,
+                name: props.defaultBranch,
+            },
+            result: "full",
         });
-        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, databaseName);
-            // Check if branch exists
-            const branchResponse = await api.get(`/organizations/${props.organizationId}/databases/${databaseName}/branches/${props.defaultBranch}`);
-            if (!branchResponse.ok && branchResponse.status === 404) {
-                // Create the branch
-                const createBranchResponse = await api.post(`/organizations/${props.organizationId}/databases/${databaseName}/branches`, {
+        if (branchResponse.error && branchResponse.error.status === 404) {
+            // Create the branch
+            await api.organizations.databases.branches.post({
+                path: {
+                    organization: props.organizationId,
+                    database: databaseName,
+                },
+                body: {
                     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, databaseName, props.defaultBranch || "main", props.clusterSize, false);
-                // Update database to use new branch as default
-                const updateResponse = await api.patch(`/organizations/${props.organizationId}/databases/${databaseName}`, {
-                    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,
+                },
+            });
+            await ensureProductionBranchClusterSize(api, props.organizationId, databaseName, props.defaultBranch || "main", props.clusterSize, false);
+            // Update database to use new branch as default
+            const updatedData = await api.organizations.databases.patch({
+                path: {
+                    organization: props.organizationId,
                     name: databaseName,
-                    state: updatedData.state,
-                    defaultBranch: updatedData.default_branch,
-                    plan: updatedData.plan,
-                    createdAt: updatedData.created_at,
-                    updatedAt: updatedData.updated_at,
-                    htmlUrl: updatedData.html_url,
-                });
-            }
+                },
+                body: {
+                    default_branch: props.defaultBranch,
+                },
+            });
+            return this({
+                ...props,
+                id: data.id,
+                name: databaseName,
+                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,
-            name: databaseName,
-            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;
     }
+    return this({
+        ...props,
+        id: data.id,
+        name: databaseName,
+        state: data.state,
+        defaultBranch: data.default_branch || "main",
+        plan: data.plan,
+        createdAt: data.created_at,
+        updatedAt: data.updated_at,
+        htmlUrl: data.html_url,
+    });
 });
 //# sourceMappingURL=database.js.map