staklink 0.4.5 → 0.4.7

package/dist/proxy-server.cjs

@@ -60411,6 +60411,23 @@ var Repo = class {
     log("nowBranch:", nowBranch);
     return nowBranch.includes(`* ${base_branch}`);
   }
+  async getDefaultBranch() {
+    try {
+      const result = await this.execCommand(
+        `git remote show origin | sed -n '/HEAD branch/s/.*: //p'`
+      );
+      return result.trim();
+    } catch {
+      try {
+        await this.execCommand(
+          "git show-ref --verify --quiet refs/heads/main"
+        );
+        return "main";
+      } catch {
+        return "master";
+      }
+    }
+  }
   async checkoutMainBranchOrBaseBranch(base_branch) {
     if (base_branch) {
       return this.execCommand(`git checkout -f ${base_branch}`);
@@ -60777,7 +60794,7 @@ var SSEManager = class {
 var sseManager = new SSEManager();
 
 // src/proxy/version.ts
-var VERSION = "0.4.5";
+var VERSION = "0.4.7";
 
 // node_modules/uuid/dist/esm/stringify.js
 var byteToHex = [];
@@ -139995,6 +140012,27 @@ var SCREENSHOT_DIR = import_path3.default.join(
   "tmp",
   "screenshots"
 );
+var RECORDING_DIR = import_path3.default.join(
+  import_os2.default.homedir(),
+  ".agent-browser",
+  "tmp",
+  "recordings"
+);
+async function cleanupAllRecordings() {
+  try {
+    const entries = await import_promises2.default.readdir(RECORDING_DIR);
+    await Promise.all(
+      entries.map(
+        (file3) => import_promises2.default.unlink(import_path3.default.join(RECORDING_DIR, file3)).catch(() => {
+        })
+      )
+    );
+    if (entries.length > 0) {
+      log(`Cleaned up ${entries.length} leftover recording(s)`);
+    }
+  } catch {
+  }
+}
 async function cleanupAllScreenshots() {
   try {
     const entries = await import_promises2.default.readdir(SCREENSHOT_DIR);
@@ -140011,6 +140049,27 @@ async function cleanupAllScreenshots() {
   } catch {
   }
 }
+async function collectRecordings(since) {
+  try {
+    const entries = await import_promises2.default.readdir(RECORDING_DIR);
+    const webms = entries.filter((e) => e.endsWith(".webm"));
+    const results = [];
+    for (const file3 of webms) {
+      const filePath = import_path3.default.join(RECORDING_DIR, file3);
+      const stat4 = await import_promises2.default.stat(filePath);
+      if (stat4.mtimeMs < since) {
+        continue;
+      }
+      results.push({
+        filename: file3,
+        url: `recordings/${file3}`
+      });
+    }
+    return results;
+  } catch {
+    return [];
+  }
+}
 async function collectScreenshots(since) {
   try {
     const entries = await import_promises2.default.readdir(SCREENSHOT_DIR);
@@ -140068,10 +140127,12 @@ var createAsyncAgentHandler = (getParams, transformResult) => {
       }).then(async (result) => {
         const finalResult = transformResult ? transformResult(result) : result;
         const screenshots = await collectScreenshots(startTime);
+        const recordings = await collectRecordings(startTime);
         finishReq(request_id, {
           success: true,
           result: finalResult,
-          ...screenshots.length > 0 && { screenshots }
+          ...screenshots.length > 0 && { screenshots },
+          ...recordings.length > 0 && { recordings }
         });
       }).catch((error88) => {
         error("Agent error:", error88);
@@ -141205,37 +141266,42 @@ async function handleBranchDiff(req, res) {
   try {
     const results = [];
     const repos = await getReposMaybe();
-    const baseBranch = req.query.base || "main";
     for (const r of repos) {
       const repo = await NewRepo(r);
       const repoName = getRepoNameFromUrl(r);
       const currentBranch = (await repo.printCurrentBranch()).trim();
-      if (currentBranch === baseBranch) {
-        continue;
-      }
-      let mergeBase;
-      try {
-        mergeBase = (await repo.execCommand(`git merge-base ${baseBranch} HEAD`)).trim();
-      } catch {
+      const baseBranch = req.query.base || await repo.getDefaultBranch();
+      const onBaseBranch = currentBranch === baseBranch;
+      let diffBase;
+      if (onBaseBranch) {
+        diffBase = "HEAD";
+      } else {
         try {
-          mergeBase = (await repo.execCommand(`git merge-base origin/${baseBranch} HEAD`)).trim();
+          diffBase = (await repo.execCommand(`git merge-base ${baseBranch} HEAD`)).trim();
         } catch {
-          warn(
-            `Could not find merge base for ${repoName}, skipping...`
-          );
-          continue;
+          try {
+            diffBase = (await repo.execCommand(
+              `git merge-base origin/${baseBranch} HEAD`
+            )).trim();
+          } catch {
+            warn(
+              `Could not find merge base for ${repoName}, skipping...`
+            );
+            continue;
+          }
         }
       }
       let diffOutput = "";
       try {
         diffOutput = await repo.execCommand(
-          `git diff --name-status ${mergeBase}`
+          `git diff --name-status ${diffBase}`
         );
       } catch (error88) {
         warn(`Error getting branch diff for repo ${repoName}:`, error88);
         continue;
       }
       const lines = diffOutput.trim().split("\n").filter((line) => line);
+      const trackedFiles = /* @__PURE__ */ new Set();
       for (const line of lines) {
         const parts = line.split("	");
         if (parts.length < 2) {
@@ -141243,6 +141309,7 @@ async function handleBranchDiff(req, res) {
         }
         const status = parts[0];
         const filePath = parts[1];
+        trackedFiles.add(filePath);
         let action;
         if (status.startsWith("A")) {
           action = "create";
@@ -141258,7 +141325,7 @@ async function handleBranchDiff(req, res) {
         } else {
           try {
             content = await repo.execCommand(
-              `git diff ${mergeBase} -- "${filePath}"`
+              `git diff ${diffBase} -- "${filePath}"`
             );
           } catch (error88) {
             warn(`Error getting branch diff for ${filePath}:`, error88);
@@ -141272,6 +141339,47 @@ async function handleBranchDiff(req, res) {
           errors: []
         });
       }
+      let untrackedOutput = "";
+      try {
+        untrackedOutput = await repo.execCommand(
+          "git ls-files --others --exclude-standard"
+        );
+      } catch (error88) {
+        warn(
+          `Error getting untracked files for repo ${repoName}:`,
+          error88
+        );
+      }
+      const untrackedLines = untrackedOutput.trim().split("\n").filter((line) => line && !trackedFiles.has(line));
+      for (const filePath of untrackedLines) {
+        let content = "";
+        const binaryContent = await binaryDiffContent(
+          repo,
+          filePath,
+          "create"
+        );
+        if (binaryContent !== null) {
+          content = binaryContent;
+        } else {
+          try {
+            const fileContent = await repo.execCommand(`cat "${filePath}"`);
+            const fileLines = fileContent.split("\n");
+            content = `--- /dev/null
++++ b/${filePath}
+@@ -0,0 +1,${fileLines.length} @@
+${fileLines.map((line) => "+" + line).join("\n")}`;
+          } catch (error88) {
+            warn(`Error reading untracked file ${filePath}:`, error88);
+          }
+        }
+        results.push({
+          file: `${repoName}/${filePath}`,
+          action: "create",
+          content,
+          repoName,
+          errors: []
+        });
+      }
     }
     res.status(200).json(results);
   } catch (error88) {
@@ -141886,6 +141994,7 @@ async function startProxyServer() {
   });
   app.use(import_express.default.json({ limit: "50mb" }));
   app.use(import_express.default.urlencoded({ extended: true }));
+  app.use("/recordings", import_express.default.static(RECORDING_DIR));
   app.get("/health", (req, res) => {
     log("===> health check");
     res.status(200).json({
@@ -142639,6 +142748,9 @@ console.log("Starting staklink proxy server...");
 cleanupAllScreenshots().catch((err) => {
   warn("Screenshot cleanup failed:", err);
 });
+cleanupAllRecordings().catch((err) => {
+  warn("Recording 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.5";
+var VERSION = "0.4.7";
 
 // src/deps.ts
 var import_child_process = require("child_process");

package/package.json

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