axhub-mcp-bridge 1.0.30

package/dist/scripts/utils.js

@@ -0,0 +1,353 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.writeFile = exports.mkdir = exports.access = void 0;
+exports.colorText = colorText;
+exports.getUserManifestPath = getUserManifestPath;
+exports.getSystemManifestPath = getSystemManifestPath;
+exports.getMainPath = getMainPath;
+exports.ensureExecutionPermissions = ensureExecutionPermissions;
+exports.createManifestContent = createManifestContent;
+exports.tryRegisterUserLevelHost = tryRegisterUserLevelHost;
+exports.registerWithElevatedPermissions = registerWithElevatedPermissions;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const os_1 = __importDefault(require("os"));
+const child_process_1 = require("child_process");
+const util_1 = require("util");
+const constant_1 = require("./constant");
+exports.access = (0, util_1.promisify)(fs_1.default.access);
+exports.mkdir = (0, util_1.promisify)(fs_1.default.mkdir);
+exports.writeFile = (0, util_1.promisify)(fs_1.default.writeFile);
+/**
+ * 打印彩色文本
+ */
+function colorText(text, color) {
+    const colors = {
+        red: '\x1b[31m',
+        green: '\x1b[32m',
+        yellow: '\x1b[33m',
+        blue: '\x1b[34m',
+        reset: '\x1b[0m',
+    };
+    return colors[color] + text + colors.reset;
+}
+/**
+ * Get user-level manifest file path
+ */
+function getUserManifestPath() {
+    if (os_1.default.platform() === 'win32') {
+        // Windows: %APPDATA%\Google\Chrome\NativeMessagingHosts\
+        return path_1.default.join(process.env.APPDATA || path_1.default.join(os_1.default.homedir(), 'AppData', 'Roaming'), 'Google', 'Chrome', 'NativeMessagingHosts', `${constant_1.HOST_NAME}.json`);
+    }
+    else if (os_1.default.platform() === 'darwin') {
+        // macOS: ~/Library/Application Support/Google/Chrome/NativeMessagingHosts/
+        return path_1.default.join(os_1.default.homedir(), 'Library', 'Application Support', 'Google', 'Chrome', 'NativeMessagingHosts', `${constant_1.HOST_NAME}.json`);
+    }
+    else {
+        // Linux: ~/.config/google-chrome/NativeMessagingHosts/
+        return path_1.default.join(os_1.default.homedir(), '.config', 'google-chrome', 'NativeMessagingHosts', `${constant_1.HOST_NAME}.json`);
+    }
+}
+/**
+ * Get system-level manifest file path
+ */
+function getSystemManifestPath() {
+    if (os_1.default.platform() === 'win32') {
+        // Windows: %ProgramFiles%\Google\Chrome\NativeMessagingHosts\
+        return path_1.default.join(process.env.ProgramFiles || 'C:\\Program Files', 'Google', 'Chrome', 'NativeMessagingHosts', `${constant_1.HOST_NAME}.json`);
+    }
+    else if (os_1.default.platform() === 'darwin') {
+        // macOS: /Library/Google/Chrome/NativeMessagingHosts/
+        return path_1.default.join('/Library', 'Google', 'Chrome', 'NativeMessagingHosts', `${constant_1.HOST_NAME}.json`);
+    }
+    else {
+        // Linux: /etc/opt/chrome/native-messaging-hosts/
+        return path_1.default.join('/etc', 'opt', 'chrome', 'native-messaging-hosts', `${constant_1.HOST_NAME}.json`);
+    }
+}
+/**
+ * Get native host startup script file path
+ */
+async function getMainPath() {
+    try {
+        const packageDistDir = path_1.default.join(__dirname, '..');
+        const wrapperScriptName = process.platform === 'win32' ? 'run_host.bat' : 'run_host.sh';
+        const absoluteWrapperPath = path_1.default.resolve(packageDistDir, wrapperScriptName);
+        return absoluteWrapperPath;
+    }
+    catch (error) {
+        console.log(colorText('Cannot find global package path, using current directory', 'yellow'));
+        throw error;
+    }
+}
+/**
+ * 确保关键文件具有执行权限
+ */
+async function ensureExecutionPermissions() {
+    try {
+        const packageDistDir = path_1.default.join(__dirname, '..');
+        if (process.platform === 'win32') {
+            // Windows 平台处理
+            await ensureWindowsFilePermissions(packageDistDir);
+            return;
+        }
+        // Unix/Linux 平台处理
+        const filesToCheck = [
+            path_1.default.join(packageDistDir, 'index.js'),
+            path_1.default.join(packageDistDir, 'run_host.sh'),
+            path_1.default.join(packageDistDir, 'cli.js'),
+        ];
+        for (const filePath of filesToCheck) {
+            if (fs_1.default.existsSync(filePath)) {
+                try {
+                    fs_1.default.chmodSync(filePath, '755');
+                    console.log(colorText(`✓ Set execution permissions for ${path_1.default.basename(filePath)}`, 'green'));
+                }
+                catch (err) {
+                    console.warn(colorText(`⚠️ Unable to set execution permissions for ${path_1.default.basename(filePath)}: ${err.message}`, 'yellow'));
+                }
+            }
+            else {
+                console.warn(colorText(`⚠️ File not found: ${filePath}`, 'yellow'));
+            }
+        }
+    }
+    catch (error) {
+        console.warn(colorText(`⚠️ Error ensuring execution permissions: ${error.message}`, 'yellow'));
+    }
+}
+/**
+ * Windows 平台文件权限处理
+ */
+async function ensureWindowsFilePermissions(packageDistDir) {
+    const filesToCheck = [
+        path_1.default.join(packageDistDir, 'index.js'),
+        path_1.default.join(packageDistDir, 'run_host.bat'),
+        path_1.default.join(packageDistDir, 'cli.js'),
+    ];
+    for (const filePath of filesToCheck) {
+        if (fs_1.default.existsSync(filePath)) {
+            try {
+                // 检查文件是否为只读，如果是则移除只读属性
+                const stats = fs_1.default.statSync(filePath);
+                if (!(stats.mode & parseInt('200', 8))) {
+                    // 检查写权限
+                    // 尝试移除只读属性
+                    fs_1.default.chmodSync(filePath, stats.mode | parseInt('200', 8));
+                    console.log(colorText(`✓ Removed read-only attribute from ${path_1.default.basename(filePath)}`, 'green'));
+                }
+                // 验证文件可读性
+                fs_1.default.accessSync(filePath, fs_1.default.constants.R_OK);
+                console.log(colorText(`✓ Verified file accessibility for ${path_1.default.basename(filePath)}`, 'green'));
+            }
+            catch (err) {
+                console.warn(colorText(`⚠️ Unable to verify file permissions for ${path_1.default.basename(filePath)}: ${err.message}`, 'yellow'));
+            }
+        }
+        else {
+            console.warn(colorText(`⚠️ File not found: ${filePath}`, 'yellow'));
+        }
+    }
+}
+/**
+ * Create Native Messaging host manifest content
+ */
+async function createManifestContent() {
+    const mainPath = await getMainPath();
+    return {
+        name: constant_1.HOST_NAME,
+        description: constant_1.DESCRIPTION,
+        path: mainPath, // Node.js可执行文件路径
+        type: 'stdio',
+        allowed_origins: [`chrome-extension://${constant_1.EXTENSION_ID}/`],
+    };
+}
+/**
+ * 验证Windows注册表项是否存在
+ */
+function verifyWindowsRegistryEntry(registryKey, expectedPath) {
+    if (os_1.default.platform() !== 'win32') {
+        return true; // 非Windows平台跳过验证
+    }
+    try {
+        const result = (0, child_process_1.execSync)(`reg query "${registryKey}" /ve`, { encoding: 'utf8', stdio: 'pipe' });
+        const lines = result.split('\n');
+        for (const line of lines) {
+            if (line.includes('REG_SZ') && line.includes(expectedPath.replace(/\\/g, '\\\\'))) {
+                return true;
+            }
+        }
+        return false;
+    }
+    catch (error) {
+        return false;
+    }
+}
+/**
+ * 尝试注册用户级别的Native Messaging主机
+ */
+async function tryRegisterUserLevelHost() {
+    try {
+        console.log(colorText('Attempting to register user-level Native Messaging host...', 'blue'));
+        // 1. 确保执行权限
+        await ensureExecutionPermissions();
+        // 2. 确定清单文件路径
+        const manifestPath = getUserManifestPath();
+        // 3. 确保目录存在
+        await (0, exports.mkdir)(path_1.default.dirname(manifestPath), { recursive: true });
+        // 4. 创建清单内容
+        const manifest = await createManifestContent();
+        console.log('manifest path==>', manifest, manifestPath);
+        // 5. 写入清单文件
+        await (0, exports.writeFile)(manifestPath, JSON.stringify(manifest, null, 2));
+        if (os_1.default.platform() === 'win32') {
+            const registryKey = `HKCU\\Software\\Google\\Chrome\\NativeMessagingHosts\\${constant_1.HOST_NAME}`;
+            try {
+                // 确保路径使用正确的转义格式
+                const escapedPath = manifestPath.replace(/\\/g, '\\\\');
+                const regCommand = `reg add "${registryKey}" /ve /t REG_SZ /d "${escapedPath}" /f`;
+                console.log(colorText(`Executing registry command: ${regCommand}`, 'blue'));
+                (0, child_process_1.execSync)(regCommand, { stdio: 'pipe' });
+                // 验证注册表项是否创建成功
+                if (verifyWindowsRegistryEntry(registryKey, manifestPath)) {
+                    console.log(colorText('✓ Successfully created Windows registry entry', 'green'));
+                }
+                else {
+                    console.log(colorText('⚠️ Registry entry created but verification failed', 'yellow'));
+                }
+            }
+            catch (error) {
+                console.log(colorText(`⚠️ Unable to create Windows registry entry: ${error.message}`, 'yellow'));
+                console.log(colorText(`Registry key: ${registryKey}`, 'yellow'));
+                console.log(colorText(`Manifest path: ${manifestPath}`, 'yellow'));
+                return false; // Windows上如果注册表项创建失败，整个注册过程应该视为失败
+            }
+        }
+        console.log(colorText('Successfully registered user-level Native Messaging host!', 'green'));
+        return true;
+    }
+    catch (error) {
+        console.log(colorText(`User-level registration failed: ${error instanceof Error ? error.message : String(error)}`, 'yellow'));
+        return false;
+    }
+}
+// 导入is-admin包（仅在Windows平台使用）
+let isAdmin = async () => false;
+if (process.platform === 'win32') {
+    try {
+        // 动态导入 ESM 模块
+        isAdmin = async () => {
+            const mod = await import('is-admin');
+            return mod.default();
+        };
+    }
+    catch (error) {
+        console.warn('缺少is-admin依赖，Windows平台下可能无法正确检测管理员权限');
+        console.warn(error);
+    }
+}
+/**
+ * 使用提升权限注册系统级清单
+ */
+async function registerWithElevatedPermissions() {
+    try {
+        console.log(colorText('Attempting to register system-level manifest...', 'blue'));
+        // 1. 确保执行权限
+        await ensureExecutionPermissions();
+        // 2. 准备清单内容
+        const manifest = await createManifestContent();
+        // 3. 获取系统级清单路径
+        const manifestPath = getSystemManifestPath();
+        // 4. 创建临时清单文件
+        const tempManifestPath = path_1.default.join(os_1.default.tmpdir(), `${constant_1.HOST_NAME}.json`);
+        await (0, exports.writeFile)(tempManifestPath, JSON.stringify(manifest, null, 2));
+        // 5. 检测是否已经有管理员权限
+        const isRoot = process.getuid && process.getuid() === 0; // Unix/Linux/Mac
+        const hasAdminRights = process.platform === 'win32' ? isAdmin() : false; // Windows平台检测管理员权限
+        const hasElevatedPermissions = isRoot || hasAdminRights;
+        // 准备命令
+        const command = os_1.default.platform() === 'win32'
+            ? `if not exist "${path_1.default.dirname(manifestPath)}" mkdir "${path_1.default.dirname(manifestPath)}" && copy "${tempManifestPath}" "${manifestPath}"`
+            : `mkdir -p "${path_1.default.dirname(manifestPath)}" && cp "${tempManifestPath}" "${manifestPath}" && chmod 644 "${manifestPath}"`;
+        if (hasElevatedPermissions) {
+            // 已经有管理员权限，直接执行命令
+            try {
+                // 创建目录
+                if (!fs_1.default.existsSync(path_1.default.dirname(manifestPath))) {
+                    fs_1.default.mkdirSync(path_1.default.dirname(manifestPath), { recursive: true });
+                }
+                // 复制文件
+                fs_1.default.copyFileSync(tempManifestPath, manifestPath);
+                // 设置权限（非Windows平台）
+                if (os_1.default.platform() !== 'win32') {
+                    fs_1.default.chmodSync(manifestPath, '644');
+                }
+                console.log(colorText('System-level manifest registration successful!', 'green'));
+            }
+            catch (error) {
+                console.error(colorText(`System-level manifest installation failed: ${error.message}`, 'red'));
+                throw error;
+            }
+        }
+        else {
+            // 没有管理员权限，打印手动操作提示
+            console.log(colorText('⚠️ Administrator privileges required for system-level installation', 'yellow'));
+            console.log(colorText('Please run one of the following commands with administrator privileges:', 'blue'));
+            if (os_1.default.platform() === 'win32') {
+                console.log(colorText('  1. Open Command Prompt as Administrator and run:', 'blue'));
+                console.log(colorText(`     ${command}`, 'cyan'));
+            }
+            else {
+                console.log(colorText('  1. Run with sudo:', 'blue'));
+                console.log(colorText(`     sudo ${command}`, 'cyan'));
+            }
+            console.log(colorText('  2. Or run the registration command with elevated privileges:', 'blue'));
+            console.log(colorText(`     sudo ${constant_1.COMMAND_NAME} register --system`, 'cyan'));
+            throw new Error('Administrator privileges required for system-level installation');
+        }
+        // 6. Windows特殊处理 - 设置系统级注册表
+        if (os_1.default.platform() === 'win32') {
+            const registryKey = `HKLM\\Software\\Google\\Chrome\\NativeMessagingHosts\\${constant_1.HOST_NAME}`;
+            // 确保路径使用正确的转义格式
+            const escapedPath = manifestPath.replace(/\\/g, '\\\\');
+            const regCommand = `reg add "${registryKey}" /ve /t REG_SZ /d "${escapedPath}" /f`;
+            console.log(colorText(`Creating system registry entry: ${registryKey}`, 'blue'));
+            console.log(colorText(`Manifest path: ${manifestPath}`, 'blue'));
+            if (hasElevatedPermissions) {
+                // 已经有管理员权限，直接执行注册表命令
+                try {
+                    (0, child_process_1.execSync)(regCommand, { stdio: 'pipe' });
+                    // 验证注册表项是否创建成功
+                    if (verifyWindowsRegistryEntry(registryKey, manifestPath)) {
+                        console.log(colorText('Windows registry entry created successfully!', 'green'));
+                    }
+                    else {
+                        console.log(colorText('⚠️ Registry entry created but verification failed', 'yellow'));
+                    }
+                }
+                catch (error) {
+                    console.error(colorText(`Windows registry entry creation failed: ${error.message}`, 'red'));
+                    console.error(colorText(`Command: ${regCommand}`, 'red'));
+                    throw error;
+                }
+            }
+            else {
+                // 没有管理员权限，打印手动操作提示
+                console.log(colorText('⚠️ Administrator privileges required for Windows registry modification', 'yellow'));
+                console.log(colorText('Please run the following command as Administrator:', 'blue'));
+                console.log(colorText(`  ${regCommand}`, 'cyan'));
+                console.log(colorText('Or run the registration command with elevated privileges:', 'blue'));
+                console.log(colorText(`  Run Command Prompt as Administrator and execute: ${constant_1.COMMAND_NAME} register --system`, 'cyan'));
+                throw new Error('Administrator privileges required for Windows registry modification');
+            }
+        }
+    }
+    catch (error) {
+        console.error(colorText(`注册失败: ${error.message}`, 'red'));
+        throw error;
+    }
+}
+//# sourceMappingURL=utils.js.map

