#!/bin/bash

# 获取脚本所在目录的绝对路径
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# 激活虚拟环境
if [ -f "${SCRIPT_DIR}/venv/bin/activate" ]; then
    source "${SCRIPT_DIR}/venv/bin/activate"
else
    echo "警告: 虚拟环境未找到，将使用系统Python环境"
fi

# 默认配置
RAP2_URL=${RAP2_URL:-"http://rap2.atmob.com"}
RAP2_SID=${RAP2_SID:-""}
RAP2_SID_SIG=${RAP2_SID_SIG:-""}

# 解析命令行参数
while [[ $# -gt 0 ]]; do
    case $1 in
        --url=*)
            RAP2_URL="${1#*=}"
            shift
            ;;
        --sid=*)
            RAP2_SID="${1#*=}"
            shift
            ;;
        --sid-sig=*)
            RAP2_SID_SIG="${1#*=}"
            shift
            ;;
        --help)
            echo "用法: ./start.sh [选项]"
            echo "选项:"
            echo "  --url=URL       设置RAP2服务器地址"
            echo "  --sid=SID       设置RAP2 koa.sid Cookie值"
            echo "  --sid-sig=SIG   设置RAP2 koa.sid.sig Cookie值"
            echo "  --help          显示此帮助信息"
            exit 0
            ;;
        *)
            echo "未知选项: $1"
            echo "使用 --help 查看帮助"
            exit 1
            ;;
    esac
done

# 构建参数
ARGS=""
if [ -n "$RAP2_URL" ]; then
    ARGS="$ARGS --rap2-url=$RAP2_URL"
fi
if [ -n "$RAP2_SID" ]; then
    ARGS="$ARGS --rap2-sid=$RAP2_SID"
fi
if [ -n "$RAP2_SID_SIG" ]; then
    ARGS="$ARGS --rap2-sid-sig=$RAP2_SID_SIG"
fi

echo "启动RAP2 MCP服务器..."
echo "RAP2服务器: $RAP2_URL"
if [ -n "$RAP2_SID" ] && [ -n "$RAP2_SID_SIG" ]; then
    echo "RAP2认证: 已配置"
else
    echo "RAP2认证: 未配置"
fi

# 运行服务器，直接传递参数
python "${SCRIPT_DIR}/rap2-mcp/rap2_mcp_server.py" $ARGS