@baitong-dev/skills-mcp 0.0.1

package/SkillsMcpServer.js

@@ -0,0 +1,201 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const mcp_js_1 = require("@modelcontextprotocol/sdk/server/mcp.js");
+const stdio_js_1 = require("@modelcontextprotocol/sdk/server/stdio.js");
+const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
+const fs_1 = __importDefault(require("fs"));
+const http_1 = require("http");
+const path_1 = __importDefault(require("path"));
+const zod_1 = __importDefault(require("zod"));
+const commander_1 = require("commander");
+//  eslint-disable-next-line @typescript-eslint/no-require-imports
+const yamlFront = require('yaml-front-matter');
+// 命令行参数处理
+const program = new commander_1.Command()
+    .option('--transport <stdio|http|sse>', 'transport type', 'stdio')
+    .option('--port <number>', 'port for HTTP/SSE transport', '3000')
+    .parse(process.argv);
+const cliOptions = program.opts();
+const allowedTransports = ['stdio', 'http'];
+if (!allowedTransports.includes(cliOptions.transport)) {
+    console.error(`Invalid --transport value: '${cliOptions.transport}'. Must be one of: stdio, http, sse.`);
+    process.exit(1);
+}
+const MCP_NAME = 'Skills MCP';
+const MCP_VERSION = '0.0.1';
+const PORT = parseInt(cliOptions.port, 10);
+const CONFIG_DIR = process.env.MCP_HOME_DIR;
+const TRANSPORT_TYPE = (cliOptions.transport || 'stdio');
+const logInfo = (...args) => {
+    if (TRANSPORT_TYPE === 'stdio') {
+        console.error(...args);
+    }
+    else {
+        console.log(...args);
+    }
+};
+function getSkillsRootDir() {
+    const dir = path_1.default.join(CONFIG_DIR, 'skills');
+    if (!fs_1.default.existsSync(dir)) {
+        fs_1.default.mkdirSync(dir, {
+            recursive: true
+        });
+    }
+    return dir;
+}
+function loadSkillsMetadatas() {
+    const skillsRootDirs = fs_1.default.readdirSync(getSkillsRootDir());
+    return skillsRootDirs
+        .map(skillName => {
+        try {
+            const { name, description } = readSkillJson(getSkillFilePath(skillName));
+            return {
+                name: name.trim(),
+                description: description.trim()
+            };
+        }
+        catch (_) {
+            return null;
+        }
+    })
+        .filter(item => item !== null);
+}
+function getSkillCwd(skillName) {
+    return path_1.default.join(getSkillsRootDir(), skillName);
+}
+function getSkillFilePath(skillName) {
+    return path_1.default.join(getSkillCwd(skillName), 'SKILL.md');
+}
+function readSkillJson(skillFilePath) {
+    const skillMd = fs_1.default.readFileSync(skillFilePath, 'utf-8');
+    return yamlFront.loadFront(skillMd);
+}
+function registerLoadSkill(server) {
+    server.registerTool('load_skill', {
+        description: `
+Available skills:
+${loadSkillsMetadatas()
+            .map(skill => `- ${skill.name}: ${skill.description}`)
+            .join('\n')}
+
+Returns the skill's prompt and context.`,
+        inputSchema: zod_1.default.object({
+            skillName: zod_1.default.string().describe('Name of skill to load')
+        })
+    }, async ({ skillName }) => {
+        // 返回skill元素据，只包含name和description
+        const skillCwd = getSkillCwd(skillName);
+        const skillFilePath = getSkillFilePath(skillName);
+        const json = readSkillJson(skillFilePath);
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: `Base directory for this skill: ${skillCwd}
+
+${json.__content.trim()}`
+                }
+            ]
+        };
+    });
+}
+// 创建MCP服务器实例
+function createServerInstance() {
+    const server = new mcp_js_1.McpServer({
+        name: MCP_NAME,
+        version: MCP_VERSION
+    }, {
+        capabilities: {
+            logging: {}
+        }
+    });
+    // 注册所有工具
+    registerLoadSkill(server);
+    return server;
+}
+async function main() {
+    if (TRANSPORT_TYPE === 'http') {
+        const httpServer = (0, http_1.createServer)(async (req, res) => {
+            const url = new URL(req.url || '', `http://${req.headers.host}`).pathname;
+            // 设置 CORS 头
+            res.setHeader('Access-Control-Allow-Origin', '*');
+            res.setHeader('Access-Control-Allow-Methods', 'GET,POST,OPTIONS,DELETE');
+            res.setHeader('Access-Control-Allow-Headers', 'Content-Type, MCP-Session-Id, mcp-session-id');
+            // 处理预检请求
+            if (req.method === 'OPTIONS') {
+                res.writeHead(200);
+                res.end();
+                return;
+            }
+            try {
+                // 为每个请求创建新的服务器实例
+                const requestServer = createServerInstance();
+                if (url === '/mcp') {
+                    const transport = new streamableHttp_js_1.StreamableHTTPServerTransport({
+                        sessionIdGenerator: undefined
+                    });
+                    await requestServer.connect(transport);
+                    await transport.handleRequest(req, res);
+                }
+                else if (url === '/health') {
+                    res.writeHead(200, { 'Content-Type': 'application/json' });
+                    res.end(JSON.stringify({ status: 'ok', transport: TRANSPORT_TYPE }));
+                }
+                else if (url === '/info') {
+                    res.writeHead(200, { 'Content-Type': 'application/json' });
+                    res.end(JSON.stringify({
+                        name: `${MCP_NAME} Server`,
+                        version: MCP_VERSION,
+                        transport: TRANSPORT_TYPE,
+                        endpoints: {
+                            mcp: '/mcp',
+                            health: '/health',
+                            info: '/info'
+                        }
+                    }));
+                }
+                else {
+                    res.writeHead(404);
+                    res.end('Not found');
+                }
+            }
+            catch (error) {
+                console.error('Error handling request: ', error);
+                if (!res.headersSent) {
+                    res.writeHead(500);
+                    res.end('Internal Server Error');
+                }
+            }
+        });
+        httpServer.listen(PORT, () => {
+            logInfo(`${MCP_NAME} ${TRANSPORT_TYPE.toUpperCase()} server started.`);
+            logInfo(`MCP endpoint: http://localhost:${PORT}/mcp`);
+            logInfo(`Health: http://localhost:${PORT}/health`);
+            logInfo(`Info: http://localhost:${PORT}/info`);
+        });
+    }
+    else {
+        // stdio 模式
+        const server = createServerInstance();
+        const transport = new stdio_js_1.StdioServerTransport();
+        await server.connect(transport);
+        logInfo(`${MCP_NAME} ${TRANSPORT_TYPE.toUpperCase()} server started.`);
+    }
+}
+// 优雅关闭
+process.on('SIGINT', async () => {
+    logInfo(`\nShutting down ${MCP_NAME} server...`);
+    process.exit(0);
+});
+process.on('SIGTERM', async () => {
+    logInfo(`\nReceived termination signal, shutting down ${MCP_NAME} server...`);
+    process.exit(0);
+});
+// 启动服务器
+main().catch(error => {
+    logInfo('Starting server failed:', error);
+    process.exit(1);
+});

package/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "@baitong-dev/skills-mcp",
+  "version": "0.0.1",
+  "main": "SkillsMcpServer.js",
+  "bin": {
+    "@baitong-dev/skills-mcp": "./SkillsMcpServer.js"
+  },
+  "files": [
+    "SkillsMcpServer.js",
+    "README.md"
+  ],
+  "scripts": {},
+  "keywords": [
+    "mcp",
+    "skills"
+  ],
+  "description": "skills-mcp",
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.25.1",
+    "commander": "^14.0.3",
+    "yaml-front-matter": "^4.1.1",
+    "zod": "^4.3.4"
+  },
+  "publishConfig": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org"
+  }
+}