viagen 0.0.11 → 0.0.12

package/README.md

@@ -74,8 +74,17 @@ help build and modify the app. Files you edit will trigger Vite HMR
 automatically. You can read .viagen/server.log to check recent Vite dev server
 output (compile errors, HMR updates, warnings). When running in a sandbox with
 git, the gh CLI is available and authenticated — you can create pull requests,
-comment on issues, and manage releases. If Vercel credentials are set, you can
-run "vercel deploy" to publish a preview and share the URL. Be concise.
+comment on issues, and manage releases.
+
+Publishing workflow:
+- If you are on a feature branch (not main/master): commit your changes, push
+  to the remote, and create a pull request using "gh pr create". Share the PR URL.
+- If you are on main/master and Vercel credentials are set ($VERCEL_TOKEN):
+  commit, push, and run "vercel deploy" to publish a preview. Share the preview URL.
+- Check your current branch with "git branch --show-current" before deciding
+  which workflow to use.
+
+Be concise.
 ```
 
 Recent build errors are automatically appended to give Claude context about what went wrong. To customize the prompt, you can replace it entirely or extend the default:

package/dist/cli.js

@@ -140,6 +140,7 @@ async function deploySandbox(opts) {
     }
     if (opts.git) {
       envMap["GITHUB_TOKEN"] = opts.git.token;
+      envMap["VIAGEN_BRANCH"] = opts.git.branch;
     }
     if (opts.vercel) {
       envMap["VERCEL_TOKEN"] = opts.vercel.token;

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import { Plugin } from 'vite';
 
-declare const DEFAULT_SYSTEM_PROMPT = "You are embedded in a Vite dev server as the \"viagen\" plugin. Your job is to help build and modify the app. Files you edit will trigger Vite HMR automatically. You can read .viagen/server.log to check recent Vite dev server output (compile errors, HMR updates, warnings). When running in a sandbox with git, the gh CLI is available and authenticated \u2014 you can create pull requests, comment on issues, and manage releases. If Vercel credentials are set, you can run \"vercel deploy\" to publish a preview and share the URL. Be concise.";
+declare const DEFAULT_SYSTEM_PROMPT = "You are embedded in a Vite dev server as the \"viagen\" plugin. Your job is to help build and modify the app. Files you edit will trigger Vite HMR automatically. You can read .viagen/server.log to check recent Vite dev server output (compile errors, HMR updates, warnings). When running in a sandbox with git, the gh CLI is available and authenticated \u2014 you can create pull requests, comment on issues, and manage releases.\n\nPublishing workflow:\n- If you are on a feature branch (not main/master): commit your changes, push to the remote, and create a pull request using \"gh pr create\". Share the PR URL.\n- If you are on main/master and Vercel credentials are set ($VERCEL_TOKEN): commit, push, and run \"vercel deploy\" to publish a preview. Share the preview URL.\n- Check your current branch with \"git branch --show-current\" before deciding which workflow to use.\n\nBe concise.";
 
 interface GitInfo {
     /** HTTPS remote URL (transformed from SSH if needed). */

package/dist/index.js

@@ -63,6 +63,7 @@ function registerHealthRoutes(server, env, errorRef) {
   server.middlewares.use("/via/health", (_req, res) => {
     const configured = !!env["ANTHROPIC_API_KEY"] || !!env["CLAUDE_ACCESS_TOKEN"];
     const git = !!env["GITHUB_TOKEN"];
+    const branch = env["VIAGEN_BRANCH"] || null;
     const sessionStart = env["VIAGEN_SESSION_START"] ? parseInt(env["VIAGEN_SESSION_START"], 10) : null;
     const sessionTimeout = env["VIAGEN_SESSION_TIMEOUT"] ? parseInt(env["VIAGEN_SESSION_TIMEOUT"], 10) : null;
     const session = sessionStart && sessionTimeout ? {
@@ -72,10 +73,10 @@ function registerHealthRoutes(server, env, errorRef) {
     } : null;
     res.setHeader("Content-Type", "application/json");
     if (configured) {
-      res.end(JSON.stringify({ status: "ok", configured: true, git, session }));
+      res.end(JSON.stringify({ status: "ok", configured: true, git, branch, session }));
     } else {
       res.end(
-        JSON.stringify({ status: "error", configured: false, git, session })
+        JSON.stringify({ status: "error", configured: false, git, branch, session })
       );
     }
   });
@@ -119,7 +120,14 @@ function readBody(req) {
     req.on("error", reject);
   });
 }
-var DEFAULT_SYSTEM_PROMPT = `You are embedded in a Vite dev server as the "viagen" plugin. Your job is to help build and modify the app. Files you edit will trigger Vite HMR automatically. You can read .viagen/server.log to check recent Vite dev server output (compile errors, HMR updates, warnings). When running in a sandbox with git, the gh CLI is available and authenticated \u2014 you can create pull requests, comment on issues, and manage releases. If Vercel credentials are set, you can run "vercel deploy" to publish a preview and share the URL. Be concise.`;
+var DEFAULT_SYSTEM_PROMPT = `You are embedded in a Vite dev server as the "viagen" plugin. Your job is to help build and modify the app. Files you edit will trigger Vite HMR automatically. You can read .viagen/server.log to check recent Vite dev server output (compile errors, HMR updates, warnings). When running in a sandbox with git, the gh CLI is available and authenticated \u2014 you can create pull requests, comment on issues, and manage releases.
+
+Publishing workflow:
+- If you are on a feature branch (not main/master): commit your changes, push to the remote, and create a pull request using "gh pr create". Share the PR URL.
+- If you are on main/master and Vercel credentials are set ($VERCEL_TOKEN): commit, push, and run "vercel deploy" to publish a preview. Share the preview URL.
+- Check your current branch with "git branch --show-current" before deciding which workflow to use.
+
+Be concise.`;
 function findClaudeBin() {
   const _require = createRequire(import.meta.url);
   const pkgPath = _require.resolve("@anthropic-ai/claude-code/package.json");
@@ -1132,7 +1140,11 @@ function buildUiHtml() {
     });
     publishBtn.addEventListener('click', function () {
       if (isStreaming) return;
-      inputEl.value = 'Commit all changes, push to the remote repository, and run vercel deploy to get a preview URL';
+      if (publishBtn.dataset.branch && publishBtn.dataset.branch !== 'main' && publishBtn.dataset.branch !== 'master') {
+        inputEl.value = 'Commit all changes, push to the remote branch, and create a pull request using gh pr create. Share the PR URL.';
+      } else {
+        inputEl.value = 'Commit all changes, push to the remote repository, and run vercel deploy to get a preview URL';
+      }
       send();
     });
 
@@ -1182,7 +1194,10 @@ function buildUiHtml() {
         var banner = document.getElementById('setup-banner');
         if (data.configured) {
           dot.className = 'status-dot ok';
-          if (data.git) publishBtn.style.display = '';
+          if (data.git) {
+            publishBtn.style.display = '';
+            if (data.branch) publishBtn.dataset.branch = data.branch;
+          }
         } else {
           dot.className = 'status-dot error';
           inputEl.disabled = true;
@@ -1272,7 +1287,7 @@ function createAuthMiddleware(token) {
       const cleanUrl = url.pathname + (url.search || "");
       res.setHeader(
         "Set-Cookie",
-        `viagen_session=${token}; HttpOnly; SameSite=Strict; Path=/`
+        `viagen_session=${token}; HttpOnly; SameSite=Lax; Path=/; Secure`
       );
       res.writeHead(302, { Location: cleanUrl });
       res.end();
@@ -1417,6 +1432,7 @@ async function deploySandbox(opts) {
     }
     if (opts.git) {
       envMap["GITHUB_TOKEN"] = opts.git.token;
+      envMap["VIAGEN_BRANCH"] = opts.git.branch;
     }
     if (opts.vercel) {
       envMap["VERCEL_TOKEN"] = opts.vercel.token;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viagen",
-  "version": "0.0.11",
+  "version": "0.0.12",
   "description": "Vite dev server plugin that exposes endpoints for chatting with Claude Code SDK",
   "type": "module",
   "main": "./dist/index.js",

package/site/index.html

@@ -458,8 +458,15 @@ will trigger Vite HMR automatically. You can read
 (compile errors, HMR updates, warnings). When running in a
 sandbox with git, the gh CLI is available and authenticated —
 you can create pull requests, comment on issues, and manage
-releases. If Vercel credentials are set, you can run
-"vercel deploy" to publish a preview and share the URL.
+releases.
+
+Publishing workflow:
+- On a feature branch: commit, push, and create a pull
+  request using "gh pr create". Share the PR URL.
+- On main/master with Vercel credentials: commit, push,
+  and run "vercel deploy" to publish a preview.
+- Check your branch with "git branch --show-current" first.
+
 Be concise.</code></pre>
                 <p
                     style="