agent

#!/bin/bash
# Simplified agent manager for parallel development with named agents

COMMAND=${1}
AGENT_NAME=${2}

# Load agent config
CONFIG_FILE="config/agents.json"

# Get agents workspace path
if command -v jq &> /dev/null && [ -f "$CONFIG_FILE" ]; then
    AGENTS_WORKSPACE_PATH=$(jq -r ".agentsWorkspacePath" "$CONFIG_FILE" | sed "s|~|$HOME|g")
else
    AGENTS_WORKSPACE_PATH="$HOME/Developer/agent54-agents"
fi

# Colors for agents
COLORS=("#00ff00" "#ff00ff" "#00ffff" "#ff8c00" "#ff0000" "#ffff00" "#9400d3" "#0000ff" "#7fff00" "#ff1493" "#40e0d0" "#ffd700")
NAMES=("Lime Green" "Hot Pink" "Cyan" "Dark Orange" "Bright Red" "Yellow" "Violet" "Blue" "Chartreuse" "Deep Pink" "Turquoise" "Gold")

# Hash agent name to get consistent color
hash_to_color_index() {
    local name=$1
    local hash=0
    for (( i=0; i<${#name}; i++ )); do
        hash=$(( (hash * 31 + $(printf "%d" "'${name:$i:1}")) % ${#COLORS[@]} ))
    done
    echo $hash
}

# Get agent color info based on name
get_agent_color() {
    local name=$1
    local color_index=$(hash_to_color_index "$name")
    echo "COLOR=${COLORS[$color_index]}"
    echo "COLOR_NAME=\"${NAMES[$color_index]}\""
}

# Calculate ports based on agent name hash
calculate_ports() {
    local name=$1
    local hash=$(hash_to_color_index "$name")
    # Ensure unique ports by using hash and name length
    local port_offset=$(( (hash * 10 + ${#name}) % 20 ))
    echo "VITE_PORT=$((1430 + (port_offset * 10)))"
    echo "MCP_PORT=$((3457 + port_offset))"
}

# Normalize agent name (handle shortcuts like "podman" -> find "plugin-podman")
normalize_agent_name() {
    local input=$1
    
    # If already has a type prefix, use as-is
    if [[ "$input" =~ ^(plugin|docs|feature|fix)- ]]; then
        echo "$input"
        return
    fi
    
    # Look for existing agents that match the shorthand
    if [ -d "$AGENTS_WORKSPACE_PATH" ]; then
        for dir in "$AGENTS_WORKSPACE_PATH"/agent-*; do
            if [ -d "$dir" ]; then
                local full_name=$(basename "$dir" | sed 's/^agent-//')
                # Check if the full name ends with the input
                if [[ "$full_name" == *"-$input" ]]; then
                    echo "$full_name"
                    return
                fi
            fi
        done
    fi
    
    # No match found, return as-is
    echo "$input"
}

# Get agent type from name
get_agent_type() {
    local name=$1
    if [[ "$name" =~ ^plugin- ]]; then
        echo "plugin"
    elif [[ "$name" =~ ^docs- ]]; then
        echo "docs"
    elif [[ "$name" =~ ^feature- ]]; then
        echo "feature"
    elif [[ "$name" =~ ^fix- ]]; then
        echo "fix"
    else
        echo "general"
    fi
}

# Get agent task from task file
get_agent_task() {
    local workspace_dir=$1
    local task_file="$workspace_dir/.agent/task.md"
    
    if [ -f "$task_file" ]; then
        task=$(cat "$task_file")
        if [ -n "$task" ] && [ "$task" != "[No task assigned yet]" ]; then
            echo "$task"
        fi
    fi
}

case $COMMAND in
  start)
    if [ -z "$AGENT_NAME" ]; then
      echo "Usage: ./agent start <name> [task]"
      echo "Examples:"
      echo "  ./agent start plugin-podman"
      echo "  ./agent start docs-review \"Review API documentation\""
      echo "  ./agent start feature-auth"
      exit 1
    fi
    
    # Get optional task from remaining arguments
    TASK="${@:3}"
    
    # Normalize the name
    AGENT_NAME=$(normalize_agent_name "$AGENT_NAME")
    WORKSPACE_DIR="$AGENTS_WORKSPACE_PATH/agent-$AGENT_NAME"
    
    # Create workspace if it doesn't exist
    if [ ! -d "$WORKSPACE_DIR" ]; then
      echo "🔧 Creating Agent '$AGENT_NAME' workspace..."
      ./scripts/agents/create-agent-workspace.sh "$AGENT_NAME"
    fi
    
    # Navigate to workspace and start
    cd "$WORKSPACE_DIR" || exit 1
    
    # Set task if provided
    if [ -n "$TASK" ]; then
      mkdir -p "$WORKSPACE_DIR/.agent"
      echo "$TASK" > "$WORKSPACE_DIR/.agent/task.md"
      echo "✅ Task set: $TASK"
      echo ""
    fi
    
    # Get ports and color info
    eval $(calculate_ports "$AGENT_NAME")
    export VITE_PORT MCP_PORT
    
    # Get color info
    eval $(get_agent_color "$AGENT_NAME")
    
    echo "🚀 Starting Agent '$AGENT_NAME'"
    echo "🎨 Color: $COLOR_NAME"
    echo "📁 Type: $(get_agent_type "$AGENT_NAME")"
    echo "📡 Vite: http://localhost:$VITE_PORT"
    echo "🔌 MCP: http://localhost:$MCP_PORT/sse"
    
    # Show current task if set
    STORED_TASK=$(get_agent_task "$WORKSPACE_DIR")
    if [ -n "$STORED_TASK" ]; then
        echo "📋 Task: $STORED_TASK"
    fi
    
    echo ""
    echo "💡 Run 'claude' in your editor to start planning"
    echo "🚀 Start dev server with: bun tauri dev"
    ;;
    
  stop)
    if [ -z "$AGENT_NAME" ]; then
      echo "Usage: ./agent stop <name>"
      exit 1
    fi
    
    AGENT_NAME=$(normalize_agent_name "$AGENT_NAME")
    
    # Just stop the running processes, keep workspace
    eval $(calculate_ports "$AGENT_NAME")
    
    if lsof -ti:$VITE_PORT >/dev/null 2>&1; then
      lsof -ti:$VITE_PORT | xargs kill -9 2>/dev/null || true
      lsof -ti:$MCP_PORT | xargs kill -9 2>/dev/null || true
      echo "✅ Agent '$AGENT_NAME' stopped"
    else
      echo "⚠️  Agent '$AGENT_NAME' was not running"
    fi
    ;;
    
  kill)
    if [ -z "$AGENT_NAME" ]; then
      echo "Usage: ./agent kill <name>"
      exit 1
    fi
    
    AGENT_NAME=$(normalize_agent_name "$AGENT_NAME")
    WORKSPACE_DIR="$AGENTS_WORKSPACE_PATH/agent-$AGENT_NAME"
    
    # Kill any running processes for this agent
    eval $(calculate_ports "$AGENT_NAME")
    lsof -ti:$VITE_PORT | xargs kill -9 2>/dev/null || true
    lsof -ti:$MCP_PORT | xargs kill -9 2>/dev/null || true
    
    # Remove workspace
    git worktree remove "$WORKSPACE_DIR" --force 2>/dev/null || true
    rm -rf "$WORKSPACE_DIR"
    
    # Clean up branches
    git branch -D "agent-$AGENT_NAME-main" 2>/dev/null || true
    
    echo "✅ Agent '$AGENT_NAME' killed and cleaned up"
    ;;
    
  task)
    if [ -z "$AGENT_NAME" ] || [ -z "$3" ]; then
      echo "Usage: ./agent task <name> \"task description\""
      exit 1
    fi
    
    AGENT_NAME=$(normalize_agent_name "$AGENT_NAME")
    TASK="${@:3}"
    WORKSPACE_DIR="$AGENTS_WORKSPACE_PATH/agent-$AGENT_NAME"
    
    if [ ! -d "$WORKSPACE_DIR" ]; then
      echo "❌ Agent '$AGENT_NAME' workspace doesn't exist"
      echo "Create it with: ./agent start $AGENT_NAME"
      exit 1
    fi
    
    # Simply write to task file
    mkdir -p "$WORKSPACE_DIR/.agent"
    echo "$TASK" > "$WORKSPACE_DIR/.agent/task.md"
    
    echo "✅ Agent '$AGENT_NAME' task set: $TASK"
    ;;
    
  status)
    echo "🤖 Agent Status"
    echo "=============="
    echo ""
    
    # List active agents with details
    if [ -d "$AGENTS_WORKSPACE_PATH" ]; then
      for dir in "$AGENTS_WORKSPACE_PATH"/agent-*; do
        if [ -d "$dir" ]; then
          AGENT_NAME=$(basename "$dir" | sed 's/^agent-//')
          eval $(calculate_ports "$AGENT_NAME")
          eval $(get_agent_color "$AGENT_NAME")
          
          # Check if running
          if lsof -ti:$VITE_PORT >/dev/null 2>&1; then
            STATUS="🟢 Running"
          else
            STATUS="⭕ Stopped"
          fi
          
          # Get current branch
          BRANCH=$(cd "$dir" && git rev-parse --abbrev-ref HEAD 2>/dev/null || echo "unknown")
          
          # Get task
          TASK=$(get_agent_task "$dir")
          if [ -z "$TASK" ]; then
            TASK="No task set"
          fi
          
          echo "Agent '$AGENT_NAME' ($COLOR_NAME)"
          echo "  Type: $(get_agent_type "$AGENT_NAME")"
          echo "  Status: $STATUS"
          echo "  Ports: Vite=$VITE_PORT, MCP=$MCP_PORT"
          echo "  Branch: $BRANCH"
          echo "  Task: $TASK"
          echo ""
        fi
      done
    else
      echo "No agent workspaces found"
      echo ""
      echo "Create your first agent with: ./agent start plugin-example"
    fi
    ;;
  
  open)
    if [ -z "$AGENT_NAME" ]; then
      echo "Usage: ./agent open <name>"
      exit 1
    fi
    
    AGENT_NAME=$(normalize_agent_name "$AGENT_NAME")
    WORKSPACE_DIR="$AGENTS_WORKSPACE_PATH/agent-$AGENT_NAME"
    
    if [ ! -d "$WORKSPACE_DIR" ]; then
      echo "❌ Agent '$AGENT_NAME' workspace doesn't exist"
      echo "Create it with: ./agent start $AGENT_NAME"
      exit 1
    fi
    
    # Get agent info
    eval $(get_agent_color "$AGENT_NAME")
    eval $(calculate_ports "$AGENT_NAME")
    
    # Open in Cursor/VSCode
    if command -v cursor &> /dev/null; then
      echo "🚀 Opening Agent '$AGENT_NAME' ($COLOR_NAME) in Cursor..."
      cursor "$WORKSPACE_DIR/xe-launcher-v2-agent-$AGENT_NAME.code-workspace"
      echo ""
      echo "📝 To start Claude in plan mode:"
      echo "   1. Open terminal in Cursor (Cmd+J)"
      echo "   2. Run: claude"
      echo ""
      echo "Agent '$AGENT_NAME' is configured with:"
      echo "  - Type: $(get_agent_type "$AGENT_NAME")"
      echo "  - Ports: Vite=$VITE_PORT, MCP=$MCP_PORT"
      echo "  - Task: Check .agent/task.md"
    elif command -v code &> /dev/null; then
      echo "🚀 Opening Agent '$AGENT_NAME' in VSCode..."
      code "$WORKSPACE_DIR/xe-launcher-v2-agent-$AGENT_NAME.code-workspace"
    else
      echo "❌ Neither Cursor nor VSCode found"
    fi
    ;;
    
  list)
    # Simple list of available agents
    echo "Available agents:"
    if [ -d "$AGENTS_WORKSPACE_PATH" ]; then
      for dir in "$AGENTS_WORKSPACE_PATH"/agent-*; do
        if [ -d "$dir" ]; then
          AGENT_NAME=$(basename "$dir" | sed 's/^agent-//')
          eval $(get_agent_color "$AGENT_NAME")
          echo "  $AGENT_NAME - $COLOR_NAME ($(get_agent_type "$AGENT_NAME"))"
        fi
      done
    else
      echo "  None"
    fi
    ;;
    
  pr)
    if [ -z "$AGENT_NAME" ]; then
      echo "Usage: ./agent pr <name> [title]"
      exit 1
    fi
    
    AGENT_NAME=$(normalize_agent_name "$AGENT_NAME")
    WORKSPACE_DIR="$AGENTS_WORKSPACE_PATH/agent-$AGENT_NAME"
    
    if [ ! -d "$WORKSPACE_DIR" ]; then
      echo "❌ Agent '$AGENT_NAME' workspace doesn't exist"
      exit 1
    fi
    
    # Get agent color
    eval $(get_agent_color "$AGENT_NAME")
    
    cd "$WORKSPACE_DIR" || exit 1
    
    # Get current branch
    CURRENT_BRANCH=$(git rev-parse --abbrev-ref HEAD)
    
    # Get task for default title
    TASK=$(get_agent_task "$WORKSPACE_DIR")
    
    # Map color names to emojis
    case "$COLOR_NAME" in
      "Lime Green") COLOR_EMOJI="🟢" ;;
      "Hot Pink") COLOR_EMOJI="🩷" ;;
      "Cyan") COLOR_EMOJI="🔵" ;;
      "Dark Orange") COLOR_EMOJI="🟠" ;;
      "Bright Red") COLOR_EMOJI="🔴" ;;
      "Yellow") COLOR_EMOJI="🟡" ;;
      "Violet") COLOR_EMOJI="🟣" ;;
      "Blue") COLOR_EMOJI="🔷" ;;
      "Chartreuse") COLOR_EMOJI="💚" ;;
      "Deep Pink") COLOR_EMOJI="💖" ;;
      "Turquoise") COLOR_EMOJI="🟦" ;;
      "Gold") COLOR_EMOJI="🟨" ;;
      *) COLOR_EMOJI="⚪" ;;
    esac
    
    DEFAULT_TITLE="$COLOR_EMOJI [$COLOR_NAME] ${TASK:-$AGENT_NAME changes}"
    TITLE="${3:-$DEFAULT_TITLE}"
    
    # Push current branch
    echo "📤 Pushing $CURRENT_BRANCH..."
    git push -u origin "$CURRENT_BRANCH"
    
    # Create PR with color in title
    echo "🔄 Creating pull request..."
    gh pr create --title "$TITLE" --body "Changes from Agent '$AGENT_NAME' ($COLOR_NAME)

