viagen 0.0.32 → 0.0.36

package/dist/cli.js

@@ -44,6 +44,57 @@ function extractHost(httpsUrl) {
     return "github.com";
   }
 }
+async function waitForServer(baseUrl, devServer) {
+  const timeout = 6e4;
+  const start = Date.now();
+  const ac = new AbortController();
+  let serverOutput = "";
+  const logStream = (async () => {
+    try {
+      for await (const log of devServer.logs({ signal: ac.signal })) {
+        const line = log.data;
+        serverOutput += line;
+        if (log.stream === "stderr") {
+          process.stderr.write(`  ${line}`);
+        } else {
+          process.stdout.write(`  ${line}`);
+        }
+      }
+    } catch {
+    }
+  })();
+  while (Date.now() - start < timeout) {
+    if (devServer.exitCode !== null) {
+      ac.abort();
+      await logStream.catch(() => {
+      });
+      return {
+        ok: false,
+        error: serverOutput.slice(-2e3) || `Process exited with code ${devServer.exitCode}`
+      };
+    }
+    try {
+      const res = await fetch(`${baseUrl}/via/health`, {
+        signal: AbortSignal.timeout(3e3)
+      });
+      if (res.ok) {
+        ac.abort();
+        await logStream.catch(() => {
+        });
+        return { ok: true };
+      }
+    } catch {
+    }
+    await new Promise((r) => setTimeout(r, 2e3));
+  }
+  ac.abort();
+  await logStream.catch(() => {
+  });
+  return {
+    ok: false,
+    error: serverOutput.slice(-2e3) || "Timed out waiting for dev server"
+  };
+}
 async function deploySandbox(opts) {
   const token = randomUUID();
   const useGit = !!opts.git;
@@ -95,7 +146,11 @@ async function deploySandbox(opts) {
         "user.email",
         opts.git.userEmail
       ]);
-      await sandbox2.runCommand("git", ["checkout", "-B", opts.git.branch]);
+      const checkout = await sandbox2.runCommand("git", ["checkout", "-B", opts.git.branch]);
+      if (checkout.exitCode !== 0) {
+        const err = await checkout.stderr();
+        throw new Error(`git checkout failed (exit ${checkout.exitCode}): ${err}`);
+      }
       await sandbox2.runCommand("bash", [
         "-c",
         `echo 'https://x-access-token:${opts.git.token}@${extractHost(opts.git.remoteUrl)}' > ~/.git-credentials`
@@ -157,20 +212,30 @@ async function deploySandbox(opts) {
         content: Buffer.from(envLines.join("\n"))
       }
     ]);
-    const install = await sandbox2.runCommand("npm", ["install"]);
+    console.log("  Installing dependencies...");
+    const install = await sandbox2.runCommand({
+      cmd: "npm",
+      args: ["install"],
+      stdout: process.stdout,
+      stderr: process.stderr
+    });
     if (install.exitCode !== 0) {
-      const stderr = await install.stderr();
-      throw new Error(
-        `npm install failed (exit ${install.exitCode}): ${stderr}`
-      );
+      throw new Error(`npm install failed (exit ${install.exitCode})`);
     }
