install.sh

#!/usr/bin/env bash
# OpenCode Agent Team Installer
# https://github.com/BrightEra/opencode-agent-team
# MIT License - Justin Carlson / BrightEra

set -euo pipefail

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
CYAN='\033[0;36m'
MAGENTA='\033[0;35m'
NC='\033[0m' # No Color

# Configuration
INSTALL_DIR="${HOME}/.config/opencode"
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
VERSION="1.0.0"
BACKUP_DIR="${HOME}/.config/opencode.backup.$(date +%Y%m%d_%H%M%S)"

# Files to install (relative to SCRIPT_DIR)
declare -a FILES_TO_INSTALL=(
    "opencode.json"
    "AGENTS.md"
    ".opencode/prompts/plan.md"
    ".opencode/prompts/build.md"
)

#==============================================================================
# Output Functions
#==============================================================================

print_banner() {
    echo -e "${CYAN}"
    cat << "EOF"
   ___                   ____          _        
  / _ \ _ __   ___ _ __ / ___|___   __| | ___   
 | | | | '_ \ / _ \ '_ \| |   / _ \ / _` |/ _ \  
 | |_| | |_) |  __/ | | | |__| (_) | (_| |  __/  
  \___/| .__/ \___|_| |_|\____\___/ \__,_|\___|  
       |_|                                        
    _                    _     _____                     
   / \   __ _  ___ _ __ | |_  |_   _|__  __ _ _ __ ___  
  / _ \ / _` |/ _ \ '_ \| __|   | |/ _ \/ _` | '_ ` _ \ 
 / ___ \ (_| |  __/ | | | |_    | |  __/ (_| | | | | | |
/_/   \_\__, |\___|_| |_|\__|   |_|\___|\__,_|_| |_| |_|
        |___/                                            
EOF
    echo -e "${NC}"
    echo -e "${BLUE}Version ${VERSION}${NC}"
    echo -e "${BLUE}https://github.com/BrightEra/opencode-agent-team${NC}"
    echo ""
}

print_success() {
    echo -e "${GREEN}✓${NC} $1"
}

print_error() {
    echo -e "${RED}✗${NC} $1" >&2
}

print_warning() {
    echo -e "${YELLOW}⚠${NC} $1"
}

print_info() {
    echo -e "${BLUE}ℹ${NC} $1"
}

print_step() {
    echo -e "${MAGENTA}▸${NC} $1"
}

#==============================================================================
# Utility Functions
#==============================================================================

check_prerequisites() {
    print_step "Checking prerequisites..."
    
    # Check if opencode CLI is installed
    if command -v opencode &> /dev/null; then
        local version=$(opencode --version 2>/dev/null || echo "unknown")
        print_success "OpenCode CLI found (${version})"
    else
        print_warning "OpenCode CLI not found"
        echo ""
        echo -e "${YELLOW}The OpenCode CLI is required to use this agent team.${NC}"
        echo -e "Install it from: ${CYAN}https://opencode.ai${NC}"
        echo ""
        read -p "Continue installation anyway? (y/N) " -n 1 -r
        echo
        if [[ ! $REPLY =~ ^[Yy]$ ]]; then
            print_error "Installation cancelled"
            exit 1
        fi
    fi
    
    # Check if git is available (for better context)
    if command -v git &> /dev/null; then
        print_success "Git found"
    else
        print_warning "Git not found (optional, but recommended)"
    fi
    
    echo ""
}

backup_existing() {
    if [ -d "$INSTALL_DIR" ]; then
        print_step "Backing up existing configuration..."
        
        # Create backup directory
        mkdir -p "$BACKUP_DIR"
        
        # Copy existing files
        cp -r "$INSTALL_DIR"/* "$BACKUP_DIR/" 2>/dev/null || true
        
        print_success "Backup created at: ${BACKUP_DIR}"
        echo ""
    fi
}

verify_source_files() {
    print_step "Verifying source files..."
    
    local missing_files=()
    for file in "${FILES_TO_INSTALL[@]}"; do
        if [ ! -f "$SCRIPT_DIR/$file" ] && [ ! -d "$SCRIPT_DIR/$file" ]; then
            missing_files+=("$file")
        fi
    done
    
    if [ ${#missing_files[@]} -gt 0 ]; then
        print_error "Missing source files:"
        for file in "${missing_files[@]}"; do
            echo "  - $file"
        done
        exit 1
    fi
    
    print_success "All source files found"
    echo ""
}

#==============================================================================
# Installation Functions
#==============================================================================

install_copy() {
    print_step "Installing files (copy mode)..."
    echo ""
    
    # Create installation directory
    mkdir -p "$INSTALL_DIR"
    mkdir -p "$INSTALL_DIR/.opencode/prompts"
    
    # Copy each file
    for file in "${FILES_TO_INSTALL[@]}"; do
        local source="$SCRIPT_DIR/$file"
        local target="$INSTALL_DIR/$file"
        local target_dir=$(dirname "$target")
        
        # Create target directory if needed
        mkdir -p "$target_dir"
        
        # Copy file
        if cp "$source" "$target"; then
            print_success "Copied: $file"
        else
            print_error "Failed to copy: $file"
            exit 1
        fi
    done
    
    echo ""
    print_success "Installation complete (copy mode)"
}

install_symlink() {
    print_step "Installing files (symlink mode)..."
    echo ""
    
    # Create installation directory
    mkdir -p "$INSTALL_DIR"
    mkdir -p "$INSTALL_DIR/.opencode/prompts"
    
    # Create symlinks for each file
    for file in "${FILES_TO_INSTALL[@]}"; do
        local source="$SCRIPT_DIR/$file"
        local target="$INSTALL_DIR/$file"
        local target_dir=$(dirname "$target")
        
        # Create target directory if needed
        mkdir -p "$target_dir"
        
        # Remove existing file/symlink if present
        if [ -e "$target" ] || [ -L "$target" ]; then
            rm -f "$target"
        fi
        
        # Create symlink
        if ln -s "$source" "$target"; then
            print_success "Linked: $file → $source"
        else
            print_error "Failed to link: $file"
            exit 1
        fi
    done
    
    echo ""
    print_success "Installation complete (symlink mode)"
    print_info "Files are linked to: $SCRIPT_DIR"
    print_info "Pull updates with: cd $SCRIPT_DIR && git pull"
}

uninstall() {
    print_step "Uninstalling OpenCode Agent Team..."
    echo ""
    
    if [ ! -d "$INSTALL_DIR" ]; then
        print_warning "No installation found at: $INSTALL_DIR"
        exit 0
    fi
    
    # Show what will be removed
    echo -e "${YELLOW}The following will be removed:${NC}"
    echo "  - $INSTALL_DIR/opencode.json"
    echo "  - $INSTALL_DIR/AGENTS.md"
    echo "  - $INSTALL_DIR/.opencode/prompts/"
    echo ""
    
    # Confirm
    read -p "Are you sure you want to uninstall? (y/N) " -n 1 -r
    echo
    if [[ ! $REPLY =~ ^[Yy]$ ]]; then
        print_info "Uninstall cancelled"
        exit 0
    fi
    
    # Create backup before removing
    backup_existing
    
    # Remove installed files
    for file in "${FILES_TO_INSTALL[@]}"; do
        local target="$INSTALL_DIR/$file"
        if [ -e "$target" ] || [ -L "$target" ]; then
            rm -f "$target"
            print_success "Removed: $file"
        fi
    done
    
    # Remove empty directories
    rmdir "$INSTALL_DIR/.opencode/prompts" 2>/dev/null || true
    rmdir "$INSTALL_DIR/.opencode" 2>/dev/null || true
    
    # Check if INSTALL_DIR is empty (except for other user files)
    if [ -z "$(ls -A $INSTALL_DIR 2>/dev/null)" ]; then
        rmdir "$INSTALL_DIR"
        print_success "Removed directory: $INSTALL_DIR"
    else
        print_info "Kept directory (contains other files): $INSTALL_DIR"
    fi
    
    echo ""
    print_success "Uninstall complete"
    print_info "Backup available at: $BACKUP_DIR"
}

#==============================================================================
# Post-Install Instructions
#==============================================================================

show_post_install() {
    echo ""
    echo -e "${GREEN}╔════════════════════════════════════════════════════════════════╗${NC}"
    echo -e "${GREEN}║                   Installation Successful!                     ║${NC}"
    echo -e "${GREEN}╚════════════════════════════════════════════════════════════════╝${NC}"
    echo ""
    
    echo -e "${CYAN}📋 Next Steps:${NC}"
    echo ""
    
    echo -e "${YELLOW}1. Set up your Anthropic API key${NC}"
    echo "   The agent team requires an Anthropic API key to function."
    echo ""
    echo "   Option A - Environment variable (recommended):"
    echo -e "   ${BLUE}export ANTHROPIC_API_KEY='your-api-key-here'${NC}"
    echo ""
    echo "   Option B - Secrets file:"
    echo -e "   ${BLUE}echo 'ANTHROPIC_API_KEY=your-api-key-here' >> ~/.secrets${NC}"
    echo ""
    
    echo -e "${YELLOW}2. (Optional) Set up enhanced search capabilities${NC}"
    echo "   For better documentation search and code context:"
    echo ""
    echo "   Exa API (web search & code context):"
    echo -e "   ${BLUE}export EXA_API_KEY='your-exa-key'${NC}"
    echo -e "   Get a key at: ${CYAN}https://exa.ai${NC}"
    echo ""
    echo "   Ref API (documentation search):"
    echo -e "   ${BLUE}export REF_API_KEY='your-ref-key'${NC}"
    echo -e "   Get a key at: ${CYAN}https://ref.tools${NC}"
    echo ""
    
    echo -e "${YELLOW}3. Start using the agent team${NC}"
    echo "   Navigate to any project directory and run:"
    echo -e "   ${BLUE}opencode${NC}"
    echo ""
    echo "   Or start with a specific agent:"
    echo -e "   ${BLUE}opencode --agent plan${NC}"
    echo -e "   ${BLUE}opencode --agent build${NC}"
    echo ""
    
    echo -e "${CYAN}📚 Available Agents:${NC}"
    echo ""
    echo "  Primary Agents:"
    echo "    @plan     - Strategic planning and coordination (Opus 4.5)"
    echo "    @build    - Full implementation with all tools (Sonnet 4.5)"
    echo ""
    echo "  Specialist Agents:"
    echo "    @scout    - Fast codebase exploration (Haiku 4.5)"
    echo "    @research - Documentation and API lookup"
    echo "    @reviewer - Code quality review"
    echo "    @tester   - Test creation and execution"
    echo "    @refactor - Code structure improvement"
    echo "    @debug    - Bug investigation"
    echo "    @devops   - Infrastructure and CI/CD"
    echo "    @security - Security auditing"
    echo "    @docs     - Documentation writing (Haiku 4.5)"
    echo ""
    
    echo -e "${CYAN}💡 Quick Tips:${NC}"
    echo ""
    echo "  • Start with @plan for complex tasks - it will coordinate subagents"
    echo "  • Use @build (or Tab) when you're ready to write code"
    echo "  • Agents can work in parallel for independent tasks"
    echo "  • Check AGENTS.md for workflow patterns and best practices"
    echo ""
    
    echo -e "${CYAN}📖 Documentation:${NC}"
    echo "  • Agent guide: $INSTALL_DIR/AGENTS.md"
    echo "  • Configuration: $INSTALL_DIR/opencode.json"
    echo "  • Project repo: https://github.com/BrightEra/opencode-agent-team"
    echo ""
    
    echo -e "${GREEN}Happy coding! 🚀${NC}"
    echo ""
}

#==============================================================================
# Help
#==============================================================================

show_help() {
    cat << EOF
OpenCode Agent Team Installer v${VERSION}

USAGE:
    ./install.sh [MODE]

MODES:
    --copy      Copy files to ~/.config/opencode (default)
                Files are independent of the source repository.
                
    --symlink   Create symlinks to source files
                Files stay in sync with the repository.
                Useful for development or easy updates via git pull.
                
    --uninstall Remove the installation
                Creates a backup before removing files.
                
    --help      Show this help message

EXAMPLES:
    # Install with copy (default)
    ./install.sh
    ./install.sh --copy
    
    # Install with symlinks (for easy updates)
    ./install.sh --symlink
    
    # Remove installation
    ./install.sh --uninstall

INSTALLATION DIRECTORY:
    ${INSTALL_DIR}

FILES INSTALLED:
    - opencode.json              (Agent configuration)
    - AGENTS.md                  (Workflow guide)
    - .opencode/prompts/plan.md  (Plan agent prompt)
    - .opencode/prompts/build.md (Build agent prompt)

REQUIREMENTS:
    - OpenCode CLI (https://opencode.ai)
    - Anthropic API key

OPTIONAL:
    - Exa API key (enhanced search)
    - Ref API key (documentation search)

MORE INFO:
    https://github.com/BrightEra/opencode-agent-team

EOF
}

#==============================================================================
# Main
#==============================================================================

main() {
    local mode="${1:-copy}"
    
    # Parse arguments
    case "$mode" in
        --copy|copy)
            print_banner
            check_prerequisites
            verify_source_files
            backup_existing
            install_copy
            show_post_install
            ;;
        --symlink|symlink)
            print_banner
            check_prerequisites
            verify_source_files
            backup_existing
            install_symlink
            show_post_install
            ;;
        --uninstall|uninstall)
            print_banner
            uninstall
            ;;
        --help|help|-h)
            show_help
            ;;
        *)
            print_error "Unknown mode: $mode"
            echo ""
            echo "Run './install.sh --help' for usage information"
            exit 1
            ;;
    esac
}

# Run main function
main "$@"