stitchdb 1.0.0 → 1.1.0

package/dist/index.d.mts

@@ -30,12 +30,21 @@ interface ExecResult {
 }
 declare class StitchDBError extends Error {
     status: number;
-    constructor(message: string, status: number);
+    constructor(message: string, status?: number);
 }
 declare class StitchDB {
     private url;
+    private wsUrl;
     private apiKey;
+    private ws;
+    private pending;
+    private msgId;
+    private connecting;
+    private useWebSocket;
     constructor(config: StitchDBConfig);
+    private connect;
+    private wsSend;
+    private httpQuery;
     /** Run a SQL query with parameterized bindings. */
     query<T = Record<string, unknown>>(sql: string, params?: unknown[]): Promise<QueryResult<T>>;
     /** Run multiple queries atomically in a single batch. */
@@ -59,7 +68,8 @@ declare class StitchDB {
         limit?: number;
         offset?: number;
     }): Promise<T[]>;
-    private post;
+    /** Close the WebSocket connection. */
+    close(): void;
 }
 /** Create a StitchDB client. */
 declare function createClient(config: StitchDBConfig): StitchDB;

package/dist/index.d.ts

@@ -30,12 +30,21 @@ interface ExecResult {
 }
 declare class StitchDBError extends Error {
     status: number;
-    constructor(message: string, status: number);
+    constructor(message: string, status?: number);
 }
 declare class StitchDB {
     private url;
+    private wsUrl;
     private apiKey;
+    private ws;
+    private pending;
+    private msgId;
+    private connecting;
+    private useWebSocket;
     constructor(config: StitchDBConfig);
+    private connect;
+    private wsSend;
+    private httpQuery;
     /** Run a SQL query with parameterized bindings. */
     query<T = Record<string, unknown>>(sql: string, params?: unknown[]): Promise<QueryResult<T>>;
     /** Run multiple queries atomically in a single batch. */
@@ -59,7 +68,8 @@ declare class StitchDB {
         limit?: number;
         offset?: number;
     }): Promise<T[]>;
-    private post;
+    /** Close the WebSocket connection. */
+    close(): void;
 }
 /** Create a StitchDB client. */
 declare function createClient(config: StitchDBConfig): StitchDB;

package/dist/index.js

