@agentlogs/pi 0.0.1

package/README.md

@@ -0,0 +1,147 @@
+# @agentlogs/pi
+
+AgentLogs extension for [pi](https://github.com/badlogic/pi-mono) - automatically captures and uploads AI coding session transcripts.
+
+## Installation
+
+```bash
+npm install -g @agentlogs/pi
+# or
+bun add -g @agentlogs/pi
+```
+
+## Setup
+
+Add the extension to your pi configuration:
+
+**Option 1: Global settings** (`~/.pi/agent/settings.json`)
+
+```json
+{
+  "extensions": ["@agentlogs/pi"]
+}
+```
+
+**Option 2: Project settings** (`.pi/settings.json`)
+
+```json
+{
+  "extensions": ["@agentlogs/pi"]
+}
+```
+
+**Option 3: Package.json**
+
+```json
+{
+  "pi": {
+    "extensions": ["@agentlogs/pi"]
+  }
+}
+```
+
+## Features
+
+### Automatic Transcript Upload
+
+When you end a pi session (Ctrl+D), your conversation transcript is automatically uploaded to AgentLogs.
+
+### Git Commit Tracking
+
+When the AI makes git commits, the extension:
+
+1. Adds a link to the transcript in the commit message footer
+2. Tracks which commits are associated with which transcripts
+3. Makes it easy to find the AI conversation that led to any commit
+
+### Branch-Aware Transcripts
+
+Pi supports conversation branching via `/tree`. The extension handles this by:
+
+- Generating unique transcript IDs for each branch
+- Only uploading the current branch (from leaf to root)
+- Preserving links to older branches when you navigate away
+
+## Configuration
+
+### Environment Variables
+
+- `AGENTLOGS_CLI_PATH` - Custom path to the agentlogs CLI (defaults to `npx -y agentlogs@latest`)
+
+### Repository Allowlist
+
+By default, transcripts are only uploaded for repositories you've explicitly allowed:
+
+```bash
+# Allow the current repo
+agentlogs allow
+
+# Set visibility
+agentlogs allow --public
+agentlogs allow --team
+agentlogs allow --private
+
+# Deny the current repo
+agentlogs deny
+```
+
+## Development Setup
+
+For local development, use the setup script:
+
+```bash
+# From the repo root
+./packages/pi/scripts/dev-setup.sh
+
+# This creates a symlink and shows the CLI path to export:
+export AGENTLOGS_CLI_PATH="bun /path/to/agentlogs/packages/cli/src/index.ts"
+
+# Now run pi - the extension loads automatically
+pi
+```
+
+To remove the dev setup:
+
+```bash
+./packages/pi/scripts/dev-teardown.sh
+unset AGENTLOGS_CLI_PATH
+```
+
+## Debug Logging
+
+Debug logs are written to `/tmp/agentlogs-pi.log` when not in production mode.
+
+```bash
+# Watch logs in real-time
+tail -f /tmp/agentlogs-pi.log
+```
+
+## CLI Commands
+
+The extension uses the `agentlogs` CLI under the hood:
+
+```bash
+# List recent sessions
+agentlogs pi upload
+
+# Upload a specific session
+agentlogs pi upload <session-id>
+agentlogs pi upload /path/to/session.jsonl
+
+# Check login status
+agentlogs status
+
+# Login to AgentLogs
+agentlogs login
+```
+
+## How It Works
+
+1. The extension registers handlers for pi's lifecycle events
+2. On `session_shutdown`, it shells out to `agentlogs pi hook` with session data
+3. The CLI converts the pi session format to AgentLogs' unified format
+4. The transcript is uploaded to the AgentLogs API
+
+## License
+
+MIT

package/dist/index.js

@@ -0,0 +1 @@
+import{appendFileSync as V}from"node:fs";import{spawn as W}from"node:child_process";var X="/tmp/agentlogs-pi.log";function D(J,M){return}async function R(J,M){let Q=process.env.AGENTLOGS_CLI_PATH,b,j;if(Q){let q=Q.split(" ");b=q[0],j=[...q.slice(1),"pi","hook"]}else b="npx",j=["-y","agentlogs@latest","pi","hook"];return D("Running hook",{command:b,args:j,payload:J.hook_event_name}),new Promise((q)=>{let z=W(b,j,{cwd:M,stdio:["pipe","pipe","pipe"]}),B="",K="";z.stdout.on("data",(A)=>{B+=A.toString()}),z.stderr.on("data",(A)=>{K+=A.toString()}),z.stdin.write(JSON.stringify(J)),z.stdin.end(),z.on("close",(A)=>{if(D("Hook process exited",{code:A,stdout:B.slice(0,500),stderr:K.slice(0,500)}),A===0&&B.trim())try{let U=JSON.parse(B.trim());q(U)}catch{q({modified:!1})}else q({modified:!1})}),z.on("error",(A)=>{D("Hook spawn error",{error:String(A)}),q({modified:!1})})})}function Y(J){D("Extension initialized");let M=new Set;J.on("tool_call",async(b,j)=>{if(b.toolName!=="bash")return;let q=b.input?.command;if(typeof q!=="string"||!/\bgit\s+commit\b/.test(q))return;D("tool_call (git commit)",{toolName:b.toolName,toolCallId:b.toolCallId});let z=j.sessionManager.getSessionId(),B=j.sessionManager.getSessionFile(),K=await R({hook_event_name:"tool_call",session_id:z,tool_call_id:b.toolCallId,tool:b.toolName,tool_input:b.input,cwd:j.cwd,session_file:B},j.cwd);if(K.modified&&K.updatedInput)D("tool_call: input modified",{modified:!0}),M.add(b.toolCallId)}),J.on("tool_result",async(b,j)=>{if(b.toolName!=="bash")return;let q=M.has(b.toolCallId),z=b.content?.map((A)=>A.type==="text"?A.text:"").join("").trim()??"",B=z.includes("agentlogs.ai/s/");if(!q&&!B)return;M.delete(b.toolCallId),D("tool_result (git commit)",{toolName:b.toolName,toolCallId:b.toolCallId,hasLink:B,wasIntercepted:q});let K=j.sessionManager.getSessionId();R({hook_event_name:"tool_result",session_id:K,tool_call_id:b.toolCallId,tool:b.toolName,tool_output:{content:z},cwd:j.cwd},j.cwd).catch((A)=>D("tool_result hook error",{error:String(A)}))});let Q=(b,j)=>{let q=j.sessionManager.getSessionId(),z=j.sessionManager.getSessionFile(),B=j.sessionManager.getLeafId();if(!z){D(`${b}: no session file (ephemeral session)`);return}D(b,{sessionId:q,sessionFile:z,leafId:B}),R({hook_event_name:b,session_id:q,session_file:z,leaf_id:B??void 0,cwd:j.cwd},j.cwd).catch((K)=>D(`${b} hook error`,{error:String(K)}))};J.on("agent_end",async(b,j)=>{Q("agent_end",j)}),J.on("session_shutdown",async(b,j)=>{Q("session_shutdown",j)})}export{Y as default,Y as agentLogsExtension};

package/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@agentlogs/pi",
+  "version": "0.0.1",
+  "description": "AgentLogs extension for Pi - automatically captures and uploads AI coding session transcripts",
+  "keywords": [
+    "agentlogs",
+    "ai-coding",
+    "extension",
+    "pi",
+    "transcript"
+  ],
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/agentlogs/agentlogs.git",
+    "directory": "packages/pi"
+  },
+  "files": [
+    "dist",
+    "src"
+  ],
+  "type": "module",
+  "main": "src/index.ts",
+  "types": "src/index.ts",
+  "publishConfig": {
+    "access": "public",
+    "main": "dist/index.js",
+    "types": "dist/index.d.ts"
+  },
+  "scripts": {
+    "build": "bun build ./src/index.ts --outdir ./dist --target node --define 'process.env.NODE_ENV=\"production\"' --minify",
+    "prepublishOnly": "bun run build",
+    "check": "tsgo --project tsconfig.json"
+  },
+  "peerDependencies": {
+    "bun": ">=1.0.0"
+  }
+}

