#!/bin/bash
# OpenClaw One-Click Installation Script
# Usage: curl -fsSL https://openclaw.ai/install | bash

set -euo pipefail

# ========== 颜色定义 =========
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# ========== 日志函数 =========
log_info() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

log_success() {
    echo -e "${GREEN}[✓]${NC} $1"
}

log_warning() {
    echo -e "${YELLOW}[⚠]${NC} $1"
}

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

# ========== 环境检测 =========
check_system() {
    log_info "检测系统环境..."
    
    if [[ ! -f /etc/os-release ]]; then
        log_error "不支持的操作系统，仅支持 Linux"
        exit 1
    fi
    
    source /etc/os-release
    OS_ID=$ID
    OS_VERSION=$VERSION_ID
    
    log_info "检测到系统：${OS_ID} ${OS_VERSION}"
    
    # 内存检测
    TOTAL_MEM=$(free -m | awk '/Mem:/ {print $2}')
    if [[ $TOTAL_MEM -lt 2048 ]]; then
        log_warning "内存仅${TOTAL_MEM}MB，建议至少 2GB"
    fi
}

# ========== Docker 安装 =========
install_docker() {
    log_info "检查 Docker 安装状态..."
    
    if command -v docker &> /dev/null; then
        log_success "Docker 已安装，跳过"
        return 0
    fi
    
    log_info "开始安装 Docker..."
    
    case $OS_ID in
        ubuntu|debian)
            apt-get update -qq
            apt-get install -y apt-transport-https ca-certificates curl gnupg lsb-release
            curl -fsSL https://download.docker.com/linux/$OS_ID/gpg | gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg
            echo "deb [arch=amd64 signed-by=/usr/share/keyrings/docker-archive-keyring.gpg] https://download.docker.com/linux/${OS_ID} ${VERSION_CODENAME} stable" > /etc/apt/sources.list.d/docker.list
            apt-get update -qq
            apt-get install -y docker-ce docker-ce-cli containerd.io
            ;;
        centos|rhel|fedora)
            yum install -y yum-utils
            yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
            yum install -y docker-ce docker-ce-cli containerd.io
            systemctl start docker
            ;;
        *)
            log_error "不支持的系统：${OS_ID}"
            exit 1
            ;;
    esac
    
    # 配置镜像加速（中国大陆）
    if [[ "${CN:-false}" == "true" ]]; then
        mkdir -p /etc/docker
        cat > /etc/docker/daemon.json <<EOF
{
  "registry-mirrors": [
    "https://docker.mirrors.ustc.edu.cn",
    "https://registry.docker-cn.com"
  ]
}
EOF
        systemctl daemon-reload
        systemctl restart docker
        log_success "已配置国内镜像加速"
    fi
    
    # 添加当前用户到 docker 组
    usermod -aG docker $SUDO_USER
    log_success "Docker 安装完成"
}

# ========== 交互式配置 =========
interactive_config() {
    log_info "开始配置 OpenClaw..."
    
    # API Key 输入
    echo -n "请输入 DeepSeek API Key: "
    read -s API_KEY
    echo
    
    if [[ ! $API_KEY =~ ^sk-[a-zA-Z0-9]{32,}$ ]]; then
        log_error "API Key 格式错误，应以 sk-开头且长度足够"
        exit 1
    fi
    
    # 端口选择
    echo -n "WebUI 端口 (默认 18789): "
    read WEBUI_PORT
    WEBUI_PORT=${WEBUI_PORT:-18789}
    
    # Token 生成
    ACCESS_TOKEN=$(openssl rand -hex 24)
    log_success "生成访问 Token: ${ACCESS_TOKEN:0:16}..."
    
    # 保存配置
    mkdir -p ~/.openclaw
    cat > ~/.openclaw/config.env <<EOF
DEEPSEEK_API_KEY=$API_KEY
WEBUI_PORT=$WEBUI_PORT
BRIDGE_PORT=18790
ACCESS_TOKEN=$ACCESS_TOKEN
EOF
    
    chmod 600 ~/.openclaw/config.env
    log_success "配置文件已保存至 ~/.openclaw/config.env"
}

# ========== Docker Compose 部署 =========
deploy_openclaw() {
    log_info "开始部署 OpenClaw..."
    
    source ~/.openclaw/config.env
    
    # 创建 docker-compose.yml
    cat > ~/.openclaw/docker-compose.yml <<EOF
version: '3.8'

services:
  openclaw-gateway:
    image: openclaw/gateway:latest
    container_name: openclaw-gateway
    ports:
      - "${BRIDGE_PORT}:18790"
    environment:
      - DEEPSEEK_API_KEY=${DEEPSEEK_API_KEY}
      - ACCESS_TOKEN=${ACCESS_TOKEN}
    volumes:
      - ./data:/app/data
    restart: unless-stopped
    networks:
      - openclaw-net

  openclaw-webui:
    image: openclaw/webui:latest
    container_name: openclaw-webui
    ports:
      - "${WEBUI_PORT}:3000"
    environment:
      - GATEWAY_URL=http://openclaw-gateway:18790
      - ACCESS_TOKEN=${ACCESS_TOKEN}
    depends_on:
      - openclaw-gateway
    restart: unless-stopped
    networks:
      - openclaw-net

networks:
  openclaw-net:
    driver: bridge
EOF
    
    # 启动服务
    cd ~/.openclaw
    docker compose up -d
    
    # 等待服务就绪
    log_info "等待服务启动..."
    sleep 10
    
    # 健康检查
    if curl -s http://localhost:$WEBUI_PORT/health > /dev/null; then
        log_success "OpenClaw 部署成功!"
        echo
        echo -e "${GREEN}==================================${NC}"
        echo -e "${GREEN}🎉 OpenClaw 已成功部署!${NC}"
        echo -e "${GREEN}==================================${NC}"
        echo
        echo -e "访问地址：http://$(hostname -I | awk '{print $1}'):${WEBUI_PORT}"
        echo -e "访问 Token: ${ACCESS_TOKEN}"
        echo
        echo -e "${YELLOW}请妥善保管访问 Token，用于登录 WebUI${NC}"
    else
        log_error "服务启动失败，请查看日志：docker logs openclaw-gateway"
        exit 1
    fi
}

# ========== 主流程 =========
main() {
    echo -e "${BLUE}"
    cat << 'BANNER'
   ____  _     _                           _ 
  / __ \| |   (_)                         | |
 | |  | | |__  _ _ __ ___  _ __   ___  ___| |_ 
 | |  | | '_ \| | '_ ` _ \| '_ \ / _ \/ __| __|
 | |__| | |_) | | | | | | | |_) |  __/\__ \ |_ 
  \____/|_.__/|_|_| |_| |_| .__/ \___||___/\__|
                          | |                  
                          |_|    One-Click Install
BANNER
    echo -e "${NC}"
    echo
    
    check_system
    install_docker
    interactive_config
    deploy_openclaw
    
    log_success "🎊 全部完成！享受你的 AI 助手吧!"
}

# 执行主函数
main "$@"