install-hooks.sh

#!/bin/bash

# CodeRocket Git Hooks 安装脚本

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

echo -e "${GREEN}=== CodeRocket Git Hooks 安装 ===${NC}"

# 检查是否在 Git 仓库中
if ! git rev-parse --git-dir > /dev/null 2>&1; then
    echo -e "${RED}错误：当前目录不是 Git 仓库${NC}"
    exit 1
fi

# 获取仓库根目录
REPO_ROOT=$(git rev-parse --show-toplevel)
echo "仓库根目录: $REPO_ROOT"

# 获取配置值的函数
get_config_value() {
    local key=$1
    local default_value=$2
    local value=""

    # 优先级：环境变量 > 项目配置 > 全局配置 > 默认值
    if [ ! -z "${!key}" ]; then
        value="${!key}"
    elif [ -f "$REPO_ROOT/.ai-config" ]; then
        value=$(grep "^$key=" "$REPO_ROOT/.ai-config" 2>/dev/null | cut -d'=' -f2)
    elif [ -f "$HOME/.coderocket/ai-config" ]; then
        value=$(grep "^$key=" "$HOME/.coderocket/ai-config" 2>/dev/null | cut -d'=' -f2)
    elif [ -f "$REPO_ROOT/.env" ]; then
        value=$(grep "^$key=" "$REPO_ROOT/.env" 2>/dev/null | cut -d'=' -f2)
    fi

    if [ -z "$value" ]; then
        value="$default_value"
    fi

    echo "$value"
}

# 获取代码审查时机配置
REVIEW_TIMING=$(get_config_value "REVIEW_TIMING" "post-commit")
echo "代码审查时机: $REVIEW_TIMING"

# 解析真实 hooks 目录（git rev-parse --git-path 兼容 linked worktree：.git 为文件时也能拿到正确路径）
HOOKS_DIR=$(git -C "$REPO_ROOT" rev-parse --git-path hooks 2>/dev/null)
if [ -z "$HOOKS_DIR" ]; then
    # 解析失败时回退到传统路径
    HOOKS_DIR="$REPO_ROOT/.git/hooks"
