@cf-vibesdk/sdk 0.0.2 → 0.0.4

package/README.md

@@ -1,6 +1,6 @@
 # @cf-vibesdk/sdk
 
-Type-safe client SDK for the VibeSDK platform.
+Client SDK for the VibeSDK platform.
 
 ## Install
 

package/dist/index.js

@@ -1,10 +1,40 @@
 // @bun
+// src/retry.ts
+function normalizeRetryConfig(retry, defaults) {
+  return {
+    enabled: retry?.enabled ?? defaults.enabled,
+    initialDelayMs: retry?.initialDelayMs ?? defaults.initialDelayMs,
+    maxDelayMs: retry?.maxDelayMs ?? defaults.maxDelayMs,
+    maxRetries: retry?.maxRetries ?? defaults.maxRetries
+  };
+}
+function computeBackoffMs(attempt, cfg) {
+  const base = Math.min(cfg.maxDelayMs, cfg.initialDelayMs * Math.pow(2, Math.max(0, attempt)));
+  const jitter = base * 0.2;
+  return Math.max(0, Math.floor(base - jitter + Math.random() * jitter * 2));
+}
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+
 // src/http.ts
+var HTTP_RETRY_DEFAULTS = {
+  enabled: true,
+  initialDelayMs: 1000,
+  maxDelayMs: 1e4,
+  maxRetries: 3
+};
+function isRetryableStatus(status) {
+  return status >= 500 && status < 600;
+}
+
 class HttpClient {
   opts;
   cachedAccessToken = null;
+  retryCfg;
   constructor(opts) {
     this.opts = opts;
+    this.retryCfg = normalizeRetryConfig(opts.retry, HTTP_RETRY_DEFAULTS);
   }
   get baseUrl() {
     return this.opts.baseUrl.replace(/\/$/, "");
@@ -34,7 +64,7 @@ class HttpClient {
       }
     });
     if (!resp.ok) {
-      const text = await resp.text().catch(() => "");
+      const text = (await resp.text().catch(() => "")).slice(0, 1000);
       if (resp.status === 401) {
         throw new Error(`HTTP 401 for /api/auth/exchange-api-key: invalid API key (regenerate in Settings \u2192 API Keys). ${text || ""}`.trim());
       }
@@ -60,21 +90,59 @@ class HttpClient {
   }
   async fetchJson(path, init) {
     const url = `${this.baseUrl}${path}`;
-    const resp = await this.fetchFn(url, init);
-    if (!resp.ok) {
-      const text = await resp.text().catch(() => "");
-      throw new Error(`HTTP ${resp.status} for ${path}: ${text || resp.statusText}`);
+    let lastError = null;
+    for (let attempt = 0;attempt <= this.retryCfg.maxRetries; attempt++) {
+      try {
+        const resp = await this.fetchFn(url, init);
+        if (!resp.ok) {
+          const text = (await resp.text().catch(() => "")).slice(0, 1000);
+          const error = new Error(`HTTP ${resp.status} for ${path}: ${text || resp.statusText}`);
+          if (this.retryCfg.enabled && isRetryableStatus(resp.status) && attempt < this.retryCfg.maxRetries) {
+            lastError = error;
+            await sleep(computeBackoffMs(attempt, this.retryCfg));
+            continue;
+          }
+          throw error;
+        }
+        return await resp.json();
+      } catch (error) {
+        if (error instanceof TypeError && this.retryCfg.enabled && attempt < this.retryCfg.maxRetries) {
+          lastError = error;
+          await sleep(computeBackoffMs(attempt, this.retryCfg));
+          continue;
+        }
+        throw error;
+      }
     }
-    return await resp.json();
+    throw lastError ?? new Error(`Failed after ${this.retryCfg.maxRetries} retries`);
   }
   async fetchRaw(path, init) {
     const url = `${this.baseUrl}${path}`;
-    const resp = await this.fetchFn(url, init);
-    if (!resp.ok) {
-      const text = await resp.text().catch(() => "");
-      throw new Error(`HTTP ${resp.status} for ${path}: ${text || resp.statusText}`);
+    let lastError = null;
+    for (let attempt = 0;attempt <= this.retryCfg.maxRetries; attempt++) {
+      try {
+        const resp = await this.fetchFn(url, init);
+        if (!resp.ok) {
+          const text = (await resp.text().catch(() => "")).slice(0, 1000);
+          const error = new Error(`HTTP ${resp.status} for ${path}: ${text || resp.statusText}`);
+          if (this.retryCfg.enabled && isRetryableStatus(resp.status) && attempt < this.retryCfg.maxRetries) {
+            lastError = error;
+            await sleep(computeBackoffMs(attempt, this.retryCfg));
+            continue;
+          }
+          throw error;
+        }
+        return resp;
+      } catch (error) {
+        if (error instanceof TypeError && this.retryCfg.enabled && attempt < this.retryCfg.maxRetries) {
+          lastError = error;
+          await sleep(computeBackoffMs(attempt, this.retryCfg));
+          continue;
+        }
+        throw error;
+      }
     }
-    return resp;
+    throw lastError ?? new Error(`Failed after ${this.retryCfg.maxRetries} retries`);
   }
 }
 
@@ -137,19 +205,25 @@ class TypedEmitter {
     for (const cb of set)
       cb(payload);
   }
+  clear() {
+    this.listeners.clear();
+    this.anyListeners.clear();
+  }
 }
 
 // src/state.ts
 var INITIAL_STATE = {
+  connection: "disconnected",
   generation: { status: "idle" },
   phase: { status: "idle" },
   preview: { status: "idle" },
   cloudflare: { status: "idle" }
 };
 function extractPhaseInfo(msg) {
+  const phase = msg?.phase;
   return {
-    name: msg.phase?.name,
-    description: msg.phase?.description
+    name: phase?.name,
+    description: phase?.description
   };
 }
 
@@ -162,6 +236,9 @@ class SessionStateStore {
   onChange(cb) {
     return this.emitter.on("change", ({ prev, next }) => cb(next, prev));
   }
+  setConnection(state) {
+    this.setState({ connection: state });
+  }
   applyWsMessage(msg) {
     switch (msg.type) {
       case "conversation_state": {
@@ -176,29 +253,57 @@ class SessionStateStore {
       }
       case "generation_started": {
         const m = msg;
-        this.setState({ generation: { status: "running", totalFiles: m.totalFiles } });
+        this.setState({
+          generation: { status: "running", totalFiles: m.totalFiles, filesGenerated: 0 },
+          currentFile: undefined
+        });
         break;
       }
       case "generation_complete": {
         const m = msg;
         const previewURL = m.previewURL;
+        const prev = this.state.generation;
+        const filesGenerated = "filesGenerated" in prev ? prev.filesGenerated : 0;
         this.setState({
           generation: {
             status: "complete",
             instanceId: m.instanceId,
-            previewURL
+            previewURL,
+            filesGenerated
           },
+          currentFile: undefined,
           ...previewURL ? { previewUrl: previewURL } : {}
         });
         break;
       }
       case "generation_stopped": {
         const m = msg;
-        this.setState({ generation: { status: "stopped", instanceId: m.instanceId } });
+        const prev = this.state.generation;
+        const filesGenerated = "filesGenerated" in prev ? prev.filesGenerated : 0;
+        this.setState({
+          generation: { status: "stopped", instanceId: m.instanceId, filesGenerated }
+        });
         break;
       }
       case "generation_resumed": {
-        this.setState({ generation: { status: "running" } });
+        const prev = this.state.generation;
+        const filesGenerated = "filesGenerated" in prev ? prev.filesGenerated : 0;
+        this.setState({ generation: { status: "running", filesGenerated } });
+        break;
+      }
+      case "file_generating": {
+        const m = msg;
+        this.setState({ currentFile: m.filePath });
+        break;
+      }
+      case "file_generated": {
+        const prev = this.state.generation;
+        if (prev.status === "running" || prev.status === "stopped") {
+          this.setState({
+            generation: { ...prev, filesGenerated: prev.filesGenerated + 1 },
+            currentFile: undefined
+          });
+        }
         break;
       }
       case "phase_generating": {
@@ -299,6 +404,10 @@ class SessionStateStore {
     this.state = next;
     this.emitter.emit("change", { prev, next });
   }
+  clear() {
+    this.state = INITIAL_STATE;
+    this.emitter.clear();
+  }
 }
 
 // src/ws.ts
@@ -308,23 +417,15 @@ function toWsCloseEvent(ev) {
     reason: typeof ev.reason === "string" ? ev.reason : ""
   };
 }
-function normalizeRetryConfig(retry) {
-  const enabled = retry?.enabled ?? true;
-  return {
-    enabled,
-    initialDelayMs: retry?.initialDelayMs ?? 1000,
-    maxDelayMs: retry?.maxDelayMs ?? 30000,
-    maxRetries: retry?.maxRetries ?? Infinity
-  };
-}
-function computeBackoffMs(attempt, cfg) {
-  const base = Math.min(cfg.maxDelayMs, cfg.initialDelayMs * Math.pow(2, Math.max(0, attempt)));
-  const jitter = base * 0.2;
-  return Math.max(0, Math.floor(base - jitter + Math.random() * jitter * 2));
-}
+var WS_RETRY_DEFAULTS = {
+  enabled: true,
+  initialDelayMs: 1000,
+  maxDelayMs: 30000,
+  maxRetries: Infinity
+};
 function createAgentConnection(url, options = {}) {
   const emitter = new TypedEmitter;
-  const retryCfg = normalizeRetryConfig(options.retry);
+  const retryCfg = normalizeRetryConfig(options.retry, WS_RETRY_DEFAULTS);
   const headers = { ...options.headers ?? {} };
   if (options.origin)
     headers.Origin = options.origin;
@@ -501,7 +602,7 @@ function createAgentConnection(url, options = {}) {
     }
     if (ws.on) {
       ws.on("open", () => onOpen());
-      ws.on("close", (code, reason) => onClose({ code, reason }));
+      ws.on("close", (code, reason) => onClose({ code: typeof code === "number" ? code : undefined, reason: typeof reason === "string" ? reason : undefined }));
       ws.on("error", (error) => onError(error));
       ws.on("message", (data) => onMessage(data));
     }
@@ -514,8 +615,12 @@ function createAgentConnection(url, options = {}) {
       return;
     }
     pendingSends.push(data);
-    if (pendingSends.length > maxPendingSends)
+    if (pendingSends.length > maxPendingSends) {
       pendingSends.shift();
+      emitter.emit("ws:error", {
+        error: new Error(`Message queue overflow: dropped oldest message (queue size: ${maxPendingSends})`)
+      });
+    }
   }
   function close() {
     closedByUser = true;
@@ -553,25 +658,24 @@ function createAgentConnection(url, options = {}) {
 function isRecord(value) {
   return typeof value === "object" && value !== null;
 }
+function isFileOutputType(value) {
+  if (!isRecord(value))
+    return false;
+  return typeof value.filePath === "string" && typeof value.fileContents === "string";
+}
 function extractGeneratedFilesFromState(state) {
   const out = [];
   for (const file of Object.values(state.generatedFilesMap ?? {})) {
-    const path = file.filePath;
-    const content = file.fileContents;
-    if (typeof path === "string" && typeof content === "string") {
-      out.push({ path, content });
-    }
+    if (!isFileOutputType(file))
+      continue;
+    out.push({ path: file.filePath, content: file.fileContents });
   }
   return out;
 }
 function extractGeneratedFileFromMessageFile(file) {
-  if (!isRecord(file))
+  if (!isFileOutputType(file))
     return null;
-  const path = file.filePath;
-  const content = file.fileContents;
-  if (typeof path !== "string" || typeof content !== "string")
-    return null;
-  return { path, content };
+  return { path: file.filePath, content: file.fileContents };
 }
 
 class WorkspaceStore {
@@ -624,6 +728,10 @@ class WorkspaceStore {
         break;
     }
   }
+  clear() {
+    this.files.clear();
+    this.emitter.clear();
+  }
 }
 
 // src/session.ts
@@ -713,11 +821,18 @@ class BuildSession {
       ...Object.keys(headers).length ? { headers } : {},
       ...webSocketFactory ? { webSocketFactory } : {}
     };
+    this.state.setConnection("connecting");
     this.connection = createAgentConnection(this.websocketUrl, connectOptions);
     this.connection.on("ws:message", (m) => {
       this.workspace.applyWsMessage(m);
       this.state.applyWsMessage(m);
     });
+    this.connection.on("ws:open", () => {
+      this.state.setConnection("connected");
+    });
+    this.connection.on("ws:close", () => {
+      this.state.setConnection("disconnected");
+    });
     const credentials = agentOptions.credentials ?? this.init.defaultCredentials;
     const shouldRequestConversationState = autoRequestConversationState ?? true;
     this.connection.on("ws:open", () => {
@@ -843,6 +958,8 @@ class BuildSession {
   close() {
     this.connection?.close();
     this.connection = null;
+    this.workspace.clear();
+    this.state.clear();
   }
   assertConnected() {
     if (!this.connection) {
@@ -983,11 +1100,114 @@ class PhasicClient extends VibeClient {
     return super.build(prompt, { ...options, behaviorType: options.behaviorType ?? "phasic" });
   }
 }
+// src/blueprint.ts
+function isRecord2(v) {
+  return Boolean(v) && typeof v === "object" && !Array.isArray(v);
+}
+function blueprintToMarkdown(bp) {
+  const lines = [];
+  const title = bp.title ?? bp.projectName ?? "Blueprint";
+  lines.push(`# ${title}`);
+  if (bp.description) {
+    lines.push("");
+    lines.push(bp.description);
+  }
+  if (bp.frameworks?.length) {
+    lines.push("");
+    lines.push("## Frameworks");
+    for (const f of bp.frameworks)
+      lines.push(`- ${f}`);
+  }
+  if (bp.detailedDescription) {
+    lines.push("");
+    lines.push("## Details");
+    lines.push(bp.detailedDescription);
+  }
+  if (bp.views?.length) {
+    lines.push("");
+    lines.push("## Views");
+    for (const v of bp.views)
+      lines.push(`- **${v.name}**: ${v.description}`);
+  }
+  if (bp.plan?.length) {
+    lines.push("");
+    lines.push("## Plan");
+    bp.plan.forEach((s, idx) => lines.push(`${idx + 1}. ${s}`));
+  }
+  if (bp.implementationRoadmap?.length) {
+    lines.push("");
+    lines.push("## Roadmap");
+    for (const p of bp.implementationRoadmap)
+      lines.push(`- **${p.phase}**: ${p.description}`);
+  }
+  return lines.join(`
+`);
+}
+function extractJsonStringField(raw, key) {
+  const re = new RegExp(`"${key}"\\s*:\\s*"([^"\\n\\r]*)"`);
+  const m = re.exec(raw);
+  return m?.[1] ?? null;
+}
+
+class BlueprintStreamParser {
+  buffer = "";
+  append(chunk) {
+    this.buffer += chunk;
+    return this.toMarkdown();
+  }
+  toMarkdown() {
+    const startsLikeJson = /^\s*[\[{]/.test(this.buffer);
+    if (!startsLikeJson) {
+      return this.buffer;
+    }
+    try {
+      const parsed = JSON.parse(this.buffer);
+      if (isRecord2(parsed)) {
+        return blueprintToMarkdown(parsed);
+      }
+    } catch {}
+    const title = extractJsonStringField(this.buffer, "title") ?? extractJsonStringField(this.buffer, "projectName") ?? "Blueprint";
+    const desc = extractJsonStringField(this.buffer, "description");
+    const lines = [`# ${title}`, "", desc ? desc : "*Generating blueprint...*"];
+    return lines.join(`
+`);
+  }
+  getRaw() {
+    return this.buffer;
+  }
+  clear() {
+    this.buffer = "";
+  }
+}
+// src/utils.ts
+class TimeoutError extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "TimeoutError";
+  }
+}
+async function withTimeout(promise, ms, message = "Operation timed out") {
+  let timeoutId;
+  const timeout = new Promise((_, reject) => {
+    timeoutId = setTimeout(() => reject(new TimeoutError(message)), ms);
+  });
+  try {
+    return await Promise.race([promise, timeout]);
+  } finally {
+    if (timeoutId !== undefined)
+      clearTimeout(timeoutId);
+  }
+}
 export {
+  withTimeout,
+  isRecord2 as isRecord,
+  blueprintToMarkdown,
   WorkspaceStore,
   VibeClient,
+  TimeoutError,
   SessionStateStore,
   PhasicClient,
   BuildSession,
+  BlueprintStreamParser,
   AgenticClient
 };

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@cf-vibesdk/sdk",
-	"version": "0.0.2",
+	"version": "0.0.4",
 	"type": "module",
 	"exports": {
 		".": {
@@ -34,5 +34,3 @@
 		"typescript": "^5.9.3"
 	}
 }
-
-