iterm_controller.sh

#!/bin/bash

# iTerm2 Controller Script
# Manages panes for Claude Code collaboration

SCRIPT_NAME="iterm_controller.sh"

show_help() {
    echo "iTerm2 Controller for Claude Code Collaboration"
    echo ""
    echo "Usage: $SCRIPT_NAME <command> [options]"
    echo ""
    echo "Commands:"
    echo "  split-current           - Split current window vertically"
    echo "  send <window> <pane> <cmd> - Send command to specific pane (1-indexed)"
    echo "  capture <window> <pane> [lines] - Capture last N lines (default 20)"
    echo "  list                    - List all windows and panes"
    echo "  focus <window> <pane>   - Focus specific pane"
    echo ""
    echo "Examples:"
    echo "  $SCRIPT_NAME split-current"
    echo "  $SCRIPT_NAME send 1 2 'cd /Users/j/Code/athena/ash_chat/tui_chat'"
    echo "  $SCRIPT_NAME send 1 2 'mix run -e \"TuiChat.CLI.main([])\"'"
    echo "  $SCRIPT_NAME capture 1 2 10"
    echo "  $SCRIPT_NAME send 1 2 $'\\003'  # Send Ctrl+C"
}

split_current_window() {
    local window_name="${1:-dev-window}"
    
    osascript <<EOF
tell application "iTerm2"
    tell current window
        tell current session
            -- Split vertically (right pane)
            set newSession to (split vertically with default profile)
        end tell
        
        -- Return window ID for reference
        return id
    end tell
end tell
EOF
}

send_command() {
    local window_index="$1"
    local pane_index="$2"
    local command="$3"
    
    if [[ -z "$window_index" || -z "$pane_index" || -z "$command" ]]; then
        echo "Error: send requires window_index, pane_index, and command"
        exit 1
    fi
    
    osascript <<EOF
tell application "iTerm2"
    tell window $window_index
        tell session $pane_index
            write text "$command"
        end tell
    end tell
end tell
EOF
}

capture_output() {
    local window_index="$1"
    local pane_index="$2"
    local lines="${3:-20}"
    
    if [[ -z "$window_index" || -z "$pane_index" ]]; then
        echo "Error: capture requires window_index and pane_index"
        exit 1
    fi
    
    osascript <<EOF
tell application "iTerm2"
    tell window $window_index
        tell session $pane_index
            get text of (lines -$lines thru -1)
        end tell
    end tell
end tell
EOF
}

list_sessions() {
    osascript <<EOF
tell application "iTerm2"
    set output to ""
    repeat with w from 1 to count of windows
        set currentWindow to window w
        set output to output & "Window " & w & ": " & (name of currentWindow) & "\\n"
        repeat with s from 1 to count of sessions of currentWindow
            set currentSession to session s of currentWindow
            set output to output & "  Session " & s & ": " & (name of currentSession) & "\\n"
        end repeat
    end repeat
    return output
end tell
EOF
}

focus_pane() {
    local window_index="$1"
    local pane_index="$2"
    
    if [[ -z "$window_index" || -z "$pane_index" ]]; then
        echo "Error: focus requires window_index and pane_index"
        exit 1
    fi
    
    osascript <<EOF
tell application "iTerm2"
    tell window $window_index
        tell session $pane_index
            select
        end tell
    end tell
end tell
EOF
}

# Main command dispatch
case "$1" in
    "split-current")
        echo "Splitting current window..."
        window_id=$(split_current_window "$2")
        echo "Created window: $window_id"
        echo "Pane 0: Claude Code (left)"
        echo "Pane 1: Available for commands (right)"
        ;;
    "send")
        send_command "$2" "$3" "$4"
        echo "Command sent to window '$2', pane $3"
        ;;
    "capture")
        capture_output "$2" "$3" "$4"
        ;;
    "list")
        list_sessions
        ;;
    "focus")
        focus_pane "$2" "$3"
        echo "Focused window '$2', pane $3"
        ;;
    "help"|"-h"|"--help"|"")
        show_help
        ;;
    *)
        echo "Unknown command: $1"
        echo "Use '$SCRIPT_NAME help' for usage information"
        exit 1
        ;;
esac