fi
# git-path 可能返回相对路径（如 ".git/hooks"），统一转为绝对路径便于比较与使用
case "$HOOKS_DIR" in
    /*) ;;
    *) HOOKS_DIR="$REPO_ROOT/$HOOKS_DIR" ;;
esac
if [ ! -d "$HOOKS_DIR" ]; then
    mkdir -p "$HOOKS_DIR"
fi

# 检测 core.hooksPath：若指向其他目录，Git 不会执行装入 $HOOKS_DIR 的 hook
CONFIGURED_HOOKS_PATH=$(git -C "$REPO_ROOT" config core.hooksPath 2>/dev/null)
if [ -n "$CONFIGURED_HOOKS_PATH" ]; then
    # 归一化（~ 展开、相对路径补全、去尾斜杠）后与解析目录比较，一致则不告警
    case "$CONFIGURED_HOOKS_PATH" in
        /*) ;;
        "~"*) CONFIGURED_HOOKS_PATH="${CONFIGURED_HOOKS_PATH/#\~/$HOME}" ;;
        *) CONFIGURED_HOOKS_PATH="$REPO_ROOT/$CONFIGURED_HOOKS_PATH" ;;
    esac
    CONFIGURED_HOOKS_PATH="${CONFIGURED_HOOKS_PATH%/}"
    HOOKS_DIR_NORM="${HOOKS_DIR%/}"
    if [ "$CONFIGURED_HOOKS_PATH" != "$HOOKS_DIR_NORM" ]; then
        echo -e "${YELLOW}⚠ 检测到 core.hooksPath 已设置为: $CONFIGURED_HOOKS_PATH${NC}"
        echo "  Git 将执行该目录下的 hooks，而非 $HOOKS_DIR_NORM"
        echo "  本次安装的 CodeRocket hooks 可能不会被触发"
        echo "  如需启用 CodeRocket hooks，可执行: git config --unset core.hooksPath"
    fi
fi

# 覆盖已存在同名 hook 前备份（存在才备份，固定文件名，重复安装覆盖旧备份不堆积）
backup_hook() {
    if [ -f "$HOOKS_DIR/$1" ]; then
        cp "$HOOKS_DIR/$1" "$HOOKS_DIR/$1.coderocket-bak"
        echo -e "${YELLOW}→ 已备份原有 $1 为 $1.coderocket-bak${NC}"
    fi
}

# 根据配置安装相应的代码审查hook
if [ "$REVIEW_TIMING" = "pre-commit" ]; then
    echo -e "${YELLOW}→ 安装 pre-commit hook (提交前审查)...${NC}"
    backup_hook "pre-commit"
    cat > "$HOOKS_DIR/pre-commit" << 'EOF'
#!/bin/bash

# 获取 Git 仓库根目录
REPO_ROOT=$(git rev-parse --show-toplevel 2>/dev/null)

# 如果不在 Git 仓库中，退出
if [ -z "$REPO_ROOT" ]; then
    echo "错误：不在 Git 仓库中"
    exit 1
fi

# 加载环境变量（如果存在）
if [ -f "$HOME/.bashrc" ]; then
    source "$HOME/.bashrc" 2>/dev/null
fi

if [ -f "$HOME/.zshrc" ]; then
    source "$HOME/.zshrc" 2>/dev/null
fi

if [ -f "$HOME/.profile" ]; then
    source "$HOME/.profile" 2>/dev/null
fi

# 尝试从项目环境文件加载
if [ -f "$REPO_ROOT/.env" ]; then
    source "$REPO_ROOT/.env" 2>/dev/null
fi

# 查找 pre-commit 脚本
PRE_COMMIT_SCRIPT=""
if [ -f "$REPO_ROOT/githooks/pre-commit" ]; then
    PRE_COMMIT_SCRIPT="$REPO_ROOT/githooks/pre-commit"
elif [ -f "$HOME/.coderocket/githooks/pre-commit" ]; then
    PRE_COMMIT_SCRIPT="$HOME/.coderocket/githooks/pre-commit"
else
    echo "错误：pre-commit 脚本不存在"
    echo "请确保 CodeRocket 已正确安装"
    exit 1
fi

# 执行 pre-commit hook
"$PRE_COMMIT_SCRIPT"
EOF
    chmod +x "$HOOKS_DIR/pre-commit"
    echo -e "${GREEN}✓ pre-commit hook 安装完成${NC}"
else
    echo -e "${YELLOW}→ 安装 post-commit hook (提交后审查)...${NC}"
    backup_hook "post-commit"
    cat > "$HOOKS_DIR/post-commit" << 'EOF'
#!/bin/bash

# 获取 Git 仓库根目录
REPO_ROOT=$(git rev-parse --show-toplevel 2>/dev/null)

# 如果不在 Git 仓库中，退出
if [ -z "$REPO_ROOT" ]; then
    echo "错误：不在 Git 仓库中"
    exit 1
fi

# 加载环境变量（如果存在）
if [ -f "$HOME/.bashrc" ]; then
    source "$HOME/.bashrc" 2>/dev/null
fi

if [ -f "$HOME/.zshrc" ]; then
    source "$HOME/.zshrc" 2>/dev/null
fi

if [ -f "$HOME/.profile" ]; then
    source "$HOME/.profile" 2>/dev/null
fi

# 尝试从项目环境文件加载
if [ -f "$REPO_ROOT/.env" ]; then
    source "$REPO_ROOT/.env" 2>/dev/null
fi

# 查找 post-commit 脚本
POST_COMMIT_SCRIPT=""
if [ -f "$REPO_ROOT/githooks/post-commit" ]; then
    POST_COMMIT_SCRIPT="$REPO_ROOT/githooks/post-commit"
elif [ -f "$HOME/.coderocket/githooks/post-commit" ]; then
    POST_COMMIT_SCRIPT="$HOME/.coderocket/githooks/post-commit"
else
    echo "错误：post-commit 脚本不存在"
    echo "请确保 CodeRocket 已正确安装"
    exit 1
fi

# 执行 post-commit hook
"$POST_COMMIT_SCRIPT"
EOF
    chmod +x "$HOOKS_DIR/post-commit"
    echo -e "${GREEN}✓ post-commit hook 安装完成${NC}"
fi

# 安装 pre-push hook
echo -e "${YELLOW}→ 安装 pre-push hook...${NC}"
backup_hook "pre-push"
cat > "$HOOKS_DIR/pre-push" << 'EOF'
#!/bin/bash

# 获取 Git 仓库根目录
REPO_ROOT=$(git rev-parse --show-toplevel 2>/dev/null)

# 如果不在 Git 仓库中，退出
if [ -z "$REPO_ROOT" ]; then
    echo "错误：不在 Git 仓库中"
    exit 1
fi

# 加载环境变量（如果存在）
if [ -f "$HOME/.bashrc" ]; then
    source "$HOME/.bashrc" 2>/dev/null
fi

if [ -f "$HOME/.zshrc" ]; then
    source "$HOME/.zshrc" 2>/dev/null
fi

if [ -f "$HOME/.profile" ]; then
    source "$HOME/.profile" 2>/dev/null
fi

# 尝试从项目环境文件加载
if [ -f "$REPO_ROOT/.env" ]; then
    source "$REPO_ROOT/.env" 2>/dev/null
fi

# 查找 pre-push 脚本
PRE_PUSH_SCRIPT=""
if [ -f "$REPO_ROOT/githooks/pre-push" ]; then
    PRE_PUSH_SCRIPT="$REPO_ROOT/githooks/pre-push"
elif [ -f "$HOME/.coderocket/githooks/pre-push" ]; then
    PRE_PUSH_SCRIPT="$HOME/.coderocket/githooks/pre-push"
else
    echo "错误：pre-push 脚本不存在"
    echo "请确保 CodeRocket 已正确安装"
    exit 1
fi

# 执行 pre-push hook，传递所有参数
"$PRE_PUSH_SCRIPT" "$@"
EOF

# 设置执行权限
chmod +x "$HOOKS_DIR/pre-push"

echo -e "${GREEN}✓ Git hooks 安装完成${NC}"

# 检查环境变量配置
echo -e "\n${YELLOW}=== 环境变量检查 ===${NC}"

if [ -z "$GITLAB_PERSONAL_ACCESS_TOKEN" ]; then
    echo -e "${YELLOW}⚠ 未检测到 GITLAB_PERSONAL_ACCESS_TOKEN 环境变量${NC}"
    echo "请按以下步骤配置："
    echo "1. 复制 .env.example 为 .env"
    echo "2. 在 .env 文件中设置你的 GitLab Personal Access Token"
    echo "3. 或者在你的 shell 配置文件中设置环境变量"
    echo ""
    echo "示例："
    echo "cp .env.example .env"
    echo "# 然后编辑 .env 文件"
else
    echo -e "${GREEN}✓ GITLAB_PERSONAL_ACCESS_TOKEN 已配置${NC}"
fi

# 检查AI服务
echo -e "\n${YELLOW}=== AI服务检查 ===${NC}"

# 检查当前配置的AI服务
if [ -f "lib/ai-service-manager.sh" ]; then
    source lib/ai-service-manager.sh
    show_ai_service_status
else
    # 备用检查
    if command -v gemini &> /dev/null; then
        echo -e "${GREEN}✓ Gemini CLI 已安装${NC}"
    else
        echo -e "${YELLOW}⚠ 未检测到 Gemini CLI${NC}"
        echo "安装 Gemini CLI: npm install -g @google/gemini-cli"
    fi

    if command -v opencode &> /dev/null; then
        echo -e "${GREEN}✓ OpenCode CLI 已安装${NC}"
    else
        echo -e "${YELLOW}⚠ 未检测到 OpenCode CLI${NC}"
        echo "安装 OpenCode CLI: npm install -g @opencode/cli"
    fi

    if command -v claudecode &> /dev/null; then
        echo -e "${GREEN}✓ ClaudeCode CLI 已安装${NC}"
    else
        echo -e "${YELLOW}⚠ 未检测到 ClaudeCode CLI${NC}"
        echo "安装 ClaudeCode CLI: npm install -g @anthropic-ai/claude-code"
    fi
fi

echo -e "\n${GREEN}=== 安装完成 ===${NC}"
echo "现在你可以："
if [ "$REVIEW_TIMING" = "pre-commit" ]; then
    echo "1. 使用 git commit 触发提交前代码审查（可能阻止有问题的提交）"
else
    echo "1. 使用 git commit 触发提交后代码审查"
fi
echo "2. 使用 git push 触发自动 MR 创建"
echo "3. 在 VS Code 和终端中都能正常工作"
echo ""
echo "💡 提示：可以使用 './lib/ai-config.sh timing' 来更改代码审查时机"