@gleapai/kai-bridge 0.6.0 → 0.7.0

package/src/workspace.mjs

@@ -15,6 +15,7 @@
 import { execFileSync } from "node:child_process";
 import { copyFileSync, existsSync, mkdirSync, readFileSync, readdirSync, rmSync, statSync, writeFileSync } from "node:fs";
 import { dirname, join } from "node:path";
+import { createHash } from 'node:crypto';
 
 import { seedNodeModules } from "./deps.mjs";
 
@@ -23,13 +24,18 @@ function git(cwd, args, opts = {}) {
 }
 
 export function sessionSlug(sessionId, title) {
+  if (process.env.KAI_HOSTED === '1') {
+    if (!/^[a-zA-Z0-9_-]{1,128}$/.test(String(sessionId))) throw new Error('Invalid session identity.');
+    return `session-${sessionId}`;
+  }
   const t = String(title || "").toLowerCase().replace(/[^a-z0-9]+/g, "-").replace(/^-|-$/g, "").slice(0, 32);
   const id = String(sessionId || "").slice(-8);
   return t ? `${t}-${id}` : `session-${id}`;
 }
 
 export function worktreePath(kaiHome, repoName, slug) {
-  return join(kaiHome, "worktrees", repoName, slug);
+  const repo = process.env.KAI_HOSTED === '1' ? createHash('sha256').update(String(repoName)).digest('hex') : repoName;
+  return join(kaiHome, "worktrees", repo, slug);
 }
 
 /**
@@ -81,7 +87,7 @@ export function materializeBinding({ kaiHome, repo, binding, sessionId, title, b
     // Resume: the worktree from the previous turn is the session state.
     return { cwd: dir, mode, branch, base, resumed: true };
   }
-  mkdirSync(join(kaiHome, "worktrees", repo.name), { recursive: true });
+  mkdirSync(dirname(dir), { recursive: true });
   git(repo.primaryPath, ["fetch", "origin", base, "--quiet"]);
   git(repo.primaryPath, ["worktree", "add", "-b", branch, dir, `origin/${base}`]);
   // A fresh worktree has no node_modules; clone the primary checkout's
@@ -170,6 +176,12 @@ export function discardChanges(cwd) {
 export function removeWorktree({ kaiHome, repo, sessionId, title }) {
   const dir = worktreePath(kaiHome, repo.name, sessionSlug(sessionId, title));
   if (!existsSync(dir)) return false;
+  if (process.env.KAI_HOSTED === '1') {
+    // Never destroy dirty or unpushed work during session cleanup.
+    if (collectChanges(dir).files.length || git(dir, ['log', '--branches', '--not', '--remotes', '--oneline'])) return false;
+    git(repo.primaryPath, ['worktree', 'remove', dir]);
+    return true;
+  }
   try {
     git(repo.primaryPath, ["worktree", "remove", "--force", dir]);
   } catch {