staklink 0.4.3 → 0.4.5

package/dist/proxy-server.cjs

@@ -60777,7 +60777,7 @@ var SSEManager = class {
 var sseManager = new SSEManager();
 
 // src/proxy/version.ts
-var VERSION = "0.4.3";
+var VERSION = "0.4.5";
 
 // node_modules/uuid/dist/esm/stringify.js
 var byteToHex = [];
@@ -60950,7 +60950,12 @@ var fs8 = __toESM(require("fs"), 1);
 var path7 = __toESM(require("path"), 1);
 var os = __toESM(require("os"), 1);
 
-// src/agent/system_suffix.ts
+// src/agent/system_prompt.ts
+var SYSTEM_CORE = `You are a software systems expert. You are working on a cloud dev server (code-server) that is running a software project. The user's application is being run with pm2. You can use the pm2 cli to inspect the running processes. There may also be supporting services running as docker containers. Because you are running on a cloud server, certain operation may not be possible, such as launching a desktop application, or opening a browser window. The user can see the running application from the pm2 service called "frontend" in an iframe on their end.
+
+Do not commit or push code to github. That is handled by a workflow after you are done working.
+
+`;
 var SYSTEM_SUFFIX = `
 You can ignore the .pod-config directory if you see it, its just config stuff
 
@@ -137820,7 +137825,8 @@ async function runAgent({
   const env2 = goose_env(apiKey, model);
   console.log("RUN goose with env", env2);
   const cleanPrompt = sanitizeShellArg(prompt);
-  let system = system_prompt ? sanitizeShellArg(system_prompt) + "\n\n" : "";
+  let system = SYSTEM_CORE;
+  system += system_prompt ? sanitizeShellArg(system_prompt) + "\n\n" : "";
   system += SYSTEM_SUFFIX;
   system = sanitizeShellArg(system);
   console.log("Using Goose CLI", cleanPrompt, system);
@@ -139430,7 +139436,8 @@ async function runAgent3({
   session,
   summarize
 }) {
-  let system = system_prompt ? system_prompt + "\n\n" : "";
+  let system = SYSTEM_CORE;
+  system += system_prompt ? system_prompt + "\n\n" : "";
   system += SYSTEM_SUFFIX;
   console.log("Codex agent using API key:", apiKey ? `${apiKey.substring(0, 8)}...` : "MISSING");
   if (apiKey) {
@@ -139463,7 +139470,7 @@ __export(repair_exports, {
   makeRepairPrompt: () => makeRepairPrompt,
   parseXML: () => parseXML
 });
-var SYSTEM2 = `You are a software systems expert. Your job is to help set up the development environment for a software project. Fix the user's pm2 config and/or docker-compose files, and use the pm2 and docker cli to get the project running. YOUR GOAL IS TO THE GET THE frontend APP IN THE PM2 CONFIG FILE TO RUN CORRECTLY! Other apps and docker containers are only supporting services. You can restart the pm2 environment with "npx -y staklink reload" after making changes to the pm2 config or docker-compose files.
+var SYSTEM2 = SYSTEM_CORE + `Your job is to help set up the development environment for a software project. Fix the user's pm2 config and/or docker-compose files, and use the pm2 and docker cli to get the project running. YOUR GOAL IS TO THE GET THE frontend APP IN THE PM2 CONFIG FILE TO RUN CORRECTLY! Other apps and docker containers are only supporting services. You can restart the pm2 environment with "npx -y staklink reload" after making changes to the pm2 config or docker-compose files.
 
 When you have gotten the frontend running, or gotten as far as you can, respond to the user by saying either "<status>COMPLETE</status>" or "<status>FAILED</status>". If COMPLETE, then tell the user the fixed config files (put the exact raw content of the files in the XML tags, no extra text like triple backticks or language label). FOR YOUR FINAL TEXT RESPONSE, USE THE THE FOLLOWING XML FORMAT:
 
@@ -139988,6 +139995,22 @@ var SCREENSHOT_DIR = import_path3.default.join(
   "tmp",
   "screenshots"
 );
+async function cleanupAllScreenshots() {
+  try {
+    const entries = await import_promises2.default.readdir(SCREENSHOT_DIR);
+    const pngs = entries.filter((e) => e.endsWith(".png"));
+    await Promise.all(
+      pngs.map(
+        (file3) => import_promises2.default.unlink(import_path3.default.join(SCREENSHOT_DIR, file3)).catch(() => {
+        })
+      )
+    );
+    if (pngs.length > 0) {
+      log(`Cleaned up ${pngs.length} leftover screenshot(s)`);
+    }
+  } catch {
+  }
+}
 async function collectScreenshots(since) {
   try {
     const entries = await import_promises2.default.readdir(SCREENSHOT_DIR);
@@ -140001,6 +140024,10 @@ async function collectScreenshots(since) {
       }
       const buf = await import_promises2.default.readFile(filePath);
       results.push({ filename: file3, base64: `data:image/png;base64,${buf.toString("base64")}` });
+      try {
+        await import_promises2.default.unlink(filePath);
+      } catch {
+      }
     }
     return results;
   } catch {
@@ -141083,6 +141110,97 @@ Stderr: ${error88.stderr}`
 };
 
 // src/proxy/git_actions.ts