Task: ${TASK:-No specific task assigned}
Branch: $CURRENT_BRANCH
Type: $(get_agent_type "$AGENT_NAME")

---
Created by $COLOR_NAME Agent ($AGENT_NAME)"
    ;;
    
  claude)
    if [ -z "$AGENT_NAME" ]; then
      echo "Usage: ./agent claude <name>"
      echo "Example: ./agent claude podman"
      exit 1
    fi
    
    AGENT_NAME=$(normalize_agent_name "$AGENT_NAME")
    WORKSPACE_DIR="$AGENTS_WORKSPACE_PATH/agent-$AGENT_NAME"
    
    if [ ! -d "$WORKSPACE_DIR" ]; then
      echo "❌ Agent '$AGENT_NAME' workspace doesn't exist"
      echo "Create it with: ./agent start $AGENT_NAME"
      exit 1
    fi
    
    # Get agent info for context
    eval $(get_agent_color "$AGENT_NAME")
    eval $(calculate_ports "$AGENT_NAME")
    TASK=$(get_agent_task "$WORKSPACE_DIR")
    
    # Change to workspace directory
    cd "$WORKSPACE_DIR" || exit 1
    
    echo "🤖 Starting Claude Code for Agent '$AGENT_NAME' ($COLOR_NAME)"
    echo "📋 Task: ${TASK:-No task assigned}"
    echo ""
    
    # Start Claude with agent context
    $HOME/.claude/local/claude --permission-mode plan "You are Agent '$AGENT_NAME' ($COLOR_NAME).

