mcp-adder-simple 1.1.2 → 1.1.3

package/cache.js

@@ -0,0 +1,65 @@
+/**
+ * 缓存模块
+ * 提供基于内存的缓存功能，支持过期时间
+ */
+
+const { CACHE_DURATION } = require('./config');
+
+const cache = new Map();
+
+/**
+ * 获取缓存数据
+ * @param {string} key - 缓存键
+ * @returns {any|null} 缓存的数据，如果不存在或已过期则返回 null
+ */
+function getCache(key) {
+  const cached = cache.get(key);
+  if (!cached) return null;
+
+  const now = Date.now();
+  if (now - cached.timestamp > CACHE_DURATION) {
+    // 缓存已过期，删除
+    cache.delete(key);
+    return null;
+  }
+
+  console.error(`[CACHE] 命中缓存: ${key}`);
+  return cached.data;
+}
+
+/**
+ * 设置缓存数据
+ * @param {string} key - 缓存键
+ * @param {any} data - 要缓存的数据
+ */
+function setCache(key, data) {
+  cache.set(key, {
+    data: data,
+    timestamp: Date.now()
+  });
+  console.error(`[CACHE] 已缓存: ${key}`);
+}
+
+/**
+ * 清除指定缓存
+ * @param {string} key - 缓存键
+ */
+function clearCache(key) {
+  cache.delete(key);
+  console.error(`[CACHE] 已清除缓存: ${key}`);
+}
+
+/**
+ * 清除所有缓存
+ */
+function clearAllCache() {
+  cache.clear();
+  console.error(`[CACHE] 已清除所有缓存`);
+}
+
+module.exports = {
+  getCache,
+  setCache,
+  clearCache,
+  clearAllCache
+};

package/config.js

@@ -0,0 +1,24 @@
+/**
+ * 配置文件
+ */
+
+module.exports = {
+  sup8Host: "https://api-fat.yesno.com.cn",
+  sup8OpenHost: "https://open-fat.yesno.com.cn",
+  AK: "d913f38847214cd0a4169a959af001de",
+  SK: "1b555b531eafff1d7d6e34c5888f05e0",
+  CACHE_DURATION: 10 * 60 * 1000, // 10分钟缓存时间
+
+  // MySQL 数据库配置
+  MYSQL_HOST: "127.0.0.1",
+  MYSQL_PORT: "3306",
+  MYSQL_USER: "root",
+  MYSQL_PASS: "123456",
+  MYSQL_DB: "supercarrier_masterdata",
+
+  serverInfo: {
+    name: 'mcp-calculator-server',
+    version: '1.0.0'
+  },
+  protocolVersion: '2024-10-07'
+};

package/handlers/initialize.js

@@ -0,0 +1,22 @@
+/**
+ * 初始化处理器
+ * 处理 MCP 的 initialize 请求
+ */
+
+const { serverInfo, protocolVersion } = require('../config');
+
+/**
+ * 处理 initialize 请求
+ * @returns {Object} 初始化响应
+ */
+function handleInitialize() {
+  return {
+    protocolVersion: protocolVersion,
+    serverInfo: serverInfo,
+    capabilities: {
+      tools: {}  // 声明支持 tools 功能
+    }
+  };
+}
+
+module.exports = { handleInitialize };

package/handlers/tools.js

