#!/bin/bash
# ============================================================
# U-Claw - Portable AI Agent
# Double-click to start / 双击启动
# ============================================================

UCLAW_DIR="$(cd "$(dirname "$0")" && pwd)"
APP_DIR="$UCLAW_DIR/app"
CORE_DIR="$APP_DIR/core"
DATA_DIR="$UCLAW_DIR/data"
SYSTEM_DIR="$UCLAW_DIR/system"

GREEN='\033[0;32m'
CYAN='\033[0;36m'
RED='\033[0;31m'
YELLOW='\033[1;33m'
NC='\033[0m'

clear
echo ""
echo -e "${CYAN}"
echo "  ╔══════════════════════════════════════╗"
echo "  ║     🦞 U-Claw v1.1                  ║"
echo "  ║     Portable AI Agent               ║"
echo "  ╚══════════════════════════════════════╝"
echo -e "${NC}"

# ---- 1. Detect CPU & set runtime ----
ARCH=$(uname -m)
if [ "$ARCH" = "arm64" ]; then
    NODE_DIR="$APP_DIR/runtime/node-mac-arm64"
    echo -e "  ${GREEN}Apple Silicon (M series)${NC}"
else
    echo -e "  ${RED}This version only supports Apple Silicon (M1-M4).${NC}"
    echo -e "  ${RED}Intel Mac is not supported in this release.${NC}"
    echo ""
    read -p "  Press Enter to exit..."
    exit 1
fi

NODE_BIN="$NODE_DIR/bin/node"
export PATH="$NODE_DIR/bin:$PATH"

# ---- 2. Remove macOS quarantine ----
if xattr -l "$NODE_BIN" 2>/dev/null | grep -q "com.apple.quarantine"; then
    echo -e "  ${YELLOW}Removing macOS security restriction...${NC}"
    xattr -rd com.apple.quarantine "$UCLAW_DIR" 2>/dev/null || true
    echo -e "  ${GREEN}Done${NC}"
fi

# ---- 3. Check runtime ----
if [ ! -f "$NODE_BIN" ]; then
    echo -e "  ${RED}Error: Node.js runtime not found${NC}"
    echo "  Please ensure app/runtime/ is complete"
    read -p "  Press Enter to exit..."
    exit 1
fi

NODE_VER=$("$NODE_BIN" --version)
echo -e "  Node.js: ${GREEN}${NODE_VER}${NC}"
echo ""

# ---- 4. Check & init data ----
mkdir -p "$DATA_DIR/memory" "$DATA_DIR/backups" "$DATA_DIR/logs"

if [ ! -f "$DATA_DIR/config.json" ] && [ ! -f "$DATA_DIR/.openclaw/openclaw.json" ]; then
    echo -e "  ${YELLOW}First run - creating default config...${NC}"
    mkdir -p "$DATA_DIR/.openclaw"
    cat > "$DATA_DIR/.openclaw/openclaw.json" << 'CFGEOF'
{
  "gateway": {
    "mode": "local",
    "auth": { "token": "uclaw" }
  }
}
CFGEOF
    echo -e "  ${GREEN}Config created${NC}"
    echo ""
fi

# ---- 5. Set environment (portable mode) ----
STATE_DIR="$DATA_DIR/.openclaw"
mkdir -p "$STATE_DIR"

# Sync config to where OpenClaw expects it
if [ -f "$DATA_DIR/config.json" ] && [ ! -f "$STATE_DIR/openclaw.json" ]; then
    cp "$DATA_DIR/config.json" "$STATE_DIR/openclaw.json"
fi

export OPENCLAW_HOME="$DATA_DIR"
export OPENCLAW_STATE_DIR="$STATE_DIR"
export OPENCLAW_CONFIG_PATH="$STATE_DIR/openclaw.json"

# ---- 6. Run migration if exists ----
if [ -f "$SYSTEM_DIR/migrate.js" ]; then
    "$NODE_BIN" "$SYSTEM_DIR/migrate.js" "$DATA_DIR" 2>/dev/null || true
