@agentmeshhq/agent 0.1.7 → 0.1.8

package/src/core/injector.ts

@@ -1,11 +1,10 @@
-import { sendKeys } from "./tmux.js";
+import { formatHandoffContextSummary, parseHandoffContext } from "../context/handoff.js";
+import type { AgentContext } from "../context/schema.js";
 import type { InboxItem } from "./registry.js";
+import { sendKeys } from "./tmux.js";
 import type { WebSocketEvent } from "./websocket.js";
 
-export function injectStartupMessage(
-  agentName: string,
-  pendingCount: number
-): void {
+export function injectStartupMessage(agentName: string, pendingCount: number): void {
   if (pendingCount === 0) {
     const message = `[AgentMesh] Connected and ready. No pending items in inbox.`;
     sendKeys(agentName, message);
@@ -18,10 +17,7 @@ Use agentmesh_check_inbox to see them, or agentmesh_accept_handoff to start work
   sendKeys(agentName, message);
 }
 
-export function injectHandoffReceived(
-  agentName: string,
-  event: WebSocketEvent
-): void {
+export function injectHandoffReceived(agentName: string, event: WebSocketEvent): void {
   const fromName =
     (event.from_agent as { display_name?: string })?.display_name ||
     (event.from_agent_id as string) ||
@@ -41,14 +37,9 @@ Accept this handoff and begin work.`;
   sendKeys(agentName, message);
 }
 
-export function injectNudge(
-  agentName: string,
-  event: WebSocketEvent
-): void {
+export function injectNudge(agentName: string, event: WebSocketEvent): void {
   const fromName =
-    (event.from as { name?: string })?.name ||
-    (event.from_name as string) ||
-    "Someone";
+    (event.from as { name?: string })?.name || (event.from_name as string) || "Someone";
   const message = (event.message as string) || "Check your inbox";
 
   const formatted = `[AgentMesh] Nudge from ${fromName}:
@@ -57,12 +48,8 @@ ${message}`;
   sendKeys(agentName, formatted);
 }
 
-export function injectBlockerResolved(
-  agentName: string,
-  event: WebSocketEvent
-): void {
-  const description =
-    (event.description as string) || "A blocker has been resolved";
+export function injectBlockerResolved(agentName: string, event: WebSocketEvent): void {
+  const description = (event.description as string) || "A blocker has been resolved";
   const resolvedBy =
     (event.resolved_by as { display_name?: string })?.display_name ||
     (event.resolved_by_name as string) ||
@@ -78,10 +65,7 @@ You can now proceed with your work.`;
   sendKeys(agentName, message);
 }
 
-export function injectInboxItems(
-  agentName: string,
-  items: InboxItem[]
-): void {
+export function injectInboxItems(agentName: string, items: InboxItem[]): void {
   if (items.length === 0) {
     sendKeys(agentName, "[AgentMesh] Your inbox is empty.");
     return;
@@ -101,10 +85,7 @@ export function injectInboxItems(
   sendKeys(agentName, message);
 }
 
-export function handleWebSocketEvent(
-  agentName: string,
-  event: WebSocketEvent
-): void {
+export function handleWebSocketEvent(agentName: string, event: WebSocketEvent): void {
   switch (event.type) {
     case "handoff_received":
     case "handoff.received":
@@ -126,3 +107,66 @@ export function handleWebSocketEvent(
       break;
   }
 }
+
+/**
+ * Injects restored context from a previous session
+ */
+export function injectRestoredContext(agentName: string, context: AgentContext): void {
+  const parts: string[] = ["[AgentMesh] Restored context from previous session:"];
+
+  // Working state
+  if (context.workingState.workdir) {
+    parts.push(`Working directory: ${context.workingState.workdir}`);
+  }
+  if (context.workingState.gitBranch) {
+    parts.push(`Git branch: ${context.workingState.gitBranch}`);
+  }
+
+  // Tasks
+  const activeTasks = context.tasks.tasks.filter(
+    (t) => t.status === "in_progress" || t.status === "pending",
+  );
+  if (activeTasks.length > 0) {
+    parts.push("");
+    parts.push("Active tasks:");
+    for (const task of activeTasks.slice(0, 5)) {
+      const statusIcon = task.status === "in_progress" ? ">" : "-";
+      parts.push(`  ${statusIcon} ${task.content}`);
+    }
+    if (activeTasks.length > 5) {
+      parts.push(`  ... and ${activeTasks.length - 5} more`);
+    }
+  }
+
+  // Current goal
+  if (context.tasks.currentGoal) {
+    parts.push("");
+    parts.push(`Goal: ${context.tasks.currentGoal}`);
+  }
+
+  // Accomplishments
+  if (context.conversation.accomplishments.length > 0) {
+    parts.push("");
+    parts.push("Recent accomplishments:");
+    for (const acc of context.conversation.accomplishments.slice(0, 3)) {
+      parts.push(`  - ${acc}`);
+    }
+  }
+
+  const message = parts.join("\n");
+  sendKeys(agentName, message);
+}
+
+/**
+ * Injects context received from a handoff
+ */
+export function injectHandoffContext(agentName: string, contextString: string): void {
+  const context = parseHandoffContext(contextString);
+  if (!context) {
+    return;
+  }
+
+  const summary = formatHandoffContextSummary(context);
+  const message = `[AgentMesh] Context from previous agent:\n\n${summary}`;
+  sendKeys(agentName, message);
+}

package/src/core/tmux.ts

@@ -176,3 +176,78 @@ export function getSessionInfo(agentName: string): { exists: boolean; command?:
     return { exists: true };
   }
 }
+
+export interface SessionContext {
+  workdir: string;
+  gitBranch?: string;
+  gitStatus?: string;
+}
+
+/**
+ * Captures current working state from the tmux session
+ */
+export function captureSessionContext(agentName: string): SessionContext | null {
+  const sessionName = getSessionName(agentName);
+
+  if (!sessionExists(sessionName)) {
+    return null;
+  }
+
+  try {
+    // Get current working directory from the pane
+    const workdir = execSync(`tmux display-message -t "${sessionName}" -p "#{pane_current_path}"`, {
+      encoding: "utf-8",
+    }).trim();
+
+    const context: SessionContext = { workdir };
+
+    // Try to get git info if in a git repo
+    try {
+      const gitBranch = execSync("git rev-parse --abbrev-ref HEAD 2>/dev/null", {
+        encoding: "utf-8",
+        cwd: workdir,
+      }).trim();
+      if (gitBranch) {
+        context.gitBranch = gitBranch;
+      }
+
+      const gitStatus = execSync("git status --short 2>/dev/null", {
+        encoding: "utf-8",
+        cwd: workdir,
+      }).trim();
+      if (gitStatus) {
+        // Truncate if too long
+        context.gitStatus =
+          gitStatus.length > 500 ? `${gitStatus.substring(0, 500)}...` : gitStatus;
+      }
+    } catch {
+      // Not a git repo or git not available
+    }
+
+    return context;
+  } catch (error) {
+    console.error(`Failed to capture session context: ${error}`);
+    return null;
+  }
+}
+
+/**
+ * Captures recent output from the tmux pane (last N lines)
+ */
+export function captureSessionOutput(agentName: string, lines = 100): string | null {
+  const sessionName = getSessionName(agentName);
+
+  if (!sessionExists(sessionName)) {
+    return null;
+  }
+
+  try {
+    const output = execSync(`tmux capture-pane -t "${sessionName}" -p -S -${lines}`, {
+      encoding: "utf-8",
+    });
+    return output;
+  } catch (error) {
+    console.error(`Failed to capture session output: ${error}`);
+    return null;
+  }
+}