@@ -0,0 +1,47 @@
+/**
+ * 工具调用处理器
+ * 根据工具名称分发到对应的工具函数
+ */
+
+const { addNumbers, multiplyNumbers } = require('../tools/calculator');
+const { sup8Decode, sup8QueryCode, queryEnterpriseNo } = require('../tools/sup8');
+const { queryEnterprise: queryEnterpriseDB, queryByEnterpriseNo } = require('../tools/database');
+
+/**
+ * 处理工具调用请求
+ * @param {Object} params - 请求参数
+ * @param {string} params.name - 工具名称
+ * @param {Object} params.arguments - 工具参数
+ * @returns {Object} 工具执行结果
+ */
+async function handleToolCall(params) {
+  const { name, arguments: args } = params;
+
+  switch (name) {
+    case 'add_numbers':
+      return addNumbers(args.a, args.b);
+
+    case 'multiply_numbers':
+      return multiplyNumbers(args.a, args.b);
+
+    case 'sup8_decode':
+      return sup8Decode(args.code, args.enterpriseNo);
+
+    case 'sup8_query_code':
+      return sup8QueryCode(args.code, args.enterpriseNo);
+
+    case 'query_enterprise_no':
+      return queryEnterpriseNo(args.enter);
+
+    case 'query_enterprise_no_db':
+      return queryEnterpriseDB(args.name);
+
+    case 'query_by_enterprise_no':
+      return queryByEnterpriseNo(args.enterpriseNo);
+
+    default:
+      throw new Error(`未知工具: ${name}`);
+  }
+}
+
+module.exports = { handleToolCall };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-adder-simple",
-  "version": "1.1.2",
+  "version": "1.1.3",
   "main": "mcp-adder.js",
   "bin": {
     "mcp-adder-simple": "./bin/mcp-adder-simple.cmd"
@@ -26,9 +26,14 @@
     "mcp-adder.js",
     "bin/mcp-adder-simple.cmd",
     "package.json",
-    "README.md"
+    "README.md",
+    "config.js",
+    "cache.js",
+    "handlers/**",
+    "tools/**",
+    "utils/**"
   ],
   "dependencies": {
     "mysql2": "^3.16.1"
   }
-}
+}

package/tools/calculator.js

@@ -0,0 +1,51 @@
+/**
+ * 计算器工具模块
+ * 提供加法和乘法功能
+ */
+
+/**
+ * 将两个数字相加
+ * @param {number} a - 第一个数
+ * @param {number} b - 第二个数
+ * @returns {Object} MCP 格式的响应
+ */
+function addNumbers(a, b) {
+  if (typeof a !== 'number' || typeof b !== 'number') {
+    throw new Error('参数必须是数字');
+  }
+
+  return {
+    content: [
+      {
+        type: 'text',
+        text: `${a} + ${b} = ${a + b}`
+      }
+    ]
+  };
+}
+
+/**
+ * 将两个数字相乘
+ * @param {number} a - 第一个数
+ * @param {number} b - 第二个数
+ * @returns {Object} MCP 格式的响应
+ */
+function multiplyNumbers(a, b) {
+  if (typeof a !== 'number' || typeof b !== 'number') {
+    throw new Error('参数必须是数字');
+  }
+
+  return {
+    content: [
+      {
+        type: 'text',
+        text: `${a} × ${b} = ${a * b}`
+      }
+    ]
+  };
+}
+
+module.exports = {
+  addNumbers,
+  multiplyNumbers
+};

package/tools/index.js

@@ -0,0 +1,72 @@
+/**
+ * 工具列表注册
+ * 定义所有可用的 MCP 工具及其参数
+ */
+
+function handleToolsList() {
+  return {
+    tools: [
+      {
+        name: 'add_numbers',
+        description: '将两个数字相加',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            a: { type: 'number', description: '第一个数' },
+            b: { type: 'number', description: '第二个数' }
+          },
+          required: ['a', 'b']
+        }
+      },
+      {
+        name: 'multiply_numbers',
+        description: '将两个数字相乘',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            a: { type: 'number', description: '第一个数' },
+            b: { type: 'number', description: '第二个数' }
+          },
+          required: ['a', 'b']
+        }
+      },
+      {
+        name: 'sup8_decode',
+        description: '解码（带10分钟缓存）',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            code: { type: 'string', description: '数码' },
+            enterpriseNo: { type: 'string', description: '企业号' }
+          },
+          required: ['code', 'enterpriseNo']
+        }
+      },
+      {
+        name: 'sup8_query_code',
+        description: '查询数码信息（带10分钟缓存）',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            code: { type: 'string', description: '数码' },
+            enterpriseNo: { type: 'string', description: '企业号' }
+          },
+          required: ['code', 'enterpriseNo']
+        }
+      },
+      {
+        name: 'sup8_query_enterprise_no',
+        description: '查询企业号信息',
+        inputSchema: {
+          type: 'object',
+          properties: {
+            enterpriseNo: { type: 'string', description: '企业号或企业名称' }
+          }
+        }
+      }
+
+    ]
+  };
+}
+
+module.exports = { handleToolsList };

package/tools/sup8.js

