@flyidea/ai-go-mcp 1.0.0

package/README.md

@@ -0,0 +1,61 @@
+# @flyidea/ai-go-mcp
+
+AI GO 官方的 Model Context Protocol (MCP) 本機代理伺服器。此套件可在您的本地環境運行 MCP Proxy 功能，協助諸如 **Cursor**, **Claude Desktop** 等支援 MCP 的智能代理，無縫連接受保護的 AI-GO 雲端（或地端）伺服器。
+
+## 🚀 快速安裝與使用
+
+### 1. 登入並驗證身份
+請在您的開發機（筆電或桌機）的終端機執行：
+```bash
+npx -y @flyidea/ai-go-mcp auth login
+```
+*這將開啟您的預設瀏覽器跳轉至 AI GO 的驗證頁面。登入成功後，您的存取憑證將被安全地存放在 `~/.ai-go/credentials.json` 之中。*
+
+### 2. 給 IDE (Cursor / Claude Desktop) 的連線設定
+登入成功後，您不需要自己跑程式，只需要將此 SDK 加入您愛用的 Desktop Agent (IDE) 的設定中。
+
+#### ➤ 若使用 Cursor IDE：
+前往 `Cursor Settings` -> `Features` -> `MCP`，點選「Add New MCP Server」：
+- **Name**: `ai-go`
+- **Type**: `command`
+- **Command**: `npx`
+- **Args**: `-y @flyidea/ai-go-mcp`
+
+#### ➤ 若使用 Claude Desktop：
+修改您的 `claude_desktop_config.json`，加入以下節點：
+```json
+{
+  "mcpServers": {
+    "ai-go": {
+      "command": "npx",
+      "args": [
+        "-y",
+        "@flyidea/ai-go-mcp"
+      ],
+      "env": {
+        "AIGO_API_URL": "https://ai-go.app" // 預設指向主站，若為內網測試可改為 localhost:8000
+      }
+    }
+  }
+}
+```
+
+## ⚙️ 環境變數 (進階)
+
+如果您是內部開發人員需要連接至其他測試伺服器，您可以在配置檔的 `env` (如上方針對 Claude Desktop 的示範) 或您的 CLI 環境中加入以下變數：
+
+- `AIGO_API_URL`：預設為 `https://ai-go.app`。您可以將其指向 `http://127.0.0.1:8000` 來測試本機伺服器。
+- `AIGO_APP_URL`：預設為 `https://ai-go.app`。這是 CLI 登入階段用來開起 Auth 授權彈窗的網址。
+
+## 📖 支援的 MCP Tools
+一旦正確配置，AI 代理將能直接探索並執行您的資料庫與 API 操作，例如：
+* `list_tables`
+* `describe_table`
+* `query_records`
+* `insert_record`
+* `update_record`
+* `delete_record`
+*(請注意：可操作的範圍嚴格遵照當前登入使用者的所屬租戶與權限進行過濾管控)*
+
+## License
+MIT

package/dist/auth.js

@@ -0,0 +1,64 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.saveCredentials = saveCredentials;
+exports.loadCredentials = loadCredentials;
+exports.clearCredentials = clearCredentials;
+exports.getValidAccessToken = getValidAccessToken;
+const promises_1 = __importDefault(require("fs/promises"));
+const path_1 = __importDefault(require("path"));
+const os_1 = __importDefault(require("os"));
+const axios_1 = __importDefault(require("axios"));
+const CONFIG_DIR = path_1.default.join(os_1.default.homedir(), '.ai-go');
+const CREDS_FILE = path_1.default.join(CONFIG_DIR, 'credentials.json');
+async function saveCredentials(creds) {
+    await promises_1.default.mkdir(CONFIG_DIR, { recursive: true });
+    await promises_1.default.writeFile(CREDS_FILE, JSON.stringify(creds, null, 2), { mode: 0o600 });
+}
+async function loadCredentials() {
+    try {
+        const data = await promises_1.default.readFile(CREDS_FILE, 'utf8');
+        return JSON.parse(data);
+    }
+    catch (e) {
+        return null; // file not found or invalid
+    }
+}
+async function clearCredentials() {
+    try {
+        await promises_1.default.unlink(CREDS_FILE);
+    }
+    catch (e) {
+        // ignore
+    }
+}
+async function getValidAccessToken(apiUrl) {
+    const creds = await loadCredentials();
+    if (!creds)
+        return null;
+    const now = Math.floor(Date.now() / 1000);
+    // Refresh 5 minutes before expiration to be safe
+    if (now > creds.expires_at - 300) {
+        try {
+            const response = await axios_1.default.post(`${apiUrl}/api/v1/auth/refresh`, {
+                refresh_token: creds.refresh_token
+            });
+            const newCreds = {
+                access_token: response.data.access_token,
+                refresh_token: response.data.refresh_token,
+                expires_at: Math.floor(Date.now() / 1000) + response.data.expires_in
+            };
+            await saveCredentials(newCreds);
+            return newCreds.access_token;
+        }
+        catch (error) {
+            console.error("\n[AI GO Auth] 無法刷新 Token，請重新登入 (npx @ai-go/mcp auth login)。");
+            if (error.response?.data)
+                console.error("詳細錯誤:", error.response.data);
+            return null;
+        }
+    }
+    return creds.access_token;
+}

