@shuchang/futu-mcp-server/bin/futu-mcp.js

富途API MCP服务器 - 完整的股票数据和交易功能MCP工具集

#!/usr/bin/env node

const { spawn, execSync } = require('child_process');
const path = require('path');
const fs = require('fs');

// 获取项目根目录
const projectRoot = path.dirname(__dirname);

// 检查虚拟环境是否存在
const venvPath = path.join(projectRoot, 'venv');
const pythonPath = process.platform === 'win32' 
    ? path.join(venvPath, 'Scripts', 'python.exe')
    : path.join(venvPath, 'bin', 'python');
const pipPath = process.platform === 'win32'
    ? path.join(venvPath, 'Scripts', 'pip')
    : path.join(venvPath, 'bin', 'pip');

// 自动设置虚拟环境函数
function setupVirtualEnv() {
    console.log('⚙️ 首次运行，正在设置Python虚拟环境...');
    
    try {
        // 检查python3是否可用
        try {
            execSync('python3 --version', { stdio: 'pipe' });
        } catch (error) {
            console.error('❌ python3 未找到，请先安装Python 3');
            process.exit(1);
        }
        
        // 创建虚拟环境
        console.log('📦 创建虚拟环境...');
        if (!fs.existsSync(venvPath)) {
            execSync(`python3 -m venv "${venvPath}"`, { 
                cwd: projectRoot, 
                stdio: 'inherit' 
            });
        }
        
        // 安装依赖
        console.log('📦 安装项目依赖...');
        const requirementsPath = path.join(projectRoot, 'requirements.txt');
        if (fs.existsSync(requirementsPath)) {
            execSync(`"${pipPath}" install -r "${requirementsPath}"`, {
                cwd: projectRoot,
                stdio: 'inherit'
            });
        }
        
        console.log('✅ 虚拟环境设置完成！');
        
    } catch (error) {
        console.error('❌ 虚拟环境设置失败:', error.message);
        console.error('\n💡 请手动运行以下命令：');
        console.error(`cd ${projectRoot}`);
        console.error('python3 -m venv venv');
        console.error('source venv/bin/activate');
        console.error('pip install -r requirements.txt');
        process.exit(1);
    }
}

// 检查并设置虚拟环境
if (!fs.existsSync(pythonPath)) {
    setupVirtualEnv();
} else {
    // 确保依赖已安装
    try {
        execSync(`"${pipPath}" freeze`, { stdio: 'pipe' });
    } catch (error) {
        console.log('📦 检测到虚拟环境可能需要更新，正在重新安装依赖...');
        setupVirtualEnv();
    }
}

// 检查MCP服务器文件是否存在
const serverPath = path.join(projectRoot, 'futu_mcp_server.py');
if (!fs.existsSync(serverPath)) {
    console.error('❌ MCP服务器文件不存在:', serverPath);
    process.exit(1);
}

// 设置环境变量
const env = {
    ...process.env,
    PYTHONPATH: projectRoot,
    // 从环境变量获取富途API配置，如果没有则使用默认值
    FUTU_API_HOST: process.env.FUTU_API_HOST || '127.0.0.1',
    FUTU_API_PORT: process.env.FUTU_API_PORT || '11111',
    // 确保 Python 使用虚拟环境中的包
    VIRTUAL_ENV: venvPath,
    PATH: `${path.dirname(pythonPath)}${path.delimiter}${process.env.PATH}`
};

// 如果设置了解锁密码
if (process.env.FUTU_UNLOCK_PASSWORD) {
    env.FUTU_UNLOCK_PASSWORD = process.env.FUTU_UNLOCK_PASSWORD;
}

console.log('🚀 启动富途 MCP 服务器...');
console.log(`📡 富途API地址: ${env.FUTU_API_HOST}:${env.FUTU_API_PORT}`);

// 启动Python MCP服务器
const child = spawn(pythonPath, [serverPath], {
    env: env,
    cwd: projectRoot,
    stdio: 'inherit'
});

// 处理进程退出
child.on('close', (code) => {
    if (code !== 0) {
        console.error(`❌ MCP服务器退出，退出码: ${code}`);
        process.exit(code);
    }
});

// 处理错误
child.on('error', (err) => {
    console.error('❌ 启动MCP服务器失败:', err.message);
    process.exit(1);
});

// 处理进程信号
process.on('SIGINT', () => {
    console.log('\n🛑 正在停止MCP服务器...');
    child.kill('SIGINT');
});

process.on('SIGTERM', () => {
    console.log('\n🛑 正在停止MCP服务器...');
    child.kill('SIGTERM');
});