@@ -0,0 +1,133 @@
+/**
+ * Sup8 API 工具模块
+ * 提供解码和查询数码信息功能（带缓存）
+ */
+
+const { sup8Host, AK, SK, sup8OpenHost } = require('../config');
+const { getCache, setCache } = require('../cache');
+const { get, post } = require('../utils/http');
+const db = require('../utils/dbhelper');
+/**
+ * Sup8 解码
+ * @param {string} code - 数码
+ * @param {string} enterpriseNo - 企业号
+ * @returns {Object} MCP 格式的响应
+ */
+async function sup8Decode(code, enterpriseNo) {
+  if (typeof code !== 'string' || typeof enterpriseNo !== 'string') {
+    throw new Error('参数必须是字符串');
+  }
+  const url = `${sup8Host}/digit/v1/decode/decode?digit=${encodeURIComponent(code)}&enterpriseNo=${encodeURIComponent(enterpriseNo)}`;
+
+  const result = await get(url);
+
+  return {
+    content: [
+      {
+        type: 'text',
+        text: JSON.stringify(result, null, 2)
+      }
+    ]
+  };
+}
+
+/**
+ * 查询数码信息（带缓存）
+ * @param {string} code - 数码
+ * @param {string} enterpriseNo - 企业号
+ * @returns {Object} MCP 格式的响应
+ */
+async function sup8QueryCode(code, enterpriseNo) {
+  if (typeof code !== 'string' || typeof enterpriseNo !== 'string') {
+    throw new Error('参数必须是字符串');
+  }
+
+  // 构建缓存键和 URL
+  const cacheKey = `query:${code}:${enterpriseNo}`;
+  const url = `${sup8Host}/digit/v2/digit/get-digit-info?code=${encodeURIComponent(code)}&enterpriseNo=${encodeURIComponent(enterpriseNo)}`;
+
+  // 先尝试从缓存获取
+  const cachedData = getCache(cacheKey);
+  if (cachedData) {
+    return {
+      content: [
+        {
+          type: 'text',
+          text: JSON.stringify(cachedData, null, 2)
+        }
+      ]
+    };
+  }
+
+  // 缓存未命中，调用 API
+  const result = await get(url);
+
+  // 将结果存入缓存
+  setCache(cacheKey, result);
+
+  return {
+    content: [
+      {
+        type: 'text',
+        text: JSON.stringify(result, null, 2)
+      }
+    ]
+  };
+}
+
+async function queryEnterpriseNo(enter) {
+
+  const sql = 'SELECT * FROM t_sup8_cc_enterprisenumber';
+  const params = [];
+
+  if (enter && enter.trim()) {
+    params.push(`%${enter}%`, `%${enter}%`);
+    const finalSql = sql + ' WHERE name LIKE ? OR enterpriseno LIKE ? ';
+    const rows = await db.query(finalSql, params); 
+    return formatMCPResponse(rows);
+  }
+  const rows = await db.query(sql, params);
+  return formatMCPResponse(rows);
+
+
+}
+
+
+async function getSup8Token() {
+  const cacheKey = 'sup8Token';
+  const url = `${sup8OpenHost}/cncop/ca/v1/applications/get-token`;
+  setCache(cacheKey, 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJhcHAiOiJ7XCJlbnRlcnByaXNlTm9cIjpcIjEzMTk4OTY1OTkxN1wiLFwiZW50ZXJwcmlzZVVzZXJJZFwiOlwiMjAwODM1MDA1NjgyNjU0NDEyOFwiLFwiYWNjZXNzS2V5XCI6XCJkOTEzZjM4ODQ3MjE0Y2QwYTQxNjlhOTU5YWYwMDFkZVwiLFwiYXBwSWRcIjpcIjIwMDkwOTMxNTg2NTQ0NTk5MDRcIixcImVudGVycHJpc2VVc2VyQWNjb3VudFwiOlwieHNfbHF4XCJ9Iiwic3ViIjoiMjAwOTA5MzE1ODY1NDQ1OTkwNCIsImlzcyI6Inllc25vLmNvbS5jbiIsImV4cCI6MTc2OTE3NjM5MDg0OSwiaWF0IjoxNzY5MTQ3NTkwODQ5LCJ2ZXJzaW9uIjoiMS4wIiwianRpIjoiMGJhZTM5NjIxYzA1NGI0ODk1MjQzYjIyMWZlZWM4MGYifQ.AapbK9QLI_pkiBBOZIaLl2ecH3vbrdNM17pmrybX1eA');
+  const cacheToken = getCache(cacheKey);
+  if (cacheToken) {
+    return cacheToken;
+  }
+
+  var body = {
+    "accessKey": AK,
+    "secretKeySign": SK
+  }
+  const result = await post(url, body)
+
+}
+
+/**
+ * 格式化为 MCP 响应格式
+ * @param {*} data - 数据对象或数组
+ * @returns {Object} MCP 格式的响应
+ */
+function formatMCPResponse(data) {
+  return {
+    content: [
+      {
+        type: 'text',
+        text: JSON.stringify(data, null, 2)
+      }
+    ]
+  };
+}
+
+module.exports = {
+  sup8Decode,
+  sup8QueryCode,
+  queryEnterpriseNo
+};

package/utils/dbhelper.js

@@ -0,0 +1,116 @@
+/**
+ * 数据库助手
+ * 提供通用的数据库操作方法
+ */
+
+const mysql = require('mysql2/promise');
+const { MYSQL_HOST, MYSQL_PORT, MYSQL_USER, MYSQL_PASS, MYSQL_DB } = require('../config');
+
+// 创建连接池
+const pool = mysql.createPool({
+  host: MYSQL_HOST,
+  port: MYSQL_PORT,
+  user: MYSQL_USER,
+  password: MYSQL_PASS,
+  database: MYSQL_DB,
+  waitForConnections: true,
+  connectionLimit: 10,
+  queueLimit: 0
+});
+
+/**
+ * 执行查询（SELECT）
+ * @param {string} sql - SQL 语句
+ * @param {Array} params - 查询参数
+ * @returns {Array} 查询结果行
+ */
+async function query(sql, params = []) {
+  try {
+    const [rows] = await pool.execute(sql, params);
+    console.log(`[DBHelper] 查询成功，返回 ${rows.length} 条记录`);
+    return rows;
+  } catch (error) {
+    console.error(`[DBHelper] 查询失败:`, error.message);
+    throw error;
+  }
+}
+
+/**
+ * 执行 INSERT/UPDATE/DELETE
+ * @param {string} sql - SQL 语句
+ * @param {Array} params - 查询参数
+ * @returns {Object} 结果信息 {affectedRows, insertId, etc.}
+ */
+async function execute(sql, params = []) {
+  try {
+    const [result] = await pool.execute(sql, params);
+    console.log(`[DBHelper] 执行成功，影响 ${result.affectedRows} 行`);
+    return result;
+  } catch (error) {
+    console.error(`[DBHelper] 执行失败:`, error.message);
+    throw error;
+  }
+}
+
+/**
+ * 查询单行数据
+ * @param {string} sql - SQL 语句
+ * @param {Array} params - 查询参数
+ * @returns {Object|null} 单行数据或 null
+ */
+async function queryOne(sql, params = []) {
+  const rows = await query(sql, params);
+  return rows.length > 0 ? rows[0] : null;
+}
+
+/**
+ * 开启事务
+ * @returns {Object} 事务连接对象
+ */
+async function beginTransaction() {
+  const conn = await pool.getConnection();
+  await conn.beginTransaction();
+  return conn;
+}
+
+/**
+ * 提交事务
+ * @param {Object} conn - 事务连接对象
+ */
+async function commit(conn) {
+  try {
+    await conn.commit();
+  } finally {
+    conn.release();
+  }
+}
+
+/**
+ * 回滚事务
+ * @param {Object} conn - 事务连接对象
+ */
+async function rollback(conn) {
+  try {
+    await conn.rollback();
+  } finally {
+    conn.release();
+  }
+}
+
+/**
+ * 关闭连接池
+ */
+async function close() {
+  await pool.end();
+  console.log('[DBHelper] 连接池已关闭');
+}
+
+module.exports = {
+  query,
+  execute,
+  queryOne,
+  beginTransaction,
+  commit,
+  rollback,
+  close
+};

package/utils/http.js

@@ -0,0 +1,141 @@
+/**
+ * HTTP 工具模块
+ * 提供 GET 和 POST 请求方法
+ */
+
+const https = require('https');
+const http = require('http');
+
+/**
+ * 通用 HTTP 请求方法
+ * @param {string} url - 请求的 URL
+ * @param {Object} options - 请求选项
+ * @param {string} [options.method='GET'] - 请求方法 (GET/POST)
+ * @param {Object} [options.headers] - 请求头
+ * @param {Object|string} [options.body] - 请求体（用于 POST）
+ * @returns {Promise<Object>} 返回 JSON 格式的响应数据
+ */
+function request(url, options = {}) {
+  return new Promise((resolve, reject) => {
+    const { method = 'GET', headers = {}, body } = options;
+
+    // 解析 URL
+    const urlObj = new URL(url);
+    const isHttps = urlObj.protocol === 'https:';
+    const requestFn = isHttps ? https.request : http.request;
+
+    // 设置默认请求头
+    const defaultHeaders = {
+      'Content-Type': 'application/json',
+      'User-Agent': 'MCP-Server/1.0.0'
+    };
+
+    const mergedHeaders = { ...defaultHeaders, ...headers };
+
+    // 如果有 body，需要设置 Content-Length
+    let bodyString = '';
+    if (body) {
+      bodyString = typeof body === 'string' ? body : JSON.stringify(body);
+      mergedHeaders['Content-Length'] = Buffer.byteLength(bodyString);
+    }
+
+    // 请求配置
+    const requestOptions = {
+      hostname: urlObj.hostname,
+      port: urlObj.port || (isHttps ? 443 : 80),
+      path: urlObj.pathname + urlObj.search,
+      method: method,
+      headers: mergedHeaders
+    };
+    
+    console.error(`[HTTP] 发送请求头:`, JSON.stringify(requestOptions.headers, null, 2));
+    console.error(`[HTTP] ${method} ${url}`);
+
+    // 发起请求
+    const req = requestFn(requestOptions, (res) => {
+      let data = '';
+
+      res.on('data', (chunk) => {
+        data += chunk;
+      });
+
+      res.on('end', () => {
+        try {
+          const jsonData = JSON.parse(data);
+          console.error(`[HTTP] 响应 (${res.statusCode}):`, JSON.stringify(jsonData, null, 2));
+          resolve(jsonData);
+        } catch (e) {
+          // 如果不是 JSON，返回原始数据
+          console.error(`[HTTP] 响应 (${res.statusCode}): ${data}`);
+          resolve({ data, statusCode: res.statusCode });
+        }
+      });
+    });
+
+    req.on('error', (err) => {
+      console.error(`[HTTP] 请求失败:`, err.message);
+      reject(new Error(`HTTP请求失败: ${err.message}`));
+    });
+
+    // 对于 POST 等有 body 的请求，写入数据
+    if (bodyString) {
+      req.write(bodyString);
+    }
+
+    req.end();
+  });
+}
+
+/**
+ * HTTP GET 请求
+ * @param {string} url - 请求的 URL
+ * @param {Object} [options] - 请求选项
+ * @param {Object} [options.headers] - 请求头
+ * @returns {Promise<Object>} 返回 JSON 格式的响应数据
+ */
+function get(url, options = {}) {
+  return request(url, { ...options, method: 'GET' });
+}
+
+/**
+ * HTTP POST 请求
+ * @param {string} url - 请求的 URL
+ * @param {Object|string} body - 请求体
+ * @param {Object} [options] - 请求选项
+ * @param {Object} [options.headers] - 请求头
+ * @returns {Promise<Object>} 返回 JSON 格式的响应数据
+ */
+function post(url, body, options = {}) {
+  return request(url, { ...options, method: 'POST', body });
+}
+
+/**
+ * HTTP PUT 请求
+ * @param {string} url - 请求的 URL
+ * @param {Object|string} body - 请求体
+ * @param {Object} [options] - 请求选项
+ * @param {Object} [options.headers] - 请求头
+ * @returns {Promise<Object>} 返回 JSON 格式的响应数据
+ */
+function put(url, body, options = {}) {
+  return request(url, { ...options, method: 'PUT', body });
+}
+
+/**
+ * HTTP DELETE 请求
+ * @param {string} url - 请求的 URL
+ * @param {Object} [options] - 请求选项
+ * @param {Object} [options.headers] - 请求头
+ * @returns {Promise<Object>} 返回 JSON 格式的响应数据
+ */
+function del(url, options = {}) {
+  return request(url, { ...options, method: 'DELETE' });
+}
+
+module.exports = {
+  request,
+  get,
+  post,
+  put,
+  delete: del
+};