wtt-connect 0.2.40 → 0.2.42

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wtt-connect",
-  "version": "0.2.40",
+  "version": "0.2.42",
   "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,158 @@ 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 pids = listenerPidsForPort(port);
+  const killed = [];
+  for (const pid of pids) {
+    try {
+      process.kill(pid, 'SIGTERM');
+      killed.push(String(pid));
+    } catch {}
+  }
+  const deadline = Date.now() + 500;
+  while (Date.now() < deadline && pids.some((pid) => isProcessAlive(pid))) {
+    Atomics.wait(new Int32Array(new SharedArrayBuffer(4)), 0, 0, 50);
+  }
+  for (const pid of pids) {
+    if (!isProcessAlive(pid)) continue;
+    try {
+      process.kill(pid, 'SIGKILL');
+      if (!killed.includes(String(pid))) killed.push(String(pid));
+    } catch {}
+  }
+  return { killed, ok: true };
+}
+
+function listenerPidsForPort(port) {
+  const inodes = listenerInodesForPort(port);
+  if (!inodes.size) return [];
+  const pids = [];
+  for (const name of safeReadDir('/proc')) {
+    if (!/^\d+$/.test(name)) continue;
+    const fdDir = `/proc/${name}/fd`;
+    for (const fd of safeReadDir(fdDir)) {
+      let target = '';
+      try {
+        target = fs.readlinkSync(path.join(fdDir, fd));
+      } catch {
+        continue;
+      }
+      const match = target.match(/^socket:\[(\d+)\]$/);
+      if (match && inodes.has(match[1])) {
+        pids.push(Number(name));
+        break;
+      }
+    }
+  }
+  return [...new Set(pids)];
+}
+
+function listenerInodesForPort(port) {
+  const targetHex = Number(port).toString(16).toUpperCase().padStart(4, '0');
+  const inodes = new Set();
+  for (const file of ['/proc/net/tcp', '/proc/net/tcp6']) {
+    let text = '';
+    try {
+      text = fs.readFileSync(file, 'utf8');
+    } catch {
+      continue;
+    }
+    for (const line of text.split('\n').slice(1)) {
+      const parts = line.trim().split(/\s+/);
+      if (parts.length < 10) continue;
+      const local = parts[1] || '';
+      const state = parts[3] || '';
+      const inode = parts[9] || '';
+      const localPort = local.split(':')[1] || '';
+      if (state === '0A' && localPort.toUpperCase() === targetHex && inode) inodes.add(inode);
+    }
+  }
+  return inodes;
+}
+
+function safeReadDir(dir) {
+  try {
+    return fs.readdirSync(dir);
+  } catch {
+    return [];
+  }
+}
+
+function isProcessAlive(pid) {
+  try {
+    process.kill(pid, 0);
+    return true;
+  } catch {
+    return false;
+  }
 }
 
 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>}'`,