package/src/index.ts

@@ -0,0 +1,300 @@
+/**
+ * AgentLogs Pi Extension
+ *
+ * Lightweight extension that shells out to the agentlogs CLI for all processing.
+ * The CLI handles transcript uploads, git commit interception, and commit tracking.
+ *
+ * @example
+ * // Install globally or in project
+ * npm install -g @agentlogs/pi
+ *
+ * // Add to pi settings.json
+ * { "extensions": ["@agentlogs/pi"] }
+ *
+ * // Or use the pi field in package.json
+ * { "pi": { "extensions": ["@agentlogs/pi"] } }
+ */
+
+import { appendFileSync } from "node:fs";
+import { spawn } from "node:child_process";
+
+// ============================================================================
+// Pi Types (inline to avoid external dependency)
+// ============================================================================
+
+interface ExtensionContext {
+  cwd: string;
+  sessionManager: {
+    getSessionId(): string;
+    getSessionFile(): string | undefined;
+    getLeafId(): string | null;
+  };
+}
+
+interface ToolCallEvent {
+  toolName: string;
+  toolCallId: string;
+  input: Record<string, unknown>;
+}
+
+interface ToolResultEvent {
+  toolName: string;
+  toolCallId: string;
+  content?: Array<{ type: string; text?: string }>;
+}
+
+interface ExtensionAPI {
+  on(event: "tool_call", handler: (event: ToolCallEvent, ctx: ExtensionContext) => void | Promise<void>): void;
+  on(event: "tool_result", handler: (event: ToolResultEvent, ctx: ExtensionContext) => void | Promise<void>): void;
+  on(event: "agent_end", handler: (event: unknown, ctx: ExtensionContext) => void | Promise<void>): void;
+  on(event: "session_shutdown", handler: (event: unknown, ctx: ExtensionContext) => void | Promise<void>): void;
+}
+
+// ============================================================================
+// Debug Logging (compiled out in production builds)
+// ============================================================================
+
+const LOG_FILE = "/tmp/agentlogs-pi.log";
+
+function log(message: string, data?: unknown): void {
+  if (process.env.NODE_ENV === "production") return;
+  const timestamp = new Date().toISOString();
+  const logLine = data
+    ? `[${timestamp}] ${message}\n${JSON.stringify(data, null, 2)}\n`
+    : `[${timestamp}] ${message}\n`;
+  try {
+    appendFileSync(LOG_FILE, logLine);
+  } catch {
+    // Ignore write errors
+  }
+}
+
+// ============================================================================
+// Types
+// ============================================================================
+
+interface HookPayload {
+  hook_event_name: string;
+  session_id: string;
+  tool_call_id?: string;
+  tool?: string;
+  cwd?: string;
+  tool_input?: Record<string, unknown>;
+  tool_output?: Record<string, unknown>;
+  session_file?: string;
+  leaf_id?: string;
+}
+
+interface HookResponse {
+  modified: boolean;
+  updatedInput?: Record<string, unknown>;
+}
+
+// ============================================================================
+// CLI Integration
+// ============================================================================
+
+/**
+ * Shell out to the agentlogs CLI hook command.
+ * Passes hook data via stdin, receives response via stdout.
+ */
+async function runHook(payload: HookPayload, cwd: string): Promise<HookResponse> {
+  const cliPath = process.env.AGENTLOGS_CLI_PATH;
+
+  let command: string;
+  let args: string[];
+
+  if (cliPath) {
+    const parts = cliPath.split(" ");
+    command = parts[0];
+    args = [...parts.slice(1), "pi", "hook"];
+  } else {
+    command = "npx";
+    args = ["-y", "agentlogs@latest", "pi", "hook"];
+  }
+
+  log("Running hook", { command, args, payload: payload.hook_event_name });
+
+  return new Promise((resolve) => {
+    const proc = spawn(command, args, {
+      cwd,
+      stdio: ["pipe", "pipe", "pipe"],
+    });
+
+    let stdout = "";
+    let stderr = "";
+
+    proc.stdout.on("data", (data) => {
+      stdout += data.toString();
+    });
+    proc.stderr.on("data", (data) => {
+      stderr += data.toString();
+    });
+
+    // Send payload via stdin
+    proc.stdin.write(JSON.stringify(payload));
+    proc.stdin.end();
+
+    proc.on("close", (code) => {
+      log("Hook process exited", { code, stdout: stdout.slice(0, 500), stderr: stderr.slice(0, 500) });
+
+      if (code === 0 && stdout.trim()) {
+        try {
+          const response = JSON.parse(stdout.trim()) as HookResponse;
+          resolve(response);
+        } catch {
+          resolve({ modified: false });
+        }
+      } else {
+        resolve({ modified: false });
+      }
+    });
+
+    proc.on("error", (err) => {
+      log("Hook spawn error", { error: String(err) });
+      resolve({ modified: false });
+    });
+  });
+}
+
+// ============================================================================
+// Main Extension
+// ============================================================================
+
+export default function agentLogsExtension(pi: ExtensionAPI) {
+  log("Extension initialized");
+
+  // Track tool call IDs where we intercepted a git commit
+  const interceptedToolCallIds = new Set<string>();
+
+  // Handle tool calls (intercept git commits)
+  pi.on("tool_call", async (event, ctx) => {
+    // Only intercept bash tool
+    if (event.toolName !== "bash") {
+      return;
+    }
+
+    // Quick check: skip if not a git commit
+    const command = event.input?.command;
+    if (typeof command !== "string" || !/\bgit\s+commit\b/.test(command)) {
+      return;
+    }
+
+    log("tool_call (git commit)", {
+      toolName: event.toolName,
+      toolCallId: event.toolCallId,
+    });
+
+    const sessionId = ctx.sessionManager.getSessionId();
+    const sessionFile = ctx.sessionManager.getSessionFile();
+
+    const response = await runHook(
+      {
+        hook_event_name: "tool_call",
+        session_id: sessionId,
+        tool_call_id: event.toolCallId,
+        tool: event.toolName,
+        tool_input: event.input as Record<string, unknown>,
+        cwd: ctx.cwd,
+        session_file: sessionFile,
+      },
+      ctx.cwd,
+    );
+
+    if (response.modified && response.updatedInput) {
+      log("tool_call: input modified", { modified: true });
+      // Track this tool call ID so we know to process the result
+      interceptedToolCallIds.add(event.toolCallId);
+
+      // Note: Pi doesn't support modifying tool input from extensions
+      // The CLI will need to handle this differently (pre-upload the transcript)
+      // For now, we just track that we saw this commit
+    }
+  });
+
+  // Handle tool results (track commits)
+  pi.on("tool_result", async (event, ctx) => {
+    // Only handle bash tool
+    if (event.toolName !== "bash") {
+      return;
+    }
+
+    // Check if we intercepted this tool call
+    const wasIntercepted = interceptedToolCallIds.has(event.toolCallId);
+
+    // Also check if output contains our transcript link
+    const outputText =
+      event.content
+        ?.map((c) => (c.type === "text" ? c.text : ""))
+        .join("")
+        .trim() ?? "";
+    const hasLink = outputText.includes("agentlogs.ai/s/");
+
+    if (!wasIntercepted && !hasLink) {
+      return;
+    }
+
+    // Clean up tracked ID
+    interceptedToolCallIds.delete(event.toolCallId);
+
+    log("tool_result (git commit)", {
+      toolName: event.toolName,
+      toolCallId: event.toolCallId,
+      hasLink,
+      wasIntercepted,
+    });
+
+    const sessionId = ctx.sessionManager.getSessionId();
+
+    // Fire and forget - CLI handles commit tracking
+    runHook(
+      {
+        hook_event_name: "tool_result",
+        session_id: sessionId,
+        tool_call_id: event.toolCallId,
+        tool: event.toolName,
+        tool_output: { content: outputText },
+        cwd: ctx.cwd,
+      },
+      ctx.cwd,
+    ).catch((err) => log("tool_result hook error", { error: String(err) }));
+  });
+
+  // Helper to trigger transcript upload
+  const uploadTranscript = (eventName: string, ctx: ExtensionContext) => {
+    const sessionId = ctx.sessionManager.getSessionId();
+    const sessionFile = ctx.sessionManager.getSessionFile();
+    const leafId = ctx.sessionManager.getLeafId();
+
+    if (!sessionFile) {
+      log(`${eventName}: no session file (ephemeral session)`);
+      return;
+    }
+
+    log(eventName, { sessionId, sessionFile, leafId });
+
+    // Fire and forget - CLI handles the upload
+    runHook(
+      {
+        hook_event_name: eventName,
+        session_id: sessionId,
+        session_file: sessionFile,
+        leaf_id: leafId ?? undefined,
+        cwd: ctx.cwd,
+      },
+      ctx.cwd,
+    ).catch((err) => log(`${eventName} hook error`, { error: String(err) }));
+  };
+
+  // Handle agent end (upload after each turn)
+  pi.on("agent_end", async (_event, ctx) => {
+    uploadTranscript("agent_end", ctx);
+  });
+
+  // Handle session shutdown (final upload)
+  pi.on("session_shutdown", async (_event, ctx) => {
+    uploadTranscript("session_shutdown", ctx);
+  });
+}
+
+export { agentLogsExtension };