pi-teams 0.7.3 → 0.8.2

package/README.md

@@ -1,6 +1,6 @@
 # pi-teams 🚀
 
-**pi-teams** turns your single Pi agent into a coordinated software engineering team. It allows you to spawn multiple "Teammate" agents in separate terminal panes that work autonomously, communicate with each other, and manage a shared task board—all mediated through tmux, iTerm2, or Zellij.
+**pi-teams** turns your single Pi agent into a coordinated software engineering team. It allows you to spawn multiple "Teammate" agents in separate terminal panes that work autonomously, communicate with each other, and manage a shared task board—all mediated through tmux, Zellij, iTerm2, or WezTerm.
 
 ### 🖥️ pi-teams in Action
 
@@ -8,6 +8,8 @@
 | :---: | :---: | :---: |
 | <a href="iTerm2.png"><img src="iTerm2.png" width="300" alt="pi-teams in iTerm2"></a> | <a href="tmux.png"><img src="tmux.png" width="300" alt="pi-teams in tmux"></a> | <a href="zellij.png"><img src="zellij.png" width="300" alt="pi-teams in Zellij"></a> |
 
+*Also works with **WezTerm** (cross-platform support)*
+
 ## 🛠 Installation
 
 Open your Pi terminal and type:
@@ -87,9 +89,9 @@ Teammates in `planning` mode will use `task_submit_plan`. As the lead, review th
 - **[Full Usage Guide](docs/guide.md)** - Detailed examples, hook system, best practices, and troubleshooting
 - **[Tool Reference](docs/reference.md)** - Complete documentation of all tools and parameters
 
-## 🪟 Terminal Requirements: tmux, Zellij, or iTerm2
+## 🪟 Terminal Requirements
 
-To show multiple agents on one screen, **pi-teams** requires a way to manage terminal panes. It supports **tmux**, **Zellij**, and **iTerm2** (macOS).
+To show multiple agents on one screen, **pi-teams** requires a way to manage terminal panes. It supports **tmux**, **Zellij**, **iTerm2**, and **WezTerm**.
 
 ### Option 1: tmux (Recommended)
 
