octomux 1.0.0

package/dist-server/terminal.js

@@ -0,0 +1,173 @@
+import { execFile as execFileCb } from 'child_process';
+import { promisify } from 'util';
+import { WebSocketServer, WebSocket } from 'ws';
+import { spawn } from 'node-pty';
+import { nanoid } from 'nanoid';
+import { getDb } from './db.js';
+import { getOrchestratorSession } from './orchestrator.js';
+const execFile = promisify(execFileCb);
+const connections = new Map();
+let wss;
+export function setupTerminalWebSocket() {
+    wss = new WebSocketServer({ noServer: true });
+}
+export function handleTerminalUpgrade(req, socket, head) {
+    // Match /ws/terminal/orchestrator
+    const orchMatch = req.url?.match(/^\/ws\/terminal\/orchestrator$/);
+    if (orchMatch) {
+        wss.handleUpgrade(req, socket, head, (ws) => {
+            handleOrchestratorConnection(ws);
+        });
+        return true;
+    }
+    // Match /ws/terminal/:taskId/:windowIndex
+    const match = req.url?.match(/^\/ws\/terminal\/([^/]+)\/(\d+)$/);
+    if (!match)
+        return false;
+    wss.handleUpgrade(req, socket, head, (ws) => {
+        const taskId = match[1];
+        const windowIndex = parseInt(match[2], 10);
+        handleConnection(ws, taskId, windowIndex);
+    });
+    return true;
+}
+function attachToTmuxSession(ws, tmuxTarget, connKey, closeReason, linkedSession, pendingMessages) {
+    let pty;
+    try {
+        pty = spawn('tmux', ['attach-session', '-t', tmuxTarget], {
+            name: 'xterm-256color',
+            cols: 120,
+            rows: 30,
+            env: process.env,
+        });
+    }
+    catch {
+        ws.close(4005, closeReason);
+        return;
+    }
+    let ptyExited = false;
+    if (!connections.has(connKey)) {
+        connections.set(connKey, []);
+    }
+    connections.get(connKey).push({ ws, pty });
+    // PTY → WebSocket
+    pty.onData((data) => {
+        if (ws.readyState === WebSocket.OPEN) {
+            ws.send(data);
+        }
+    });
+    const handleMessage = (data) => {
+        if (ptyExited)
+            return;
+        const msg = typeof data === 'string' ? data : data.toString();
+        // Handle resize messages
+        try {
+            const parsed = JSON.parse(msg);
+            if (parsed.type === 'resize' && parsed.cols && parsed.rows) {
+                pty.resize(parsed.cols, parsed.rows);
+                return;
+            }
+        }
+        catch {
+            // Not JSON, treat as terminal input
+        }
+        try {
+            pty.write(msg);
+        }
+        catch {
+            // PTY already exited
+        }
+    };
+    // Replace any buffering handler with the real one
+    ws.removeAllListeners('message');
+    ws.on('message', handleMessage);
+    // Replay any messages that arrived before the PTY was ready
+    if (pendingMessages) {
+        for (const msg of pendingMessages) {
+            handleMessage(msg);
+        }
+    }
+    const cleanupLinkedSession = () => {
+        if (linkedSession) {
+            execFile('tmux', ['kill-session', '-t', linkedSession]).catch(() => { });
+        }
+    };
+    // Cleanup on WebSocket close
+    ws.on('close', () => {
+        if (!ptyExited) {
+            pty.kill();
+        }
+        cleanupLinkedSession();
+        const conns = connections.get(connKey);
+        if (conns) {
+            const idx = conns.findIndex((c) => c.ws === ws);
+            if (idx >= 0)
+                conns.splice(idx, 1);
+            if (conns.length === 0)
+                connections.delete(connKey);
+        }
+    });
+    // Cleanup on PTY exit
+    pty.onExit(() => {
+        ptyExited = true;
+        cleanupLinkedSession();
+        if (ws.readyState === WebSocket.OPEN) {
+            ws.close(4006, 'Terminal process exited');
+        }
+    });
+}
+async function handleConnection(ws, taskId, windowIndex) {
+    // Buffer messages that arrive while we set up the tmux session.
+    // Without this, the client's initial resize message (sent on WS open) would be
+    // lost because the message handler isn't registered until after the async tmux
+    // setup completes — Node.js EventEmitter discards events with no listeners.
+    const pendingMessages = [];
+    ws.on('message', (data) => {
+        pendingMessages.push(data);
+    });
+    const db = getDb();
+    const task = db.prepare('SELECT * FROM tasks WHERE id = ?').get(taskId);
+    if (!task || !task.tmux_session) {
+        ws.close(4004, 'Task not found or no tmux session');
+        return;
+    }
+    // Create a grouped session so each viewer has independent window selection.
+    // Without this, all clients attached to the same session share the active window,
+    // meaning switching tabs in one browser would affect all other viewers.
+    const linkedSession = `${task.tmux_session}-v-${nanoid(6)}`;
+    try {
+        await execFile('tmux', ['new-session', '-d', '-t', task.tmux_session, '-s', linkedSession]);
+        // Prevent grouped sessions from constraining window size to the smallest client
+        await execFile('tmux', ['set-option', '-t', linkedSession, 'aggressive-resize', 'on']).catch(() => { });
+        await execFile('tmux', ['select-window', '-t', `${linkedSession}:${windowIndex}`]);
+    }
+    catch {
+        ws.close(4005, 'Failed to create terminal view session');
+        return;
+    }
+    const connKey = `${taskId}:${windowIndex}`;
+    attachToTmuxSession(ws, linkedSession, connKey, 'Failed to attach to tmux session', linkedSession, pendingMessages);
+}
+function handleOrchestratorConnection(ws) {
+    const session = getOrchestratorSession();
+    attachToTmuxSession(ws, session, 'orchestrator', 'Failed to attach to orchestrator session');
+}
+export function getActiveConnections() {
+    return connections;
+}
+export function cleanupAllConnections() {
+    for (const [, conns] of connections) {
+        for (const { ws, pty } of conns) {
+            try {
+                pty.kill();
+            }
+            catch {
+                // already dead
+            }
+            if (ws.readyState === WebSocket.OPEN) {
+                ws.close(1001, 'Server shutting down');
+            }
+        }
+    }
+    connections.clear();
+}

