#!/bin/bash
# ============================================================
# U-Claw 虾盘 - macOS 安装到电脑
# 将 OpenClaw 及运行环境永久安装到电脑上
# ============================================================

UCLAW_DIR="$(cd "$(dirname "$0")" && pwd)"
INSTALL_DIR="$HOME/.uclaw"

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 安装到电脑               ║"
echo "  ║     OpenClaw 永久安装                ║"
echo "  ╚══════════════════════════════════════╝"
echo -e "${NC}"
echo ""
echo "  安装位置: $INSTALL_DIR"
echo "  安装内容: Node.js + OpenClaw + 所有依赖"
echo ""
read -p "  是否继续安装? (y/n) " -n 1 -r
echo ""
echo ""

if [[ ! $REPLY =~ ^[Yy]$ ]]; then
    echo "  已取消安装"
    exit 0
fi

ARCH=$(uname -m)
if [ "$ARCH" = "arm64" ]; then
    NODE_SRC="$UCLAW_DIR/runtime/node-mac-arm64"
    echo -e "  ${GREEN}Apple Silicon (M系列)${NC}"
else
    NODE_SRC="$UCLAW_DIR/runtime/node-mac-x64"
    echo -e "  ${GREEN}Intel Mac${NC}"
fi

echo -e "  ${CYAN}[1/4] 创建安装目录...${NC}"
mkdir -p "$INSTALL_DIR"

echo -e "  ${CYAN}[2/4] 安装 Node.js 运行环境...${NC}"
rm -rf "$INSTALL_DIR/node"
cp -R "$NODE_SRC" "$INSTALL_DIR/node"
echo -e "  ${GREEN}Node.js $($INSTALL_DIR/node/bin/node --version) 已安装${NC}"

echo -e "  ${CYAN}[3/4] 安装 OpenClaw...${NC}"
if [ -d "$INSTALL_DIR/openclaw" ] && [ -f "$INSTALL_DIR/openclaw/.env" ]; then
    cp "$INSTALL_DIR/openclaw/.env" "/tmp/.uclaw-env-backup" 2>/dev/null || true
fi
rm -rf "$INSTALL_DIR/openclaw"
cp -R "$UCLAW_DIR/openclaw" "$INSTALL_DIR/openclaw"
if [ -f "/tmp/.uclaw-env-backup" ]; then
    cp "/tmp/.uclaw-env-backup" "$INSTALL_DIR/openclaw/.env"
    rm -f "/tmp/.uclaw-env-backup"
fi

# 复制用户数据
[ -d "$UCLAW_DIR/memory" ] && [ "$(ls -A "$UCLAW_DIR/memory" 2>/dev/null)" ] && cp -R "$UCLAW_DIR/memory" "$INSTALL_DIR/"
[ -d "$UCLAW_DIR/persona" ] && [ "$(ls -A "$UCLAW_DIR/persona" 2>/dev/null)" ] && cp -R "$UCLAW_DIR/persona" "$INSTALL_DIR/"

echo -e "  ${CYAN}[4/4] 配置环境变量...${NC}"

# macOS Catalina+ 默认 zsh，始终写 .zshrc
if [ "$(basename "$SHELL")" = "zsh" ] || [ ! -f "$HOME/.bashrc" -a ! -f "$HOME/.bash_profile" ]; then
    SHELL_RC="$HOME/.zshrc"
else
    SHELL_RC="$HOME/.bashrc"
fi
touch "$SHELL_RC"

if ! grep -q ".uclaw/node/bin" "$SHELL_RC" 2>/dev/null; then
    echo "" >> "$SHELL_RC"
    echo "# U-Claw OpenClaw" >> "$SHELL_RC"
    echo 'export PATH="$HOME/.uclaw/node/bin:$PATH"' >> "$SHELL_RC"
    echo "alias openclaw='\$HOME/.uclaw/node/bin/node \$HOME/.uclaw/openclaw/openclaw.mjs'" >> "$SHELL_RC"
    echo -e "  ${GREEN}已添加到 $SHELL_RC${NC}"
else
    echo -e "  ${YELLOW}PATH 已配置，跳过${NC}"
fi

# 创建全局命令
cat > "$INSTALL_DIR/uclaw" << 'SCRIPT'
#!/bin/bash
UCLAW_HOME="$HOME/.uclaw"
exec "$UCLAW_HOME/node/bin/node" "$UCLAW_HOME/openclaw/openclaw.mjs" "$@"
SCRIPT
chmod +x "$INSTALL_DIR/uclaw"

if [ -d "/usr/local/bin" ] && [ -w "/usr/local/bin" ]; then
    ln -sf "$INSTALL_DIR/uclaw" /usr/local/bin/uclaw 2>/dev/null || true
    ln -sf "$INSTALL_DIR/uclaw" /usr/local/bin/openclaw 2>/dev/null || true
    echo -e "  ${GREEN}已创建全局命令: uclaw / openclaw${NC}"
else
    echo -e "  ${YELLOW}提示: 运行 sudo ln -sf $INSTALL_DIR/uclaw /usr/local/bin/uclaw${NC}"
fi

echo ""
echo -e "  ${GREEN}╔══════════════════════════════════════╗${NC}"
echo -e "  ${GREEN}║     安装完成!                        ║${NC}"
echo -e "  ${GREEN}╚══════════════════════════════════════╝${NC}"
echo ""
echo "  安装位置: $INSTALL_DIR"
echo "  Node.js:  $($INSTALL_DIR/node/bin/node --version)"
echo ""
echo "  使用方法:"
echo "    1. 打开新的终端窗口"
echo "    2. 运行: openclaw"
echo "    或: uclaw"
echo ""
echo "  首次使用请运行:"
echo "    openclaw onboard --install-daemon"
echo ""
read -p "  按回车键关闭..."