@@ -27,7 +27,7 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 var StitchDBError = class extends Error {
-  constructor(message, status) {
+  constructor(message, status = 0) {
     super(message);
     this.name = "StitchDBError";
     this.status = status;
@@ -35,20 +35,132 @@ var StitchDBError = class extends Error {
 };
 var StitchDB = class {
   constructor(config) {
+    this.ws = null;
+    this.pending = /* @__PURE__ */ new Map();
+    this.msgId = 0;
+    this.connecting = null;
     this.url = (config.url || "https://db.stitchdb.com").replace(/\/$/, "");
+    this.wsUrl = this.url.replace("https://", "wss://").replace("http://", "ws://");
     this.apiKey = config.apiKey;
+    this.useWebSocket = typeof WebSocket !== "undefined";
+  }
+  async connect() {
+    if (this.ws?.readyState === 1) return;
+    if (this.connecting) return this.connecting;
+    this.connecting = new Promise((resolve, reject) => {
+      try {
+        const ws = new WebSocket(`${this.wsUrl}/ws/query?key=${this.apiKey}`);
+        ws.onopen = () => {
+          this.ws = ws;
+          this.connecting = null;
+          resolve();
+        };
+        ws.onmessage = (event) => {
+          try {
+            const data = JSON.parse(typeof event.data === "string" ? event.data : "");
+            const pending = this.pending.get(data.id);
+            if (pending) {
+              this.pending.delete(data.id);
+              if (data.error) {
+                pending.reject(new StitchDBError(data.error));
+              } else {
+                pending.resolve(data);
+              }
+            }
+          } catch {
+          }
+        };
+        ws.onclose = () => {
+          this.ws = null;
+          this.connecting = null;
+          for (const [id, p] of this.pending) {
+            p.reject(new StitchDBError("Connection closed"));
+            this.pending.delete(id);
+          }
+        };
+        ws.onerror = () => {
+          this.ws = null;
+          this.connecting = null;
+          reject(new StitchDBError("WebSocket connection failed"));
+        };
+      } catch {
+        this.connecting = null;
+        this.useWebSocket = false;
+        resolve();
+      }
+    });
+    return this.connecting;
+  }
+  async wsSend(msg) {
+    await this.connect();
+    if (!this.ws || this.ws.readyState !== 1) {
+      this.useWebSocket = false;
+      return this.httpQuery(msg);
+    }
+    const id = String(++this.msgId);
+    msg.id = id;
+    return new Promise((resolve, reject) => {
+      const timeout = setTimeout(() => {
+        this.pending.delete(id);
+        reject(new StitchDBError("Query timeout"));
+      }, 3e4);
+      this.pending.set(id, {
+        resolve: (v) => {
+          clearTimeout(timeout);
+          resolve(v);
+        },
+        reject: (e) => {
+          clearTimeout(timeout);
+          reject(e);
+        }
+      });
+      this.ws.send(JSON.stringify(msg));
+    });
+  }
+  async httpQuery(msg) {
+    let path = "/v1/query";
+    let body = { sql: msg.sql, params: msg.params };
+    if (msg.action === "batch") {
+      path = "/v1/batch";
+      body = { queries: msg.queries };
+    } else if (msg.action === "exec") {
+      path = "/v1/exec";
+      body = { sql: msg.sql };
+    }
+    const res = await fetch(`${this.url}${path}`, {
+      method: "POST",
+      headers: {
+        "Authorization": `Bearer ${this.apiKey}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify(body)
+    });
+    const data = await res.json();
+    if (!res.ok || data.error) {
+      throw new StitchDBError(data.error || `Request failed: ${res.status}`, res.status);
+    }
+    return data;
   }
   /** Run a SQL query with parameterized bindings. */
   async query(sql, params) {
-    return this.post("/v1/query", { sql, params });
+    if (this.useWebSocket) {
+      return this.wsSend({ action: "query", sql, params });
+    }
+    return this.httpQuery({ action: "query", sql, params });
   }
   /** Run multiple queries atomically in a single batch. */
   async batch(queries) {
-    return this.post("/v1/batch", { queries });
+    if (this.useWebSocket) {
+      return this.wsSend({ action: "batch", queries });
+    }
+    return this.httpQuery({ action: "batch", queries });
   }
   /** Run a DDL statement (CREATE TABLE, ALTER TABLE, DROP TABLE, etc.) */
   async run(sql) {
-    return this.post("/v1/exec", { sql });
+    if (this.useWebSocket) {
+      return this.wsSend({ action: "exec", sql });
+    }
+    return this.httpQuery({ action: "exec", sql });
   }
   /** Insert a row. */
   async insert(table, data) {
@@ -80,20 +192,12 @@ var StitchDB = class {
     const { results } = await this.query(sql, params);
     return results;
   }
-  async post(path, body) {
-    const res = await fetch(`${this.url}${path}`, {
-      method: "POST",
-      headers: {
-        "Authorization": `Bearer ${this.apiKey}`,
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(body)
-    });
-    const data = await res.json();
-    if (!res.ok || data.error) {
-      throw new StitchDBError(data.error || `Request failed: ${res.status}`, res.status);
+  /** Close the WebSocket connection. */
+  close() {
+    if (this.ws) {
+      this.ws.close();
+      this.ws = null;
     }
-    return data;
   }
 };
 function createClient(config) {

package/dist/index.mjs

@@ -1,6 +1,6 @@
 // src/index.ts
 var StitchDBError = class extends Error {
-  constructor(message, status) {
+  constructor(message, status = 0) {
     super(message);
     this.name = "StitchDBError";
     this.status = status;
@@ -8,20 +8,132 @@ var StitchDBError = class extends Error {
 };
 var StitchDB = class {
   constructor(config) {
+    this.ws = null;
+    this.pending = /* @__PURE__ */ new Map();
+    this.msgId = 0;
+    this.connecting = null;
     this.url = (config.url || "https://db.stitchdb.com").replace(/\/$/, "");
+    this.wsUrl = this.url.replace("https://", "wss://").replace("http://", "ws://");
     this.apiKey = config.apiKey;
+    this.useWebSocket = typeof WebSocket !== "undefined";
+  }
+  async connect() {
+    if (this.ws?.readyState === 1) return;
+    if (this.connecting) return this.connecting;
+    this.connecting = new Promise((resolve, reject) => {
+      try {
+        const ws = new WebSocket(`${this.wsUrl}/ws/query?key=${this.apiKey}`);
+        ws.onopen = () => {
+          this.ws = ws;
+          this.connecting = null;
+          resolve();
+        };
+        ws.onmessage = (event) => {
+          try {
+            const data = JSON.parse(typeof event.data === "string" ? event.data : "");
+            const pending = this.pending.get(data.id);
+            if (pending) {
+              this.pending.delete(data.id);
+              if (data.error) {
+                pending.reject(new StitchDBError(data.error));
+              } else {
+                pending.resolve(data);
+              }
+            }
+          } catch {
+          }
+        };
+        ws.onclose = () => {
+          this.ws = null;
+          this.connecting = null;
+          for (const [id, p] of this.pending) {
+            p.reject(new StitchDBError("Connection closed"));
+            this.pending.delete(id);
+          }
+        };
+        ws.onerror = () => {
+          this.ws = null;
+          this.connecting = null;
+          reject(new StitchDBError("WebSocket connection failed"));
+        };
+      } catch {
+        this.connecting = null;
+        this.useWebSocket = false;
+        resolve();
+      }
+    });
+    return this.connecting;
+  }
+  async wsSend(msg) {
+    await this.connect();
+    if (!this.ws || this.ws.readyState !== 1) {
+      this.useWebSocket = false;
+      return this.httpQuery(msg);
+    }
+    const id = String(++this.msgId);
+    msg.id = id;
+    return new Promise((resolve, reject) => {
+      const timeout = setTimeout(() => {
+        this.pending.delete(id);
+        reject(new StitchDBError("Query timeout"));
+      }, 3e4);
+      this.pending.set(id, {
+        resolve: (v) => {
+          clearTimeout(timeout);
+          resolve(v);
+        },
+        reject: (e) => {
+          clearTimeout(timeout);
+          reject(e);
+        }
+      });
+      this.ws.send(JSON.stringify(msg));
+    });
+  }
+  async httpQuery(msg) {
+    let path = "/v1/query";
+    let body = { sql: msg.sql, params: msg.params };
+    if (msg.action === "batch") {
+      path = "/v1/batch";
+      body = { queries: msg.queries };
+    } else if (msg.action === "exec") {
+      path = "/v1/exec";
+      body = { sql: msg.sql };
+    }
+    const res = await fetch(`${this.url}${path}`, {
+      method: "POST",
+      headers: {
+        "Authorization": `Bearer ${this.apiKey}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify(body)
+    });
+    const data = await res.json();
+    if (!res.ok || data.error) {
+      throw new StitchDBError(data.error || `Request failed: ${res.status}`, res.status);
+    }
+    return data;
   }
   /** Run a SQL query with parameterized bindings. */
   async query(sql, params) {
-    return this.post("/v1/query", { sql, params });
+    if (this.useWebSocket) {
+      return this.wsSend({ action: "query", sql, params });
+    }
+    return this.httpQuery({ action: "query", sql, params });
   }
   /** Run multiple queries atomically in a single batch. */
   async batch(queries) {
-    return this.post("/v1/batch", { queries });
+    if (this.useWebSocket) {
+      return this.wsSend({ action: "batch", queries });
+    }
+    return this.httpQuery({ action: "batch", queries });
   }
   /** Run a DDL statement (CREATE TABLE, ALTER TABLE, DROP TABLE, etc.) */
   async run(sql) {
-    return this.post("/v1/exec", { sql });
+    if (this.useWebSocket) {
+      return this.wsSend({ action: "exec", sql });
+    }
+    return this.httpQuery({ action: "exec", sql });
   }
   /** Insert a row. */
   async insert(table, data) {
@@ -53,20 +165,12 @@ var StitchDB = class {
     const { results } = await this.query(sql, params);
     return results;
   }
-  async post(path, body) {
-    const res = await fetch(`${this.url}${path}`, {
-      method: "POST",
-      headers: {
-        "Authorization": `Bearer ${this.apiKey}`,
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(body)
-    });
-    const data = await res.json();
-    if (!res.ok || data.error) {
-      throw new StitchDBError(data.error || `Request failed: ${res.status}`, res.status);
+  /** Close the WebSocket connection. */
+  close() {
+    if (this.ws) {
+      this.ws.close();
+      this.ws = null;
     }
-    return data;
   }
 };
 function createClient(config) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "stitchdb",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "StitchDB client for JavaScript and TypeScript",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -19,7 +19,13 @@
     "build": "tsup src/index.ts --format cjs,esm --dts",
     "prepublishOnly": "npm run build"
   },
-  "keywords": ["stitchdb", "database", "sql", "edge", "serverless"],
+  "keywords": [
+    "stitchdb",
+    "database",
+    "sql",
+    "edge",
+    "serverless"
+  ],
   "license": "MIT",
   "author": {
     "name": "StitchDB",
@@ -35,4 +41,4 @@
     "tsup": "^8.0.0",
     "typescript": "^5.0.0"
   }
-}
+}