pgbeam 0.1.0 → 0.2.1

package/dist/{index.js → index.mjs}

@@ -1,3 +1,5 @@
+import { _ } from '@swc/helpers/_/_extends';
+
 /**
  * Generated by scripts/generate-maps.ts — do not edit manually.
  * Run: pnpm --filter pgbeam generate
@@ -6,9 +8,17 @@
         exportAccountData: {
             method: "GET",
             path: "/v1/account/export"
+        },
+        getOnboardingProgress: {
+            method: "GET",
+            path: "/v1/organizations/{org_id}/onboarding"
+        },
+        updateOnboardingProgress: {
+            method: "PATCH",
+            path: "/v1/organizations/{org_id}/onboarding"
         }
     },
-    usage: {
+    analytics: {
         getOrganizationActivity: {
             method: "GET",
             path: "/v1/organizations/{org_id}/activity"
@@ -21,6 +31,14 @@
             method: "GET",
             path: "/v1/organizations/{org_id}/plan"
         },
+        updateSpendLimit: {
+            method: "PUT",
+            path: "/v1/organizations/{org_id}/spend-limit"
+        },
+        submitCancellationFeedback: {
+            method: "POST",
+            path: "/v1/organizations/{org_id}/cancellation-feedback"
+        },
         getProjectInsights: {
             method: "GET",
             path: "/v1/projects/{project_id}/insights"
@@ -34,43 +52,15 @@
             path: "/v1/projects/{project_id}/usage"
         }
     },
-    catalog: {
-        listQueryCatalog: {
+    projects: {
+        listCacheRules: {
             method: "GET",
-            path: "/v1/projects/{project_id}/databases/{database_id}/catalog"
+            path: "/v1/projects/{project_id}/databases/{database_id}/cache-rules"
         },
         updateCacheRule: {
             method: "PUT",
-            path: "/v1/projects/{project_id}/databases/{database_id}/catalog/{query_hash}/cache-rule"
-        }
-    },
-    databases: {
-        listDatabases: {
-            method: "GET",
-            path: "/v1/projects/{project_id}/databases"
-        },
-        createDatabase: {
-            method: "POST",
-            path: "/v1/projects/{project_id}/databases"
-        },
-        getDatabase: {
-            method: "GET",
-            path: "/v1/projects/{project_id}/databases/{database_id}"
-        },
-        updateDatabase: {
-            method: "PATCH",
-            path: "/v1/projects/{project_id}/databases/{database_id}"
+            path: "/v1/projects/{project_id}/databases/{database_id}/cache-rules/{query_hash}"
         },
-        deleteDatabase: {
-            method: "DELETE",
-            path: "/v1/projects/{project_id}/databases/{database_id}"
-        },
-        testDatabaseConnection: {
-            method: "POST",
-            path: "/v1/projects/{project_id}/databases/{database_id}/test-connection"
-        }
-    },
-    domains: {
         listCustomDomains: {
             method: "GET",
             path: "/v1/projects/{project_id}/domains"
@@ -86,25 +76,7 @@
         verifyCustomDomain: {
             method: "POST",
             path: "/v1/projects/{project_id}/domains/{domain_id}/verify"
-        }
-    },
-    health: {
-        getHealth: {
-            method: "GET",
-            path: "/v1/health"
-        }
-    },
-    onboarding: {
-        getOnboardingProgress: {
-            method: "GET",
-            path: "/v1/organizations/{org_id}/onboarding"
         },
-        updateOnboardingProgress: {
-            method: "PATCH",
-            path: "/v1/organizations/{org_id}/onboarding"
-        }
-    },
-    projects: {
         listProjects: {
             method: "GET",
             path: "/v1/projects"
@@ -128,15 +100,7 @@
         getProjectMetrics: {
             method: "GET",
             path: "/v1/projects/{project_id}/metrics"
-        }
-    },
-    regions: {
-        listRegions: {
-            method: "GET",
-            path: "/v1/regions"
-        }
-    },
-    replicas: {
+        },
         listReplicas: {
             method: "GET",
             path: "/v1/databases/{database_id}/replicas"
@@ -150,10 +114,40 @@
             path: "/v1/databases/{database_id}/replicas/{replica_id}"
         }
     },
-    waitlist: {
-        joinWaitlist: {
+    databases: {
+        listDatabases: {
+            method: "GET",
+            path: "/v1/projects/{project_id}/databases"
+        },
+        createDatabase: {
+            method: "POST",
+            path: "/v1/projects/{project_id}/databases"
+        },
+        getDatabase: {
+            method: "GET",
+            path: "/v1/projects/{project_id}/databases/{database_id}"
+        },
+        updateDatabase: {
+            method: "PATCH",
+            path: "/v1/projects/{project_id}/databases/{database_id}"
+        },
+        deleteDatabase: {
+            method: "DELETE",
+            path: "/v1/projects/{project_id}/databases/{database_id}"
+        },
+        testDatabaseConnection: {
             method: "POST",
-            path: "/v1/waitlist"
+            path: "/v1/projects/{project_id}/databases/{database_id}/test-connection"
+        }
+    },
+    platform: {
+        getHealth: {
+            method: "GET",
+            path: "/v1/health"
+        },
+        listRegions: {
+            method: "GET",
+            path: "/v1/regions"
         }
     }
 };
@@ -178,14 +172,24 @@ const operationsByPath = {
         path: "/v1/organizations/{org_id}/plan",
         operationId: "getOrganizationPlan"
     },
-    "GET /v1/projects/{project_id}/databases/{database_id}/catalog": {
+    "PUT /v1/organizations/{org_id}/spend-limit": {
+        method: "PUT",
+        path: "/v1/organizations/{org_id}/spend-limit",
+        operationId: "updateSpendLimit"
+    },
+    "POST /v1/organizations/{org_id}/cancellation-feedback": {
+        method: "POST",
+        path: "/v1/organizations/{org_id}/cancellation-feedback",
+        operationId: "submitCancellationFeedback"
+    },
+    "GET /v1/projects/{project_id}/databases/{database_id}/cache-rules": {
         method: "GET",
-        path: "/v1/projects/{project_id}/databases/{database_id}/catalog",
-        operationId: "listQueryCatalog"
+        path: "/v1/projects/{project_id}/databases/{database_id}/cache-rules",
+        operationId: "listCacheRules"
     },
-    "PUT /v1/projects/{project_id}/databases/{database_id}/catalog/{query_hash}/cache-rule": {
+    "PUT /v1/projects/{project_id}/databases/{database_id}/cache-rules/{query_hash}": {
         method: "PUT",
-        path: "/v1/projects/{project_id}/databases/{database_id}/catalog/{query_hash}/cache-rule",
+        path: "/v1/projects/{project_id}/databases/{database_id}/cache-rules/{query_hash}",
         operationId: "updateCacheRule"
     },
     "GET /v1/projects/{project_id}/databases": {
@@ -317,11 +321,6 @@ const operationsByPath = {
         method: "GET",
         path: "/v1/projects/{project_id}/usage",
         operationId: "getProjectUsage"
-    },
-    "POST /v1/waitlist": {
-        method: "POST",
-        path: "/v1/waitlist",
-        operationId: "joinWaitlist"
     }
 };
 
@@ -347,7 +346,62 @@ class ApiError extends Error {
     if (typeof obj.message === "string") return obj.message;
     return undefined;
 }
+const RETRYABLE_STATUS_CODES = new Set([
+    408,
+    429,
+    502,
+    503,
+    504
+]);
+const DEFAULT_RETRY = {
+    maxRetries: 5,
+    initialDelayMs: 500,
+    maxDelayMs: 30000,
+    idempotencyKeys: true
+};
+function resolveRetry(config) {
+    if (!config) return _({}, DEFAULT_RETRY, {
+        maxRetries: 0
+    });
+    return _({}, DEFAULT_RETRY, config);
+}
+/** Parse Retry-After header as milliseconds. Returns undefined if unparseable. */ function parseRetryAfter(response) {
+    const header = response.headers.get("Retry-After");
+    if (!header) return undefined;
+    // Integer seconds
+    const seconds = Number(header);
+    if (Number.isFinite(seconds) && seconds >= 0) {
+        return seconds * 1000;
+    }
+    // HTTP-date
+    const date = Date.parse(header);
+    if (Number.isFinite(date)) {
+        const ms = date - Date.now();
+        return ms > 0 ? ms : 0;
+    }
+    return undefined;
+}
+/** Compute exponential backoff with jitter. */ function backoff(attempt, initialDelayMs, maxDelayMs) {
+    const delay = Math.min(initialDelayMs * 2 ** attempt, maxDelayMs);
+    return delay * (0.5 + Math.random());
+}
+/** Generate a UUID v4 using crypto.randomUUID if available, otherwise fallback. */ function generateUUID() {
+    if (typeof crypto !== "undefined" && typeof crypto.randomUUID === "function") {
+        return crypto.randomUUID();
+    }
+    // Fallback for environments without crypto.randomUUID
+    return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, (c)=>{
+        const r = Math.random() * 16 | 0;
+        const v = c === "x" ? r : r & 0x3 | 0x8;
+        return v.toString(16);
+    });
+}
+function isMutatingMethod(method) {
+    const upper = method.toUpperCase();
+    return upper === "POST" || upper === "PATCH";
+}
 async function fetcher(options) {
+    var _options_fetchImpl;
     let url = options.path;
     if (options.pathParams) {
         for (const [key, value] of Object.entries(options.pathParams)){
@@ -369,29 +423,68 @@ async function fetcher(options) {
     if (options.body !== undefined) {
         headers["Content-Type"] = "application/json";
     }
-    var _options_fetchImpl;
     const fetchFn = (_options_fetchImpl = options.fetchImpl) != null ? _options_fetchImpl : globalThis.fetch;
-    const response = await fetchFn(fullUrl.toString(), {
-        method: options.method,
-        headers,
-        body: options.body !== undefined ? JSON.stringify(options.body) : undefined
-    });
-    if (options.onResponse) {
-        await options.onResponse(response);
-    }
-    if (!response.ok) {
-        const errorBody = await response.json().catch(()=>null);
-        throw new ApiError(response.status, response.statusText, errorBody);
+    const retry = resolveRetry(options.retry);
+    const urlStr = fullUrl.toString();
+    const fetchBody = options.body !== undefined ? JSON.stringify(options.body) : undefined;
+    // Generate idempotency key once, reused across all retry attempts.
+    const idempotencyKey = retry.maxRetries > 0 && retry.idempotencyKeys && isMutatingMethod(options.method) ? generateUUID() : undefined;
+    for(let attempt = 0; attempt <= retry.maxRetries; attempt++){
+        const reqHeaders = _({}, headers);
+        // Attach idempotency key on all attempts for mutating requests.
+        if (idempotencyKey) {
+            reqHeaders["Idempotency-Key"] = idempotencyKey;
+        }
+        try {
+            var _parseRetryAfter;
+            const response = await fetchFn(urlStr, {
+                method: options.method,
+                headers: reqHeaders,
+                body: fetchBody
+            });
+            if (options.onResponse) {
+                await options.onResponse(response);
+            }
+            if (response.ok) {
+                return parseResponseBody(response);
+            }
+            // Non-retryable error or last attempt — throw immediately.
+            if (!RETRYABLE_STATUS_CODES.has(response.status) || attempt === retry.maxRetries) {
+                const errorBody = await response.json().catch(()=>null);
+                throw new ApiError(response.status, response.statusText, errorBody);
+            }
+            // Retryable error — wait and retry.
+            const delay = (_parseRetryAfter = parseRetryAfter(response)) != null ? _parseRetryAfter : backoff(attempt, retry.initialDelayMs, retry.maxDelayMs);
+            await sleep(delay);
+        } catch (err) {
+            // If it's already an ApiError, rethrow (we threw it above).
+            if (err instanceof ApiError) throw err;
+            // Network error — retry if we have attempts left.
+            if (attempt === retry.maxRetries) throw err;
+            const delay = backoff(attempt, retry.initialDelayMs, retry.maxDelayMs);
+            await sleep(delay);
+        }
     }
+    // Unreachable — the loop either returns or throws.
+    throw new Error("fetcher: exhausted all retry attempts");
+}
+async function parseResponseBody(response) {
     if (response.status === 204) return undefined;
     const contentType = response.headers.get("content-type");
-    if (contentType && contentType.includes("application/json")) {
+    if (contentType == null ? void 0 : contentType.includes("application/json")) {
         return response.json();
     }
     // Fallback: try JSON, return undefined if body is empty/non-JSON
     const text = await response.text();
     if (!text) return undefined;
-    return JSON.parse(text);
+    try {
+        return JSON.parse(text);
+    } catch (unused) {
+        return undefined;
+    }
+}
+function sleep(ms) {
+    return new Promise((resolve)=>setTimeout(resolve, ms));
 }
 
 class PgBeamClient {
@@ -413,7 +506,8 @@ class PgBeamClient {
                 baseUrl: this._baseUrl,
                 token,
                 fetchImpl: this._fetchImpl,
-                onResponse: this._onResponse
+                onResponse: this._onResponse,
+                retry: this._retry
             });
         };
     }
@@ -432,7 +526,8 @@ class PgBeamClient {
                 baseUrl: this._baseUrl,
                 token,
                 fetchImpl: this._fetchImpl,
-                onResponse: this._onResponse
+                onResponse: this._onResponse,
+                retry: this._retry
             });
         };
         this._api = new Proxy({}, {
@@ -453,10 +548,16 @@ class PgBeamClient {
         return this._api;
     }
     constructor(options){
+        var _options_retry;
         this._baseUrl = options.baseUrl;
         this._tokenOrFn = options.token;
         this._fetchImpl = options.fetch;
         this._onResponse = options.onResponse;
+        this._retry = options.retry === false ? {
+            maxRetries: 0
+        } : (_options_retry = options.retry) != null ? _options_retry : {
+            maxRetries: 5
+        };
     }
 }