httpServer.js•20.5 kB
#!/usr/bin/env node
import express from "express";
import cors from "cors";
import { randomUUID } from "node:crypto";
import { runWithRequestContext } from "./config.js";
// 工具导入
import { financeNews } from "./tools/financeNews.js";
import { stockData } from "./tools/stockData.js";
import { stockDataMinutes } from "./tools/stockDataMinutes.js";
import { indexData } from "./tools/indexData.js";
import { macroEcon } from "./tools/macroEcon.js";
import { companyPerformance } from "./tools/companyPerformance.js";
import { fundData } from "./tools/fundData.js";
import { fundManagerByName, runFundManagerByName } from "./tools/fundManagerByName.js";
import { convertibleBond } from "./tools/convertibleBond.js";
import { blockTrade } from "./tools/blockTrade.js";
import { moneyFlow } from "./tools/moneyFlow.js";
import { marginTrade } from "./tools/marginTrade.js";
import { companyPerformance_hk } from "./tools/companyPerformance_hk.js";
import { companyPerformance_us } from "./tools/companyPerformance_us.js";
import { csiIndexConstituents } from "./tools/csiIndexConstituents.js";
import { dragonTigerInst } from "./tools/dragonTigerInst.js";
import { hotNews } from "./tools/hotNews.js";
// 时间戳工具（保留）
const timestampTool = {
name: "current_timestamp",
description: "获取当前东八区（中国时区）的时间戳，包括年月日时分秒信息",
parameters: {
type: "object",
properties: {
format: {
type: "string",
description: "时间格式，可选值：datetime(完整日期时间，默认)、date(仅日期)、time(仅时间)、timestamp(Unix时间戳)、readable(可读格式)"
}
}
},
async run(args) {
const now = new Date();
const chinaTime = new Date(now.getTime() + (8 * 60 * 60 * 1000));
const format = args?.format || 'datetime';
const pad = (n) => n.toString().padStart(2, '0');
const y = chinaTime.getUTCFullYear();
const m = pad(chinaTime.getUTCMonth() + 1);
const d = pad(chinaTime.getUTCDate());
const hh = pad(chinaTime.getUTCHours());
const mm = pad(chinaTime.getUTCMinutes());
const ss = pad(chinaTime.getUTCSeconds());
const weekdays = ['星期日', '星期一', '星期二', '星期三', '星期四', '星期五', '星期六'];
const wd = weekdays[chinaTime.getUTCDay()];
let result = `${y}-${m}-${d} ${hh}:${mm}:${ss}`;
if (format === 'date')
result = `${y}-${m}-${d}`;
if (format === 'time')
result = `${hh}:${mm}:${ss}`;
if (format === 'timestamp')
result = Math.floor(chinaTime.getTime() / 1000).toString();
if (format === 'readable')
result = `${y}年${m}月${d}日 ${wd} ${hh}时${mm}分${ss}秒`;
return { content: [{ type: 'text', text: `## 🕐 当前东八区时间\n\n格式: ${format}\n时间: ${result}\n星期: ${wd}` }] };
}
};
const toolList = [
{ name: timestampTool.name, description: timestampTool.description, inputSchema: timestampTool.parameters },
{ name: financeNews.name, description: financeNews.description, inputSchema: financeNews.parameters },
{ name: stockData.name, description: stockData.description, inputSchema: stockData.parameters },
{ name: stockDataMinutes.name, description: stockDataMinutes.description, inputSchema: stockDataMinutes.parameters },
{ name: indexData.name, description: indexData.description, inputSchema: indexData.parameters },
{ name: macroEcon.name, description: macroEcon.description, inputSchema: macroEcon.parameters },
{ name: companyPerformance.name, description: companyPerformance.description, inputSchema: companyPerformance.parameters },
{ name: fundData.name, description: fundData.description, inputSchema: fundData.parameters },
{ name: fundManagerByName.name, description: fundManagerByName.description, inputSchema: fundManagerByName.inputSchema },
{ name: convertibleBond.name, description: convertibleBond.description, inputSchema: convertibleBond.parameters },
{ name: blockTrade.name, description: blockTrade.description, inputSchema: blockTrade.parameters },
{ name: moneyFlow.name, description: moneyFlow.description, inputSchema: moneyFlow.parameters },
{ name: marginTrade.name, description: marginTrade.description, inputSchema: marginTrade.parameters },
{ name: companyPerformance_hk.name, description: companyPerformance_hk.description, inputSchema: companyPerformance_hk.parameters },
{ name: companyPerformance_us.name, description: companyPerformance_us.description, inputSchema: companyPerformance_us.parameters },
{ name: csiIndexConstituents.name, description: csiIndexConstituents.description, inputSchema: csiIndexConstituents.parameters },
{ name: dragonTigerInst.name, description: dragonTigerInst.description, inputSchema: dragonTigerInst.parameters },
{ name: hotNews.name, description: hotNews.description, inputSchema: hotNews.parameters }
];
const sessions = new Map();
function extractTokenFromHeaders(req) {
const h = req.headers;
// 1. 尝试从标准请求头读取
const tokenHeader = (h['x-tushare-token'] || h['x-api-key']);
if (tokenHeader && tokenHeader.trim()) {
console.log(`[TOKEN] Found in X-Tushare-Token/X-Api-Key header`);
return tokenHeader.trim();
}
// 2. 尝试从 Authorization Bearer 读取
const auth = h['authorization'];
if (typeof auth === 'string' && auth.toLowerCase().startsWith('bearer ')) {
console.log(`[TOKEN] Found in Authorization Bearer header`);
return auth.slice(7).trim();
}
// 3. 🔍 尝试从 Smithery 特殊头读取（可能的头名称）
const smitheryConfig = h['x-smithery-config'] || h['x-config'] || h['x-session-config'];
if (smitheryConfig) {
console.log(`[TOKEN] Found Smithery config header:`, smitheryConfig);
try {
const config = JSON.parse(smitheryConfig);
if (config.TUSHARE_TOKEN) {
console.log(`[TOKEN] Extracted from Smithery config`);
return config.TUSHARE_TOKEN;
}
}
catch (e) {
console.log(`[TOKEN] Failed to parse Smithery config:`, e);
}
}
// 4. 🔍 尝试从查询参数读取
const query = req.query;
if (query.tushare_token || query.TUSHARE_TOKEN) {
console.log(`[TOKEN] Found in query parameters`);
return (query.tushare_token || query.TUSHARE_TOKEN);
}
console.log(`[TOKEN] Not found in request, falling back to environment variable`);
return undefined;
}
// 移除 CoinGecko 头的解析（已改为 Binance 公共行情，无需 Key）
const app = express();
const PORT = Number(process.env.PORT || 3000);
// 日志中间件：记录所有请求
app.use((req, res, next) => {
const timestamp = new Date().toISOString();
const method = req.method;
const url = req.url;
const ip = req.ip || req.socket.remoteAddress;
console.log(`[${timestamp}] ${method} ${url} - IP: ${ip}`);
// 🔍 详细记录所有请求头，用于调试 Smithery 配置传递
console.log(`[DEBUG] Request Headers:`, JSON.stringify(req.headers, null, 2));
// 记录请求完成时的状态码
const originalSend = res.send;
res.send = function (data) {
console.log(`[${timestamp}] ${method} ${url} - Status: ${res.statusCode}`);
return originalSend.call(this, data);
};
next();
});
app.use(cors({
origin: '*',
methods: ['GET', 'POST', 'OPTIONS'],
allowedHeaders: [
'Content-Type', 'Accept', 'Authorization', 'Mcp-Session-Id', 'Last-Event-ID',
'X-Tenant-Id', 'X-Api-Key', 'X-Tushare-Token',
'X-Smithery-Config', 'X-Config', 'X-Session-Config' // Smithery 可能的配置头
],
exposedHeaders: ['Content-Type', 'Mcp-Session-Id']
}));
app.use(express.json({ limit: '10mb' }));
app.get('/health', (_req, res) => {
res.json({ status: 'healthy', transport: 'streamable-http', activeSessions: sessions.size });
});
app.get('/mcp', (req, res) => {
const accept = req.headers.accept || '';
const forceSse = req.query.sse === '1' || req.query.sse === 'true';
console.log(`📡 [MCP-SSE] Client connecting - Accept: ${accept}, Force SSE: ${forceSse}`);
if (forceSse || (typeof accept === 'string' && accept.includes('text/event-stream'))) {
res.writeHead(200, {
'Content-Type': 'text/event-stream; charset=utf-8',
'Cache-Control': 'no-cache, no-transform',
'Connection': 'keep-alive',
'X-Accel-Buffering': 'no'
});
// 仅发送注释型心跳，避免发送非 JSON-RPC 的 data 事件
res.write(': stream established\n\n');
console.log(`✅ [MCP-SSE] Stream established`);
const keep = setInterval(() => res.write(': keepalive\n\n'), 30000);
req.on('close', () => {
clearInterval(keep);
console.log(`🔌 [MCP-SSE] Client disconnected`);
});
return;
}
console.log(`❌ [MCP-SSE] Invalid Accept header`);
return res.status(400).json({ jsonrpc: '2.0', error: { code: -32600, message: 'Accept must include text/event-stream' }, id: null });
});
app.post('/mcp', async (req, res) => {
const body = req.body;
if (!body)
return res.status(400).json({ jsonrpc: '2.0', error: { code: -32600, message: 'Empty body' }, id: null });
const isNotification = (body.id === undefined || body.id === null) && typeof body.method === 'string' && body.method.startsWith('notifications/');
if (isNotification) {
const sid = req.headers['mcp-session-id'];
console.log(`🔔 [MCP-Notification] ${body.method} - Session: ${sid || 'none'}`);
if (sid && sessions.has(sid))
sessions.get(sid).lastActivity = new Date();
return res.status(204).end();
}
const method = body.method;
console.log(`🔧 [MCP-${method}] Request ID: ${body.id}`);
if (method === 'initialize') {
const newId = randomUUID();
sessions.set(newId, { id: newId, createdAt: new Date(), lastActivity: new Date() });
res.setHeader('Mcp-Session-Id', newId);
console.log(`✅ [MCP-initialize] New session created: ${newId}`);
return res.json({ jsonrpc: '2.0', result: { protocolVersion: '2024-11-05', capabilities: { tools: {} }, serverInfo: { name: 'FinanceMCP', version: '1.0.0' } }, id: body.id });
}
if (method === 'tools/list') {
console.log(`📋 [MCP-tools/list] Returning ${toolList.length} tools`);
return res.json({ jsonrpc: '2.0', result: { tools: toolList }, id: body.id });
}
// 明确表示不支持 resources 和 prompts（返回空列表而不是错误）
if (method === 'resources/list') {
console.log(`📋 [MCP-resources/list] Not supported, returning empty list`);
return res.json({ jsonrpc: '2.0', result: { resources: [] }, id: body.id });
}
if (method === 'prompts/list') {
console.log(`📋 [MCP-prompts/list] Not supported, returning empty list`);
return res.json({ jsonrpc: '2.0', result: { prompts: [] }, id: body.id });
}
if (method === 'tools/call') {
const { name, arguments: args } = body.params || {};
const token = extractTokenFromHeaders(req);
const startTime = Date.now();
console.log(`🚀 [MCP-tools/call] Tool: ${name} | Has Token: ${!!token}`);
try {
const result = await runWithRequestContext({ tushareToken: token }, async () => {
switch (name) {
case 'current_timestamp':
return await timestampTool.run({ format: args?.format ? String(args.format) : undefined });
case 'finance_news':
return await financeNews.run({
query: String(args?.query)
});
case 'stock_data':
return await stockData.run({
code: String(args?.code),
market_type: String(args?.market_type),
start_date: args?.start_date ? String(args.start_date) : undefined,
end_date: args?.end_date ? String(args.end_date) : undefined,
indicators: args?.indicators ? String(args.indicators) : undefined,
});
case 'stock_data_minutes':
return await stockDataMinutes.run({
code: String(args?.code),
market_type: String(args?.market_type),
start_datetime: String(args?.start_datetime),
end_datetime: String(args?.end_datetime),
freq: String(args?.freq)
});
case 'index_data':
return await indexData.run({
code: String(args?.code),
start_date: args?.start_date ? String(args.start_date) : undefined,
end_date: args?.end_date ? String(args.end_date) : undefined,
});
case 'macro_econ':
return await macroEcon.run({
indicator: String(args?.indicator),
start_date: args?.start_date ? String(args.start_date) : undefined,
end_date: args?.end_date ? String(args.end_date) : undefined,
});
case 'company_performance':
return await companyPerformance.run({
ts_code: String(args?.ts_code),
data_type: String(args?.data_type),
start_date: String(args?.start_date),
end_date: String(args?.end_date),
period: args?.period ? String(args.period) : undefined,
});
case 'fund_data':
return await fundData.run({
ts_code: args?.ts_code ? String(args.ts_code) : undefined,
data_type: String(args?.data_type),
start_date: args?.start_date ? String(args.start_date) : undefined,
end_date: args?.end_date ? String(args.end_date) : undefined,
period: args?.period ? String(args.period) : undefined,
});
case 'fund_manager_by_name':
return await runFundManagerByName({
name: String(args?.name),
ann_date: args?.ann_date ? String(args.ann_date) : undefined,
});
case 'convertible_bond':
return await convertibleBond.run({
ts_code: args?.ts_code ? String(args.ts_code) : undefined,
data_type: String(args?.data_type),
start_date: args?.start_date ? String(args.start_date) : undefined,
end_date: args?.end_date ? String(args.end_date) : undefined,
});
case 'block_trade':
return await blockTrade.run({
code: args?.code ? String(args.code) : undefined,
start_date: String(args?.start_date),
end_date: String(args?.end_date),
});
case 'money_flow':
return await moneyFlow.run({
ts_code: args?.ts_code ? String(args.ts_code) : undefined,
start_date: String(args?.start_date),
end_date: String(args?.end_date),
});
case 'margin_trade':
return await marginTrade.run({
data_type: String(args?.data_type),
ts_code: args?.ts_code ? String(args.ts_code) : undefined,
start_date: String(args?.start_date),
end_date: args?.end_date ? String(args.end_date) : undefined,
exchange: args?.exchange ? String(args.exchange) : undefined,
});
case 'company_performance_hk':
return await companyPerformance_hk.run({
ts_code: String(args?.ts_code),
data_type: String(args?.data_type),
start_date: String(args?.start_date),
end_date: String(args?.end_date),
period: args?.period ? String(args.period) : undefined,
ind_name: args?.ind_name ? String(args.ind_name) : undefined,
});
case 'company_performance_us':
return await companyPerformance_us.run({
ts_code: String(args?.ts_code),
data_type: String(args?.data_type),
start_date: String(args?.start_date),
end_date: String(args?.end_date),
period: args?.period ? String(args.period) : undefined,
});
case 'csi_index_constituents':
return await csiIndexConstituents.run({
index_code: String(args?.index_code),
start_date: String(args?.start_date),
end_date: String(args?.end_date),
});
case 'dragon_tiger_inst':
return await dragonTigerInst.run({
trade_date: String(args?.trade_date),
ts_code: args?.ts_code ? String(args.ts_code) : undefined,
});
case 'hot_news_7x24':
return await hotNews.run({});
default:
throw new Error(`Unknown tool: ${name}`);
}
});
const duration = Date.now() - startTime;
console.log(`✅ [MCP-tools/call] Tool: ${name} completed in ${duration}ms`);
return res.json({ jsonrpc: '2.0', result, id: body.id });
}
catch (error) {
const duration = Date.now() - startTime;
const message = error?.message || String(error);
console.error(`❌ [MCP-tools/call] Tool: ${name} failed after ${duration}ms - Error: ${message}`);
return res.status(400).json({ jsonrpc: '2.0', error: { code: -32000, message }, id: body.id });
}
}
console.error(`❌ [MCP] Unknown method: ${method}`);
return res.status(400).json({ jsonrpc: '2.0', error: { code: -32601, message: `Method not found: ${method}` }, id: body.id });
});
// 兼容性终止路由：部分客户端在结束会话时会调用此端点
app.post('/mcp/terminate', (_req, res) => {
return res.status(200).json({ ok: true });
});
// 备用别名
app.post('/terminate', (_req, res) => {
return res.status(200).json({ ok: true });
});
// 兼容 GET 终止
app.get('/mcp/terminate', (_req, res) => {
return res.status(200).json({ ok: true });
});
app.get('/terminate', (_req, res) => {
return res.status(200).json({ ok: true });
});
app.listen(PORT, () => {
console.log('\n' + '='.repeat(60));
console.log('🚀 FinanceMCP Streamable HTTP Server Started');
console.log('='.repeat(60));
console.log(`📍 Server URL: http://localhost:${PORT}`);
console.log(`📡 MCP Endpoint: http://localhost:${PORT}/mcp`);
console.log(`💚 Health Check: http://localhost:${PORT}/health`);
console.log(`📊 Active Sessions: ${sessions.size}`);
console.log(`🔧 Available Tools: ${toolList.length}`);
console.log('='.repeat(60));
console.log('📝 Server is ready to accept connections');
console.log('='.repeat(60) + '\n');
});