@zonease/happy 0.6.40 → 0.6.42

package/dist/setupOfflineReconnection-B6coH9QN.cjs

@@ -0,0 +1,461 @@
+'use strict';
+
+var logger = require('./persistence-BcQmjobZ.cjs');
+var fs = require('node:fs');
+var os = require('node:os');
+var path = require('node:path');
+var projectPath = require('./projectPath-ChLhdOyt.cjs');
+var api = require('./api-CIE4_cWd.cjs');
+var gitWorktree = require('./gitWorktree-C6jLoaMw.cjs');
+
+class BasePermissionHandler {
+  pendingRequests = /* @__PURE__ */ new Map();
+  session;
+  pushClient;
+  isResetting = false;
+  isUserInteractionTool(toolName) {
+    return ["AskUserQuestion", "ExitPlanMode", "exit_plan_mode"].includes(toolName);
+  }
+  constructor(session, pushClient) {
+    this.session = session;
+    this.pushClient = pushClient;
+    this.setupRpcHandler();
+  }
+  /**
+   * Update the session reference (used after offline reconnection swaps sessions).
+   * This is critical for avoiding stale session references after onSessionSwap.
+   */
+  updateSession(newSession) {
+    logger.logger.debug(`${this.getLogPrefix()} Session reference updated`);
+    this.session = newSession;
+    this.setupRpcHandler();
+  }
+  /**
+   * Setup RPC handler for permission responses.
+   */
+  setupRpcHandler() {
+    this.session.rpcHandlerManager.registerHandler(
+      "permission",
+      async (response) => {
+        const pending = this.pendingRequests.get(response.id);
+        if (!pending) {
+          logger.logger.debug(`${this.getLogPrefix()} Permission request not found or already resolved`);
+          return;
+        }
+        this.pendingRequests.delete(response.id);
+        const result = response.approved ? { decision: response.decision === "approved_for_session" ? "approved_for_session" : "approved" } : { decision: response.decision === "denied" ? "denied" : "abort" };
+        pending.resolve(result);
+        if (response.approved && (response.mode === "bypassPermissions" || response.decision === "approved_for_session")) {
+          this.approvePendingRequestsForSession(response);
+        }
+        this.session.updateAgentState((currentState) => {
+          const request = currentState.requests?.[response.id];
+          if (!request) return currentState;
+          const { [response.id]: _, ...remainingRequests } = currentState.requests || {};
+          let res = {
+            ...currentState,
+            requests: remainingRequests,
+            completedRequests: {
+              ...currentState.completedRequests,
+              [response.id]: {
+                ...request,
+                completedAt: Date.now(),
+                status: response.approved ? "approved" : "denied",
+                decision: result.decision,
+                mode: response.mode,
+                allowedTools: response.allowTools
+              }
+            }
+          };
+          return res;
+        });
+        logger.logger.debug(`${this.getLogPrefix()} Permission ${response.approved ? "approved" : "denied"} for ${pending.toolName}`);
+      }
+    );
+  }
+  approvePendingRequestsForSession(response) {
+    const pendingSnapshot = Array.from(this.pendingRequests.entries());
+    const completedAt = Date.now();
+    for (const [id, pending] of pendingSnapshot) {
+      if (this.isUserInteractionTool(pending.toolName)) {
+        continue;
+      }
+      this.pendingRequests.delete(id);
+      pending.resolve({ decision: "approved_for_session" });
+    }
+    if (pendingSnapshot.length === 0) {
+      return;
+    }
+    this.session.updateAgentState((currentState) => {
+      const requests = { ...currentState.requests || {} };
+      const completedRequests = { ...currentState.completedRequests };
+      let changed = false;
+      for (const [id, pending] of pendingSnapshot) {
+        if (this.isUserInteractionTool(pending.toolName)) {
+          continue;
+        }
+        const request = requests[id];
+        if (!request) {
+          continue;
+        }
+        delete requests[id];
+        completedRequests[id] = {
+          ...request,
+          completedAt,
+          status: "approved",
+          decision: "approved_for_session",
+          mode: response.mode,
+          allowedTools: response.allowTools
+        };
+        changed = true;
+      }
+      if (!changed) {
+        return currentState;
+      }
+      return {
+        ...currentState,
+        requests,
+        completedRequests
+      };
+    });
+  }
+  /**
+   * Add a pending request to the agent state and send a push notification.
+   */
+  addPendingRequestToState(toolCallId, toolName, input) {
+    this.pushClient.sendToAllDevices(
+      "Permission Request",
+      `${this.getAgentName()} wants to ${toolName}`,
+      {
+        sessionId: this.session.sessionId,
+        requestId: toolCallId,
+        tool: toolName,
+        type: "permission_request"
+      }
+    );
+    this.session.updateAgentState((currentState) => ({
+      ...currentState,
+      requests: {
+        ...currentState.requests,
+        [toolCallId]: {
+          tool: toolName,
+          arguments: input,
+          createdAt: Date.now()
+        }
+      }
+    }));
+  }
+  /**
+   * Reset state for new sessions.
+   * This method is idempotent - safe to call multiple times.
+   */
+  reset() {
+    if (this.isResetting) {
+      logger.logger.debug(`${this.getLogPrefix()} Reset already in progress, skipping`);
+      return;
+    }
+    this.isResetting = true;
+    try {
+      const pendingSnapshot = Array.from(this.pendingRequests.entries());
+      this.pendingRequests.clear();
+      for (const [id, pending] of pendingSnapshot) {
+        try {
+          pending.reject(new Error("Session reset"));
+        } catch (err) {
+          logger.logger.debug(`${this.getLogPrefix()} Error rejecting pending request ${id}:`, err);
+        }
+      }
+      this.session.updateAgentState((currentState) => {
+        const pendingRequests = currentState.requests || {};
+        const completedRequests = { ...currentState.completedRequests };
+        for (const [id, request] of Object.entries(pendingRequests)) {
+          completedRequests[id] = {
+            ...request,
+            completedAt: Date.now(),
+            status: "canceled",
+            reason: "Session reset"
+          };
+        }
+        return {
+          ...currentState,
+          requests: {},
+          completedRequests
+        };
+      });
+      logger.logger.debug(`${this.getLogPrefix()} Permission handler reset`);
+    } finally {
+      this.isResetting = false;
+    }
+  }
+}
+
+const DEFAULT_MAX_SKILLS = 400;
+const DEFAULT_MAX_DEPTH = 10;
+const SKILL_FILE = "SKILL.md";
+const IGNORED_DIRECTORIES = /* @__PURE__ */ new Set([
+  ".git",
+  "node_modules",
+  "dist",
+  "build",
+  "references",
+  "assets",
+  "templates"
+]);
+function parseFrontMatter(content) {
+  if (!content.startsWith("---")) return {};
+  const end = content.indexOf("\n---", 3);
+  if (end < 0) return {};
+  const block = content.slice(3, end).split(/\r?\n/);
+  const result = {};
+  for (let i = 0; i < block.length; i++) {
+    const line = block[i];
+    const match = /^([A-Za-z0-9_-]+):\s*(.*)$/.exec(line.trim());
+    if (!match) continue;
+    const [, key, rawValue] = match;
+    let value = rawValue.trim().replace(/^['"]|['"]$/g, "");
+    if ((rawValue.trim() === ">-" || rawValue.trim() === ">" || rawValue.trim() === "|-" || rawValue.trim() === "|") && key === "description") {
+      const continuation = [];
+      while (i + 1 < block.length && /^\s+/.test(block[i + 1])) {
+        i++;
+        continuation.push(block[i].trim());
+      }
+      value = continuation.join(rawValue.trim().startsWith("|") ? "\n" : " ").trim();
+    }
+    if (key === "name" && value) result.name = value;
+    if (key === "description" && value) result.description = value;
+  }
+  return result;
+}
+function deriveSkillName(skillFilePath) {
+  return path.basename(path.dirname(skillFilePath));
+}
+function collectSkillFiles(root, maxDepth, maxFiles) {
+  const files = [];
+  function walk(dir, depth) {
+    if (files.length >= maxFiles || depth > maxDepth) return;
+    let entries;
+    try {
+      entries = fs.readdirSync(dir, { withFileTypes: true });
+    } catch {
+      return;
+    }
+    for (const entry of entries) {
+      if (files.length >= maxFiles) return;
+      const fullPath = path.join(dir, entry.name);
+      if (entry.isFile() && entry.name === SKILL_FILE) {
+        files.push(fullPath);
+        continue;
+      }
+      if (!entry.isDirectory()) continue;
+      if (IGNORED_DIRECTORIES.has(entry.name)) continue;
+      walk(fullPath, depth + 1);
+    }
+  }
+  if (fs.existsSync(root)) {
+    walk(root, 0);
+  }
+  return files;
+}
+function skillRoots(homeDir, includePluginCache) {
+  const roots = [
+    { path: path.join(homeDir, ".agents", "skills"), source: "agents" },
+    { path: path.join(homeDir, ".codex", "skills"), source: "codex" },
+    { path: path.join(homeDir, ".claude", "skills"), source: "claude" }
+  ];
+  if (includePluginCache) {
+    roots.push(
+      { path: path.join(homeDir, ".codex", "plugins", "cache"), source: "codex-plugin-cache" },
+      { path: path.join(homeDir, ".codex", ".tmp", "plugins", "plugins"), source: "codex-plugin-marketplace" }
+    );
+  }
+  return roots;
+}
+function discoverCodexSkills(options = {}) {
+  const homeDir = options.homeDir ?? os.homedir();
+  const maxSkills = options.maxSkills ?? DEFAULT_MAX_SKILLS;
+  const maxDepth = options.maxDepth ?? DEFAULT_MAX_DEPTH;
+  const includePluginCache = options.includePluginCache ?? true;
+  const result = [];
+  const seen = /* @__PURE__ */ new Set();
+  for (const root of skillRoots(homeDir, includePluginCache)) {
+    if (result.length >= maxSkills) break;
+    const skillFiles = collectSkillFiles(root.path, maxDepth, maxSkills - result.length);
+    for (const skillFile of skillFiles) {
+      if (result.length >= maxSkills) break;
+      let content = "";
+      try {
+        content = fs.readFileSync(skillFile, "utf8");
+      } catch {
+        continue;
+      }
+      const frontMatter = parseFrontMatter(content);
+      const name = (frontMatter.name || deriveSkillName(skillFile)).trim();
+      if (!name) continue;
+      const key = name.toLowerCase();
+      if (seen.has(key)) continue;
+      seen.add(key);
+      result.push({
+        name,
+        description: frontMatter.description,
+        source: root.source
+      });
+    }
+  }
+  return result;
+}
+
+function detectWorkspaceSubdirectories(cwd) {
+  try {
+    const entries = fs.readdirSync(cwd, { withFileTypes: true });
+    const repos = [];
+    for (const entry of entries) {
+      if (!entry.isDirectory() || entry.name.startsWith(".")) continue;
+      const subDir = path.join(cwd, entry.name);
+      const info = gitWorktree.detectGitWorktree(subDir);
+      if (info.isWorktree && info.worktreeBasePath && info.worktreeBranchName) {
+        repos.push({
+          path: subDir,
+          basePath: info.worktreeBasePath,
+          branchName: info.worktreeBranchName,
+          displayName: entry.name
+        });
+      }
+    }
+    if (repos.length > 0) {
+      return {
+        isWorktree: true,
+        workspaceRepos: repos,
+        workspacePath: cwd
+      };
+    }
+  } catch {
+  }
+  return {};
+}
+function detectWorktreeMetadata() {
+  if (process.env.HAPPY_WORKSPACE_REPOS) {
+    try {
+      const repos = JSON.parse(process.env.HAPPY_WORKSPACE_REPOS);
+      return {
+        isWorktree: true,
+        workspaceRepos: repos,
+        workspacePath: process.env.HAPPY_WORKSPACE_PATH
+      };
+    } catch {
+    }
+  }
+  if (process.env.HAPPY_WORKTREE_BASE_PATH) {
+    return {
+      isWorktree: true,
+      worktreeBasePath: process.env.HAPPY_WORKTREE_BASE_PATH,
+      worktreeBranchName: process.env.HAPPY_WORKTREE_BRANCH_NAME
+    };
+  }
+  const info = gitWorktree.detectGitWorktree(process.cwd());
+  if (info.isWorktree) {
+    return {
+      isWorktree: true,
+      worktreeBasePath: info.worktreeBasePath,
+      worktreeBranchName: info.worktreeBranchName
+    };
+  }
+  const workspaceResult = detectWorkspaceSubdirectories(process.cwd());
+  if (workspaceResult.isWorktree) {
+    return workspaceResult;
+  }
+  return {};
+}
+function createSessionMetadata(opts) {
+  const state = {
+    controlledByUser: false
+  };
+  const metadata = {
+    path: process.cwd(),
+    host: os.hostname(),
+    version: logger.packageJson.version,
+    os: os.platform(),
+    machineId: opts.machineId,
+    homeDir: os.homedir(),
+    happyHomeDir: logger.configuration.happyHomeDir,
+    happyLibDir: projectPath.projectPath(),
+    happyToolsDir: api.resolveToolsDir(),
+    startedFromDaemon: opts.startedBy === "daemon",
+    hostPid: process.pid,
+    startedBy: opts.startedBy || "terminal",
+    lifecycleState: "running",
+    lifecycleStateSince: Date.now(),
+    flavor: opts.flavor,
+    // Worktree metadata: env vars from daemon take priority, otherwise detect via git
+    ...detectWorktreeMetadata()
+  };
+  if (opts.flavor === "codex") {
+    metadata.skills = discoverCodexSkills();
+  }
+  return { state, metadata };
+}
+
+function createOfflineSessionStub(sessionTag) {
+  return {
+    sessionId: `offline-${sessionTag}`,
+    sendCodexMessage: () => {
+    },
+    sendAgentMessage: () => {
+    },
+    sendClaudeSessionMessage: () => {
+    },
+    keepAlive: () => {
+    },
+    sendSessionEvent: () => {
+    },
+    sendSessionDeath: () => {
+    },
+    updateLifecycleState: () => {
+    },
+    requestControlTransfer: async () => {
+    },
+    flush: async () => {
+    },
+    close: async () => {
+    },
+    updateMetadata: () => {
+    },
+    updateAgentState: () => {
+    },
+    onUserMessage: () => {
+    },
+    rpcHandlerManager: {
+      registerHandler: () => {
+      }
+    }
+  };
+}
+
+function setupOfflineReconnection(opts) {
+  const { api: api$1, sessionTag, metadata, state, response, onSessionSwap } = opts;
+  let session;
+  let reconnectionHandle = null;
+  if (!response) {
+    session = createOfflineSessionStub(sessionTag);
+    reconnectionHandle = api.startOfflineReconnection({
+      serverUrl: logger.configuration.serverUrl,
+      onReconnected: async () => {
+        const resp = await api$1.getOrCreateSession({ tag: sessionTag, metadata, state });
+        if (!resp) throw new Error("Server unavailable");
+        const realSession = api$1.sessionSyncClient(resp);
+        onSessionSwap(realSession);
+        return realSession;
+      },
+      onNotify: (msg) => {
+        console.log(msg);
+      }
+    });
+    return { session, reconnectionHandle, isOffline: true };
+  } else {
+    session = api$1.sessionSyncClient(response);
+    return { session, reconnectionHandle: null, isOffline: false };
+  }
+}
+
+exports.BasePermissionHandler = BasePermissionHandler;
+exports.createSessionMetadata = createSessionMetadata;
+exports.setupOfflineReconnection = setupOfflineReconnection;