package/dist/bin/cli.js

@@ -0,0 +1,87 @@
+#!/usr/bin/env node
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const commander_1 = require("commander");
+const express_1 = __importDefault(require("express"));
+const cors_1 = __importDefault(require("cors"));
+const open_1 = __importDefault(require("open"));
+const auth_js_1 = require("../auth.js");
+// Get version from package.json or hardcode for now
+const program = new commander_1.Command();
+program
+    .name('ai-go-mcp')
+    .description('AI GO MCP CLI utility')
+    .version('1.0.0');
+const auth = program.command('auth').description('Manage authorization for the MCP server');
+auth
+    .command('login')
+    .description('Login and authorize local MCP server')
+    .action(async () => {
+    const app = (0, express_1.default)();
+    app.use((0, cors_1.default)());
+    app.use(express_1.default.json());
+    const server = app.listen(0, async () => {
+        const address = server.address();
+        const port = typeof address === 'string' ? 0 : address?.port;
+        if (!port) {
+            console.error("無法取得啟動的 port。");
+            process.exit(1);
+        }
+        // 根據預設環境開啟網頁
+        const targetUrl = process.env.AIGO_APP_URL || 'https://ai-go.app';
+        const authUrl = `${targetUrl}/auth/cli?port=${port}`;
+        console.log(`\n正在開啟瀏覽器授權頁面...`);
+        console.log(`如果沒有自動跳轉，請手動複製以下連結：\n\n  ${authUrl}\n`);
+        try {
+            await (0, open_1.default)(authUrl);
+        }
+        catch (e) {
+            console.log("無法自動開啟瀏覽器，請手動前往上述連結。");
+        }
+    });
+    // 接收 Token callback
+    app.post('/callback', async (req, res) => {
+        const { access_token, refresh_token, expires_at } = req.body;
+        if (!access_token || !refresh_token) {
+            res.status(400).send("Bad Request: Missing tokens");
+            return;
+        }
+        try {
+            await (0, auth_js_1.saveCredentials)({ access_token, refresh_token, expires_at });
+            res.status(200).send("OK");
+            console.log("✅ 授權成功！");
+            console.log("您現在可以在 Claude Code 等 AI 工具中配置 AI GO MCP Server。");
+            console.log("\n範例 claude_desktop_config.json 格式:");
+            console.log(JSON.stringify({
+                mcpServers: {
+                    "ai-go": {
+                        "command": "npx",
+                        "args": ["-y", "@ai-go/mcp"],
+                        "env": {
+                            "AIGO_API_URL": "https://api.ai-go.app"
+                        }
+                    }
+                }
+            }, null, 2));
+            // 關閉 server 結束 process
+            server.close(() => process.exit(0));
+        }
+        catch (e) {
+            console.error("保存失敗", e);
+            res.status(500).send("Internal Server Error");
+            server.close(() => process.exit(1));
+        }
+    });
+});
+auth
+    .command('logout')
+    .description('Clear credentials')
+    .action(async () => {
+    await (0, auth_js_1.clearCredentials)();
+    console.log("✅ 已登出並清除本機授權。");
+    process.exit(0);
+});
+program.parse(process.argv);

package/dist/index.js

@@ -0,0 +1,52 @@
+#!/usr/bin/env node
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const readline_1 = __importDefault(require("readline"));
+const transport_js_1 = require("./transport.js");
+async function main() {
+    const apiUrl = process.env.AIGO_API_URL || 'https://ai-go.app';
+    const transport = new transport_js_1.McpTransport(apiUrl);
+    try {
+        await transport.connect();
+    }
+    catch (e) {
+        console.error("Failed to connect to AI GO MCP Server:", e.message);
+        process.exit(1);
+    }
+    // 當從伺服器收到 JSON-RPC 回應時，印到 stdout
+    transport.onMessage = (message) => {
+        console.log(JSON.stringify(message));
+    };
+    // 監聽 stdin (AI 工具發送的 JSON-RPC 請求)
+    const rl = readline_1.default.createInterface({
+        input: process.stdin,
+        output: process.stdout,
+        terminal: false
+    });
+    rl.on('line', async (line) => {
+        if (!line.trim())
+            return;
+        try {
+            const message = JSON.parse(line);
+            await transport.sendMessage(message);
+        }
+        catch (e) {
+            console.error("Failed to parse or send message:", e);
+        }
+    });
+    process.on('SIGINT', () => {
+        transport.close();
+        process.exit(0);
+    });
+    process.on('SIGTERM', () => {
+        transport.close();
+        process.exit(0);
+    });
+}
+main().catch(err => {
+    console.error("Fatal error:", err);
+    process.exit(1);
+});

package/dist/transport.js