Your mission: ${TASK:-Check .agent/task.md}

⚡ WORKFLOW REMINDER:
1. Research thoroughly before exiting plan mode
2. Stay focused on your assigned scope  
3. Keep your branch rebased with main
4. Test before marking complete
5. You're part of a team - respect boundaries

Key info:
- Type: $(get_agent_type "$AGENT_NAME")
- Vite port: $VITE_PORT
- MCP port: $MCP_PORT  
- Branch: agent-$AGENT_NAME-main

Read docs/AGENT_WORKFLOW.md for your complete workflow guide."
    ;;
    
  reset-all)
    echo "🔥 WARNING: This will remove ALL agent workspaces and branches!"
    echo ""
    
    # List what will be removed
    if [ -d "$AGENTS_WORKSPACE_PATH" ]; then
      echo "Workspaces to remove:"
      for dir in "$AGENTS_WORKSPACE_PATH"/agent-*; do
        if [ -d "$dir" ]; then
          AGENT_NAME=$(basename "$dir" | sed 's/^agent-//')
          eval $(get_agent_color "$AGENT_NAME")
          echo "  - Agent '$AGENT_NAME' ($COLOR_NAME)"
        fi
      done
      echo ""
    fi
    
    # Check for --yes flag
    if [[ "$2" == "--yes" || "$2" == "-y" ]]; then
      REPLY="yes"
    else
      read -p "Are you sure? Type 'yes' to confirm: " -r
    fi
    
    if [[ $REPLY == "yes" ]]; then
      echo ""
      echo "🧹 Cleaning up all agents..."
      
      # Stop all running agents
      for port in $(seq 1430 10 1630); do
        lsof -ti:$port | xargs kill -9 2>/dev/null || true
      done
      
      # Remove all agent workspaces
      if [ -d "$AGENTS_WORKSPACE_PATH" ]; then
        for dir in "$AGENTS_WORKSPACE_PATH"/agent-*; do
          if [ -d "$dir" ]; then
            AGENT_NAME=$(basename "$dir" | sed 's/^agent-//')
            echo "Removing Agent '$AGENT_NAME'..."
            git worktree remove "$dir" --force 2>/dev/null || true
            rm -rf "$dir"
          fi
        done
      fi
      
      # Clean up all agent branches
      echo "🔧 Cleaning up agent branches..."
      git branch | grep -E "agent-.*-main" | xargs -n 1 git branch -D 2>/dev/null || true
      
      # Clear agent registry
      echo '{"agents": {}, "lastUpdated": ""}' > .agent-registry.json
      
      # Prune worktree list
      git worktree prune
      
      echo "✅ All agents have been reset!"
    else
      echo "❌ Reset cancelled"
    fi
    ;;
    
  *)
    echo "🤖 Agent Manager - Named Agents"
    echo "==============================="
    echo ""
    echo "Commands:"
    echo "  ./agent start <name> [task]  Start agent with optional task"
    echo "  ./agent stop <name>          Stop agent (keeps workspace)"
    echo "  ./agent kill <name>          Remove agent and workspace"
    echo "  ./agent task <name> \"desc\"   Set agent task"
    echo "  ./agent open <name>          Open in editor (shortcuts work: podman → plugin-podman)"
    echo "  ./agent claude <name>        Start Claude Code in agent workspace"
    echo "  ./agent pr <name> [title]    Create PR with color emoji + label"
    echo "  ./agent status               Show all agents"
    echo "  ./agent list                 List agent names"
    echo "  ./agent reset-all [-y]       Remove ALL agents"
    echo ""
    echo "Agent Types:"
    echo "  plugin-*   Plugin development (e.g. plugin-podman, plugin-ollama)"
    echo "  docs-*     Documentation (e.g. docs-review, docs-api)"
    echo "  feature-*  Feature development (e.g. feature-auth, feature-ui)"
    echo "  fix-*      Bug fixes (e.g. fix-build-loop)"
    echo ""
    echo "Example workflow:"
    echo "  ./agent start plugin-podman \"Implement Podman container management\""
    echo "  ./agent open podman       # shortcut works!"
    echo "  ./agent claude podman     # start Claude in agent context"
    echo "  # ... make changes ..."
    echo "  ./agent pr podman"
    echo ""
    echo "Or set task separately:"
    echo "  ./agent start docs-api"
    echo "  ./agent task docs-api \"Document all API endpoints\""
    echo ""
    echo "Agents start in plan mode - run 'claude' in editor terminal"
    ;;
esac