chaimi-keep-mcp 3.1.24

package/install.sh

@@ -0,0 +1,268 @@
+#!/bin/bash
+
+# 柴米记账 MCP Server 一键安装脚本
+# 支持 macOS 和 Linux
+# 用法: curl -fsSL https://chaihuo.io/install.sh | bash
+
+set -e
+
+# 颜色定义
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+NC='\033[0m' # No Color
+
+# 配置
+INSTALL_DIR="$HOME/.chaihuo-mcp"
+WORKBUDDY_DIR="$HOME/WorkBuddy"
+PYTHON_VERSION="3.10"
+SCF_URL="https://1412172089-4wbwsop8pe.ap-shanghai.tencentscf.com"
+
+echo -e "${GREEN}=== 柴米记账 MCP Server 安装脚本 ===${NC}"
+echo ""
+
+# 检测操作系统
+OS="$(uname -s)"
+case "${OS}" in
+    Linux*)     PLATFORM=Linux;;
+    Darwin*)    PLATFORM=Mac;;
+    *)          PLATFORM="UNKNOWN:${OS}"
+esac
+
+echo -e "${YELLOW}检测到操作系统: ${PLATFORM}${NC}"
+
+# 检查并安装 Python 3.10+
+check_python() {
+    echo -e "${YELLOW}检查 Python 版本...${NC}"
+    
+    if command -v python3.10 &> /dev/null; then
+        PYTHON_CMD="python3.10"
+    elif command -v python3.11 &> /dev/null; then
+        PYTHON_CMD="python3.11"
+    elif command -v python3.12 &> /dev/null; then
+        PYTHON_CMD="python3.12"
+    elif command -v python3 &> /dev/null; then
+        PY_VERSION=$(python3 --version 2>&1 | awk '{print $2}' | cut -d. -f1,2)
+        if (( $(echo "$PY_VERSION >= 3.10" | bc -l) )); then
+            PYTHON_CMD="python3"
+        else
+            PYTHON_CMD=""
+        fi
+    else
+        PYTHON_CMD=""
+    fi
+    
+    if [ -z "$PYTHON_CMD" ]; then
+        echo -e "${RED}未检测到 Python 3.10+${NC}"
+        install_python
+    else
+        echo -e "${GREEN}找到 Python: $PYTHON_CMD${NC}"
+    fi
+}
+
+# 安装 Python 3.10
+install_python() {
+    echo -e "${YELLOW}正在安装 Python 3.10...${NC}"
+    
+    if [ "$PLATFORM" == "Mac" ]; then
+        # macOS
+        if command -v brew &> /dev/null; then
+            echo "使用 Homebrew 安装 Python 3.10..."
+            brew install python@3.10
+            PYTHON_CMD="/opt/homebrew/bin/python3.10"
+        else
+            echo -e "${RED}请先安装 Homebrew: https://brew.sh${NC}"
+            exit 1
+        fi
+    elif [ "$PLATFORM" == "Linux" ]; then
+        # Linux
+        if command -v apt-get &> /dev/null; then
+            echo "使用 apt 安装 Python 3.10..."
+            sudo apt-get update
+            sudo apt-get install -y python3.10 python3.10-venv python3.10-pip
+            PYTHON_CMD="python3.10"
+        elif command -v yum &> /dev/null; then
+            echo "使用 yum 安装 Python 3.10..."
+            sudo yum install -y python310
+            PYTHON_CMD="python3.10"
+        else
+            echo -e "${RED}不支持的 Linux 发行版，请手动安装 Python 3.10+${NC}"
+            exit 1
+        fi
+    else
+        echo -e "${RED}不支持的操作系统: ${PLATFORM}${NC}"
+        exit 1
+    fi
+}
+
+# 创建安装目录
+setup_install_dir() {
+    echo -e "${YELLOW}创建安装目录...${NC}"
+    
+    if [ -d "$INSTALL_DIR" ]; then
+        echo -e "${YELLOW}目录已存在，清理旧文件...${NC}"
+        rm -rf "$INSTALL_DIR"
+    fi
+    
+    mkdir -p "$INSTALL_DIR"
+    echo -e "${GREEN}安装目录: $INSTALL_DIR${NC}"
+}
+
+# 创建虚拟环境
+create_venv() {
+    echo -e "${YELLOW}创建 Python 虚拟环境...${NC}"
+    
+    $PYTHON_CMD -m venv "$INSTALL_DIR/venv"
+    source "$INSTALL_DIR/venv/bin/activate"
+    
+    echo -e "${GREEN}虚拟环境创建成功${NC}"
+}
+
+# 安装依赖
+install_dependencies() {
+    echo -e "${YELLOW}安装依赖...${NC}"
+    
+    pip install --upgrade pip
+    pip install fastmcp>=0.4.0 requests
+    
+    echo -e "${GREEN}依赖安装完成${NC}"
+}
+
+# 下载 MCP Server 文件
+download_server() {
+    echo -e "${YELLOW}下载 MCP Server...${NC}"
+    
+    # 复制本地文件（如果是本地安装）
+    if [ -f "server.py" ]; then
+        cp server.py "$INSTALL_DIR/"
+        echo -e "${GREEN}使用本地 server.py${NC}"
+    else
+        # 从远程下载
+        echo "从远程下载..."
+        curl -fsSL -o "$INSTALL_DIR/server.py" "https://raw.githubusercontent.com/chaihuo/mcp-server/main/server.py"
+        echo -e "${GREEN}下载完成${NC}"
+    fi
+}
+
+# 配置 WorkBuddy
+configure_workbuddy() {
+    echo -e "${YELLOW}配置 WorkBuddy...${NC}"
+    
+    if [ ! -d "$WORKBUDDY_DIR" ]; then
+        echo -e "${RED}未找到 WorkBuddy 目录: $WORKBUDDY_DIR${NC}"
+        echo "请确认 WorkBuddy 已安装，或手动配置 MCP"
+        return 1
+    fi
+    
+    # 创建 MCP 配置
+    mkdir -p "$WORKBUDDY_DIR/mcp"
+    
+    # 复制 server.py 到 WorkBuddy 目录
+    cp "$INSTALL_DIR/server.py" "$WORKBUDDY_DIR/mcp/chaihuo-server.py"
+    
+    # 创建配置文件
+    cat > "$WORKBUDDY_DIR/mcp_config.json" << EOF
+{
+  "mcpServers": {
+    "chaihuo": {
+      "command": "$INSTALL_DIR/venv/bin/python",
+      "args": ["$WORKBUDDY_DIR/mcp/chaihuo-server.py"],
+      "env": {
+        "SCF_URL": "$SCF_URL"
+      }
+    }
+  }
+}
+EOF
+    
+    echo -e "${GREEN}WorkBuddy 配置完成${NC}"
+    echo -e "${YELLOW}配置文件: $WORKBUDDY_DIR/mcp_config.json${NC}"
+}
+
+# 生成 JWT Secret
+generate_jwt_secret() {
+    echo -e "${YELLOW}生成 JWT Secret...${NC}"
+    
+    JWT_SECRET=$(openssl rand -base64 32)
+    
+    # 更新配置文件
+    cat > "$WORKBUDDY_DIR/mcp_config.json" << EOF
+{
+  "mcpServers": {
+    "chaihuo": {
+      "command": "$INSTALL_DIR/venv/bin/python",
+      "args": ["$WORKBUDDY_DIR/mcp/chaihuo-server.py"],
+      "env": {
+        "SCF_URL": "$SCF_URL",
+        "JWT_SECRET": "$JWT_SECRET"
+      }
+    }
+  }
+}
+EOF
+    
+    echo -e "${GREEN}JWT Secret 生成成功${NC}"
+    echo -e "${YELLOW}请保存以下密钥（用于生成用户 Token）:${NC}"
+    echo -e "${GREEN}$JWT_SECRET${NC}"
+}
+
+# 测试安装
+test_installation() {
+    echo -e "${YELLOW}测试安装...${NC}"
+    
+    source "$INSTALL_DIR/venv/bin/activate"
+    
+    if python -c "import fastmcp" 2>/dev/null; then
+        echo -e "${GREEN}✓ fastmcp 安装成功${NC}"
+    else
+        echo -e "${RED}✗ fastmcp 安装失败${NC}"
+        return 1
+    fi
+    
+    if [ -f "$INSTALL_DIR/server.py" ]; then
+        echo -e "${GREEN}✓ server.py 存在${NC}"
+    else
+        echo -e "${RED}✗ server.py 不存在${NC}"
+        return 1
+    fi
+    
+    echo -e "${GREEN}安装测试通过！${NC}"
+}
+
+# 打印完成信息
+print_finish() {
+    echo ""
+    echo -e "${GREEN}=== 安装完成！===${NC}"
+    echo ""
+    echo -e "${YELLOW}下一步操作:${NC}"
+    echo "1. 重启 WorkBuddy"
+    echo "2. 在 WorkBuddy 中测试记账功能"
+    echo ""
+    echo -e "${YELLOW}配置信息:${NC}"
+    echo "- 安装目录: $INSTALL_DIR"
+    echo "- 配置文件: $WORKBUDDY_DIR/mcp_config.json"
+    echo "- SCF 地址: $SCF_URL"
+    echo ""
+    echo -e "${YELLOW}如需卸载，运行:${NC}"
+    echo "rm -rf $INSTALL_DIR"
+    echo ""
+}
+
+# 主流程
+main() {
+    echo -e "${GREEN}开始安装柴米记账 MCP Server...${NC}"
+    echo ""
+    
+    check_python
+    setup_install_dir
+    create_venv
+    install_dependencies
+    download_server
+    configure_workbuddy
+    generate_jwt_secret
+    test_installation
+    print_finish
+}
+
+# 运行主流程
+main