install.sh

#!/bin/bash
# Auto-fix Windows CRLF line endings (skip when piped via curl)
if [ -f "$0" ] && LC_ALL=C grep -q $'\r' "$0"; then
    tmp=$(mktemp); tr -d '\r' < "$0" > "$tmp"; chmod +x "$tmp"; exec bash "$tmp" "$@"
fi

# install.sh - TrinityClaw Installation Wizard (Linux/Mac)

echo ""
echo -e "\033[32m╔══════════════════════════════════════════════════════╗"
echo -e "║        TrinityClaw AI Agent - Installation Wizard    ║"
echo -e "╚══════════════════════════════════════════════════════╝\033[0m"
echo ""

# ─────────────────────────────────────────────────────────
# Detect platform
# ─────────────────────────────────────────────────────────
IS_MAC=false
if [[ "$(uname)" == "Darwin" ]]; then
    IS_MAC=true
    echo -e "   \033[36m🍎 macOS detected\033[0m"
    # Ensure brew bins are in PATH
    if [ -f /opt/homebrew/bin/brew ]; then
        eval "$(/opt/homebrew/bin/brew shellenv)"
    elif [ -f /usr/local/bin/brew ]; then
        eval "$(/usr/local/bin/brew shellenv)"
    fi
    export PATH="/opt/homebrew/bin:/usr/local/bin:$PATH"
    hash -r
fi

# ─────────────────────────────────────────────────────────
# [1/5] Check prerequisites
# ─────────────────────────────────────────────────────────
echo -e "\033[33m[1/5] Checking prerequisites...\033[0m"

# On Mac: start or install Docker Desktop if not running
if [ "$IS_MAC" = true ]; then
    if ! docker info &>/dev/null 2>&1; then
        # Try to start Docker Desktop if the app is installed
        if [ -d "/Applications/Docker.app" ]; then
            echo "   🚀 Starting Docker Desktop..."
            open /Applications/Docker.app
            echo "   ⏳ Waiting for Docker to start (up to 90 seconds)..."
            echo "      Watch for the whale 🐳 icon in your Mac menu bar."
            for i in $(seq 1 45); do
                sleep 2
                docker info &>/dev/null 2>&1 && break
                printf "   [%02d/45] still starting...\r" "$i"
            done
            echo ""
        fi
        if ! docker info &>/dev/null 2>&1; then
            echo ""
            if [ -d "/Applications/Docker.app" ]; then
                echo "   ❌ Docker did not start in time."
                echo "   👉 Open Docker Desktop manually from Applications,"
                echo "      wait for the whale 🐳 icon in the menu bar, then run:"
                echo ""
                echo "      bash install.sh"
            else
                echo "   ❌ Docker Desktop is not installed."
                echo ""
                echo "════════════════════════════════════════════════════════"
                echo "   📥  INSTALL DOCKER DESKTOP (it's free)"
                echo ""
                echo "   1. Go to:  https://www.docker.com/products/docker-desktop/"
                echo "   2. Click 'Download for Mac'"
                echo "   3. Open the .dmg file → drag Docker to Applications"
                echo "   4. Open Docker Desktop from Applications"
                echo "   5. Wait for the whale 🐳 icon in your menu bar"
                echo "   6. Come back here and run:"
                echo ""
                echo "         bash install.sh"
                echo ""
                echo "════════════════════════════════════════════════════════"
                open "https://www.docker.com/products/docker-desktop/" 2>/dev/null || true
            fi
            echo ""
            exit 1
        fi
    fi

# On Windows (Git Bash)
elif grep -qi microsoft /proc/version 2>/dev/null; then
    if ! command -v docker &>/dev/null; then
        echo "   ❌ Docker Desktop is not installed."
        echo "   👉 Download from: https://www.docker.com/products/docker-desktop/"
        echo "   Then restart Git Bash and re-run this script."
        exit 1
    fi

# On Linux
else
    if ! command -v docker &>/dev/null; then
        echo "   📥 Installing Docker Engine..."
        curl -fsSL https://get.docker.com | sh
        sudo systemctl enable --now docker
        sudo usermod -aG docker "$USER"
        newgrp docker
    fi