package/dist/scripts/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/scripts/utils.ts"],"names":[],"mappings":";;;;;;AAcA,8BAUC;AAKD,kDA+BC;AAKD,sDAiBC;AAKD,kCAUC;AAKD,gEAuCC;AAgDD,sDAUC;AA2BD,4DA0DC;AAoBD,0EAoIC;AApbD,4CAAoB;AACpB,gDAAwB;AACxB,4CAAoB;AACpB,iDAAyC;AACzC,+BAAiC;AACjC,yCAAgF;AAEnE,QAAA,MAAM,GAAG,IAAA,gBAAS,EAAC,YAAE,CAAC,MAAM,CAAC,CAAC;AAC9B,QAAA,KAAK,GAAG,IAAA,gBAAS,EAAC,YAAE,CAAC,KAAK,CAAC,CAAC;AAC5B,QAAA,SAAS,GAAG,IAAA,gBAAS,EAAC,YAAE,CAAC,SAAS,CAAC,CAAC;AAEjD;;GAEG;AACH,SAAgB,SAAS,CAAC,IAAY,EAAE,KAAa;IACnD,MAAM,MAAM,GAA2B;QACrC,GAAG,EAAE,UAAU;QACf,KAAK,EAAE,UAAU;QACjB,MAAM,EAAE,UAAU;QAClB,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,SAAS;KACjB,CAAC;IAEF,OAAO,MAAM,CAAC,KAAK,CAAC,GAAG,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC;AAC7C,CAAC;AAED;;GAEG;AACH,SAAgB,mBAAmB;IACjC,IAAI,YAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,EAAE,CAAC;QAC9B,yDAAyD;QACzD,OAAO,cAAI,CAAC,IAAI,CACd,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,cAAI,CAAC,IAAI,CAAC,YAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,SAAS,CAAC,EACpE,QAAQ,EACR,QAAQ,EACR,sBAAsB,EACtB,GAAG,oBAAS,OAAO,CACpB,CAAC;IACJ,CAAC;SAAM,IAAI,YAAE,CAAC,QAAQ,EAAE,KAAK,QAAQ,EAAE,CAAC;QACtC,2EAA2E;QAC3E,OAAO,cAAI,CAAC,IAAI,CACd,YAAE,CAAC,OAAO,EAAE,EACZ,SAAS,EACT,qBAAqB,EACrB,QAAQ,EACR,QAAQ,EACR,sBAAsB,EACtB,GAAG,oBAAS,OAAO,CACpB,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,uDAAuD;QACvD,OAAO,cAAI,CAAC,IAAI,CACd,YAAE,CAAC,OAAO,EAAE,EACZ,SAAS,EACT,eAAe,EACf,sBAAsB,EACtB,GAAG,oBAAS,OAAO,CACpB,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,qBAAqB;IACnC,IAAI,YAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,EAAE,CAAC;QAC9B,8DAA8D;QAC9D,OAAO,cAAI,CAAC,IAAI,CACd,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,mBAAmB,EAC/C,QAAQ,EACR,QAAQ,EACR,sBAAsB,EACtB,GAAG,oBAAS,OAAO,CACpB,CAAC;IACJ,CAAC;SAAM,IAAI,YAAE,CAAC,QAAQ,EAAE,KAAK,QAAQ,EAAE,CAAC;QACtC,sDAAsD;QACtD,OAAO,cAAI,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,sBAAsB,EAAE,GAAG,oBAAS,OAAO,CAAC,CAAC;IAChG,CAAC;SAAM,CAAC;QACN,iDAAiD;QACjD,OAAO,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,wBAAwB,EAAE,GAAG,oBAAS,OAAO,CAAC,CAAC;IAC3F,CAAC;AACH,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,WAAW;IAC/B,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAClD,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,aAAa,CAAC;QACxF,MAAM,mBAAmB,GAAG,cAAI,CAAC,OAAO,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC;QAC5E,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,0DAA0D,EAAE,QAAQ,CAAC,CAAC,CAAC;QAC7F,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,0BAA0B;IAC9C,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAElD,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;YACjC,eAAe;YACf,MAAM,4BAA4B,CAAC,cAAc,CAAC,CAAC;YACnD,OAAO;QACT,CAAC;QAED,kBAAkB;QAClB,MAAM,YAAY,GAAG;YACnB,cAAI,CAAC,IAAI,CAAC,cAAc,EAAE,UAAU,CAAC;YACrC,cAAI,CAAC,IAAI,CAAC,cAAc,EAAE,aAAa,CAAC;YACxC,cAAI,CAAC,IAAI,CAAC,cAAc,EAAE,QAAQ,CAAC;SACpC,CAAC;QAEF,KAAK,MAAM,QAAQ,IAAI,YAAY,EAAE,CAAC;YACpC,IAAI,YAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC5B,IAAI,CAAC;oBACH,YAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;oBAC9B,OAAO,CAAC,GAAG,CACT,SAAS,CAAC,mCAAmC,cAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,EAAE,OAAO,CAAC,CACjF,CAAC;gBACJ,CAAC;gBAAC,OAAO,GAAQ,EAAE,CAAC;oBAClB,OAAO,CAAC,IAAI,CACV,SAAS,CACP,8CAA8C,cAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,GAAG,CAAC,OAAO,EAAE,EACvF,QAAQ,CACT,CACF,CAAC;gBACJ,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,sBAAsB,QAAQ,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC;YACtE,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,4CAA4C,KAAK,CAAC,OAAO,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC;IACjG,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,4BAA4B,CAAC,cAAsB;IAChE,MAAM,YAAY,GAAG;QACnB,cAAI,CAAC,IAAI,CAAC,cAAc,EAAE,UAAU,CAAC;QACrC,cAAI,CAAC,IAAI,CAAC,cAAc,EAAE,cAAc,CAAC;QACzC,cAAI,CAAC,IAAI,CAAC,cAAc,EAAE,QAAQ,CAAC;KACpC,CAAC;IAEF,KAAK,MAAM,QAAQ,IAAI,YAAY,EAAE,CAAC;QACpC,IAAI,YAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC;gBACH,uBAAuB;gBACvB,MAAM,KAAK,GAAG,YAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBACpC,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACvC,QAAQ;oBACR,WAAW;oBACX,YAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;oBACxD,OAAO,CAAC,GAAG,CACT,SAAS,CAAC,sCAAsC,cAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,EAAE,OAAO,CAAC,CACpF,CAAC;gBACJ,CAAC;gBAED,UAAU;gBACV,YAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,YAAE,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;gBAC3C,OAAO,CAAC,GAAG,CACT,SAAS,CAAC,qCAAqC,cAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,EAAE,OAAO,CAAC,CACnF,CAAC;YACJ,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,OAAO,CAAC,IAAI,CACV,SAAS,CACP,4CAA4C,cAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,GAAG,CAAC,OAAO,EAAE,EACrF,QAAQ,CACT,CACF,CAAC;YACJ,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,sBAAsB,QAAQ,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;AACH,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,qBAAqB;IACzC,MAAM,QAAQ,GAAG,MAAM,WAAW,EAAE,CAAC;IAErC,OAAO;QACL,IAAI,EAAE,oBAAS;QACf,WAAW,EAAE,sBAAW;QACxB,IAAI,EAAE,QAAQ,EAAE,iBAAiB;QACjC,IAAI,EAAE,OAAO;QACb,eAAe,EAAE,CAAC,sBAAsB,uBAAY,GAAG,CAAC;KACzD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,0BAA0B,CAAC,WAAmB,EAAE,YAAoB;IAC3E,IAAI,YAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,CAAC,iBAAiB;IAChC,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAA,wBAAQ,EAAC,cAAc,WAAW,OAAO,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/F,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACjC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,EAAE,CAAC;gBAClF,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,wBAAwB;IAC5C,IAAI,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,4DAA4D,EAAE,MAAM,CAAC,CAAC,CAAC;QAE7F,YAAY;QACZ,MAAM,0BAA0B,EAAE,CAAC;QAEnC,cAAc;QACd,MAAM,YAAY,GAAG,mBAAmB,EAAE,CAAC;QAE3C,YAAY;QACZ,MAAM,IAAA,aAAK,EAAC,cAAI,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAE7D,YAAY;QACZ,MAAM,QAAQ,GAAG,MAAM,qBAAqB,EAAE,CAAC;QAE/C,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;QAExD,YAAY;QACZ,MAAM,IAAA,iBAAS,EAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAEjE,IAAI,YAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,EAAE,CAAC;YAC9B,MAAM,WAAW,GAAG,yDAAyD,oBAAS,EAAE,CAAC;YACzF,IAAI,CAAC;gBACH,gBAAgB;gBAChB,MAAM,WAAW,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBACxD,MAAM,UAAU,GAAG,YAAY,WAAW,uBAAuB,WAAW,MAAM,CAAC;gBAEnF,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,+BAA+B,UAAU,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;gBAC5E,IAAA,wBAAQ,EAAC,UAAU,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;gBAExC,eAAe;gBACf,IAAI,0BAA0B,CAAC,WAAW,EAAE,YAAY,CAAC,EAAE,CAAC;oBAC1D,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,+CAA+C,EAAE,OAAO,CAAC,CAAC,CAAC;gBACnF,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,mDAAmD,EAAE,QAAQ,CAAC,CAAC,CAAC;gBACxF,CAAC;YACH,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CACT,SAAS,CAAC,+CAA+C,KAAK,CAAC,OAAO,EAAE,EAAE,QAAQ,CAAC,CACpF,CAAC;gBACF,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,iBAAiB,WAAW,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC;gBACjE,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,YAAY,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC;gBACnE,OAAO,KAAK,CAAC,CAAC,kCAAkC;YAClD,CAAC;QACH,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,2DAA2D,EAAE,OAAO,CAAC,CAAC,CAAC;QAC7F,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CACT,SAAS,CACP,mCAAmC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAC3F,QAAQ,CACT,CACF,CAAC;QACF,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,6BAA6B;AAC7B,IAAI,OAAO,GAAG,KAAK,IAAI,EAAE,CAAC,KAAK,CAAC;AAChC,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;IACjC,IAAI,CAAC;QACH,cAAc;QACd,OAAO,GAAG,KAAK,IAAI,EAAE;YACnB,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YACrC,OAAO,GAAG,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;QACrD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;AACH,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,+BAA+B;IACnD,IAAI,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,iDAAiD,EAAE,MAAM,CAAC,CAAC,CAAC;QAElF,YAAY;QACZ,MAAM,0BAA0B,EAAE,CAAC;QAEnC,YAAY;QACZ,MAAM,QAAQ,GAAG,MAAM,qBAAqB,EAAE,CAAC;QAE/C,eAAe;QACf,MAAM,YAAY,GAAG,qBAAqB,EAAE,CAAC;QAE7C,cAAc;QACd,MAAM,gBAAgB,GAAG,cAAI,CAAC,IAAI,CAAC,YAAE,CAAC,MAAM,EAAE,EAAE,GAAG,oBAAS,OAAO,CAAC,CAAC;QACrE,MAAM,IAAA,iBAAS,EAAC,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAErE,kBAAkB;QAClB,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,iBAAiB;QAC1E,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,mBAAmB;QAC5F,MAAM,sBAAsB,GAAG,MAAM,IAAI,cAAc,CAAC;QAExD,OAAO;QACP,MAAM,OAAO,GACX,YAAE,CAAC,QAAQ,EAAE,KAAK,OAAO;YACvB,CAAC,CAAC,iBAAiB,cAAI,CAAC,OAAO,CAAC,YAAY,CAAC,YAAY,cAAI,CAAC,OAAO,CAAC,YAAY,CAAC,cAAc,gBAAgB,MAAM,YAAY,GAAG;YACtI,CAAC,CAAC,aAAa,cAAI,CAAC,OAAO,CAAC,YAAY,CAAC,YAAY,gBAAgB,MAAM,YAAY,mBAAmB,YAAY,GAAG,CAAC;QAE9H,IAAI,sBAAsB,EAAE,CAAC;YAC3B,kBAAkB;YAClB,IAAI,CAAC;gBACH,OAAO;gBACP,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,cAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;oBAC/C,YAAE,CAAC,SAAS,CAAC,cAAI,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAChE,CAAC;gBAED,OAAO;gBACP,YAAE,CAAC,YAAY,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;gBAEhD,mBAAmB;gBACnB,IAAI,YAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,EAAE,CAAC;oBAC9B,YAAE,CAAC,SAAS,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;gBACpC,CAAC;gBAED,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,gDAAgD,EAAE,OAAO,CAAC,CAAC,CAAC;YACpF,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,OAAO,CAAC,KAAK,CACX,SAAS,CAAC,8CAA8C,KAAK,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAChF,CAAC;gBACF,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;aAAM,CAAC;YACN,mBAAmB;YACnB,OAAO,CAAC,GAAG,CACT,SAAS,CAAC,oEAAoE,EAAE,QAAQ,CAAC,CAC1F,CAAC;YACF,OAAO,CAAC,GAAG,CACT,SAAS,CACP,yEAAyE,EACzE,MAAM,CACP,CACF,CAAC;YAEF,IAAI,YAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,EAAE,CAAC;gBAC9B,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,oDAAoD,EAAE,MAAM,CAAC,CAAC,CAAC;gBACrF,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,OAAO,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;YACpD,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC,CAAC;gBACtD,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,aAAa,OAAO,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;YACzD,CAAC;YAED,OAAO,CAAC,GAAG,CACT,SAAS,CAAC,gEAAgE,EAAE,MAAM,CAAC,CACpF,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,aAAa,uBAAY,oBAAoB,EAAE,MAAM,CAAC,CAAC,CAAC;YAE9E,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;QACrF,CAAC;QAED,4BAA4B;QAC5B,IAAI,YAAE,CAAC,QAAQ,EAAE,KAAK,OAAO,EAAE,CAAC;YAC9B,MAAM,WAAW,GAAG,yDAAyD,oBAAS,EAAE,CAAC;YACzF,gBAAgB;YAChB,MAAM,WAAW,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACxD,MAAM,UAAU,GAAG,YAAY,WAAW,uBAAuB,WAAW,MAAM,CAAC;YAEnF,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,mCAAmC,WAAW,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;YACjF,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,YAAY,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;YAEjE,IAAI,sBAAsB,EAAE,CAAC;gBAC3B,qBAAqB;gBACrB,IAAI,CAAC;oBACH,IAAA,wBAAQ,EAAC,UAAU,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;oBAExC,eAAe;oBACf,IAAI,0BAA0B,CAAC,WAAW,EAAE,YAAY,CAAC,EAAE,CAAC;wBAC1D,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,8CAA8C,EAAE,OAAO,CAAC,CAAC,CAAC;oBAClF,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,mDAAmD,EAAE,QAAQ,CAAC,CAAC,CAAC;oBACxF,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAU,EAAE,CAAC;oBACpB,OAAO,CAAC,KAAK,CACX,SAAS,CAAC,2CAA2C,KAAK,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAC7E,CAAC;oBACF,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,YAAY,UAAU,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC;oBAC1D,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,mBAAmB;gBACnB,OAAO,CAAC,GAAG,CACT,SAAS,CACP,wEAAwE,EACxE,QAAQ,CACT,CACF,CAAC;gBACF,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,oDAAoD,EAAE,MAAM,CAAC,CAAC,CAAC;gBACrF,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,UAAU,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;gBAClD,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,2DAA2D,EAAE,MAAM,CAAC,CAAC,CAAC;gBAC5F,OAAO,CAAC,GAAG,CACT,SAAS,CACP,sDAAsD,uBAAY,oBAAoB,EACtF,MAAM,CACP,CACF,CAAC;gBAEF,MAAM,IAAI,KAAK,CAAC,qEAAqE,CAAC,CAAC;YACzF,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,KAAK,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC;QAC1D,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/server/index.d.ts