+var IMAGE_EXTENSIONS = /* @__PURE__ */ new Set([
+  ".png",
+  ".jpg",
+  ".jpeg",
+  ".gif",
+  ".bmp",
+  ".webp",
+  ".ico",
+  ".svg",
+  ".tiff",
+  ".tif",
+  ".avif"
+]);
+var MIME_TYPES = {
+  ".png": "image/png",
+  ".jpg": "image/jpeg",
+  ".jpeg": "image/jpeg",
+  ".gif": "image/gif",
+  ".bmp": "image/bmp",
+  ".webp": "image/webp",
+  ".ico": "image/x-icon",
+  ".svg": "image/svg+xml",
+  ".tiff": "image/tiff",
+  ".tif": "image/tiff",
+  ".avif": "image/avif"
+};
+function isImageFile(filePath) {
+  return IMAGE_EXTENSIONS.has(import_path4.default.extname(filePath).toLowerCase());
+}
+async function binaryDiffContent(repo, filePath, action) {
+  const isBinary = await isGitBinaryFile(repo, filePath);
+  if (!isBinary) {
+    return null;
+  }
+  if (action === "delete") {
+    if (isImageFile(filePath)) {
+      return `Binary image file deleted: ${filePath}`;
+    }
+    return `Binary file deleted: ${filePath}`;
+  }
+  if (isImageFile(filePath)) {
+    try {
+      const absPath = import_path4.default.join(repo.printcwd(), filePath);
+      const buffer = import_fs4.default.readFileSync(absPath);
+      const ext2 = import_path4.default.extname(filePath).toLowerCase();
+      const mime = MIME_TYPES[ext2] || "application/octet-stream";
+      const b64 = buffer.toString("base64");
+      return `--- ${action === "create" ? "/dev/null" : `a/${filePath}`}
++++ b/${filePath}
+Binary image file (${mime}, ${formatBytes(buffer.length)})
+data:${mime};base64,${b64}`;
+    } catch {
+      return `Binary image file changed: ${filePath} (could not read file)`;
+    }
+  }
+  return `Binary file changed: ${filePath}`;
+}
+async function isGitBinaryFile(repo, filePath) {
+  try {
+    const numstat = await repo.execCommand(
+      `git diff --numstat HEAD -- "${filePath}"`
+    );
+    if (numstat.trim().startsWith("-	-")) {
+      return true;
+    }
+  } catch {
+  }
+  try {
+    const absPath = import_path4.default.join(repo.printcwd(), filePath);
+    const buf = Buffer.alloc(8192);
+    const fd = import_fs4.default.openSync(absPath, "r");
+    const bytesRead = import_fs4.default.readSync(fd, buf, 0, 8192, 0);
+    import_fs4.default.closeSync(fd);
+    for (let i = 0; i < bytesRead; i++) {
+      if (buf[i] === 0) {
+        return true;
+      }
+    }
+  } catch {
+  }
+  return false;
+}
+function formatBytes(bytes) {
+  if (bytes < 1024) {
+    return `${bytes} B`;
+  }
+  if (bytes < 1024 * 1024) {
+    return `${(bytes / 1024).toFixed(1)} KB`;
+  }
+  return `${(bytes / (1024 * 1024)).toFixed(1)} MB`;
+}
 async function handleBranchDiff(req, res) {
   try {
     const results = [];
@@ -141134,12 +141252,17 @@ async function handleBranchDiff(req, res) {
           action = "modify";
         }
         let content = "";
-        try {
-          content = await repo.execCommand(
-            `git diff ${mergeBase} -- "${filePath}"`
-          );
-        } catch (error88) {
-          warn(`Error getting branch diff for ${filePath}:`, error88);
+        const binaryContent = await binaryDiffContent(repo, filePath, action);
+        if (binaryContent !== null) {
+          content = binaryContent;
+        } else {
+          try {
+            content = await repo.execCommand(
+              `git diff ${mergeBase} -- "${filePath}"`
+            );
+          } catch (error88) {
+            warn(`Error getting branch diff for ${filePath}:`, error88);
+          }
         }
         results.push({
           file: `${repoName}/${filePath}`,
@@ -141190,15 +141313,24 @@ async function handleDiff(_req, res) {
           action = "modify";
         }
         let content = "";
-        try {
-          if (action === "delete") {
-            content = await repo.execCommand(`git diff HEAD -- "${filePath}"`);
-          } else {
-            log("Getting diff for", filePath);
-            content = await repo.execCommand(`git diff HEAD -- "${filePath}"`);
+        const binaryContent = await binaryDiffContent(repo, filePath, action);
+        if (binaryContent !== null) {
+          content = binaryContent;
+        } else {
+          try {
+            if (action === "delete") {
+              content = await repo.execCommand(
+                `git diff HEAD -- "${filePath}"`
+              );
+            } else {
+              log("Getting diff for", filePath);
+              content = await repo.execCommand(
+                `git diff HEAD -- "${filePath}"`
+              );
+            }
+          } catch (error88) {
+            warn(`Error getting diff for ${filePath}:`, error88);
           }
-        } catch (error88) {
-          warn(`Error getting diff for ${filePath}:`, error88);
         }
         results.push({
           file: `${repoName}/${filePath}`,
@@ -141211,15 +141343,24 @@ async function handleDiff(_req, res) {
       const untrackedLines = untrackedOutput.trim().split("\n").filter((line) => line);
       for (const filePath of untrackedLines) {
         let content = "";
-        try {
-          const fileContent = await repo.execCommand(`cat "${filePath}"`);
-          const lines2 = fileContent.split("\n");
-          content = `--- /dev/null
+        const binaryContent = await binaryDiffContent(
+          repo,
+          filePath,
+          "create"
+        );
+        if (binaryContent !== null) {
+          content = binaryContent;
+        } else {
+          try {
+            const fileContent = await repo.execCommand(`cat "${filePath}"`);
+            const lines2 = fileContent.split("\n");
+            content = `--- /dev/null
 +++ b/${filePath}
 @@ -0,0 +1,${lines2.length} @@
 ${lines2.map((line) => "+" + line).join("\n")}`;
-        } catch (error88) {
-          warn(`Error reading untracked file ${filePath}:`, error88);
+          } catch (error88) {
+            warn(`Error reading untracked file ${filePath}:`, error88);
+          }
         }
         results.push({
           file: `${repoName}/${filePath}`,
@@ -142495,6 +142636,9 @@ ${logs}
 
 // src/proxy/main.ts
 console.log("Starting staklink proxy server...");
+cleanupAllScreenshots().catch((err) => {
+  warn("Screenshot cleanup failed:", err);
+});
 startProxyServer().catch((err) => {
   error("Failed to start proxy server:", err);
   process.exit(1);

package/dist/staklink-cli.cjs

@@ -10967,7 +10967,7 @@ var glob = Object.assign(glob_, {
 glob.glob = glob;
 
 // src/proxy/version.ts
-var VERSION = "0.4.3";
+var VERSION = "0.4.5";
 
 // src/deps.ts
 var import_child_process = require("child_process");
@@ -10977,11 +10977,18 @@ var import_os = require("os");
 var import_path2 = require("path");
 var execAsync = (0, import_util2.promisify)(import_child_process.exec);
 var GOOSE_VERSION = "1.22.0";
+var SKILLS = [
+  { name: "agent-browser", repo: "https://github.com/vercel-labs/agent-browser" },
+  { name: "code-simplifier", repo: "https://github.com/getsentry/skills" },
+  { name: "frontend-design", repo: "https://github.com/anthropics/claude-code" }
+];
 async function ensureDepsInstalled() {
   await ensureGooseInstalled();
   await ensureAgentBrowserInstalled();
-  await ensureAgentBrowserSkillInstalled();
-  await migrateAgentBrowserSkill();
+  for (const skill of SKILLS) {
+    await ensureSkillInstalled(skill.name, skill.repo);
+    await migrateSkill(skill.name);
+  }
 }
 async function ensureGooseInstalled() {
   try {
@@ -11032,42 +11039,35 @@ async function installAgentBrowser() {
     console.error("\u26A0\uFE0F Failed to install agent-browser:", error2);
   }
 }
-var AGENT_BROWSER_SKILL_NAME = "agent-browser";
-var AGENT_BROWSER_SKILL_REPO = "https://github.com/vercel-labs/agent-browser";
-async function ensureAgentBrowserSkillInstalled() {
-  const skillPath = (0, import_path2.join)((0, import_os.homedir)(), ".config", "agents", "skills", AGENT_BROWSER_SKILL_NAME);
-  if ((0, import_fs3.existsSync)(skillPath)) {
-    console.log(`\u2705 Skill "${AGENT_BROWSER_SKILL_NAME}" is already installed`);
-    return;
-  }
-  console.log(`\u{1F4E6} Installing ${AGENT_BROWSER_SKILL_NAME} skill...`);
-  await installAgentBrowserSkill();
-}
-async function migrateAgentBrowserSkill() {
+async function migrateSkill(skillName) {
   const home = (0, import_os.homedir)();
-  const oldPath = (0, import_path2.join)(home, ".agents", "skills", "agent-browser");
+  const oldPath = (0, import_path2.join)(home, ".agents", "skills", skillName);
   const newDir = (0, import_path2.join)(home, ".config", "agents", "skills");
-  const newPath = (0, import_path2.join)(newDir, "agent-browser");
+  const newPath = (0, import_path2.join)(newDir, skillName);
   if (!(0, import_fs3.existsSync)(oldPath) || (0, import_fs3.existsSync)(newPath)) {
     return;
   }
   try {
-    console.log("\u{1F504} Migrating agent-browser skill to new location...");
+    console.log(`\u{1F504} Copying ${skillName} skill to new location...`);
     await execAsync(`mkdir -p "${newDir}"`);
-    await execAsync(`mv "${oldPath}" "${newPath}"`);
-    console.log("\u2705 agent-browser skill migrated successfully");
+    await execAsync(`cp -r "${oldPath}" "${newPath}"`);
+    console.log(`\u2705 ${skillName} skill copied to new location`);
   } catch (error2) {
-    console.error("\u26A0\uFE0F Failed to migrate agent-browser skill:", error2);
+    console.error(`\u26A0\uFE0F Failed to copy ${skillName} skill:`, error2);
   }
 }
-async function installAgentBrowserSkill() {
+async function ensureSkillInstalled(skillName, repo) {
+  const skillPath = (0, import_path2.join)((0, import_os.homedir)(), ".config", "agents", "skills", skillName);
+  if ((0, import_fs3.existsSync)(skillPath)) {
+    console.log(`\u2705 Skill "${skillName}" is already installed`);
+    return;
+  }
+  console.log(`\u{1F4E6} Installing ${skillName} skill...`);
   try {
-    await execAsync(
-      `npx skills add ${AGENT_BROWSER_SKILL_REPO} --skill ${AGENT_BROWSER_SKILL_NAME} -g -y`
-    );
-    console.log(`\u2705 Skill "${AGENT_BROWSER_SKILL_NAME}" installed successfully`);
+    await execAsync(`npx skills add ${repo} --skill ${skillName} -g -y`);
+    console.log(`\u2705 Skill "${skillName}" installed successfully`);
   } catch (error2) {
-    console.error(`\u26A0\uFE0F Failed to install ${AGENT_BROWSER_SKILL_NAME} skill:`, error2);
+    console.error(`\u26A0\uFE0F Failed to install ${skillName} skill:`, error2);
   }
 }
 

package/package.json

@@ -2,7 +2,7 @@
   "name": "staklink",
   "displayName": "staklink",
   "description": "staklink process manager",
-  "version": "0.4.3",
+  "version": "0.4.5",
   "type": "module",
   "publisher": "stakwork",
   "engines": {