fi

echo "   ✅ Docker installed"

# Find docker-compose — check plugin, standalone, and known brew paths
DC=""
if docker compose version &>/dev/null 2>&1; then
    DC="docker compose"
elif command -v docker-compose &>/dev/null; then
    DC="docker-compose"
elif [ -f /opt/homebrew/bin/docker-compose ]; then
    DC="/opt/homebrew/bin/docker-compose"
elif [ -f /usr/local/bin/docker-compose ]; then
    DC="/usr/local/bin/docker-compose"
fi

# On Mac: install if still not found, then link as plugin
if [ -z "$DC" ] && [ "$IS_MAC" = true ]; then
    echo "   📥 Installing docker-compose..."
    brew install docker-compose
    export PATH="/opt/homebrew/bin:/usr/local/bin:$PATH"
    hash -r
    DC="/opt/homebrew/bin/docker-compose"
fi

if [ -z "$DC" ]; then
    echo "   ❌ Docker Compose not found."
    exit 1
fi

# Link as Docker CLI plugin if not already working
if ! docker compose version &>/dev/null 2>&1; then
    mkdir -p ~/.docker/cli-plugins
    COMPOSE_PATH="${DC##* }"  # get binary path (strip "docker " if present)
    [ -f "$COMPOSE_PATH" ] && ln -sfn "$COMPOSE_PATH" ~/.docker/cli-plugins/docker-compose 2>/dev/null || true
fi

echo "   ✅ Docker Compose ready"

# ─────────────────────────────────────────────────────────
# [1b/5] Download repository if not already present
# ─────────────────────────────────────────────────────────
if [ ! -f "./docker-compose.yml" ] && [ ! -f "./docker-compose.mac.yml" ]; then
    INSTALL_DIR="$HOME/trinity-claw"
    echo ""
    echo "   📥 Downloading TrinityClaw files to $INSTALL_DIR..."
    mkdir -p "$INSTALL_DIR"
    if curl -fsSL https://github.com/TrinityClaw/trinity-claw/archive/refs/heads/main.tar.gz \
        | tar -xz -C "$INSTALL_DIR" --strip-components=1; then
        echo "   ✅ Files ready at $INSTALL_DIR"
    else
        echo "   ❌ Download failed. Check your internet and try again."
        exit 1
    fi
    cd "$INSTALL_DIR"
    echo "   ✅ Working in $INSTALL_DIR"
fi

# ─────────────────────────────────────────────────────────
# [2/5] Choose model source
# ─────────────────────────────────────────────────────────
echo ""
echo -e "\033[33m[2/5] Model source...\033[0m"
echo ""
echo "   Options:"
echo "   [cloud]  Use a cloud provider (OpenAI, NVIDIA, Anthropic, etc.)"
echo "   [local]  Use a local Ollama model (qwen3.5:9b, ~6.6GB, no API key needed)"
echo ""
read -p "   Choose model source [cloud/local]: " model_source
model_source=$(echo "$model_source" | tr '[:upper:]' '[:lower:]' | xargs)

# ─────────────────────────────────────────────────────────
# [3/5] Configure environment
# ─────────────────────────────────────────────────────────
echo ""
echo -e "\033[33m[3/5] Configuring environment...\033[0m"

two_key=$(python3 -c "import secrets; print(secrets.token_hex(16))" 2>/dev/null)
[ -z "$two_key" ] && two_key=$(LC_ALL=C tr -dc 'a-zA-Z0-9' < /dev/urandom | head -c 32)

