wtt-connect 0.2.40 → 0.2.41

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wtt-connect",
-  "version": "0.2.40",
+  "version": "0.2.41",
   "private": false,
   "description": "WTT-native connector daemon for Codex, Claude Code, Cursor, Gemini, ACP, and other coding agent surfaces.",
   "type": "module",

package/src/main.js

@@ -36,6 +36,7 @@ export async function main(args) {
   if (cmd === 'upload-file' || cmd === 'send-file') return uploadFile(loadConfig(argv), argv);
   if (cmd === 'upload-artifact' || cmd === 'opendesign-upload') return uploadArtifact(loadConfig(argv), argv);
   if (cmd === 'preview-port' || cmd === 'sandbox-preview') return previewPort(loadConfig(argv), argv);
+  if (cmd === 'cleanup-previews' || cmd === 'preview-cleanup') return cleanupPreviewsCommand(loadConfig(argv), argv);
   if (cmd === 'start') {
     const config = loadConfig(argv);
     if (!config.agentId) throw new Error('WTT_AGENT_ID is required');
@@ -95,6 +96,7 @@ function parseArgs(args) {
     else if (a === '--port') out.port = Number(args[++i]);
     else if (a === '--preview-name') out.previewName = args[++i];
     else if (a === '--preview-token') out.previewToken = args[++i];
+    else if (a === '--keep-last') out.keepLast = Number(args[++i]);
     else if (a === '--timeout') out.timeout = Number(args[++i]) * 1000;
     else if (a === '--sender-agent-id') out.senderAgentId = args[++i];
     else if (a === '--sender-token') out.senderToken = args[++i];
@@ -134,6 +136,7 @@ Commands:
                                   Alias for upload-artifact
   preview-port --port <port> [--topic-id <id>]
                                   Create a Cloud Sandbox port preview URL through sandbox outbox
+  cleanup-previews                Stop preview servers previously registered by this agent
   help                           Show this help
 `);
 }
@@ -143,6 +146,10 @@ async function previewPort(config, argv) {
   if (!Number.isInteger(port) || port < 1024 || port > 65535 || port === 3000) {
     throw new Error('preview-port requires --port <1024-65535>, excluding 3000');
   }
+  const cleanup = cleanupOldPreviews(config, {
+    currentPort: port,
+    keepLast: Number.isFinite(Number(argv.keepLast)) ? Number(argv.keepLast) : 0,
+  });
   const preview = await createSandboxPreviewFromOutbox(config, port, {
     name: String(argv.previewName || argv.title || '').trim(),
     token: String(argv.previewToken || '').trim(),
@@ -162,12 +169,89 @@ async function previewPort(config, argv) {
         type: 'cloud_sandbox_preview',
         port,
         preview_url: url,
+        preview_cleanup: cleanup,
       });
     } finally {
       await wtt.close();
     }
   }
-  console.log(JSON.stringify({ ok: true, port, url, preview_url: url, markdown, published }, null, 2));
+  recordPreview(config, { port, url, name: preview.name || title });
+  console.log(JSON.stringify({ ok: true, port, url, preview_url: url, markdown, cleanup, published }, null, 2));
+}
+
+async function cleanupPreviewsCommand(config, argv) {
+  const keepLast = Number.isFinite(Number(argv.keepLast)) ? Number(argv.keepLast) : 0;
+  const cleanup = cleanupOldPreviews(config, { currentPort: 0, keepLast });
+  console.log(JSON.stringify({ ok: true, cleanup }, null, 2));
+}
+
+function cleanupOldPreviews(config, { currentPort = 0, keepLast = 0 } = {}) {
+  const registry = readPreviewRegistry(config);
+  const entries = registry
+    .filter((entry) => Number(entry.port) !== Number(currentPort))
+    .sort((a, b) => Number(b.created_at_ms || 0) - Number(a.created_at_ms || 0));
+  const keep = Math.max(0, Number(keepLast) || 0);
+  const toKeep = entries.slice(0, keep);
+  const toStop = entries.slice(keep);
+  const stopped = [];
+  for (const entry of toStop) {
+    const result = stopPreviewPort(Number(entry.port));
+    stopped.push({ port: Number(entry.port), url: entry.url || '', ...result });
+  }
+  writePreviewRegistry(config, currentPort ? toKeep : toKeep);
+  return { stopped, kept: toKeep.map((entry) => ({ port: Number(entry.port), url: entry.url || '' })) };
+}
+
+function recordPreview(config, entry) {
+  const registry = readPreviewRegistry(config).filter((item) => Number(item.port) !== Number(entry.port));
+  registry.unshift({
+    port: Number(entry.port),
+    url: String(entry.url || ''),
+    name: String(entry.name || ''),
+    created_at: new Date().toISOString(),
+    created_at_ms: Date.now(),
+  });
+  writePreviewRegistry(config, registry.slice(0, 10));
+}
+
+function previewRegistryFile(config) {
+  return path.join(config.stateDir || path.join(config.workDir || process.cwd(), '.wtt-connect'), 'previews.json');
+}
+
+function readPreviewRegistry(config) {
+  try {
+    const file = previewRegistryFile(config);
+    if (!fs.existsSync(file)) return [];
+    const parsed = JSON.parse(fs.readFileSync(file, 'utf8'));
+    return Array.isArray(parsed) ? parsed : [];
+  } catch {
+    return [];
+  }
+}
+
+function writePreviewRegistry(config, registry) {
+  const file = previewRegistryFile(config);
+  fs.mkdirSync(path.dirname(file), { recursive: true });
+  fs.writeFileSync(file, `${JSON.stringify(registry, null, 2)}\n`);
+}
+
+function stopPreviewPort(port) {
+  if (!Number.isInteger(port) || port < 1024 || port > 65535) return { killed: [], reason: 'invalid_port' };
+  const script = [
+    `pids=$(ss -ltnp 2>/dev/null | awk '/:${port} / {print $NF}' | sed -n 's/.*pid=\\([0-9][0-9]*\\).*/\\1/p' | sort -u)`,
+    'killed=""',
+    'for pid in $pids; do',
+    '  if [ -n "$pid" ] && kill -0 "$pid" 2>/dev/null; then kill "$pid" 2>/dev/null || true; killed="$killed $pid"; fi',
+    'done',
+    'sleep 0.2',
+    'for pid in $pids; do',
+    '  if [ -n "$pid" ] && kill -0 "$pid" 2>/dev/null; then kill -9 "$pid" 2>/dev/null || true; fi',
+    'done',
+    'printf "%s" "$killed"',
+  ].join('; ');
+  const result = spawnSync('bash', ['-lc', script], { encoding: 'utf8', timeout: 5000 });
+  const killed = String(result.stdout || '').trim().split(/\s+/).filter(Boolean);
+  return { killed, ok: result.status === 0 };
 }
 
 async function createSandboxPreviewFromOutbox(config, port, { name = '', token = '', timeoutMs = 15000 } = {}) {

package/src/runner.js

@@ -939,9 +939,11 @@ function renderCloudSandboxStorageInstruction(config, topicId = '') {
     '- If a generated user-facing file is stored in R2/persistent output storage and should appear in WTT chat, still publish it with `wtt-connect upload-file` or a WTT artifact marker.',
     '- If the user asks for a visual page, HTML artifact, chart, dashboard, animation, or browser preview, build it in the workspace, start a local web server, create a Cloudflare Sandbox preview URL, and return that preview URL to WTT.',
     '- The Cloud Sandbox preview URL feature is only for WTT Cloud Sandbox agents. Do not use WTT backend artifact/media preview flows for live sandbox web servers.',
+    '- Before starting a new preview server, run `wtt-connect cleanup-previews` to stop older preview servers for this agent and free sandbox resources.',
     '- Preview servers must keep running after your command finishes. Start them in the background with `nohup ... >/tmp/<name>.log 2>&1 &` or an equivalent long-lived process, and bind to `0.0.0.0`.',
     '- Before publishing a preview URL, verify the server is actually listening and serving content with `curl -fsS http://127.0.0.1:<port>/ >/dev/null`.',
     '- If local curl fails, fix or restart the web server first. Never publish a preview URL for a dead port.',
+    '- `wtt-connect preview-port` automatically stops older preview servers registered by this agent before publishing the new preview, so prefer one active preview per agent unless the user asks otherwise.',
     '- If you start a web server in the sandbox and the user should preview it, call the Cloudflare Sandbox outbound Worker directly with curl and include the returned `preview_url` in your reply as `[preview_url:Short Title](<preview_url>)`.',
     '- Preview ports must be 1024-65535 and cannot be 3000. For Vite/Next-style dev servers prefer 5173, 4173, or 8080.',
     `- Preview curl rule: curl -sS -X POST "\${WTT_SANDBOX_OUTBOX_URL:-http://wtt.preview}/preview-port" -H 'content-type: application/json' -d '{"agent_id":"'\${WTT_AGENT_ID:-cloud-agent}'","port":<port>}'`,