@@ -0,0 +1,162 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.McpTransport = void 0;
+const http = __importStar(require("http"));
+const https = __importStar(require("https"));
+const axios_1 = __importDefault(require("axios"));
+const auth_1 = require("./auth");
+class McpTransport {
+    apiUrl;
+    sessionId = null;
+    req = null;
+    res = null;
+    // Callback when a JSON-RPC message is received from the server
+    onMessage;
+    // Callback when connection closes
+    onClose;
+    constructor(apiUrl) {
+        this.apiUrl = apiUrl.replace(/\/+$/, '');
+    }
+    async connect() {
+        const token = await (0, auth_1.getValidAccessToken)(this.apiUrl);
+        if (!token) {
+            throw new Error("Token 缺失或已過期，請執行 npx @ai-go/mcp auth login");
+        }
+        const sseUrl = `${this.apiUrl}/api/v1/mcp/sse`;
+        const urlObj = new URL(sseUrl);
+        const requestModule = urlObj.protocol === 'https:' ? https : http;
+        return new Promise((resolve, reject) => {
+            this.req = requestModule.get(sseUrl, {
+                headers: {
+                    'Authorization': `Bearer ${token}`,
+                    'Accept': 'text/event-stream'
+                }
+            }, (res) => {
+                this.res = res;
+                if (res.statusCode !== 200) {
+                    reject(new Error(`Failed to connect to SSE: HTTP ${res.statusCode}`));
+                    return;
+                }
+                let buffer = '';
+                res.on('data', (chunk) => {
+                    buffer += chunk.toString();
+                    const lines = buffer.split('\n');
+                    buffer = lines.pop() || ''; // keep the last incomplete line
+                    let currentEvent = '';
+                    let currentData = '';
+                    for (const line of lines) {
+                        if (line.trim() === '') {
+                            // Empty line means dispatch event
+                            if (currentEvent === 'endpoint' && currentData) {
+                                try {
+                                    const url = new URL(`${this.apiUrl}${currentData}`);
+                                    this.sessionId = url.searchParams.get('session_id');
+                                    resolve();
+                                }
+                                catch (err) {
+                                    reject(err);
+                                }
+                            }
+                            else if (currentEvent === 'message' && currentData) {
+                                try {
+                                    const parsed = JSON.parse(currentData);
+                                    if (this.onMessage)
+                                        this.onMessage(parsed);
+                                }
+                                catch (e) {
+                                    console.error("Parse error on message:", e);
+                                }
+                            }
+                            currentEvent = '';
+                            currentData = '';
+                        }
+                        else if (line.startsWith('event:')) {
+                            currentEvent = line.slice(6).trim();
+                        }
+                        else if (line.startsWith('data:')) {
+                            currentData += line.slice(5).trim();
+                        }
+                    }
+                });
+                res.on('end', () => {
+                    if (this.onClose)
+                        this.onClose();
+                });
+            });
+            this.req.on('error', (err) => {
+                if (!this.sessionId)
+                    reject(err);
+            });
+        });
+    }
+    async sendMessage(message) {
+        if (!this.sessionId) {
+            throw new Error("Not connected");
+        }
+        const token = await (0, auth_1.getValidAccessToken)(this.apiUrl);
+        if (!token) {
+            throw new Error("Token 缺失或已過期");
+        }
+        try {
+            await axios_1.default.post(`${this.apiUrl}/api/v1/mcp/message?session_id=${this.sessionId}`, message, {
+                headers: {
+                    'Authorization': `Bearer ${token}`,
+                    'Content-Type': 'application/json'
+                }
+            });
+        }
+        catch (error) {
+            console.error("Failed to send message:", error.response?.data || error.message);
+        }
+    }
+    close() {
+        if (this.res) {
+            this.res.destroy();
+            this.res = null;
+        }
+        if (this.req) {
+            this.req.destroy();
+            this.req = null;
+        }
+        if (this.onClose) {
+            this.onClose();
+        }
+    }
+}
+exports.McpTransport = McpTransport;

package/package.json

@@ -0,0 +1,36 @@
+{
+  "name": "@flyidea/ai-go-mcp",
+  "version": "1.0.0",
+  "description": "AI GO MCP (Model Context Protocol) CLI & Proxy Server",
+  "main": "dist/index.js",
+  "bin": "dist/bin/cli.js",
+  "files": [
+    "dist"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc",
+    "start": "node dist/index.js",
+    "dev": "tsc -w"
+  },
+  "author": "Urfit Tech",
+  "license": "MIT",
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.1",
+    "axios": "^1.6.8",
+    "commander": "^12.0.0",
+    "cors": "^2.8.5",
+    "dotenv": "^16.4.5",
+    "express": "^4.19.2",
+    "open": "^10.1.0"
+  },
+  "devDependencies": {
+    "@types/cors": "^2.8.17",
+    "@types/eventsource": "^1.1.15",
+    "@types/express": "^4.17.21",
+    "@types/node": "^20.12.7",
+    "typescript": "^5.4.5"
+  }
+}