if [ "$model_source" = "local" ]; then

    echo ""
    echo -e "   \033[32m✅ Local mode selected — Ollama will be used.\033[0m"

    if [ "$IS_MAC" = true ]; then
        echo -e "   \033[36mℹ️  On Mac, Ollama runs natively for full Metal GPU acceleration.\033[0m"
        echo ""

        if ! command -v ollama &>/dev/null; then
            echo "   📥 Installing Ollama..."
            brew install ollama
            export PATH="/opt/homebrew/bin:/usr/local/bin:$PATH"
            hash -r
        else
            echo "   ✅ Ollama already installed"
        fi

        echo "   📥 Pulling qwen3.5:9b (~6.6GB, this may take several minutes)..."
        ollama pull qwen3.5:9b || { echo "   ❌ Failed to pull qwen3.5:9b. Check your internet and try again."; exit 1; }

        if ! curl -s http://localhost:11434/api/tags >/dev/null 2>&1; then
            echo "   🚀 Starting Ollama service..."
            ollama serve &>/dev/null &
            sleep 3
        else
            echo "   ✅ Ollama already running"
        fi

        OLLAMA_API_BASE="http://host.docker.internal:11434"
        COMPOSE_FILE="docker-compose.mac.yml"
    else
        echo -e "   \033[33mℹ️  The model 'qwen3.5:9b' (~6.6GB) will be pulled on first startup.\033[0m"
        OLLAMA_API_BASE="http://ollama:11434"
        COMPOSE_FILE="docker-compose.yml"
    fi

    cat > .env << EOF
# TrinityClaw Secrets
LITELLM_MASTER_KEY=sk-trinity-local-key
MODEL_SOURCE=local
OLLAMA_MODEL=qwen3.5:9b
TRINITY_API_KEY=${two_key}
EOF

    mkdir -p config
    cat > config/litellm_config.yaml << EOF
model_list:
  - model_name: trinity-default
    litellm_params:
      model: ollama/qwen3.5:9b
      api_base: ${OLLAMA_API_BASE}

  - model_name: trinity-vision
    litellm_params:
      model: ollama/qwen3.5:9b
      api_base: ${OLLAMA_API_BASE}

general_settings:
  master_key: os.environ/LITELLM_MASTER_KEY
EOF

    echo "   ✅ Local configuration created!"
    LOCAL_MODE=true