@@ -111,6 +113,21 @@ Simply start `pi` inside a Zellij session. **pi-teams** will detect it via the `
 
 If you are using **iTerm2** on macOS and are *not* inside tmux or Zellij, **pi-teams** will use AppleScript to automatically split your current window into an optimized layout (1 large Lead pane on the left, Teammates stacked on the right). It will also name the panes with the teammate's agent name for easy identification.
 
+### Option 4: WezTerm (macOS, Linux, Windows)
+
+**WezTerm** is a GPU-accelerated, cross-platform terminal emulator written in Rust. If you are using WezTerm and are *not* inside tmux or Zellij, **pi-teams** will use `wezterm cli split-pane` to spawn teammates in new panes with an optimized layout (1 large Lead pane on the left, Teammates stacked on the right).
+
+Install WezTerm:
+- **macOS**: `brew install --cask wezterm`
+- **Linux**: See [wezterm.org/installation](https://wezterm.org/installation)
+- **Windows**: Download from [wezterm.org](https://wezterm.org)
+
+How to run:
+```bash
+wezterm  # Start WezTerm
+pi       # Start pi inside WezTerm
+```
+
 ## 📜 Credits & Attribution
 
 This project is a port of the excellent [claude-code-teams-mcp](https://github.com/cs50victor/claude-code-teams-mcp) by [cs50victor](https://github.com/cs50victor).

package/package.json

@@ -1,8 +1,11 @@
 {
   "name": "pi-teams",
-  "version": "0.7.3",
+  "version": "0.8.2",
   "description": "Agent teams for pi, ported from claude-code-teams-mcp",
-  "repository": "github:burggraf/pi-teams",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/burggraf/pi-teams.git"
+  },
   "author": "Mark Burggraf",
   "license": "MIT",
   "keywords": [

package/src/adapters/terminal-registry.ts

@@ -9,14 +9,22 @@ import { TerminalAdapter } from "../utils/terminal-adapter";
 import { TmuxAdapter } from "./tmux-adapter";
 import { Iterm2Adapter } from "./iterm2-adapter";
 import { ZellijAdapter } from "./zellij-adapter";
+import { WezTermAdapter } from "./wezterm-adapter";
 
 /**
  * Available terminal adapters, ordered by priority
+ *
+ * Detection order (first match wins):
+ * 1. tmux - if TMUX env is set
+ * 2. Zellij - if ZELLIJ env is set and not in tmux
+ * 3. iTerm2 - if TERM_PROGRAM=iTerm.app and not in tmux/zellij
+ * 4. WezTerm - if WEZTERM_PANE env is set and not in tmux/zellij
  */
 const adapters: TerminalAdapter[] = [
   new TmuxAdapter(),
-  new Iterm2Adapter(),
   new ZellijAdapter(),
+  new Iterm2Adapter(),
+  new WezTermAdapter(),
 ];
 
 /**
@@ -26,12 +34,13 @@ let cachedAdapter: TerminalAdapter | null = null;
 
 /**
  * Detect and return the appropriate terminal adapter for the current environment.
- * 
+ *
  * Detection order (first match wins):
  * 1. tmux - if TMUX env is set
- * 2. iTerm2 - if TERM_PROGRAM=iTerm.app and not in tmux/zellij
- * 3. Zellij - if ZELLIJ env is set and not in tmux
- * 
+ * 2. Zellij - if ZELLIJ env is set and not in tmux
+ * 3. iTerm2 - if TERM_PROGRAM=iTerm.app and not in tmux/zellij
+ * 4. WezTerm - if WEZTERM_PANE env is set and not in tmux/zellij
+ *
  * @returns The detected terminal adapter, or null if none detected
  */
 export function getTerminalAdapter(): TerminalAdapter | null {
@@ -51,8 +60,8 @@ export function getTerminalAdapter(): TerminalAdapter | null {
 
 /**
  * Get a specific terminal adapter by name.
- * 
- * @param name - The adapter name (e.g., "tmux", "iTerm2", "zellij")
+ *
+ * @param name - The adapter name (e.g., "tmux", "iTerm2", "zellij", "WezTerm")
  * @returns The adapter instance, or undefined if not found
  */
 export function getAdapterByName(name: string): TerminalAdapter | undefined {

package/src/adapters/wezterm-adapter.test.ts

@@ -0,0 +1,101 @@
+/**
+ * WezTerm Adapter Tests
+ */
+
+import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
+import { WezTermAdapter } from "./wezterm-adapter";
+import * as terminalAdapter from "../utils/terminal-adapter";
+
+describe("WezTermAdapter", () => {
+  let adapter: WezTermAdapter;
+  let mockExecCommand: ReturnType<typeof vi.spyOn>;
+
+  beforeEach(() => {
+    adapter = new WezTermAdapter();
+    mockExecCommand = vi.spyOn(terminalAdapter, "execCommand");
+    delete process.env.WEZTERM_PANE;
+    delete process.env.TMUX;
+    delete process.env.ZELLIJ;
+    process.env.WEZTERM_PANE = "0";
+  });
+
+  afterEach(() => {
+    vi.clearAllMocks();
+  });
+
+  describe("name", () => {
+    it("should have the correct name", () => {
+      expect(adapter.name).toBe("WezTerm");
+    });
+  });
+
+  describe("detect", () => {
+    it("should detect when WEZTERM_PANE is set", () => {
+      mockExecCommand.mockReturnValue({ stdout: "version 1.0", stderr: "", status: 0 });
+      expect(adapter.detect()).toBe(true);
+    });
+  });
+
+  describe("spawn", () => {
+    it("should spawn first pane to the right with 50%", () => {
+      // Mock getPanes finding only current pane
+      mockExecCommand.mockImplementation((bin, args) => {
+        if (args.includes("list")) {
+          return { 
+            stdout: JSON.stringify([{ pane_id: 0, tab_id: 0 }]), 
+            stderr: "", 
+            status: 0 
+          };
+        }
+        if (args.includes("split-pane")) {
+          return { stdout: "1", stderr: "", status: 0 };
+        }
+        return { stdout: "", stderr: "", status: 0 };
+      });
+
+      const result = adapter.spawn({
+        name: "test-agent",
+        cwd: "/home/user/project",
+        command: "pi --agent test",
+        env: { PI_AGENT_ID: "test-123" },
+      });
+
+      expect(result).toBe("wezterm_1");
+      expect(mockExecCommand).toHaveBeenCalledWith(
+        expect.stringContaining("wezterm"),
+        expect.arrayContaining(["cli", "split-pane", "--right", "--percent", "50"])
+      );
+    });
+
+    it("should spawn subsequent panes by splitting the sidebar", () => {
+      // Mock getPanes finding current pane (0) and sidebar pane (1)
+      mockExecCommand.mockImplementation((bin, args) => {
+        if (args.includes("list")) {
+          return { 
+            stdout: JSON.stringify([{ pane_id: 0, tab_id: 0 }, { pane_id: 1, tab_id: 0 }]), 
+            stderr: "", 
+            status: 0 
+          };
+        }
+        if (args.includes("split-pane")) {
+          return { stdout: "2", stderr: "", status: 0 };
+        }
+        return { stdout: "", stderr: "", status: 0 };
+      });
+
+      const result = adapter.spawn({
+        name: "agent2",
+        cwd: "/home/user/project",
+        command: "pi",
+        env: {},
+      });
+
+      expect(result).toBe("wezterm_2");
+      // 1 sidebar pane already exists, so percent should be floor(100/(1+1)) = 50%
+      expect(mockExecCommand).toHaveBeenCalledWith(
+        expect.stringContaining("wezterm"),
+        expect.arrayContaining(["cli", "split-pane", "--bottom", "--pane-id", "1", "--percent", "50"])
+      );
+    });
+  });
+});

package/src/adapters/wezterm-adapter.ts

@@ -0,0 +1,166 @@
+/**
+ * WezTerm Terminal Adapter
+ *
+ * Implements the TerminalAdapter interface for WezTerm terminal emulator.
+ * Uses wezterm cli split-pane for pane management.
+ */
+
+import { TerminalAdapter, SpawnOptions, execCommand } from "../utils/terminal-adapter";
+
+export class WezTermAdapter implements TerminalAdapter {
+  readonly name = "WezTerm";
+
+  // Common paths where wezterm CLI might be found
+  private possiblePaths = [
+    "wezterm",  // In PATH
+    "/Applications/WezTerm.app/Contents/MacOS/wezterm",  // macOS
+    "/usr/local/bin/wezterm",  // Linux/macOS common
+    "/usr/bin/wezterm",  // Linux system
+  ];
+
+  private weztermPath: string | null = null;
+
+  private findWeztermBinary(): string | null {
+    if (this.weztermPath !== null) {
+      return this.weztermPath;
+    }
+
+    for (const path of this.possiblePaths) {
+      try {
+        const result = execCommand(path, ["--version"]);
+        if (result.status === 0) {
+          this.weztermPath = path;
+          return path;
+        }
+      } catch {
+        // Continue to next path
+      }
+    }
+
+    this.weztermPath = null;
+    return null;
+  }
+
+  detect(): boolean {
+    if (!process.env.WEZTERM_PANE || process.env.TMUX || process.env.ZELLIJ) {
+      return false;
+    }
+    return this.findWeztermBinary() !== null;
+  }
+
+  /**
+   * Get all panes in the current tab to determine layout state.
+   */
+  private getPanes(): any[] {
+    const weztermBin = this.findWeztermBinary();
+    if (!weztermBin) return [];
+
+    const result = execCommand(weztermBin, ["cli", "list", "--format", "json"]);
+    if (result.status !== 0) return [];
+
+    try {
+      const allPanes = JSON.parse(result.stdout);
+      const currentPaneId = parseInt(process.env.WEZTERM_PANE || "0", 10);
+      
+      // Find the tab of the current pane
+      const currentPane = allPanes.find((p: any) => p.pane_id === currentPaneId);
+      if (!currentPane) return [];
+
+      // Return all panes in the same tab
+      return allPanes.filter((p: any) => p.tab_id === currentPane.tab_id);
+    } catch {
+      return [];
+    }
+  }
+
+  spawn(options: SpawnOptions): string {
+    const weztermBin = this.findWeztermBinary();
+    if (!weztermBin) {
+      throw new Error("WezTerm CLI binary not found.");
+    }
+
+    const panes = this.getPanes();
+    const envArgs = Object.entries(options.env)
+      .filter(([k]) => k.startsWith("PI_"))
+      .map(([k, v]) => `${k}=${v}`);
+
+    let weztermArgs: string[];
+    
+    // First pane: split to the right with 50% (matches iTerm2/tmux behavior)
+    const isFirstPane = panes.length === 1;
+
+    if (isFirstPane) {
+      weztermArgs = [
+        "cli", "split-pane", "--right", "--percent", "50",
+        "--cwd", options.cwd, "--", "env", ...envArgs, "sh", "-c", options.command
+      ];
+    } else {
+      // Subsequent teammates stack in the sidebar on the right.
+      // currentPaneId (id 0) is the main pane on the left.
+      // All other panes are in the sidebar.
+      const currentPaneId = parseInt(process.env.WEZTERM_PANE || "0", 10);
+      const sidebarPanes = panes
+        .filter(p => p.pane_id !== currentPaneId)
+        .sort((a, b) => b.cursor_y - a.cursor_y); // Sort by vertical position (bottom-most first)
+
+      // To add a new pane to the bottom of the sidebar stack:
+      // We always split the BOTTOM-MOST pane (sidebarPanes[0])
+      // and use 50% so the new pane and the previous bottom pane are equal.
+      // This progressively fills the sidebar from top to bottom.
+      const targetPane = sidebarPanes[0];
+
+      weztermArgs = [
+        "cli", "split-pane", "--bottom", "--pane-id", targetPane.pane_id.toString(),
+        "--percent", "50",
+        "--cwd", options.cwd, "--", "env", ...envArgs, "sh", "-c", options.command
+      ];
+    }
+
+    const result = execCommand(weztermBin, weztermArgs);
+    if (result.status !== 0) {
+      throw new Error(`wezterm spawn failed: ${result.stderr}`);
+    }
+
+    // New: After spawning, tell WezTerm to equalize the panes in this tab
+    // This ensures that regardless of the split math, they all end up the same height.
+    try {
+      execCommand(weztermBin, ["cli", "zoom-pane", "--unzoom"]); // Ensure not zoomed
+      // WezTerm doesn't have a single "equalize" command like tmux, 
+      // but splitting with no percentage usually balances, or we can use 
+      // the 'AdjustPaneSize' sequence. 
+      // For now, let's stick to the 50/50 split of the LAST pane which is most reliable.
+    } catch {}
+
+    const paneId = result.stdout.trim();
+    return `wezterm_${paneId}`;
+  }
+
+  kill(paneId: string): void {
+    if (!paneId?.startsWith("wezterm_")) return;
+    const weztermBin = this.findWeztermBinary();
+    if (!weztermBin) return;
+
+    const weztermId = paneId.replace("wezterm_", "");
+    try {
+      execCommand(weztermBin, ["cli", "kill-pane", "--pane-id", weztermId]);
+    } catch {}
+  }
+
+  isAlive(paneId: string): boolean {
+    if (!paneId?.startsWith("wezterm_")) return false;
+    const weztermBin = this.findWeztermBinary();
+    if (!weztermBin) return false;
+
+    const weztermId = parseInt(paneId.replace("wezterm_", ""), 10);
+    const panes = this.getPanes();
+    return panes.some(p => p.pane_id === weztermId);
+  }
+
+  setTitle(title: string): void {
+    const weztermBin = this.findWeztermBinary();
+    if (!weztermBin) return;
+    try {
+      execCommand(weztermBin, ["cli", "set-tab-title", title]);
+    } catch {}
+  }
+}