wispy-cli 1.1.2 → 1.2.1

package/core/engine.mjs

@@ -23,6 +23,7 @@ import { MemoryManager } from "./memory.mjs";
 import { SubAgentManager } from "./subagents.mjs";
 import { PermissionManager } from "./permissions.mjs";
 import { AuditLog, EVENT_TYPES } from "./audit.mjs";
+import { Harness } from "./harness.mjs";
 
 const MAX_TOOL_ROUNDS = 10;
 const MAX_CONTEXT_CHARS = 40_000;
@@ -38,7 +39,11 @@ export class WispyEngine {
     this.subagents = new SubAgentManager(this, this.sessions);
     this.permissions = new PermissionManager(config.permissions ?? {});
     this.audit = new AuditLog(WISPY_DIR);
+    this.harness = new Harness(this.tools, this.permissions, this.audit, config);
     this._initialized = false;
+    this._workMdContent = null;
+    this._workMdLoaded = false;
+    this._workMdPath = null;
     this._activeWorkstream = config.workstream
       ?? process.env.WISPY_WORKSTREAM
       ?? process.argv.find((a, i) => (process.argv[i-1] === "-w" || process.argv[i-1] === "--workstream"))
@@ -67,9 +72,15 @@ export class WispyEngine {
     // Register memory tools
     this._registerMemoryTools();
 
+    // Register work.md tool
+    this._registerWorkContextTool();
+
     // Register node tools
     this._registerNodeTools();
 
+    // Re-wire harness after tools are registered
+    this.harness = new Harness(this.tools, this.permissions, this.audit, this.config);
+
     // Initialize MCP
     if (!opts.skipMcp) {
       await ensureDefaultMcpConfig(this.mcpManager.configPath);
@@ -289,68 +300,33 @@ export class WispyEngine {
   }
 
   /**
-   * Execute a tool, handling engine-level tools (spawn_agent, etc.) specially.
+   * Execute a tool, routing through the harness for permissions/audit/receipts.
+   * Engine-level tools that need conversation context are dispatched via executeToolFn.
    */
   async _executeTool(name, args, messages, session, opts) {
-    // ── Permission check ─────────────────────────────────────────────────────
-    const permResult = await this.permissions.check(name, args, { session });
-    const permLevel = permResult.level ?? "auto";
-
-    if (!permResult.allowed) {
-      const reason = permResult.reason ?? "Permission denied";
-      this.audit.log({
-        type: EVENT_TYPES.APPROVAL_DENIED,
-        sessionId: session?.id,
-        tool: name,
-        args,
-      }).catch(() => {});
-      return { success: false, error: reason, denied: true };
-    }
-
-    if (permResult.needsApproval && permResult.approved) {
-      this.audit.log({
-        type: EVENT_TYPES.APPROVAL_GRANTED,
-        sessionId: session?.id,
-        tool: name,
-        args,
-      }).catch(() => {});
-    }
-
-    // ── Audit: log tool call ──────────────────────────────────────────────────
-    const callStart = Date.now();
-    this.audit.log({
-      type: EVENT_TYPES.TOOL_CALL,
+    const engineTools = new Set([
+      "spawn_agent", "list_agents", "get_agent_result", "update_plan",
+      "pipeline", "spawn_async_agent", "ralph_loop",
+      "memory_save", "memory_search", "memory_list", "memory_get",
+      "memory_append", "memory_delete",
+      "spawn_subagent", "list_subagents", "get_subagent_result",
+      "kill_subagent", "steer_subagent",
+      "node_list", "node_status", "node_execute",
+      "update_work_context",
+    ]);
+
+    const harnessResult = await this.harness.execute(name, args, {
       sessionId: session?.id,
-      tool: name,
-      args,
-      permissionLevel: permLevel,
-    }).catch(() => {});
-
-    // ── Execute ───────────────────────────────────────────────────────────────
-    let result;
-    try {
-      result = await this._executeToolInner(name, args, messages, session, opts);
-    } catch (err) {
-      this.audit.log({
-        type: EVENT_TYPES.ERROR,
-        sessionId: session?.id,
-        tool: name,
-        message: err.message,
-        duration: Date.now() - callStart,
-      }).catch(() => {});
-      throw err;
-    }
-
-    // ── Audit: log tool result ────────────────────────────────────────────────
-    this.audit.log({
-      type: EVENT_TYPES.TOOL_RESULT,
-      sessionId: session?.id,
-      tool: name,
-      result: JSON.stringify(result).slice(0, 500),
-      duration: Date.now() - callStart,
-    }).catch(() => {});
+      channel: opts?.channel,
+      dryRun: opts?.dryRun,
+      // Engine-level tools use inner dispatch
+      executeToolFn: engineTools.has(name)
+        ? (n, a) => this._executeToolInner(n, a, messages, session, opts)
+        : null,
+    });
 
-    return result;
+    if (opts?.onReceipt) opts.onReceipt(harnessResult.receipt);
+    return harnessResult.result;
   }
 
   /**
@@ -365,6 +341,8 @@ export class WispyEngine {
         return this._toolListAgents();
       case "get_agent_result":
         return this._toolGetAgentResult(args);
+      case "update_work_context":
+        return this._toolUpdateWorkContext(args);
       case "update_plan":
         return this._toolUpdatePlan(args);
       case "pipeline":
@@ -759,6 +737,12 @@ export class WispyEngine {
       parts.push("## Project Context (WISPY.md)", wispyMd, "");
     }
 
+    // Load work.md context for active workstream
+    const workMd = await this._loadWorkMd();
+    if (workMd) {
+      parts.push("## Project context from work.md", workMd, "");
+    }
+
     // Load memories via MemoryManager
     try {
       const memories = await this.memory.getContextForPrompt(lastUserMessage);
@@ -791,6 +775,76 @@ export class WispyEngine {
     return null;
   }
 
+  async _loadWorkMd() {
+    if (this._workMdLoaded) return this._workMdContent;
+    this._workMdLoaded = true;
+
+    const ws = this._activeWorkstream;
+    const { default: osModule } = await import("node:os");
+    const homedir = osModule.homedir();
+    const candidates = [
+      path.resolve("work.md"),
+      path.resolve(ws, "work.md"),
+      path.join(homedir, ".wispy", "workstreams", ws, "work.md"),
+    ];
+
+    for (const p of candidates) {
+      try {
+        const content = await readFile(p, "utf8");
+        if (content?.trim()) {
+          this._workMdContent = content.trim().slice(0, 8000);
+          this._workMdPath = p;
+          return this._workMdContent;
+        }
+      } catch { /* not found */ }
+    }
+    this._workMdContent = null;
+    return null;
+  }
+
+  // ── work.md tool ─────────────────────────────────────────────────────────────
+
+  _registerWorkContextTool() {
+    this.tools._definitions.set("update_work_context", {
+      name: "update_work_context",
+      description: "Update the work.md context file for the current workstream with project notes, decisions, and context.",
+      parameters: {
+        type: "object",
+        properties: {
+          content: { type: "string", description: "New content for work.md" },
+          append: { type: "boolean", description: "If true, append instead of replace" },
+        },
+        required: ["content"],
+      },
+    });
+  }
+
+  async _toolUpdateWorkContext(args) {
+    const { mkdir: mkdirFn, writeFile: writeFn, appendFile: appendFn } = await import("node:fs/promises");
+    const { default: osModule } = await import("node:os");
+    const ws = this._activeWorkstream;
+
+    let targetPath = this._workMdPath;
+    if (!targetPath) {
+      targetPath = path.join(osModule.homedir(), ".wispy", "workstreams", ws, "work.md");
+    }
+
+    await mkdirFn(path.dirname(targetPath), { recursive: true });
+
+    if (args.append) {
+      await appendFn(targetPath, "\n" + args.content, "utf8");
+    } else {
+      await writeFn(targetPath, args.content, "utf8");
+    }
+
+    // Invalidate cache
+    this._workMdLoaded = false;
+    this._workMdContent = null;
+    this._workMdPath = targetPath;
+
+    return { success: true, message: `work.md updated at ${targetPath}`, path: targetPath };
+  }
+
   async _loadMemories() {
     const types = ["user", "feedback", "project", "references"];
     const sections = [];