@wong2kim/wmux 1.0.0 → 1.0.1

package/src/main/pty/AgentDetector.ts

@@ -1,247 +0,0 @@
-// Terminal agent status detection — monitors PTY output for known AI agent
-// prompt patterns and status indicators. This is status display only;
-// no content is captured, stored, or transmitted.
-
-export interface AgentEvent {
-  agent: string;
-  status: 'completed' | 'waiting' | 'running' | 'error';
-  message: string;
-}
-
-export interface CriticalEvent {
-  action: string;
-  riskLevel: 'review' | 'critical';
-}
-
-type AgentEventCallback = (event: AgentEvent) => void;
-type CriticalEventCallback = (event: CriticalEvent) => void;
-
-interface AgentPattern {
-  agent: string;
-  patterns: { regex: RegExp; status: AgentEvent['status']; message: string }[];
-}
-
-// ---------------------------------------------------------------------------
-// Common cross-agent terminal patterns
-// ---------------------------------------------------------------------------
-
-/** Shared completion indicators (✓ ✔ Done Complete Finished Success) */
-const COMMON_COMPLETE: AgentPattern['patterns'] = [
-  { regex: /[✓✔]\s+(.+)/,               status: 'completed', message: 'Task completed' },
-  { regex: /\b(Done|Complete(?:d)?|Finished|Success(?:ful)?)\b/, status: 'completed', message: 'Task completed' },
-];
-
-/** Shared error indicators (✗ ✘ Error Failed error:) */
-const COMMON_ERROR: AgentPattern['patterns'] = [
-  { regex: /[✗✘]\s+(.+)/,               status: 'error',     message: 'Error occurred' },
-  { regex: /\bFailed\b/,                 status: 'error',     message: 'Task failed' },
-  { regex: /\berror:\s+(.+)/i,           status: 'error',     message: 'Error occurred' },
-];
-
-/** Shared waiting indicators (? Waiting for Press y/n [Y/n]) */
-const COMMON_WAITING: AgentPattern['patterns'] = [
-  { regex: /\?\s+(.+)/,                  status: 'waiting',   message: 'Waiting for input' },
-  { regex: /Waiting for\s+(.+)/i,        status: 'waiting',   message: 'Waiting for input' },
-  { regex: /Press\s+.+\s+to\s+/i,        status: 'waiting',   message: 'Waiting for key press' },
-  { regex: /\[Y\/n\]|\(y\/n\)/i,         status: 'waiting',   message: 'Waiting for confirmation' },
-];
-
-// ---------------------------------------------------------------------------
-// Per-agent patterns
-// ---------------------------------------------------------------------------
-
-const AGENT_PATTERNS: AgentPattern[] = [
-  // ── Claude Code ────────────────────────────────────────────────────────────
-  {
-    agent: 'Claude Code',
-    patterns: [
-      { regex: /⏳\s+(.+)/,              status: 'running',   message: 'Processing...' },
-      { regex: /❌\s+(.+)/,              status: 'error',     message: 'Error occurred' },
-      { regex: /Do you want to/,         status: 'waiting',   message: 'Waiting for confirmation' },
-      ...COMMON_COMPLETE,
-      ...COMMON_ERROR,
-      ...COMMON_WAITING,
-    ],
-  },
-
-  // ── Cursor Agent ──────────────────────────────────────────────────────────
-  {
-    agent: 'Cursor Agent',
-    patterns: [
-      { regex: /Applied \d+ changes?/,   status: 'completed', message: 'Changes applied' },
-      { regex: /Thinking\.\.\./,         status: 'running',   message: 'Thinking...' },
-      ...COMMON_COMPLETE,
-      ...COMMON_ERROR,
-      ...COMMON_WAITING,
-    ],
-  },
-
-  // ── Aider ─────────────────────────────────────────────────────────────────
-  {
-    agent: 'Aider',
-    patterns: [
-      { regex: /Applied edit to/,        status: 'completed', message: 'Edit applied' },
-      { regex: /aider>/,                 status: 'waiting',   message: 'Waiting for input' },
-      ...COMMON_COMPLETE,
-      ...COMMON_ERROR,
-      ...COMMON_WAITING,
-    ],
-  },
-
-  // ── Codex CLI ─────────────────────────────────────────────────────────────
-  {
-    agent: 'Codex CLI',
-    patterns: [
-      { regex: /codex>/,                 status: 'waiting',   message: 'Waiting for input' },
-      { regex: /Codex:\s+(.+)/,          status: 'running',   message: 'Processing...' },
-      ...COMMON_COMPLETE,
-      ...COMMON_ERROR,
-      ...COMMON_WAITING,
-    ],
-  },
-
-  // ── Gemini CLI ────────────────────────────────────────────────────────────
-  {
-    agent: 'Gemini CLI',
-    patterns: [
-      { regex: /gemini>/,                status: 'waiting',   message: 'Waiting for input' },
-      { regex: /Gemini:\s+(.+)/,         status: 'running',   message: 'Processing...' },
-      ...COMMON_COMPLETE,
-      ...COMMON_ERROR,
-      ...COMMON_WAITING,
-    ],
-  },
-
-  // ── OpenCode ──────────────────────────────────────────────────────────────
-  {
-    agent: 'OpenCode',
-    patterns: [
-      { regex: /opencode>/,              status: 'waiting',   message: 'Waiting for input' },
-      ...COMMON_COMPLETE,
-      ...COMMON_ERROR,
-      ...COMMON_WAITING,
-    ],
-  },
-
-  // ── GitHub Copilot CLI ────────────────────────────────────────────────────
-  {
-    agent: 'GitHub Copilot CLI',
-    patterns: [
-      { regex: /copilot>/,               status: 'waiting',   message: 'Waiting for input' },
-      { regex: /gh copilot\s+(.+)/,      status: 'running',   message: 'Processing...' },
-      ...COMMON_COMPLETE,
-      ...COMMON_ERROR,
-      ...COMMON_WAITING,
-    ],
-  },
-];
-
-// ---------------------------------------------------------------------------
-// Critical action patterns — require approval before execution
-// ---------------------------------------------------------------------------
-
-interface CriticalPattern {
-  regex: RegExp;
-  riskLevel: 'review' | 'critical';
-  label: string;
-}
-
-const CRITICAL_PATTERNS: CriticalPattern[] = [
-  // Destructive git operations
-  { regex: /git\s+push\s+(?:.*--force|-f)\b/i,          riskLevel: 'critical', label: 'git push --force' },
-  { regex: /git\s+reset\s+--hard\b/i,                   riskLevel: 'critical', label: 'git reset --hard' },
-  { regex: /git\s+clean\s+.*-f\b/i,                     riskLevel: 'critical', label: 'git clean -f' },
-  // File system wipe
-  { regex: /\brm\s+(?:.*-r.*-f|-f.*-r|-rf|-fr)\s+/i,   riskLevel: 'critical', label: 'rm -rf' },
-  { regex: /\brmdir\s+\/[sS]\s+/,                       riskLevel: 'critical', label: 'rmdir /S' },
-  // Database destructive
-  { regex: /\bDROP\s+(?:TABLE|DATABASE|SCHEMA)\b/i,     riskLevel: 'critical', label: 'DROP TABLE/DATABASE' },
-  { regex: /\bDELETE\s+FROM\b/i,                        riskLevel: 'review',   label: 'DELETE FROM' },
-  { regex: /\bTRUNCATE\s+TABLE\b/i,                     riskLevel: 'critical', label: 'TRUNCATE TABLE' },
-  // NPM publishing
-  { regex: /\bnpm\s+publish\b/i,                        riskLevel: 'critical', label: 'npm publish' },
-  { regex: /\bnpx\s+.*--publish\b/i,                    riskLevel: 'review',   label: 'npx publish' },
-  // Cloud resource destruction
-  { regex: /\bterraform\s+destroy\b/i,                  riskLevel: 'critical', label: 'terraform destroy' },
-  { regex: /\bkubectl\s+delete\b/i,                     riskLevel: 'review',   label: 'kubectl delete' },
-  { regex: /\baws\s+.*\s+delete\b/i,                    riskLevel: 'review',   label: 'aws delete' },
-  // Disk formatting
-  { regex: /\bformat\s+[A-Za-z]:\\/i,                   riskLevel: 'critical', label: 'format disk' },
-  { regex: /\bmkfs\b/i,                                 riskLevel: 'critical', label: 'mkfs' },
-];
-
-const MAX_BUFFER = 16 * 1024; // 16 KB
-
-export class AgentDetector {
-  private callbacks: AgentEventCallback[] = [];
-  private criticalCallbacks: CriticalEventCallback[] = [];
-  private lineBuffer = '';
-  private lastEmittedKey = '';
-
-  onEvent(callback: AgentEventCallback): void {
-    this.callbacks.push(callback);
-  }
-
-  onCritical(callback: CriticalEventCallback): void {
-    this.criticalCallbacks.push(callback);
-  }
-
-  feed(data: string): void {
-    // Accumulate lines
-    this.lineBuffer += data;
-    // Prevent unbounded buffer growth
-    if (this.lineBuffer.length > MAX_BUFFER) {
-      this.lineBuffer = this.lineBuffer.slice(-MAX_BUFFER);
-    }
-    const lines = this.lineBuffer.split(/\r?\n/);
-    // Keep the last incomplete line in buffer
-    this.lineBuffer = lines.pop() || '';
-
-    for (const line of lines) {
-      this.processLine(line);
-    }
-  }
-
-  private processLine(line: string): void {
-    // Strip ANSI escape codes for pattern matching
-    const clean = line.replace(/\x1b\[[0-9;]*[a-zA-Z]/g, '').trim();
-    if (!clean) return;
-
-    // Check critical patterns first
-    for (const cp of CRITICAL_PATTERNS) {
-      if (cp.regex.test(clean)) {
-        const key = `critical:${cp.label}:${clean.slice(0, 80)}`;
-        if (key !== this.lastEmittedKey) {
-          this.lastEmittedKey = key;
-          const event: CriticalEvent = { action: cp.label, riskLevel: cp.riskLevel };
-          for (const cb of this.criticalCallbacks) {
-            cb(event);
-          }
-        }
-        return;
-      }
-    }
-
-    for (const ap of AGENT_PATTERNS) {
-      for (const p of ap.patterns) {
-        const match = clean.match(p.regex);
-        if (match) {
-          // Deduplicate: don't emit the same event twice in a row
-          const key = `${ap.agent}:${p.status}:${match[0]}`;
-          if (key === this.lastEmittedKey) return;
-          this.lastEmittedKey = key;
-
-          const event: AgentEvent = {
-            agent: ap.agent,
-            status: p.status,
-            message: match[1] || p.message,
-          };
-          for (const cb of this.callbacks) {
-            cb(event);
-          }
-          return;
-        }
-      }
-    }
-  }
-}

package/src/main/pty/OscParser.ts

@@ -1,81 +0,0 @@
-export interface OscEvent {
-  code: number;
-  data: string;
-}
-
-export type OscCallback = (event: OscEvent) => void;
-
-/**
- * Parses OSC (Operating System Command) sequences from terminal data.
- * Handles OSC 7 (CWD), OSC 9/99/777 (notifications).
- */
-const MAX_BUFFER = 64 * 1024; // 64 KB
-
-export class OscParser {
-  private buffer = '';
-  private inOsc = false;
-  private callbacks: OscCallback[] = [];
-
-  onOsc(callback: OscCallback): void {
-    this.callbacks.push(callback);
-  }
-
-  /**
-   * Process terminal data, extract OSC sequences, return cleaned data.
-   */
-  process(data: string): string {
-    let result = '';
-    let i = 0;
-
-    while (i < data.length) {
-      if (this.inOsc) {
-        // Look for ST (String Terminator): BEL (\x07) or ESC \ (\x1b\x5c)
-        if (data[i] === '\x07') {
-          this.emitOsc(this.buffer);
-          this.buffer = '';
-          this.inOsc = false;
-          i++;
-        } else if (data[i] === '\x1b' && i + 1 < data.length && data[i + 1] === '\\') {
-          this.emitOsc(this.buffer);
-          this.buffer = '';
-          this.inOsc = false;
-          i += 2;
-        } else {
-          this.buffer += data[i];
-          // Prevent unbounded buffer growth
-          if (this.buffer.length > MAX_BUFFER) {
-            this.buffer = '';
-            this.inOsc = false;
-          }
-          i++;
-        }
-      } else if (data[i] === '\x1b' && i + 1 < data.length && data[i + 1] === ']') {
-        // OSC start: ESC ]
-        this.inOsc = true;
-        this.buffer = '';
-        i += 2;
-      } else {
-        result += data[i];
-        i++;
-      }
-    }
-
-    return result;
-  }
-
-  private emitOsc(raw: string): void {
-    // OSC format: code;data
-    const semicolonIdx = raw.indexOf(';');
-    if (semicolonIdx === -1) return;
-
-    const codeStr = raw.substring(0, semicolonIdx);
-    const code = parseInt(codeStr, 10);
-    if (isNaN(code)) return;
-
-    const data = raw.substring(semicolonIdx + 1);
-
-    for (const cb of this.callbacks) {
-      cb({ code, data });
-    }
-  }
-}

package/src/main/pty/PTYBridge.ts

@@ -1,88 +0,0 @@
-import { BrowserWindow } from 'electron';
-import { PTYManager } from './PTYManager';
-import { OscParser } from './OscParser';
-import { AgentDetector } from './AgentDetector';
-import { ToastManager } from '../notification/ToastManager';
-import { IPC } from '../../shared/constants';
-
-export class PTYBridge {
-  private oscParsers = new Map<string, OscParser>();
-  private agentDetectors = new Map<string, AgentDetector>();
-  private toastManager = new ToastManager();
-
-  constructor(
-    private ptyManager: PTYManager,
-    private getWindow: () => BrowserWindow | null,
-  ) {}
-
-  setupDataForwarding(ptyId: string): void {
-    const instance = this.ptyManager.get(ptyId);
-    if (!instance) return;
-
-    const oscParser = new OscParser();
-    this.oscParsers.set(ptyId, oscParser);
-
-    const agentDetector = new AgentDetector();
-    this.agentDetectors.set(ptyId, agentDetector);
-
-    // Handle OSC events
-    oscParser.onOsc((event) => {
-      const win = this.getWindow();
-      if (!win || win.isDestroyed()) return;
-
-      switch (event.code) {
-        case 7: {
-          // CWD changed — data is typically file://host/path
-          const cwd = event.data.replace(/^file:\/\/[^/]*/, '');
-          win.webContents.send(IPC.CWD_CHANGED, ptyId, cwd);
-          break;
-        }
-        case 9:   // Windows Terminal notification
-        case 99:  // iTerm2 notification
-        case 777: // rxvt-unicode notification
-          // Silently ignore — no notification, no sound
-          break;
-      }
-    });
-
-    // Handle agent detection events — status tracking only, no notification/sound
-    agentDetector.onEvent(() => {
-      // Agent status is tracked internally by AgentDetector.
-      // No notification or sound — these fire too frequently and flood the UI.
-    });
-
-    // Handle critical action events — send approval request to renderer
-    agentDetector.onCritical((criticalEvent) => {
-      const win = this.getWindow();
-      if (!win || win.isDestroyed()) return;
-
-      win.webContents.send(IPC.APPROVAL_REQUEST, ptyId, {
-        action: criticalEvent.action,
-        riskLevel: criticalEvent.riskLevel,
-      });
-    });
-
-    instance.process.onData((data: string) => {
-      const win = this.getWindow();
-      if (win && !win.isDestroyed()) {
-        // Process data through OscParser (strips OSC sequences)
-        oscParser.process(data);
-        // Feed data to AgentDetector
-        agentDetector.feed(data);
-        // Forward raw data to renderer (xterm handles OSC itself)
-        win.webContents.send(IPC.PTY_DATA, ptyId, data);
-      }
-    });
-
-    instance.process.onExit(({ exitCode }) => {
-      const win = this.getWindow();
-      if (win && !win.isDestroyed()) {
-        win.webContents.send(IPC.PTY_EXIT, ptyId, exitCode);
-      }
-      this.oscParsers.delete(ptyId);
-      this.agentDetectors.delete(ptyId);
-      // Process already exited — remove from map without calling kill()
-      this.ptyManager.remove(ptyId);
-    });
-  }
-}

package/src/main/pty/PTYManager.ts

@@ -1,104 +0,0 @@
-import * as pty from 'node-pty';
-import os from 'node:os';
-import { getPipeName, ENV_KEYS } from '../../shared/constants';
-
-export interface PTYInstance {
-  id: string;
-  process: pty.IPty;
-  shell: string;
-}
-
-export class PTYManager {
-  private instances = new Map<string, PTYInstance>();
-  private nextId = 0;
-  private _authToken: string | undefined;
-
-  setAuthToken(token: string): void {
-    this._authToken = token;
-  }
-
-  create(options?: {
-    shell?: string;
-    cwd?: string;
-    cols?: number;
-    rows?: number;
-    workspaceId?: string;
-    surfaceId?: string;
-    authToken?: string;
-  }): PTYInstance {
-    const id = `pty-${++this.nextId}`;
-    const shell = options?.shell || this.getDefaultShell();
-    const cwd = options?.cwd || os.homedir();
-
-    // Filter out ELECTRON_ variables to prevent leaking internal state to child processes
-    const env: Record<string, string> = {};
-    for (const [key, value] of Object.entries(globalThis.process.env)) {
-      if (value === undefined) continue;
-      if (key.startsWith('ELECTRON_')) continue;
-      env[key] = value;
-    }
-    env[ENV_KEYS.SOCKET_PATH] = getPipeName();
-    if (options?.workspaceId) env[ENV_KEYS.WORKSPACE_ID] = options.workspaceId;
-    if (options?.surfaceId) env[ENV_KEYS.SURFACE_ID] = options.surfaceId;
-    const authToken = options?.authToken || this._authToken;
-    if (authToken) env[ENV_KEYS.AUTH_TOKEN] = authToken;
-
-    const process = pty.spawn(shell, [], {
-      name: 'xterm-256color',
-      cols: options?.cols || 80,
-      rows: options?.rows || 24,
-      cwd,
-      env,
-      useConpty: true,
-    });
-
-    const instance: PTYInstance = { id, process, shell };
-    this.instances.set(id, instance);
-    return instance;
-  }
-
-  write(id: string, data: string): void {
-    const instance = this.instances.get(id);
-    if (instance) {
-      instance.process.write(data);
-    }
-  }
-
-  resize(id: string, cols: number, rows: number): void {
-    const instance = this.instances.get(id);
-    if (instance) {
-      instance.process.resize(cols, rows);
-    }
-  }
-
-  dispose(id: string): void {
-    const instance = this.instances.get(id);
-    if (instance) {
-      try { instance.process.kill(); } catch { /* already dead */ }
-      this.instances.delete(id);
-    }
-  }
-
-  /** Remove an entry from the map without killing — use when the process has already exited. */
-  remove(id: string): void {
-    this.instances.delete(id);
-  }
-
-  get(id: string): PTYInstance | undefined {
-    return this.instances.get(id);
-  }
-
-  disposeAll(): void {
-    for (const [id] of this.instances) {
-      this.dispose(id);
-    }
-  }
-
-  private getDefaultShell(): string {
-    if (process.platform === 'win32') {
-      // PowerShell 우선 (cd로 드라이브 전환 자동 지원)
-      return 'powershell.exe';
-    }
-    return process.env.SHELL || '/bin/bash';
-  }
-}

package/src/main/pty/ShellDetector.ts

@@ -1,63 +0,0 @@
-import fs from 'node:fs';
-import path from 'node:path';
-
-export interface ShellInfo {
-  name: string;
-  path: string;
-  args?: string[];
-}
-
-export class ShellDetector {
-  detect(): ShellInfo[] {
-    const shells: ShellInfo[] = [];
-
-    // PowerShell 7+ (pwsh)
-    const pwshPaths = [
-      'C:\\Program Files\\PowerShell\\7\\pwsh.exe',
-      path.join(process.env.LOCALAPPDATA || '', 'Microsoft\\WindowsApps\\pwsh.exe'),
-    ];
-    for (const p of pwshPaths) {
-      if (fs.existsSync(p)) {
-        shells.push({ name: 'PowerShell 7', path: p });
-        break;
-      }
-    }
-
-    // Windows PowerShell 5.1
-    const ps5 = path.join(process.env.SystemRoot || 'C:\\Windows', 'System32\\WindowsPowerShell\\v1.0\\powershell.exe');
-    if (fs.existsSync(ps5)) {
-      shells.push({ name: 'Windows PowerShell', path: ps5 });
-    }
-
-    // Git Bash
-    const gitBashPaths = [
-      'C:\\Program Files\\Git\\bin\\bash.exe',
-      'C:\\Program Files (x86)\\Git\\bin\\bash.exe',
-    ];
-    for (const p of gitBashPaths) {
-      if (fs.existsSync(p)) {
-        shells.push({ name: 'Git Bash', path: p, args: ['--login', '-i'] });
-        break;
-      }
-    }
-
-    // WSL
-    const wslPath = path.join(process.env.SystemRoot || 'C:\\Windows', 'System32\\wsl.exe');
-    if (fs.existsSync(wslPath)) {
-      shells.push({ name: 'WSL', path: wslPath });
-    }
-
-    // cmd.exe
-    const cmd = process.env.COMSPEC || path.join(process.env.SystemRoot || 'C:\\Windows', 'System32\\cmd.exe');
-    if (fs.existsSync(cmd)) {
-      shells.push({ name: 'Command Prompt', path: cmd });
-    }
-
-    return shells;
-  }
-
-  getDefault(): string {
-    const shells = this.detect();
-    return shells.length > 0 ? shells[0].path : 'powershell.exe';
-  }
-}

package/src/main/session/SessionManager.ts

@@ -1,53 +0,0 @@
-import { app } from 'electron';
-import fs from 'node:fs';
-import path from 'node:path';
-import type { SessionData } from '../../shared/types';
-
-export class SessionManager {
-  private filePath: string;
-
-  constructor() {
-    this.filePath = path.join(app.getPath('userData'), 'session.json');
-  }
-
-  save(data: SessionData): void {
-    try {
-      const dir = path.dirname(this.filePath);
-      if (!fs.existsSync(dir)) {
-        fs.mkdirSync(dir, { recursive: true });
-      }
-      fs.writeFileSync(this.filePath, JSON.stringify(data, null, 2), 'utf-8');
-    } catch (err) {
-      console.error('Failed to save session:', err);
-    }
-  }
-
-  load(): SessionData | null {
-    try {
-      if (!fs.existsSync(this.filePath)) return null;
-      const raw = fs.readFileSync(this.filePath, 'utf-8');
-
-      // Guard against prototype pollution via JSON reviver
-      const parsed: unknown = JSON.parse(raw, (key, value) => {
-        if (key === '__proto__' || key === 'constructor' || key === 'prototype') return undefined;
-        return value;
-      });
-
-      // Basic schema validation
-      if (
-        typeof parsed !== 'object' ||
-        parsed === null ||
-        !Array.isArray((parsed as Record<string, unknown>)['workspaces']) ||
-        typeof (parsed as Record<string, unknown>)['activeWorkspaceId'] !== 'string'
-      ) {
-        console.warn('[SessionManager] Session file failed schema validation — discarding.');
-        return null;
-      }
-
-      return parsed as SessionData;
-    } catch (err) {
-      console.error('Failed to load session:', err);
-      return null;
-    }
-  }
-}