else

    echo ""
    echo -e "   \033[36mCloud API Configuration\033[0m"
    echo ""
    echo "   Provider examples:"
    echo "   - OpenAI:     openai/gpt-4o"
    echo "   - NVIDIA:     openai/moonshotai/kimi-k2-instruct"
    echo "   - Anthropic:  anthropic/claude-3-5-sonnet-20241022"
    echo ""
    read -p "   1. Model name: " model
    model=${model:-gpt-4o}

    echo ""
    echo "   API Base examples:"
    echo "   - OpenAI:  https://api.openai.com/v1"
    echo "   - NVIDIA:  https://integrate.api.nvidia.com/v1"
    echo ""
    read -p "   2. API Base URL (default: https://api.openai.com/v1): " api_base
    api_base=${api_base:-https://api.openai.com/v1}

    echo ""
    echo "   Example: NVIDIA_API_KEY, OPENAI_API_KEY, ANTHROPIC_API_KEY"
    read -p "   3. API Key variable name: " api_key_name
    api_key_name=${api_key_name:-OPENAI_API_KEY}

    echo ""
    read -s -p "   4. API Key value (hidden): " api_key_value
    echo ""

    cat > .env << EOF
# TrinityClaw Secrets
LITELLM_MASTER_KEY=sk-trinity-local-key
MODEL_SOURCE=cloud
${api_key_name}=${api_key_value}
TRINITY_API_KEY=${two_key}
EOF

    mkdir -p config
    cat > config/litellm_config.yaml << EOF
model_list:
  - model_name: trinity-default
    litellm_params:
      model: ${model}
      api_key: os.environ/${api_key_name}
      api_base: ${api_base}

  - model_name: trinity-vision
    litellm_params:
      model: ${model}
      api_key: os.environ/${api_key_name}
      api_base: ${api_base}

general_settings:
  master_key: os.environ/LITELLM_MASTER_KEY
EOF

    echo "   ✅ Cloud configuration created!"
    LOCAL_MODE=false
    COMPOSE_FILE="docker-compose.yml"
fi

# ── Append optional integrations template to .env (both modes) ───────────────
cat >> .env << 'OPTIONAL_KEYS'

# ── Optional integrations (uncomment and fill in to enable) ──────────────────

# Web search — Tavily (best results, free tier: 1000 searches/month)
# Get key at: https://tavily.com → Dashboard → API Keys
# Without this, search falls back to DuckDuckGo → Bing (may be rate-limited)
# TAVILY_API_KEY=tvly-xxxxx

# Telegram — chat with your agent via text, voice, and photos
# 1. Create bot: Telegram → @BotFather → /newbot → copy token
# 2. Get your Chat ID: Telegram → @userinfobot → send any message
# TELEGRAM_BOT_TOKEN=xxxxxx:xxxxxxxxxxxxxxxxxxxxxxxxx
# TELEGRAM_CHAT_ID=123456789

# Email sending — Gmail SMTP (needs an App Password, not your regular password)
# Enable 2FA at myaccount.google.com/security, then:
# Generate App Password at myaccount.google.com/apppasswords
# EMAIL_PROVIDER=smtp
# EMAIL_FROM=you@gmail.com
# SMTP_HOST=smtp.gmail.com
# SMTP_PORT=587
# SMTP_USER=you@gmail.com
# SMTP_PASSWORD=xxxx xxxx xxxx xxxx

# ─────────────────────────────────────────────────────────────────────────────
OPTIONAL_KEYS
echo "   ✅ .env created at $(pwd)/.env"
echo "      → Add Tavily, Telegram, SMTP and other optional keys there anytime."

# ─────────────────────────────────────────────────────────
# [4/5] Building containers
# ─────────────────────────────────────────────────────────
echo ""
echo -e "\033[33m[4/5] Building containers...\033[0m"
$DC -f "$COMPOSE_FILE" build

# ─────────────────────────────────────────────────────────
# [5/5] Starting services
# ─────────────────────────────────────────────────────────
echo ""
echo -e "\033[33m[5/5] Starting services...\033[0m"

if [ "$LOCAL_MODE" = true ] && [ "$IS_MAC" = false ]; then
    $DC -f "$COMPOSE_FILE" --profile local up -d
else
    $DC -f "$COMPOSE_FILE" up -d
fi

# ─────────────────────────────────────────────────────────
# Done
# ─────────────────────────────────────────────────────────
echo ""
echo -e "\033[32m✅ TrinityClaw Installed!\033[0m"
echo ""
echo "   🌐 Web UI:  http://localhost:8080"
echo "   🔌 API:     http://localhost:8001"
echo "   📖 Docs:    http://localhost:8001/docs"
echo ""
# Save the key to a plain text file so it is never lost
echo "${two_key}" > trinity-key.txt

echo "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "   🔑  AGENT API KEY — copy and save this:"
echo ""
echo "       ${two_key}"
echo ""
echo "   Enter it in: Settings ⚙️ → Agent Security → Agent API Key"
echo "   (also saved to trinity-key.txt in this folder)"
echo "   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

if [ "$LOCAL_MODE" = true ] && [ "$IS_MAC" = false ]; then
    echo -e "   \033[33m⚠️  qwen3.5:9b (~6.6GB) is downloading in the background.\033[0m"
    echo -e "   \033[33m   Monitor: docker logs trinity-claw-ollama-init-1 -f\033[0m"
    echo ""
fi

if [ "$IS_MAC" = true ]; then
    INSTALL_PATH="$(pwd)"

    # ── 1. Auto-start Ollama on boot ──────────────────────────────────────────
    if [ "$LOCAL_MODE" = true ]; then
        echo "   🔄 Configuring Ollama to start automatically on boot..."
        brew services start ollama &>/dev/null && \
            echo "   ✅ Ollama will start automatically after every reboot" || true
    fi

    # ── 2. Enable Docker Desktop start at login ───────────────────────────────
    echo "   🔄 Enabling Docker Desktop auto-start at login..."
    defaults write com.docker.docker autoStart -bool true 2>/dev/null || true
    for settings_path in \
        "$HOME/Library/Group Containers/group.com.docker/settings-store.json" \
        "$HOME/Library/Group Containers/group.com.docker/settings.json"; do
        if [ -f "$settings_path" ]; then
            python3 -c "
import json, sys
path = sys.argv[1]
with open(path) as f: s = json.load(f)
s['autoStart'] = True
with open(path, 'w') as f: json.dump(s, f, indent=2)
" "$settings_path" 2>/dev/null && echo "   ✅ Docker Desktop set to start at login" || true
            break
        fi
    done

    # ── 3. Create startup.sh (called by LaunchAgent on every login) ───────────
    cat > "$INSTALL_PATH/startup.sh" << 'STARTUP_SCRIPT'
#!/bin/bash
# Wait for Docker Desktop to be ready (up to 2.5 minutes)
for i in $(seq 1 30); do
    docker info &>/dev/null 2>&1 && break
    sleep 5
done
STARTUP_SCRIPT
    echo "cd \"$INSTALL_PATH\" && docker compose -f \"$COMPOSE_FILE\" up -d" \
        >> "$INSTALL_PATH/startup.sh"
    chmod +x "$INSTALL_PATH/startup.sh"

    # ── 4. Install LaunchAgent (auto-runs startup.sh on every login) ──────────
    PLIST_DIR="$HOME/Library/LaunchAgents"
    mkdir -p "$PLIST_DIR"
    cat > "$PLIST_DIR/com.trinityclaw.autostart.plist" << PLIST_END
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
    <key>Label</key>
    <string>com.trinityclaw.autostart</string>
    <key>ProgramArguments</key>
    <array>
        <string>/bin/bash</string>
        <string>${INSTALL_PATH}/startup.sh</string>
    </array>
    <key>RunAtLoad</key>
    <true/>
    <key>StandardOutPath</key>
    <string>/tmp/trinityclaw-startup.log</string>
    <key>StandardErrorPath</key>
    <string>/tmp/trinityclaw-error.log</string>
</dict>
</plist>
PLIST_END
    launchctl load "$PLIST_DIR/com.trinityclaw.autostart.plist" 2>/dev/null || true
    echo "   ✅ LaunchAgent installed — containers auto-start after every login"

    # ── 5. Create double-clickable Desktop launcher (manual backup) ───────────
    LAUNCHER="$HOME/Desktop/Start TrinityClaw.command"
    cat > "$LAUNCHER" << LAUNCHER_END
#!/bin/bash
echo "🚀 Starting TrinityClaw..."
if ! docker info &>/dev/null 2>&1; then
    echo "   Opening Docker Desktop — please wait..."
    open /Applications/Docker.app
    echo "   Waiting for Docker to be ready..."
    for i in \$(seq 1 45); do
        sleep 2
        docker info &>/dev/null 2>&1 && break
        printf "."
    done
    echo ""
fi
cd "${INSTALL_PATH}"
docker compose -f "${COMPOSE_FILE}" up -d
echo ""
echo "✅ TrinityClaw is running!"
echo "   Opening http://localhost:8080 ..."
sleep 2
open http://localhost:8080
LAUNCHER_END
    chmod +x "$LAUNCHER"
    echo "   ✅ Desktop launcher created: ~/Desktop/Start TrinityClaw.command"

    echo ""
    echo -e "   \033[32m🎉 After every Mac restart, TrinityClaw starts AUTOMATICALLY!\033[0m"
    echo -e "   \033[36m   Docker Desktop starts → containers start → ready in ~30 sec\033[0m"
    echo -e "   \033[36m   Need a manual restart? Double-click 'Start TrinityClaw' on your Desktop.\033[0m"
    echo ""
    echo -e "   \033[33m⚙️  To add Tavily, Telegram, email or other optional integrations:\033[0m"
    echo -e "   \033[33m   Edit: ${INSTALL_PATH}/.env\033[0m"
    echo -e "   \033[33m   Then restart: docker compose -f ${COMPOSE_FILE} restart\033[0m"
    echo ""
fi

echo -e "   \033[33m🎤 Voice: Whisper (~150MB) downloads on first voice message.\033[0m"
echo -e "   \033[33m📸 Vision: uses trinity-vision model in litellm_config.yaml.\033[0m"
echo -e "   \033[33m🌐 Browser: Playwright + Chromium installed automatically.\033[0m"
echo ""