@@ -0,0 +1,24 @@
+import { FastifyInstance } from 'fastify';
+import { NativeMessagingHost } from '../native-messaging-host';
+export declare class Server {
+    private fastify;
+    isRunning: boolean;
+    private nativeHost;
+    private transportsMap;
+    constructor();
+    /**
+     * Initialize Fastify instance with plugins and routes
+     */
+    private initializeFastify;
+    /**
+     * Associate NativeMessagingHost instance
+     */
+    setNativeHost(nativeHost: NativeMessagingHost): void;
+    private setupPlugins;
+    private setupRoutes;
+    start(port: number | undefined, nativeHost: NativeMessagingHost): Promise<void>;
+    stop(): Promise<void>;
+    getInstance(): FastifyInstance;
+}
+declare const serverInstance: Server;
+export default serverInstance;

package/dist/server/index.js

@@ -0,0 +1,266 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Server = void 0;
+const fastify_1 = __importDefault(require("fastify"));
+const cors_1 = __importDefault(require("@fastify/cors"));
+const constant_1 = require("../constant");
+const sse_js_1 = require("@modelcontextprotocol/sdk/server/sse.js");
+const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
+const node_crypto_1 = require("node:crypto");
+const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
+const mcp_server_1 = require("../mcp/mcp-server");
+class Server {
+    constructor() {
+        this.isRunning = false; // Changed to public or provide a getter
+        this.nativeHost = null;
+        this.transportsMap = new Map();
+        this.fastify = (0, fastify_1.default)({ logger: constant_1.SERVER_CONFIG.LOGGER_ENABLED });
+        this.setupPlugins();
+        this.setupRoutes();
+    }
+    /**
+     * Initialize Fastify instance with plugins and routes
+     */
+    initializeFastify() {
+        this.fastify = (0, fastify_1.default)({ logger: constant_1.SERVER_CONFIG.LOGGER_ENABLED });
+        this.setupPlugins();
+        this.setupRoutes();
+    }
+    /**
+     * Associate NativeMessagingHost instance
+     */
+    setNativeHost(nativeHost) {
+        this.nativeHost = nativeHost;
+    }
+    async setupPlugins() {
+        await this.fastify.register(cors_1.default, {
+            origin: constant_1.SERVER_CONFIG.CORS_ORIGIN,
+        });
+    }
+    setupRoutes() {
+        // for ping
+        this.fastify.get('/ask-extension', async (request, reply) => {
+            if (!this.nativeHost) {
+                return reply
+                    .status(constant_1.HTTP_STATUS.INTERNAL_SERVER_ERROR)
+                    .send({ error: constant_1.ERROR_MESSAGES.NATIVE_HOST_NOT_AVAILABLE });
+            }
+            if (!this.isRunning) {
+                return reply
+                    .status(constant_1.HTTP_STATUS.INTERNAL_SERVER_ERROR)
+                    .send({ error: constant_1.ERROR_MESSAGES.SERVER_NOT_RUNNING });
+            }
+            try {
+                // wait from extension message
+                const extensionResponse = await this.nativeHost.sendRequestToExtensionAndWait(request.query, 'process_data', constant_1.TIMEOUTS.EXTENSION_REQUEST_TIMEOUT);
+                return reply.status(constant_1.HTTP_STATUS.OK).send({ status: 'success', data: extensionResponse });
+            }
+            catch (error) {
+                if (error.message.includes('timed out')) {
+                    return reply
+                        .status(constant_1.HTTP_STATUS.GATEWAY_TIMEOUT)
+                        .send({ status: 'error', message: constant_1.ERROR_MESSAGES.REQUEST_TIMEOUT });
+                }
+                else {
+                    return reply.status(constant_1.HTTP_STATUS.INTERNAL_SERVER_ERROR).send({
+                        status: 'error',
+                        message: `Failed to get response from extension: ${error.message}`,
+                    });
+                }
+            }
+        });
+        // Compatible with SSE
+        this.fastify.get('/sse', async (_, reply) => {
+            try {
+                // Set SSE headers
+                reply.raw.writeHead(constant_1.HTTP_STATUS.OK, {
+                    'Content-Type': 'text/event-stream',
+                    'Cache-Control': 'no-cache',
+                    Connection: 'keep-alive',
+                });
+                // Create SSE transport
+                const transport = new sse_js_1.SSEServerTransport('/messages', reply.raw);
+                this.transportsMap.set(transport.sessionId, transport);
+                reply.raw.on('close', () => {
+                    this.transportsMap.delete(transport.sessionId);
+                });
+                const server = (0, mcp_server_1.getMcpServer)();
+                await server.connect(transport);
+                // Keep connection open
+                reply.raw.write(':\n\n');
+            }
+            catch (error) {
+                if (!reply.sent) {
+                    reply.code(constant_1.HTTP_STATUS.INTERNAL_SERVER_ERROR).send(constant_1.ERROR_MESSAGES.INTERNAL_SERVER_ERROR);
+                }
+            }
+        });
+        // Compatible with SSE
+        this.fastify.post('/messages', async (req, reply) => {
+            try {
+                const { sessionId } = req.query;
+                const transport = this.transportsMap.get(sessionId);
+                if (!sessionId || !transport) {
+                    reply.code(constant_1.HTTP_STATUS.BAD_REQUEST).send('No transport found for sessionId');
+                    return;
+                }
+                await transport.handlePostMessage(req.raw, reply.raw, req.body);
+            }
+            catch (error) {
+                if (!reply.sent) {
+                    reply.code(constant_1.HTTP_STATUS.INTERNAL_SERVER_ERROR).send(constant_1.ERROR_MESSAGES.INTERNAL_SERVER_ERROR);
+                }
+            }
+        });
+        // POST /mcp: Handle client-to-server messages
+        this.fastify.post('/mcp', async (request, reply) => {
+            const sessionId = request.headers['mcp-session-id'];
+            let transport = this.transportsMap.get(sessionId || '');
+            if (transport) {
+                // transport found, do nothing
+                //} else if (!sessionId && isInitializeRequest(request.body)) {
+                // 方便测试，临时去掉sessionId的限制
+            }
+            else if ((0, types_js_1.isInitializeRequest)(request.body)) {
+                const newSessionId = (0, node_crypto_1.randomUUID)(); // Generate session ID
+                transport = new streamableHttp_js_1.StreamableHTTPServerTransport({
+                    sessionIdGenerator: () => newSessionId, // Use pre-generated ID
+                    onsessioninitialized: (initializedSessionId) => {
+                        // Ensure transport instance exists and session ID matches
+                        if (transport && initializedSessionId === newSessionId) {
+                            this.transportsMap.set(initializedSessionId, transport);
+                        }
+                    },
+                });
+                transport.onclose = () => {
+                    if ((transport === null || transport === void 0 ? void 0 : transport.sessionId) && this.transportsMap.get(transport.sessionId)) {
+                        this.transportsMap.delete(transport.sessionId);
+                    }
+                };
+                await (0, mcp_server_1.getMcpServer)().connect(transport);
+            }
+            else {
+                reply.code(constant_1.HTTP_STATUS.BAD_REQUEST).send({ error: constant_1.ERROR_MESSAGES.INVALID_MCP_REQUEST });
+                return;
+            }
+            try {
+                await transport.handleRequest(request.raw, reply.raw, request.body);
+            }
+            catch (error) {
+                if (!reply.sent) {
+                    reply
+                        .code(constant_1.HTTP_STATUS.INTERNAL_SERVER_ERROR)
+                        .send({ error: constant_1.ERROR_MESSAGES.MCP_REQUEST_PROCESSING_ERROR });
+                }
+            }
+        });
+        this.fastify.get('/mcp', async (request, reply) => {
+            const sessionId = request.headers['mcp-session-id'];
+            const transport = sessionId
+                ? this.transportsMap.get(sessionId)
+                : undefined;
+            if (!transport) {
+                reply.code(constant_1.HTTP_STATUS.BAD_REQUEST).send({ error: constant_1.ERROR_MESSAGES.INVALID_SSE_SESSION });
+                return;
+            }
+            reply.raw.setHeader('Content-Type', 'text/event-stream');
+            reply.raw.setHeader('Cache-Control', 'no-cache');
+            reply.raw.setHeader('Connection', 'keep-alive');
+            reply.raw.flushHeaders(); // Ensure headers are sent immediately
+            try {
+                // transport.handleRequest will take over the response stream
+                await transport.handleRequest(request.raw, reply.raw);
+                if (!reply.sent) {
+                    // If transport didn't send anything (unlikely for SSE initial handshake)
+                    reply.hijack(); // Prevent Fastify from automatically sending response
+                }
+            }
+            catch (error) {
+                if (!reply.raw.writableEnded) {
+                    reply.raw.end();
+                }
+            }
+            request.socket.on('close', () => {
+                request.log.info(`SSE client disconnected for session: ${sessionId}`);
+                // transport's onclose should handle its own cleanup
+            });
+        });
+        this.fastify.delete('/mcp', async (request, reply) => {
+            const sessionId = request.headers['mcp-session-id'];
+            const transport = sessionId
+                ? this.transportsMap.get(sessionId)
+                : undefined;
+            if (!transport) {
+                reply.code(constant_1.HTTP_STATUS.BAD_REQUEST).send({ error: constant_1.ERROR_MESSAGES.INVALID_SESSION_ID });
+                return;
+            }
+            try {
+                await transport.handleRequest(request.raw, reply.raw);
+                // Assume transport.handleRequest will send response or transport.onclose will cleanup
+                if (!reply.sent) {
+                    reply.code(constant_1.HTTP_STATUS.NO_CONTENT).send();
+                }
+            }
+            catch (error) {
+                if (!reply.sent) {
+                    reply
+                        .code(constant_1.HTTP_STATUS.INTERNAL_SERVER_ERROR)
+                        .send({ error: constant_1.ERROR_MESSAGES.MCP_SESSION_DELETION_ERROR });
+                }
+            }
+        });
+    }
+    async start(port = constant_1.NATIVE_SERVER_PORT, nativeHost) {
+        if (!this.nativeHost) {
+            this.nativeHost = nativeHost; // Ensure nativeHost is set
+        }
+        else if (this.nativeHost !== nativeHost) {
+            this.nativeHost = nativeHost; // Update to the passed instance
+        }
+        if (this.isRunning) {
+            return;
+        }
+        try {
+            // Ensure Fastify instance is ready before starting
+            if (!this.fastify) {
+                this.initializeFastify();
+            }
+            await this.fastify.listen({ port, host: constant_1.SERVER_CONFIG.HOST });
+            this.isRunning = true; // Update running status
+            // No need to return, Promise resolves void by default
+        }
+        catch (err) {
+            this.isRunning = false; // Startup failed, reset status
+            // Throw error instead of exiting directly, let caller (possibly NativeHost) handle
+            throw err; // or return Promise.reject(err);
+            // process.exit(1); // Not recommended to exit directly here
+        }
+    }
+    async stop() {
+        if (!this.isRunning) {
+            return;
+        }
+        try {
+            await this.fastify.close();
+            this.isRunning = false; // Update running status
+            // Recreate Fastify instance to allow restart
+            // This is necessary because Fastify instances cannot be restarted after close()
+            this.initializeFastify();
+        }
+        catch (err) {
+            // Even if closing fails, mark as not running, but log the error
+            this.isRunning = false;
+            throw err; // Throw error
+        }
+    }
+    getInstance() {
+        return this.fastify;
+    }
+}
+exports.Server = Server;
+const serverInstance = new Server();
+exports.default = serverInstance;
+//# sourceMappingURL=index.js.map