-    await sandbox2.runCommand({
+    const devServer = await sandbox2.runCommand({
       cmd: "npm",
       args: ["run", "dev", "--", "--host", "0.0.0.0"],
       detached: true
     });
+    console.log("  Starting dev server...");
     const baseUrl = sandbox2.domain(5173);
     const url = `${baseUrl}/t/${token}`;
+    const ready = await waitForServer(baseUrl, devServer);
+    if (!ready.ok) {
+      throw new Error(
+        `Dev server failed to start: ${ready.error}`
+      );
+    }
     return {
       url,
       token,
@@ -816,7 +881,7 @@ function parseFlag(args, flag) {
   if (idx !== -1 && idx + 1 < args.length) return args[idx + 1];
   return void 0;
 }
-async function sandbox(args) {
+async function sandbox(args, options) {
   const subcommand = args[0];
   if (subcommand === "stop") {
     const sandboxId = args[1];
@@ -832,9 +897,12 @@ async function sandbox(args) {
   const branchOverride = parseFlag(args, "--branch") || parseFlag(args, "-b");
   const timeoutFlag = parseFlag(args, "--timeout") || parseFlag(args, "-t");
   const timeoutMinutes = timeoutFlag ? parseInt(timeoutFlag, 10) : void 0;
-  const prompt = parseFlag(args, "--prompt") || parseFlag(args, "-p");
+  const prompt = options?.promptOverride || parseFlag(args, "--prompt") || parseFlag(args, "-p");
   const cwd = process.cwd();
   const dotenv = loadDotenv(cwd);
+  if (options?.extraEnvVars) {
+    Object.assign(dotenv, options.extraEnvVars);
+  }
   for (const [key, val] of Object.entries(dotenv)) {
     if (!process.env[key]) process.env[key] = val;
   }
@@ -1277,6 +1345,43 @@ async function requireClient() {
   }
   return createViagen({ baseUrl: creds.baseUrl, token: creds.token, orgId: creds.orgId });
 }
+async function requireProjectId(client) {
+  const cwd = process.cwd();
+  const env = loadDotenv(cwd);
+  let projectId = env["VIAGEN_PROJECT_ID"];
+  if (projectId) {
+    try {
+      await client.projects.get(projectId);
+      return projectId;
+    } catch {
+      console.log("Project from .env not found. Choose a project:");
+      console.log("");
+      projectId = void 0;
+    }
+  }
+  const projects2 = await client.projects.list();
+  if (projects2.length === 0) {
+    console.error("No projects. Create one with `viagen projects create <name>` or `viagen sync`.");
+    process.exit(1);
+  }
+  if (projects2.length === 1) {
+    return projects2[0].id;
+  }
+  console.log("Select a project:");
+  console.log("");
+  for (let i = 0; i < projects2.length; i++) {
+    const repo = projects2[i].githubRepo ? ` (${projects2[i].githubRepo})` : "";
+    console.log(`  ${i + 1}) ${projects2[i].name}${repo}`);
+  }
+  console.log("");
+  const choice = await promptUser("Choose: ");
+  const idx = parseInt(choice, 10) - 1;
+  if (idx < 0 || idx >= projects2.length) {
+    console.error("Invalid selection.");
+    process.exit(1);
+  }
+  return projects2[idx].id;
+}
 async function teams() {
   const client = await requireClient();
   const memberships = await client.orgs.list() ?? [];
@@ -1393,6 +1498,156 @@ async function projects(args) {
     console.log(`  ${p.name}${repo}  ${p.id}`);
   }
 }
+function formatTaskStatus(status) {
+  const icons = {
+    ready: "\u25CB",
+    running: "\u25CF",
+    completed: "\u2713",
+    failed: "\u2717",
+    validating: "\u25CE"
+  };
+  return `${icons[status] || "?"} ${status}`;
+}
+async function tasksList(args) {
+  const client = await requireClient();
+  const projectId = await requireProjectId(client);
+  const status = parseFlag(args, "--status") || parseFlag(args, "-s");
+  const list = await client.tasks.list(projectId, status);
+  if (list.length === 0) {
+    console.log(
+      status ? `No tasks with status "${status}".` : "No tasks. Create one with `viagen tasks create <prompt>`."
+    );
+    return;
+  }
+  console.log(
+    `${"ID".padEnd(10)} ${"STATUS".padEnd(14)} ${"BRANCH".padEnd(20)} ${"CREATED".padEnd(12)} PROMPT`
+  );
+  console.log("-".repeat(80));
+  for (const t of list) {
+    const id = t.id.slice(0, 8);
+    const st = formatTaskStatus(t.status).padEnd(14);
+    const branch = (t.branch || "-").slice(0, 18).padEnd(20);
+    const created = t.createdAt.slice(0, 10).padEnd(12);
+    const prompt = t.prompt.length > 40 ? t.prompt.slice(0, 37) + "..." : t.prompt;
+    console.log(`${id.padEnd(10)} ${st} ${branch} ${created} ${prompt}`);
+  }
+  console.log("");
+  console.log(`${list.length} task(s)`);
+}
+async function tasksCreate(args) {
+  const branch = parseFlag(args, "--branch") || parseFlag(args, "-b");
+  const positional = [];
+  for (let i = 0; i < args.length; i++) {
+    if (args[i] === "--branch" || args[i] === "-b") {
+      i++;
+      continue;
+    }
+    positional.push(args[i]);
+  }
+  let prompt = positional.join(" ");
+  if (!prompt) {
+    prompt = await promptUser("Task prompt: ");
+    if (!prompt) {
+      console.log("Cancelled.");
+      return;
+    }
+  }
+  const client = await requireClient();
+  const projectId = await requireProjectId(client);
+  const input = { prompt };
+  if (branch) input.branch = branch;
+  const task = await client.tasks.create(projectId, input);
+  console.log(`Task created: ${task.id}`);
+  console.log(`  Prompt: ${task.prompt}`);
+  console.log(`  Branch: ${task.branch}`);
+  console.log(`  Status: ${task.status}`);
+  console.log("");
+  console.log(`Run with: viagen tasks run ${task.id}`);
+}
+async function tasksGet(args) {
+  const taskId = args[0];
+  if (!taskId) {
+    console.error("Usage: viagen tasks get <id>");
+    process.exit(1);
+  }
+  const client = await requireClient();
+  const projectId = await requireProjectId(client);
+  const task = await client.tasks.get(projectId, taskId);
+  console.log(`  ID:         ${task.id}`);
+  console.log(`  Status:     ${formatTaskStatus(task.status)}`);
+  console.log(`  Prompt:     ${task.prompt}`);
+  console.log(`  Branch:     ${task.branch}`);
+  console.log(`  Model:      ${task.model}`);
+  console.log(`  Created:    ${task.createdAt}`);
+  console.log(`  Created By: ${task.createdBy}`);
+  if (task.startedAt) console.log(`  Started:    ${task.startedAt}`);
+  if (task.completedAt) console.log(`  Completed:  ${task.completedAt}`);
+  if (task.durationMs) {
+    const secs = (task.durationMs / 1e3).toFixed(1);
+    console.log(`  Duration:   ${secs}s`);
+  }
+  if (task.inputTokens || task.outputTokens) {
+    console.log(
+      `  Tokens:     ${(task.inputTokens || 0).toLocaleString()} in / ${(task.outputTokens || 0).toLocaleString()} out`
+    );
+  }
+  if (task.prUrl) console.log(`  PR:         ${task.prUrl}`);
+  if (task.result) console.log(`  Result:     ${task.result}`);
+  if (task.error) console.log(`  Error:      ${task.error}`);
+}
+async function tasksRun(args) {
+  const taskId = args[0];
+  if (!taskId) {
+    console.error("Usage: viagen tasks run <id>");
+    process.exit(1);
+  }
+  const client = await requireClient();
+  const projectId = await requireProjectId(client);
+  const task = await client.tasks.get(projectId, taskId);
+  console.log(`Running task: ${task.id}`);
+  console.log(`  Prompt: ${task.prompt}`);
+  console.log(`  Branch: ${task.branch}`);
+  console.log("");
+  await client.tasks.update(projectId, taskId, { status: "running" });
+  const sandboxArgs = [];
+  if (task.branch) {
+    sandboxArgs.push("--branch", task.branch);
+  }
+  const timeout = parseFlag(args, "--timeout") || parseFlag(args, "-t");
+  if (timeout) {
+    sandboxArgs.push("--timeout", timeout);
+  }
+  try {
+    await sandbox(sandboxArgs, {
+      promptOverride: task.prompt,
+      extraEnvVars: {
+        VIAGEN_TASK_ID: task.id,
+        VIAGEN_PROJECT_ID: projectId
+      }
+    });
+  } catch (err) {
+    console.error(`Sandbox deploy failed: ${err instanceof Error ? err.message : String(err)}`);
+    await client.tasks.update(projectId, taskId, { status: "ready" }).catch(() => {
+    });
+    process.exit(1);
+  }
+}
+async function tasks(args) {
+  const sub = args[0];
+  if (sub === "create") {
+    await tasksCreate(args.slice(1));
+    return;
+  }
+  if (sub === "get") {
+    await tasksGet(args.slice(1));
+    return;
+  }
+  if (sub === "run") {
+    await tasksRun(args.slice(1));
+    return;
+  }
+  await tasksList(args);
+}
 function help() {
   console.log("viagen \u2014 Claude Code in your Vite dev server");
   console.log("");
@@ -1411,6 +1666,10 @@ function help() {
   console.log("  projects create <name>         Create a new project");
   console.log("  projects get <id>              Show project details");
   console.log("  projects delete <id>           Delete a project");
+  console.log("  tasks                          List tasks for current project");
+  console.log("  tasks create <prompt>          Create a new task");
+  console.log("  tasks get <id>                 Show task details");
+  console.log("  tasks run <id>                 Run a task in a sandbox");
   console.log("  dev                            Start Vite and open the split view");
   console.log("  setup                          Set up .env with API keys and tokens");
   console.log("  sandbox [-b branch] [-t min]   Deploy your project to a Vercel Sandbox");
@@ -1422,6 +1681,10 @@ function help() {
   console.log("  -b, --branch <name>   Branch to clone (default: current branch)");
   console.log("  -t, --timeout <min>   Sandbox timeout in minutes (default: 30)");
   console.log("");
+  console.log("Task options:");
+  console.log("  -s, --status <status>   Filter tasks by status (list)");
+  console.log("  -b, --branch <name>     Branch for the task (create)");
+  console.log("");
   console.log("Getting started:");
   console.log("  1. npm install viagen");
   console.log("  2. Add viagen() to your vite.config.ts plugins");
@@ -1483,6 +1746,8 @@ async function main() {
     await orgs(args.slice(1));
   } else if (command === "projects") {
     await projects(args.slice(1));
+  } else if (command === "tasks") {
+    await tasks(args.slice(1));
   } else if (command === "dev") {
     dev();
     return;

package/dist/index.js

@@ -1,7 +1,7 @@
 // src/index.ts
 import { execSync as execSync2 } from "child_process";
 import { mkdirSync as mkdirSync2, writeFileSync as writeFileSync4 } from "fs";
-import { join as join6 } from "path";
+import { join as join7 } from "path";
 import { loadEnv } from "vite";
 
 // src/logger.ts
@@ -59,6 +59,8 @@ function wrapLogger(logger, buffer) {
 }
 
 // src/health.ts
+import { readFileSync } from "fs";
+import { join as join2 } from "path";
 function registerHealthRoutes(server, env, errorRef) {
   server.middlewares.use("/via/error", (_req, res) => {
     res.setHeader("Content-Type", "application/json");
@@ -84,20 +86,81 @@ function registerHealthRoutes(server, env, errorRef) {
       timeoutSeconds: sessionTimeout
     } : null;
     const prompt = env["VIAGEN_PROMPT"] || null;
+    const taskId = env["VIAGEN_TASK_ID"] || null;
+    const projectId = env["VIAGEN_PROJECT_ID"] || null;
     res.setHeader("Content-Type", "application/json");
     if (configured) {
-      res.end(JSON.stringify({ status: "ok", configured: true, git, vercel, branch, session, prompt, missing }));
+      res.end(JSON.stringify({ status: "ok", configured: true, git, vercel, branch, session, prompt, taskId, projectId, missing }));
     } else {
       res.end(
-        JSON.stringify({ status: "error", configured: false, git, vercel, branch, session, prompt, missing })
+        JSON.stringify({ status: "error", configured: false, git, vercel, branch, session, prompt, taskId, projectId, missing })
+      );
+    }
+  });
+  let currentVersion = null;
+  let versionCache = null;
+  function getCurrentVersion() {
+    if (currentVersion) return currentVersion;
+    try {
+      const pkg = JSON.parse(
+        readFileSync(join2(__dirname, "..", "package.json"), "utf-8")
+      );
+      currentVersion = pkg.version;
+    } catch {
+      try {
+        const pkg = JSON.parse(
+          readFileSync(
+            join2(__dirname, "package.json"),
+            "utf-8"
+          )
+        );
+        currentVersion = pkg.version;
+      } catch {
+        currentVersion = "0.0.0";
+      }
+    }
+    return currentVersion;
+  }
+  server.middlewares.use("/via/version", (_req, res) => {
+    res.setHeader("Content-Type", "application/json");
+    const current = getCurrentVersion();
+    if (versionCache && Date.now() - versionCache.ts < 3e5) {
+      res.end(
+        JSON.stringify({
+          current,
+          latest: versionCache.latest,
+          updateAvailable: versionCache.latest !== current
+        })
       );
+      return;
     }
+    fetch("https://registry.npmjs.org/viagen/latest", {
+      headers: { Accept: "application/json" }
+    }).then((r) => r.json()).then((data) => {
+      const latest = data.version ?? current;
+      versionCache = { latest, ts: Date.now() };
+      res.end(
+        JSON.stringify({
+          current,
+          latest,
+          updateAvailable: latest !== current
+        })
+      );
+    }).catch(() => {
+      res.end(
+        JSON.stringify({
+          current,
+          latest: null,
+          updateAvailable: false
+        })
+      );
+    });
   });
 }
 
 // src/chat.ts
-import { readFileSync, writeFileSync as writeFileSync2, appendFileSync, existsSync } from "fs";
-import { join as join2 } from "path";
+import { readFileSync as readFileSync2, writeFileSync as writeFileSync2, appendFileSync, existsSync } from "fs";
+import { join as join3 } from "path";
 import {
   query
 } from "@anthropic-ai/claude-agent-sdk";
@@ -197,7 +260,7 @@ var ChatSession = class {
   lastUsedToken;
   constructor(opts) {
     this.opts = opts;
-    this.chatLogPath = join2(opts.projectRoot, ".viagen", "chat.log");
+    this.chatLogPath = join3(opts.projectRoot, ".viagen", "chat.log");
   }
   reset() {
     this.sessionId = void 0;
@@ -208,7 +271,7 @@ var ChatSession = class {
   }
   getHistory() {
     try {
-      const raw = readFileSync(this.chatLogPath, "utf-8");
+      const raw = readFileSync2(this.chatLogPath, "utf-8");
       const entries = [];
       for (const line of raw.split("\n")) {
         if (!line.trim()) continue;
@@ -245,9 +308,9 @@ var ChatSession = class {
         this.opts.env["CLAUDE_TOKEN_EXPIRES"] = String(
           nowSec + tokens.expires_in
         );
-        const envPath = join2(this.opts.projectRoot, ".env");
+        const envPath = join3(this.opts.projectRoot, ".env");
         if (existsSync(envPath)) {
-          let content = readFileSync(envPath, "utf-8");
+          let content = readFileSync2(envPath, "utf-8");
           const replacements = {
             CLAUDE_ACCESS_TOKEN: tokens.access_token,
             CLAUDE_REFRESH_TOKEN: tokens.refresh_token,
@@ -414,7 +477,19 @@ var ChatSession = class {
             sink?.({ type: "error", text: err });
           }
         }
-        sink?.({ type: "done" });
+        const doneEvent = { type: "done" };
+        if ("total_cost_usd" in msg) {
+          doneEvent.costUsd = msg.total_cost_usd;
+        }
+        if ("duration_ms" in msg) {
+          doneEvent.durationMs = msg.duration_ms;
+        }
+        if ("usage" in msg && msg.usage) {
+          const u = msg.usage;
+          doneEvent.inputTokens = u.input_tokens ?? 0;
+          doneEvent.outputTokens = u.output_tokens ?? 0;
+        }
+        sink?.(doneEvent);
         this.currentDoneResolve?.();
         this.currentDoneResolve = null;
         this.currentEventSink = null;
@@ -1374,6 +1449,30 @@ function buildUiHtml(opts) {
       color: #d4d4d8;
       font-size: 11px;
     }
+    .update-banner {
+      padding: 6px 12px;
+      border-bottom: 1px solid #27272a;
+      background: #18181b;
+      font-family: ui-monospace, monospace;
+      font-size: 11px;
+      color: #a1a1aa;
+      flex-shrink: 0;
+      display: none;
+      align-items: center;
+      gap: 8px;
+      cursor: pointer;
+      transition: background 0.15s;
+    }
+    .update-banner:hover { background: #1e1e22; }
+    .update-banner .update-badge {
+      font-size: 9px;
+      font-weight: 700;
+      padding: 1px 5px;
+      border-radius: 3px;
+      background: #365314;
+      color: #86efac;
+      text-transform: uppercase;
+    }
     .btn {
       padding: 5px 10px;
       border: 1px solid #3f3f46;
@@ -1800,13 +1899,14 @@ function buildUiHtml(opts) {
   </div>` : ""}
   <div id="chat-view" style="display:flex;flex-direction:column;flex:1;overflow:hidden;">
     <div class="setup-banner" id="setup-banner"></div>
+    <div class="update-banner" id="update-banner"><span class="update-badge">update</span><span id="update-text"></span></div>
     <div class="activity-bar" id="activity-bar"></div>
     <div class="messages" id="messages"></div>
     <div class="input-area">
       <input type="text" id="input" placeholder="What do you want to build?" autofocus />
       <button class="send-btn" id="send-btn">Send</button>
     </div>
-    ${hasGit ? '<div class="status-bar" id="status-bar" style="display:none;"><span id="status-branch"></span><span id="status-diff"></span></div>' : ""}
+    <div class="status-bar" id="status-bar">${hasGit ? '<span id="status-branch"></span><span id="status-diff"></span>' : ""}<span id="status-cost" style="display:none;margin-left:auto;"></span></div>
   </div>
   ${editor ? editor.html : ""}
   ${hasGit ? `<div id="changes-view" style="display:none;flex-direction:column;flex:1;overflow:hidden;">
@@ -1895,6 +1995,12 @@ function buildUiHtml(opts) {
       } catch(e) {}
     }
 
+    function scrollToBottom() {
+      requestAnimationFrame(function() {
+        scrollToBottom();
+      });
+    }
+
     function formatDuration(ms) {
       if (ms < 1000) return ms + 'ms';
       var secs = Math.round(ms / 1000);
@@ -1919,14 +2025,41 @@ function buildUiHtml(opts) {
       activityTimer = setInterval(updateActivityBar, 1000);
     }
 
-    function hideActivity() {
+    var sessionCostUsd = 0;
+    var sessionInputTokens = 0;
+    var sessionOutputTokens = 0;
+
+    function formatCost(usd) {
+      if (usd < 0.01) return '<$0.01';
+      return '$' + usd.toFixed(2);
+    }
+
+    function formatTokens(n) {
+      if (n >= 1000000) return (n / 1000000).toFixed(1) + 'M';
+      if (n >= 1000) return (n / 1000).toFixed(1) + 'k';
+      return String(n);
+    }
+
+    function hideActivity(usage) {
       if (activityTimer) { clearInterval(activityTimer); activityTimer = null; }
       var elapsed = formatDuration(Date.now() - sendStartTime);
       var parts = ['Done in ' + elapsed];
       if (toolCount > 0) parts.push(toolCount + (toolCount === 1 ? ' action' : ' actions'));
-      activityBar.textContent = parts.join(' \xB7 ');
+      if (usage && usage.costUsd != null) {
+        parts.push(formatCost(usage.costUsd));
+        sessionCostUsd += usage.costUsd;
+        sessionInputTokens += (usage.inputTokens || 0);
+        sessionOutputTokens += (usage.outputTokens || 0);
+      }
+      activityBar.textContent = parts.join(' \\u00b7 ');
       activityBar.classList.add('done');
       setTimeout(function() { activityBar.style.display = 'none'; }, 5000);
+      // Update status bar with session total
+      var costEl = document.getElementById('status-cost');
+      if (costEl && sessionCostUsd > 0) {
+        costEl.textContent = formatCost(sessionCostUsd) + ' (' + formatTokens(sessionInputTokens + sessionOutputTokens) + ' tokens)';
+        costEl.style.display = '';
+      }
     }
     window.addEventListener('beforeunload', function() { unloading = true; stopHistoryPolling(); });
     window.addEventListener('pagehide', function() { unloading = true; });
@@ -2110,7 +2243,7 @@ function buildUiHtml(opts) {
       chatLog.push({ type: 'user', content: text });
 
       renderUserMessage(text);
-      messagesEl.scrollTop = messagesEl.scrollHeight;
+      scrollToBottom();
     }
 
     function appendText(text) {
@@ -2131,7 +2264,7 @@ function buildUiHtml(opts) {
       }
       var fullText = chatLog[chatLog.length - 1].content;
       currentTextSpan.innerHTML = renderMarkdown(fullText);
-      messagesEl.scrollTop = messagesEl.scrollHeight;
+      scrollToBottom();
     }
 
     function addToolBlock(name, input) {
@@ -2140,7 +2273,7 @@ function buildUiHtml(opts) {
       chatLog.push({ type: 'tool', content: label });
 
       renderToolBlock(label);
-      messagesEl.scrollTop = messagesEl.scrollHeight;
+      scrollToBottom();
     }
 
     function renderToolResult(text) {
@@ -2167,7 +2300,7 @@ function buildUiHtml(opts) {
       chatLog.push({ type: 'error', content: text });
 
       renderErrorBlock(text);
-      messagesEl.scrollTop = messagesEl.scrollHeight;
+      scrollToBottom();
     }
 
     function setStreaming(v) {
@@ -2179,18 +2312,7 @@ function buildUiHtml(opts) {
       else startHistoryPolling();
     }
 
-    async function send() {
-      var text = inputEl.value.trim();
-      if (!text || isStreaming) return;
-
-      addUserMessage(text);
-      inputEl.value = '';
-      setStreaming(true);
-      currentTextSpan = null;
-      sendStartTime = Date.now();
-      toolCount = 0;
-      showActivity();
-
+    async function sendRaw(text) {
       try {
         var res = await fetch('/via/chat', {
           method: 'POST',
@@ -2201,6 +2323,7 @@ function buildUiHtml(opts) {
         var reader = res.body.getReader();
         var decoder = new TextDecoder();
         var buffer = '';
+        var lastUsage = null;
 
         while (true) {
           var result = await reader.read();
@@ -2218,6 +2341,7 @@ function buildUiHtml(opts) {
               else if (data.type === 'tool_use') { toolCount++; updateActivityBar(); addToolBlock(data.name, data.input); }
               else if (data.type === 'tool_result') addToolResult(data.text);
               else if (data.type === 'error') addErrorBlock(data.text);
+              else if (data.type === 'done') lastUsage = data;
             } catch (e) {}
           }
         }
@@ -2225,14 +2349,29 @@ function buildUiHtml(opts) {
         if (!unloading) addErrorBlock('Connection failed');
       }
 
-      hideActivity();
+      hideActivity(lastUsage);
       playDoneSound();
-      // Advance timestamp so polling doesn't re-render messages from this stream
       historyTimestamp = Date.now();
       setStreaming(false);
       inputEl.focus();
     }
 
+    async function send() {
+      var text = inputEl.value.trim();
+      if (!text || isStreaming) return;
+
+      addUserMessage(text);
+      inputEl.value = '';
+      setStreaming(true);
+      currentTextSpan = null;
+      sendStartTime = Date.now();
+      toolCount = 0;
+      showActivity();
+      scrollToBottom();
+
+      await sendRaw(text);
+    }
+
     sendBtn.addEventListener('click', send);
     inputEl.addEventListener('keydown', function (e) {
       if (e.key === 'Enter' && !e.shiftKey) { e.preventDefault(); send(); }
@@ -2383,7 +2522,6 @@ function buildUiHtml(opts) {
             var statusBar = document.getElementById('status-bar');
             var statusBranch = document.getElementById('status-branch');
             if (statusBar && statusBranch) {
-              statusBar.style.display = 'flex';
               statusBranch.textContent = '\\u2387 ' + d.branch;
               if (branchUrl) {
                 statusBranch.addEventListener('click', function() { window.open(branchUrl, '_blank'); });
@@ -2416,10 +2554,43 @@ function buildUiHtml(opts) {
           }).catch(function() {});
         }
 
+        // Check for viagen updates
+        fetch('/via/version').then(function(r) { return r.json(); }).then(function(v) {
+          if (v.updateAvailable && v.latest) {
+            var updateBanner = document.getElementById('update-banner');
+            var updateText = document.getElementById('update-text');
+            if (updateBanner && updateText) {
+              updateText.textContent = 'viagen ' + v.latest + ' available (current: ' + v.current + ')';
+              updateBanner.style.display = 'flex';
+              updateBanner.addEventListener('click', function() {
+                updateBanner.style.display = 'none';
+                inputEl.value = 'Update viagen to v' + v.latest + ' (npm install viagen@' + v.latest + ') and restart the dev server.';
+                send();
+              });
+            }
+          }
+        }).catch(function() {});
+
         // Only auto-send prompt if no history exists (first boot)
         if (data.prompt && data.configured && chatLog.length === 0) {
-          inputEl.value = data.prompt;
-          send();
+          if (data.taskId) {
+            // Task mode: show link instead of raw prompt
+            var taskUrl = 'https://app.viagen.dev' + (data.projectId ? '/' + data.projectId : '') + '/' + data.taskId;
+            var div = document.createElement('div');
+            div.className = 'msg msg-user';
+            div.innerHTML = '<span class="label">Task</span><span class="text">Received instructions from <a href="' + escapeHtml(taskUrl) + '" target="_blank" style="color:#93c5fd;text-decoration:underline;">Viagen Task</a></span>';
+            messagesEl.appendChild(div);
+            scrollToBottom();
+            // Send the prompt silently (don't show it as a user message)
+            showActivity();
+            setStreaming(true);
+            sendStartTime = Date.now();
+            toolCount = 0;
+            sendRaw(data.prompt);
+          } else {
+            inputEl.value = data.prompt;
+            send();
+          }
         }
       })
       .catch(function() {
@@ -2804,11 +2975,11 @@ function createAuthMiddleware(token) {
 // src/files.ts
 import {
   readdirSync,
-  readFileSync as readFileSync2,
+  readFileSync as readFileSync3,
   writeFileSync as writeFileSync3,
   statSync
 } from "fs";
-import { join as join3, resolve, relative } from "path";
+import { join as join4, resolve, relative } from "path";
 function readBody2(req) {
   return new Promise((resolve3, reject) => {
     let body = "";
@@ -2823,7 +2994,7 @@ function collectFiles(dir, projectRoot) {
   const results = [];
   try {
     for (const entry of readdirSync(dir, { withFileTypes: true })) {
-      const fullPath = join3(dir, entry.name);
+      const fullPath = join4(dir, entry.name);
       if (entry.isDirectory()) {
         if (entry.name === "node_modules" || entry.name === ".git") continue;
         results.push(...collectFiles(fullPath, projectRoot));
@@ -2911,7 +3082,7 @@ function registerFileRoutes(server, opts) {
     const ext = filePath.split(".").pop()?.toLowerCase() ?? "";
     const mime = MIME_TYPES[ext] ?? "application/octet-stream";
     try {
-      const data = readFileSync2(abs);
+      const data = readFileSync3(abs);
       res.setHeader("Content-Type", mime);
       res.setHeader("Cache-Control", "no-cache");
       res.end(data);
@@ -2949,7 +3120,7 @@ function registerFileRoutes(server, opts) {
       }
       const abs = resolve(opts.projectRoot, filePath);
       try {
-        const content = readFileSync2(abs, "utf-8");
+        const content = readFileSync3(abs, "utf-8");
         res.setHeader("Content-Type", "application/json");
         res.end(JSON.stringify({ path: filePath, content }));
       } catch {
@@ -3057,8 +3228,8 @@ function createInjectionMiddleware() {
 }
 
 // src/git.ts
-import { readFileSync as readFileSync3 } from "fs";
-import { join as join4, resolve as resolve2 } from "path";
+import { readFileSync as readFileSync4 } from "fs";
+import { join as join5, resolve as resolve2 } from "path";
 import {
   simpleGit
 } from "simple-git";
@@ -3101,7 +3272,7 @@ async function getDiffStats(git, repoRoot, untrackedFiles) {
     }
     for (const filePath of untrackedFiles) {
       try {
-        const content = readFileSync3(join4(repoRoot, filePath), "utf-8");
+        const content = readFileSync4(join5(repoRoot, filePath), "utf-8");
         const lines = content.split("\n").length;
         stats.set(filePath, { ins: lines, del: 0 });
       } catch {
@@ -3127,7 +3298,7 @@ async function getFileDiff(git, repoRoot, filePath) {
     if (staged) return staged;
     if (unstaged) return unstaged;
     try {
-      const content = readFileSync3(join4(repoRoot, filePath), "utf-8");
+      const content = readFileSync4(join5(repoRoot, filePath), "utf-8");
       const lines = content.split("\n");
       const added = lines.map((l) => `+${l}`).join("\n");
       return `--- /dev/null
@@ -3332,8 +3503,8 @@ function registerLogRoutes(server, opts) {
 
 // src/sandbox.ts
 import { randomUUID } from "crypto";
-import { readFileSync as readFileSync4, readdirSync as readdirSync2 } from "fs";
-import { join as join5, relative as relative2 } from "path";
+import { readFileSync as readFileSync5, readdirSync as readdirSync2 } from "fs";
+import { join as join6, relative as relative2 } from "path";
 import { Sandbox } from "@vercel/sandbox";
 var SKIP_DIRS = /* @__PURE__ */ new Set([
   "node_modules",
@@ -3349,13 +3520,13 @@ function collectFiles2(dir, base) {
   for (const entry of readdirSync2(dir, { withFileTypes: true })) {
     if (entry.name.startsWith(".") && SKIP_DIRS.has(entry.name)) continue;
     if (SKIP_DIRS.has(entry.name)) continue;
-    const fullPath = join5(dir, entry.name);
+    const fullPath = join6(dir, entry.name);
     const relPath = relative2(base, fullPath);
     if (entry.isDirectory()) {
       files.push(...collectFiles2(fullPath, base));
     } else if (entry.isFile()) {
       if (SKIP_FILES.has(entry.name)) continue;
-      files.push({ path: relPath, content: readFileSync4(fullPath) });
+      files.push({ path: relPath, content: readFileSync5(fullPath) });
     }
   }
   return files;
@@ -3367,6 +3538,57 @@ function extractHost(httpsUrl) {
     return "github.com";
   }
 }
+async function waitForServer(baseUrl, devServer) {
+  const timeout = 6e4;
+  const start = Date.now();
+  const ac = new AbortController();
+  let serverOutput = "";
+  const logStream = (async () => {
+    try {
+      for await (const log of devServer.logs({ signal: ac.signal })) {
+        const line = log.data;
+        serverOutput += line;
+        if (log.stream === "stderr") {
+          process.stderr.write(`  ${line}`);
+        } else {
+          process.stdout.write(`  ${line}`);
+        }
+      }
+    } catch {
+    }
+  })();
+  while (Date.now() - start < timeout) {
+    if (devServer.exitCode !== null) {
+      ac.abort();
+      await logStream.catch(() => {
+      });
+      return {
+        ok: false,
+        error: serverOutput.slice(-2e3) || `Process exited with code ${devServer.exitCode}`
+      };
+    }
+    try {
+      const res = await fetch(`${baseUrl}/via/health`, {
+        signal: AbortSignal.timeout(3e3)
+      });
+      if (res.ok) {
+        ac.abort();
+        await logStream.catch(() => {
+        });
+        return { ok: true };
+      }
+    } catch {
+    }
+    await new Promise((r) => setTimeout(r, 2e3));
+  }
+  ac.abort();
+  await logStream.catch(() => {
+  });
+  return {
+    ok: false,
+    error: serverOutput.slice(-2e3) || "Timed out waiting for dev server"
+  };
+}
 async function deploySandbox(opts) {
   const token = randomUUID();
   const useGit = !!opts.git;
@@ -3418,7 +3640,11 @@ async function deploySandbox(opts) {
         "user.email",
         opts.git.userEmail
       ]);
-      await sandbox.runCommand("git", ["checkout", "-B", opts.git.branch]);
+      const checkout = await sandbox.runCommand("git", ["checkout", "-B", opts.git.branch]);
+      if (checkout.exitCode !== 0) {
+        const err = await checkout.stderr();
+        throw new Error(`git checkout failed (exit ${checkout.exitCode}): ${err}`);
+      }
       await sandbox.runCommand("bash", [
         "-c",
         `echo 'https://x-access-token:${opts.git.token}@${extractHost(opts.git.remoteUrl)}' > ~/.git-credentials`
@@ -3480,20 +3706,30 @@ async function deploySandbox(opts) {
         content: Buffer.from(envLines.join("\n"))
       }
     ]);
-    const install = await sandbox.runCommand("npm", ["install"]);
+    console.log("  Installing dependencies...");
+    const install = await sandbox.runCommand({
+      cmd: "npm",
+      args: ["install"],
+      stdout: process.stdout,
+      stderr: process.stderr
+    });
     if (install.exitCode !== 0) {
-      const stderr = await install.stderr();
-      throw new Error(
-        `npm install failed (exit ${install.exitCode}): ${stderr}`
-      );
+      throw new Error(`npm install failed (exit ${install.exitCode})`);
     }
-    await sandbox.runCommand({
+    const devServer = await sandbox.runCommand({
       cmd: "npm",
       args: ["run", "dev", "--", "--host", "0.0.0.0"],
       detached: true
     });
+    console.log("  Starting dev server...");
     const baseUrl = sandbox.domain(5173);
     const url = `${baseUrl}/t/${token}`;
+    const ready = await waitForServer(baseUrl, devServer);
+    if (!ready.ok) {
+      throw new Error(
+        `Dev server failed to start: ${ready.error}`
+      );
+    }
     return {
       url,
       token,
@@ -3535,10 +3771,10 @@ function viagen(options) {
       projectRoot = config.root;
       logBuffer.init(projectRoot);
       wrapLogger(config.logger, logBuffer);
-      const viagenDir = join6(projectRoot, ".viagen");
+      const viagenDir = join7(projectRoot, ".viagen");
       mkdirSync2(viagenDir, { recursive: true });
       writeFileSync4(
-        join6(viagenDir, "config.json"),
+        join7(viagenDir, "config.json"),
         JSON.stringify({
           sandboxFiles: options?.sandboxFiles ?? [],
           editable: options?.editable ?? []

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viagen",
-  "version": "0.0.32",
+  "version": "0.0.36",
   "description": "Vite dev server plugin that exposes endpoints for chatting with Claude Code SDK",
   "type": "module",
   "main": "./dist/index.js",
@@ -44,7 +44,7 @@
     "@vercel/sandbox": "^1",
     "lucide-react": "^0.564.0",
     "simple-git": "^3.31.1",
-    "viagen-sdk": "^0.0.2"
+    "viagen-sdk": "^0.0.4"
   },
   "license": "MIT",
   "repository": {