@empiricalrun/playwright-utils 0.34.0 → 0.34.1

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @empiricalrun/playwright-utils
 
+## 0.34.1
+
+### Patch Changes
+
+- 510c376: fix: get or create behavior in postgres client
+
 ## 0.34.0
 
 ### Minor Changes

package/dist/postgres.d.ts

@@ -1,18 +1,16 @@
-type PostgresDatabase = {
+export type PostgresDatabase = {
     projectId: string;
     connectionUri: string;
 };
 type GetOptions = {
-    name?: string;
     pgVersion?: number;
-    ttl?: number;
 };
 export declare class PostgresClient {
     private client;
-    private kv;
     constructor();
     get(name: string, options?: GetOptions): Promise<PostgresDatabase>;
-    delete(name: string): Promise<void>;
+    private getConnectionUri;
+    delete(projectId: string): Promise<void>;
     query<T = Record<string, unknown>>(connectionUri: string, sql: string): Promise<T[]>;
     execute(connectionUri: string, sql: string): Promise<{
         statement: string;

package/dist/postgres.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../src/postgres.ts"],"names":[],"mappings":"AA6BA,KAAK,gBAAgB,GAAG;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,KAAK,UAAU,GAAG;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd,CAAC;AAIF,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,EAAE,CAAW;;IAYf,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAuClE,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAwBnC,KAAK,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,GACV,OAAO,CAAC,CAAC,EAAE,CAAC;IAWT,OAAO,CACX,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,GACV,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAsBrD,OAAO,CAAC,eAAe;CAsBxB"}
+{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../src/postgres.ts"],"names":[],"mappings":"AAgDA,MAAM,MAAM,gBAAgB,GAAG;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,KAAK,UAAU,GAAG;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAqB;;IAW7B,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC,gBAAgB,CAAC;YAqD1D,gBAAgB;IAwBxB,MAAM,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBxC,KAAK,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,GACV,OAAO,CAAC,CAAC,EAAE,CAAC;IAWT,OAAO,CACX,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,GACV,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAsBrD,OAAO,CAAC,eAAe;CAsBxB"}

package/dist/postgres.js

@@ -3,11 +3,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.PostgresClient = void 0;
 const dashboard_1 = require("@empiricalrun/test-gen/dashboard");
 const pg_1 = require("pg");
-const kv_1 = require("./kv");
-const DEFAULT_TTL = 3600;
 class PostgresClient {
     client;
-    kv;
     constructor() {
         if (!process.env.EMPIRICALRUN_API_KEY) {
             throw new Error("EMPIRICALRUN_API_KEY environment variable is required");
@@ -15,47 +12,66 @@ class PostgresClient {
         this.client = new dashboard_1.DashboardAPIClient({
             authType: "project-api-key",
         });
-        this.kv = new kv_1.KVClient();
     }
     async get(name, options) {
-        const kvKey = `postgres:${name}`;
-        const existing = await this.kv.get(kvKey);
-        if (existing) {
-            return existing;
+        const listResponse = await this.client.request("/api/neon/proxy", {
+            method: "POST",
+            body: {
+                action: "list",
+            },
+        });
+        if (listResponse.error) {
+            throw new Error(`Postgres list failed: ${listResponse.error}`);
         }
-        const response = await this.client.request("/api/neon/proxy", {
+        const existingProject = listResponse.data.projects.find((p) => p.name === name);
+        if (existingProject) {
+            const connectionUri = await this.getConnectionUri(existingProject.id);
+            return { projectId: existingProject.id, connectionUri };
+        }
+        const createResponse = await this.client.request("/api/neon/proxy", {
             method: "POST",
             body: {
                 action: "create",
                 payload: {
-                    name: options?.name ?? name,
+                    name,
                     pg_version: options?.pgVersion,
                 },
             },
         });
-        if (response.error) {
-            throw new Error(`Postgres create failed: ${response.error}`);
+        if (createResponse.error) {
+            throw new Error(`Postgres create failed: ${createResponse.error}`);
         }
-        const projectId = response.data.project.id;
-        const connectionUri = response.data.connection_uris?.[0]?.connection_uri;
+        const projectId = createResponse.data.project.id;
+        const connectionUri = createResponse.data.connection_uris?.[0]?.connection_uri;
         if (!connectionUri) {
             throw new Error("No connection URI returned from Neon API");
         }
-        const db = { projectId, connectionUri };
-        await this.kv.set(kvKey, db, options?.ttl ?? DEFAULT_TTL);
-        return db;
+        return { projectId, connectionUri };
     }
-    async delete(name) {
-        const kvKey = `postgres:${name}`;
-        const existing = await this.kv.get(kvKey);
-        if (!existing) {
-            return;
+    async getConnectionUri(projectId) {
+        const response = await this.client.request("/api/neon/proxy", {
+            method: "POST",
+            body: {
+                action: "getConnectionUri",
+                projectId,
+                payload: {
+                    databaseName: "neondb",
+                    roleName: "neondb_owner",
+                    pooled: false,
+                },
+            },
+        });
+        if (response.error) {
+            throw new Error(`Postgres getConnectionUri failed: ${response.error}`);
         }
+        return response.data.uri;
+    }
+    async delete(projectId) {
         const response = await this.client.request("/api/neon/proxy", {
             method: "POST",
             body: {
                 action: "delete",
-                projectId: existing.projectId,
+                projectId,
             },
         });
         if (response.error) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/playwright-utils",
-  "version": "0.34.0",
+  "version": "0.34.1",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -44,8 +44,8 @@
     "console-log-level": "^1.4.1",
     "puppeteer-extra-plugin-recaptcha": "^3.6.8",
     "rimraf": "^6.0.1",
-    "@empiricalrun/r2-uploader": "^0.5.0",
     "@empiricalrun/llm": "^0.25.1",
+    "@empiricalrun/r2-uploader": "^0.5.0",
     "@empiricalrun/test-gen": "^0.78.3"
   },
   "scripts": {