package/dist-server/types.d.ts

@@ -0,0 +1,73 @@
+export type TaskStatus = 'draft' | 'setting_up' | 'running' | 'closed' | 'error';
+export type AgentStatus = 'running' | 'idle' | 'waiting' | 'stopped';
+export type HookActivity = 'active' | 'idle' | 'waiting';
+export type DerivedTaskStatus = 'working' | 'needs_attention' | 'done';
+export interface Task {
+    id: string;
+    title: string;
+    description: string;
+    repo_path: string;
+    status: TaskStatus;
+    branch: string | null;
+    base_branch: string | null;
+    worktree: string | null;
+    tmux_session: string | null;
+    pr_url: string | null;
+    pr_number: number | null;
+    user_window_index: number | null;
+    initial_prompt: string | null;
+    error: string | null;
+    created_at: string;
+    updated_at: string;
+    agents?: Agent[];
+    pending_prompts?: PermissionPrompt[];
+    derived_status?: DerivedTaskStatus | null;
+}
+export interface Agent {
+    id: string;
+    task_id: string;
+    window_index: number;
+    label: string;
+    status: AgentStatus;
+    claude_session_id: string | null;
+    hook_activity: HookActivity;
+    hook_activity_updated_at: string | null;
+    created_at: string;
+}
+export interface PermissionPrompt {
+    id: string;
+    task_id: string;
+    agent_id: string | null;
+    agent_label: string;
+    session_id: string;
+    tool_name: string;
+    tool_input: Record<string, unknown>;
+    status: 'pending' | 'resolved';
+    created_at: string;
+    resolved_at: string | null;
+}
+export interface CreateTaskRequest {
+    title: string;
+    description: string;
+    repo_path: string;
+    branch?: string;
+    base_branch?: string;
+    initial_prompt?: string;
+    draft?: boolean;
+}
+export interface OrchestratorStatus {
+    running: boolean;
+    session: string;
+}
+export interface AddAgentRequest {
+    prompt?: string;
+}
+export interface UpdateTaskRequest {
+    status?: 'closed' | 'running';
+    title?: string;
+    description?: string;
+    repo_path?: string;
+    branch?: string;
+    base_branch?: string;
+    initial_prompt?: string;
+}

package/dist-server/types.js

@@ -0,0 +1 @@
+export {};

package/package.json