fi

# ---- 7. Check dependencies ----
if [ ! -d "$CORE_DIR/node_modules" ]; then
    echo -e "  ${YELLOW}First run - installing dependencies...${NC}"
    echo "  (Using China mirror)"
    cd "$CORE_DIR"
    "$NODE_BIN" "$NODE_DIR/bin/npm" install --registry=https://registry.npmmirror.com 2>&1
    echo -e "  ${GREEN}Dependencies installed${NC}"
    echo ""
fi

# OpenClaw doesn't need a separate build step when installed via npm

# ---- 8. Find available port ----
PORT=18789
while lsof -i :$PORT >/dev/null 2>&1; do
    echo -e "  ${YELLOW}Port $PORT in use, trying next...${NC}"
    PORT=$((PORT + 1))
    if [ $PORT -gt 18799 ]; then
        echo -e "  ${RED}No available port (18789-18799)${NC}"
        read -p "  Press Enter to exit..."
        exit 1
    fi
done

# Update config with correct port if changed
if [ $PORT -ne 18789 ]; then
    "$NODE_BIN" -e "
        const fs = require('fs');
        const p = '$DATA_DIR/config.json';
        const c = JSON.parse(fs.readFileSync(p, 'utf8'));
        c.gateway = c.gateway || {};
        c.gateway.port = $PORT;
        fs.writeFileSync(p, JSON.stringify(c, null, 2));
    " 2>/dev/null || true
fi

# ---- 9. Start gateway ----
echo -e "  ${CYAN}Starting OpenClaw on port $PORT...${NC}"
echo "  Do NOT close this window."
echo ""

cd "$CORE_DIR"

TOKEN=$(python3 -c "import json,os; p='$STATE_DIR/openclaw.json' if os.path.exists('$STATE_DIR/openclaw.json') else '$DATA_DIR/config.json'; d=json.load(open(p)); print(d.get('gateway',{}).get('auth',{}).get('token','uclaw'))" 2>/dev/null || echo "uclaw")

OPENCLAW_MJS="$CORE_DIR/node_modules/openclaw/openclaw.mjs"
"$NODE_BIN" "$OPENCLAW_MJS" gateway run --allow-unconfigured --force --port $PORT &
GW_PID=$!

# ---- 10. Wait & open browser ----
for i in $(seq 1 30); do
    sleep 0.5
    if curl -s -o /dev/null -w '' "http://127.0.0.1:$PORT/" 2>/dev/null; then
        DASHBOARD_URL="http://127.0.0.1:$PORT/#token=${TOKEN}"
        CONFIG_PAGE="$UCLAW_DIR/Config.html?port=$PORT"
        echo ""
        echo -e "  ${GREEN}✅ Started successfully!${NC}"
        echo ""

        # Check if model is configured
        HAS_MODEL=$(python3 -c "
import json,os
for p in ['$STATE_DIR/openclaw.json','$DATA_DIR/config.json']:
    if os.path.exists(p):
        d=json.load(open(p))
        if d.get('agent',{}).get('model'):
            print('yes'); break
" 2>/dev/null)

        if [ "$HAS_MODEL" = "yes" ]; then
            echo -e "  ${CYAN}Dashboard: ${DASHBOARD_URL}${NC}"
            echo ""
            open "$DASHBOARD_URL" 2>/dev/null
        else
            echo -e "  ${YELLOW}首次使用，打开配置页面...${NC}"
            echo -e "  ${CYAN}配置页面: Config.html${NC}"
            echo -e "  ${CYAN}控制台: ${DASHBOARD_URL}${NC}"
            echo ""
            open "$CONFIG_PAGE" 2>/dev/null
        fi
        break
    fi
done

wait $GW_PID

echo ""
echo -e "  ${YELLOW}OpenClaw stopped.${NC}"
read -p "  Press Enter to close..."