@colmbus72/yeehaw 0.1.0

package/dist/lib/tmux.js

@@ -0,0 +1,397 @@
+import { execaSync } from 'execa';
+import { spawnSync } from 'child_process';
+import { fileURLToPath } from 'url';
+import { dirname, join } from 'path';
+import { writeTmuxConfig, TMUX_CONFIG_PATH } from './tmux-config.js';
+import { shellEscape } from './shell.js';
+// Get the path to the MCP server (it's in dist/, not dist/lib/)
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+const MCP_SERVER_PATH = join(__dirname, '..', 'mcp-server.js');
+export const YEEHAW_SESSION = 'yeehaw';
+// Remote mode state tracking
+let remoteWindowIndex = null;
+// Keys to unbind when entering remote mode (so they pass through to inner tmux)
+const REMOTE_MODE_UNBIND_KEYS = ['C-h', 'C-l', 'C-y'];
+export function hasTmux() {
+    try {
+        execaSync('which', ['tmux']);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+export function isInsideYeehawSession() {
+    const tmuxEnv = process.env.TMUX;
+    if (!tmuxEnv)
+        return false;
+    // Check if we're in the yeehaw session
+    try {
+        const result = execaSync('tmux', ['display-message', '-p', '#{session_name}']);
+        return result.stdout.trim() === YEEHAW_SESSION;
+    }
+    catch {
+        return false;
+    }
+}
+export function yeehawSessionExists() {
+    try {
+        execaSync('tmux', ['has-session', '-t', YEEHAW_SESSION]);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+export function createYeehawSession() {
+    // Write the tmux config
+    writeTmuxConfig();
+    // Create the session with window 0 named "yeehaw"
+    execaSync('tmux', [
+        'new-session',
+        '-d',
+        '-s', YEEHAW_SESSION,
+        '-n', 'yeehaw',
+    ]);
+    // Source the config
+    execaSync('tmux', ['source-file', TMUX_CONFIG_PATH]);
+    // Set up hook to hide status bar in window 0
+    setupStatusBarHooks();
+}
+export function setupStatusBarHooks() {
+    // Hide status bar when in window 0, show in other windows
+    try {
+        // Start with status off (we begin in window 0)
+        execaSync('tmux', ['set', '-t', YEEHAW_SESSION, 'status', 'off']);
+        // Hook for when window changes
+        execaSync('tmux', [
+            'set-hook', '-t', YEEHAW_SESSION,
+            'after-select-window',
+            'if-shell -F "#{==:#{window_index},0}" "set status off" "set status on"'
+        ]);
+    }
+    catch {
+        // Hooks might fail on older tmux versions, not critical
+    }
+}
+export function attachToYeehaw() {
+    // This replaces the current process
+    spawnSync('tmux', ['attach-session', '-t', YEEHAW_SESSION], {
+        stdio: 'inherit',
+    });
+    process.exit(0);
+}
+// All yeehaw MCP tools that should be auto-approved
+const YEEHAW_MCP_TOOLS = [
+    'mcp__yeehaw__list_projects',
+    'mcp__yeehaw__get_project',
+    'mcp__yeehaw__create_project',
+    'mcp__yeehaw__update_project',
+    'mcp__yeehaw__delete_project',
+    'mcp__yeehaw__add_livestock',
+    'mcp__yeehaw__remove_livestock',
+    'mcp__yeehaw__list_barns',
+    'mcp__yeehaw__get_barn',
+    'mcp__yeehaw__create_barn',
+    'mcp__yeehaw__update_barn',
+    'mcp__yeehaw__delete_barn',
+    'mcp__yeehaw__get_wiki',
+    'mcp__yeehaw__add_wiki_section',
+    'mcp__yeehaw__update_wiki_section',
+    'mcp__yeehaw__delete_wiki_section',
+];
+export function createClaudeWindow(workingDir, windowName) {
+    // Build MCP config for yeehaw server
+    const mcpConfig = JSON.stringify({
+        mcpServers: {
+            yeehaw: {
+                command: 'node',
+                args: [MCP_SERVER_PATH],
+            },
+        },
+    });
+    // Build allowed tools list for auto-approval
+    const allowedTools = YEEHAW_MCP_TOOLS.join(',');
+    // Create new window running claude with yeehaw MCP server (-a appends after current window)
+    // Use shell escaping to safely handle special characters in JSON
+    const claudeCmd = `claude --mcp-config ${shellEscape(mcpConfig)} --allowedTools ${shellEscape(allowedTools)}`;
+    execaSync('tmux', [
+        'new-window',
+        '-a',
+        '-t', YEEHAW_SESSION,
+        '-n', windowName,
+        '-c', workingDir,
+        claudeCmd,
+    ]);
+    // Get the window index we just created (new window is now current)
+    const result = execaSync('tmux', [
+        'display-message', '-p', '#{window_index}'
+    ]);
+    return parseInt(result.stdout.trim(), 10);
+}
+export function createShellWindow(workingDir, windowName, shell = '/bin/zsh') {
+    // Create new window running shell (-a appends after current window)
+    execaSync('tmux', [
+        'new-window',
+        '-a',
+        '-t', YEEHAW_SESSION,
+        '-n', windowName,
+        '-c', workingDir,
+        shell,
+    ]);
+    // Get the window index we just created (new window is now current)
+    const result = execaSync('tmux', [
+        'display-message', '-p', '#{window_index}'
+    ]);
+    return parseInt(result.stdout.trim(), 10);
+}
+export function createSshWindow(windowName, host, user, port, identityFile, remotePath) {
+    // Two levels of escaping needed:
+    // 1. Remote shell sees: cd /path && exec $SHELL -l
+    // 2. Local shell (tmux) sees: ssh ... -t 'cd /path && ...'
+    // Build the remote command - escape remotePath for the remote shell
+    const remoteCmd = `cd ${shellEscape(remotePath)} && exec $SHELL -l`;
+    // Build SSH command parts - escape for local shell (tmux passes to sh)
+    const sshParts = [
+        'ssh',
+        '-p', String(port),
+        '-i', shellEscape(identityFile),
+        shellEscape(`${user}@${host}`),
+        '-t',
+        shellEscape(remoteCmd) // Double-escaped: once for remote, once for local
+    ];
+    const sshCmd = sshParts.join(' ');
+    execaSync('tmux', [
+        'new-window',
+        '-a',
+        '-t', YEEHAW_SESSION,
+        '-n', windowName,
+        sshCmd,
+    ]);
+    const result = execaSync('tmux', [
+        'display-message', '-p', '#{window_index}'
+    ]);
+    return parseInt(result.stdout.trim(), 10);
+}
+export function detachFromSession() {
+    execaSync('tmux', ['detach-client']);
+}
+export function killYeehawSession() {
+    try {
+        execaSync('tmux', ['kill-session', '-t', YEEHAW_SESSION]);
+    }
+    catch {
+        // Session might already be dead
+    }
+}
+export function switchToWindow(windowIndex) {
+    execaSync('tmux', ['select-window', '-t', `${YEEHAW_SESSION}:${windowIndex}`]);
+}
+export function listYeehawWindows() {
+    try {
+        // Use tab as delimiter since pane_title can contain colons
+        const result = execaSync('tmux', [
+            'list-windows',
+            '-t', YEEHAW_SESSION,
+            '-F', '#{window_index}\t#{window_name}\t#{window_active}\t#{pane_title}\t#{pane_current_command}\t#{window_activity}',
+        ]);
+        return result.stdout
+            .split('\n')
+            .filter(Boolean)
+            .map((line) => {
+            const [index, name, active, paneTitle, paneCurrentCommand, windowActivity] = line.split('\t');
+            return {
+                index: parseInt(index, 10),
+                name,
+                active: active === '1',
+                paneTitle: paneTitle || '',
+                paneCurrentCommand: paneCurrentCommand || '',
+                windowActivity: parseInt(windowActivity, 10) || 0,
+            };
+        });
+    }
+    catch {
+        return [];
+    }
+}
+export function killWindow(windowIndex) {
+    try {
+        execaSync('tmux', ['kill-window', '-t', `${YEEHAW_SESSION}:${windowIndex}`]);
+    }
+    catch {
+        // Window might already be dead
+    }
+}
+/**
+ * Format relative time from a Unix timestamp
+ */
+function formatRelativeTime(timestamp) {
+    const now = Math.floor(Date.now() / 1000);
+    const diff = now - timestamp;
+    if (diff < 60)
+        return 'now';
+    if (diff < 3600)
+        return `${Math.floor(diff / 60)}m`;
+    if (diff < 86400)
+        return `${Math.floor(diff / 3600)}h`;
+    return `${Math.floor(diff / 86400)}d`;
+}
+/**
+ * Check if a pane title indicates Claude is actively working (has spinner)
+ */
+function isClaudeWorking(paneTitle) {
+    // Braille spinner characters used by Claude Code
+    const spinnerChars = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏', '⠂', '⠐', '⠈'];
+    return spinnerChars.some(char => paneTitle.startsWith(char));
+}
+/**
+ * Get formatted status text for a tmux window
+ */
+export function getWindowStatus(window) {
+    const isClaudeSession = window.name.includes('claude') && window.paneCurrentCommand === 'node';
+    const relativeTime = formatRelativeTime(window.windowActivity);
+    if (isClaudeSession) {
+        // For Claude sessions, use pane title (contains task context)
+        if (window.paneTitle) {
+            const working = isClaudeWorking(window.paneTitle);
+            if (working) {
+                return window.paneTitle; // Shows spinner + task description
+            }
+            // Not working - show title with idle time if stale
+            if (relativeTime !== 'now' && relativeTime !== '1m') {
+                return `${window.paneTitle} (${relativeTime})`;
+            }
+            return window.paneTitle;
+        }
+        return relativeTime === 'now' ? 'active' : `idle ${relativeTime}`;
+    }
+    // For shell sessions, show current command
+    const cmd = window.paneCurrentCommand;
+    if (cmd && cmd !== 'zsh' && cmd !== 'bash' && cmd !== 'sh' && cmd !== 'fish') {
+        // Running a specific command
+        return cmd;
+    }
+    // At shell prompt - show idle time
+    return relativeTime === 'now' ? 'ready' : `idle ${relativeTime}`;
+}
+export function updateStatusBar(projectName) {
+    const left = projectName
+        ? `#[bold] YEEHAW | ${projectName} `
+        : '#[bold] YEEHAW ';
+    try {
+        execaSync('tmux', ['set', '-t', YEEHAW_SESSION, 'status-left', left]);
+    }
+    catch {
+        // Not critical if this fails
+    }
+}
+export function enterRemoteMode(barnName, host, user, port, identityFile) {
+    // 1. Create SSH window that attaches to remote yeehaw tmux
+    const windowName = `remote:${barnName}`;
+    const remoteCmd = 'tmux attach -t yeehaw';
+    const sshParts = [
+        'ssh',
+        '-p', String(port),
+        '-i', shellEscape(identityFile),
+        '-t', // Force TTY allocation
+        shellEscape(`${user}@${host}`),
+        shellEscape(remoteCmd)
+    ];
+    const sshCmd = sshParts.join(' ');
+    execaSync('tmux', [
+        'new-window',
+        '-a',
+        '-t', YEEHAW_SESSION,
+        '-n', windowName,
+        sshCmd,
+    ]);
+    const result = execaSync('tmux', [
+        'display-message', '-p', '#{window_index}'
+    ]);
+    const windowIndex = parseInt(result.stdout.trim(), 10);
+    remoteWindowIndex = windowIndex;
+    // 2. Unbind navigation keys so they pass through to inner tmux
+    for (const key of REMOTE_MODE_UNBIND_KEYS) {
+        try {
+            execaSync('tmux', ['unbind-key', '-n', key]);
+        }
+        catch {
+            // Key might not be bound, ignore
+        }
+    }
+    // 3. Bind Ctrl-\ as escape hatch
+    try {
+        execaSync('tmux', [
+            'bind-key', '-n', 'C-\\',
+            `run-shell "tmux kill-window -t ${YEEHAW_SESSION}:${windowIndex}; exit 0"`
+        ]);
+    }
+    catch {
+        // Ignore errors
+    }
+    // 4. Show minimal status bar with connection info
+    try {
+        execaSync('tmux', ['set', '-t', YEEHAW_SESSION, 'status', 'on']);
+        execaSync('tmux', ['set', '-t', YEEHAW_SESSION, 'status-left', `#[bold] Connected to: ${barnName} `]);
+        execaSync('tmux', ['set', '-t', YEEHAW_SESSION, 'status-right', ' Ctrl-\\ back ']);
+    }
+    catch {
+        // Not critical
+    }
+    // 5. Set up hook to restore when remote window closes
+    try {
+        execaSync('tmux', [
+            'set-hook', '-t', YEEHAW_SESSION,
+            'window-unlinked',
+            `if-shell "[ ! -z \\"#{@remote_mode}\\" ]" "run-shell \\"tmux set -u @remote_mode; tmux source-file ${TMUX_CONFIG_PATH}; tmux select-window -t ${YEEHAW_SESSION}:0; tmux set status off\\""`
+        ]);
+        // Mark that we're in remote mode
+        execaSync('tmux', ['set', '-t', YEEHAW_SESSION, '@remote_mode', '1']);
+    }
+    catch {
+        // Hooks might fail on older tmux
+    }
+    return windowIndex;
+}
+export function exitRemoteMode() {
+    // Kill the remote window if it exists
+    if (remoteWindowIndex !== null) {
+        try {
+            execaSync('tmux', ['kill-window', '-t', `${YEEHAW_SESSION}:${remoteWindowIndex}`]);
+        }
+        catch {
+            // Window might already be dead
+        }
+        remoteWindowIndex = null;
+    }
+    // Restore keybindings by re-sourcing config
+    try {
+        execaSync('tmux', ['source-file', TMUX_CONFIG_PATH]);
+    }
+    catch {
+        // Not critical
+    }
+    // Unbind the escape hatch
+    try {
+        execaSync('tmux', ['unbind-key', '-n', 'C-\\']);
+    }
+    catch {
+        // Ignore
+    }
+    // Hide status bar and switch to window 0
+    try {
+        execaSync('tmux', ['set', '-t', YEEHAW_SESSION, 'status', 'off']);
+        execaSync('tmux', ['select-window', '-t', `${YEEHAW_SESSION}:0`]);
+        execaSync('tmux', ['set', '-u', '-t', YEEHAW_SESSION, '@remote_mode']);
+    }
+    catch {
+        // Not critical
+    }
+}
+export function isInRemoteMode() {
+    return remoteWindowIndex !== null;
+}
+export function getRemoteWindowIndex() {
+    return remoteWindowIndex;
+}

package/dist/mcp-server.d.ts

@@ -0,0 +1,23 @@
+#!/usr/bin/env node
+/**
+ * Yeehaw MCP Server
+ *
+ * Exposes Yeehaw project/barn/livestock management to Claude Code sessions.
+ *
+ * Terminology:
+ * - Project: A codebase you're working on
+ * - Barn: A server you manage
+ * - Livestock: Deployed instances of your apps (local or on barns)
+ * - Critter: System services that support livestock (nginx, mysql, etc.)
+ *
+ * Usage: Add to Claude Code's MCP config:
+ * {
+ *   "mcpServers": {
+ *     "yeehaw": {
+ *       "command": "node",
+ *       "args": ["/path/to/yeehaw/dist/mcp-server.js"]
+ *     }
+ *   }
+ * }
+ */
+export {};