@@ -0,0 +1,113 @@
+{
+  "name": "octomux",
+  "version": "1.0.0",
+  "type": "module",
+  "bin": {
+    "octomux": "./bin/octomux.js"
+  },
+  "engines": {
+    "node": ">=20"
+  },
+  "files": [
+    "bin/",
+    "dist/",
+    "dist-server/",
+    "!dist-server/*.test.*",
+    "!dist-server/test-helpers.*",
+    "cli/dist/",
+    "cli/package.json"
+  ],
+  "scripts": {
+    "dev": "concurrently \"bun:dev:server\" \"bun:dev:client\"",
+    "dev:server": "tsx watch server/index.ts",
+    "dev:client": "vite",
+    "build": "vite build && bun run build:server && bun run cli:build",
+    "build:server": "tsup server/index.ts --format esm --minify --out-dir dist-server --external better-sqlite3 --external node-pty --external ws",
+    "start": "NODE_ENV=production node dist-server/index.js",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "test:e2e": "playwright test",
+    "test:e2e:ui": "playwright test --ui",
+    "lint": "eslint .",
+    "lint:fix": "eslint . --fix",
+    "format": "prettier --write .",
+    "format:check": "prettier --check .",
+    "cli:build": "cd cli && tsc",
+    "typecheck": "tsc --noEmit",
+    "prepublishOnly": "bun run build",
+    "postinstall": "chmod +x node_modules/node-pty/prebuilds/darwin-*/spawn-helper 2>/dev/null || true",
+    "prepare": "husky",
+    "changelog": "conventional-changelog -p conventionalcommits -i CHANGELOG.md -s"
+  },
+  "lint-staged": {
+    "*.{ts,tsx}": [
+      "eslint --fix",
+      "prettier --write"
+    ],
+    "*.{json,md,css,html}": [
+      "prettier --write"
+    ]
+  },
+  "trustedDependencies": [
+    "better-sqlite3",
+    "husky"
+  ],
+  "dependencies": {
+    "@base-ui/react": "^1.2.0",
+    "@fontsource-variable/geist": "^5.2.8",
+    "better-sqlite3": "^11.7.0",
+    "class-variance-authority": "^0.7.1",
+    "clsx": "^2.1.1",
+    "express": "^5.0.1",
+    "nanoid": "^5.0.9",
+    "node-pty": "^1.0.0",
+    "shadcn": "latest",
+    "sonner": "^2.0.7",
+    "tailwind-merge": "^3.5.0",
+    "tw-animate-css": "^1.4.0",
+    "ws": "^8.18.0",
+    "chalk": "^5.6.2",
+    "commander": "^14.0.3"
+  },
+  "devDependencies": {
+    "@commitlint/cli": "^20.0.0",
+    "@commitlint/config-conventional": "^20.0.0",
+    "@eslint/js": "^9.0.0",
+    "@playwright/test": "^1.58.2",
+    "@tailwindcss/vite": "^4.0.0",
+    "@testing-library/jest-dom": "^6.9.1",
+    "@testing-library/react": "^16.3.2",
+    "@testing-library/user-event": "^14.6.1",
+    "@types/better-sqlite3": "^7.6.0",
+    "@types/express": "^5.0.0",
+    "@types/node": "^22.0.0",
+    "@types/react": "^19.0.0",
+    "@types/react-dom": "^19.0.0",
+    "@types/supertest": "^7.2.0",
+    "@types/ws": "^8.5.0",
+    "@vitejs/plugin-react": "^4.3.0",
+    "@xterm/addon-fit": "^0.10.0",
+    "@xterm/addon-web-links": "^0.11.0",
+    "@xterm/xterm": "^5.5.0",
+    "concurrently": "^9.1.0",
+    "conventional-changelog-cli": "^5.0.0",
+    "eslint": "^9.0.0",
+    "globals": "^16.0.0",
+    "husky": "^9.0.0",
+    "jsdom": "^28.1.0",
+    "lint-staged": "^16.0.0",
+    "prettier": "^3.5.0",
+    "react": "^19.0.0",
+    "react-dom": "^19.0.0",
+    "react-router": "^7.0.0",
+    "react-router-dom": "^7.0.0",
+    "supertest": "^7.2.2",
+    "tailwindcss": "^4.0.0",
+    "tsup": "^8.5.1",
+    "tsx": "^4.19.0",
+    "typescript": "^5.7.0",
+    "typescript-eslint": "^8.0.0",
+    "vite": "^6.0.0",
+    "vitest": "^3.2.4"
+  }
+}