#!/usr/bin/env node
import fs from 'fs';
import dotenv from 'dotenv';
import { Command } from 'commander';
import { currentVersion } from './utils/version';
import { initStdioServer, initSSEServer, initMcpServer } from './mcp-server';
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
import { RecallTool } from './mcp-tool/document-tool/recall';
import { OAPI_MCP_DEFAULT_ARGS, OAPI_MCP_ENV_ARGS } from './utils/constants';
dotenv.config();
const program = new Command();
program.name('lark-mcp').description('Feishu/Lark MCP Tool').version(currentVersion);
program
.command('mcp')
.description('Start Feishu/Lark MCP Service')
.option('-a, --app-id <appId>', 'Feishu/Lark App ID')
.option('-s, --app-secret <appSecret>', 'Feishu/Lark App Secret')
.option('-d, --domain <domain>', 'Feishu/Lark Domain (default: "https://open.feishu.cn")')
.option('-t, --tools <tools>', 'Allowed Tools List, separated by commas')
.option('-c, --tool-name-case <toolNameCase>', 'Tool Name Case, snake or camel or kebab or dot (default: "snake")')
.option('-l, --language <language>', 'Tools Language, zh or en (default: "en")')
.option('-u, --user-access-token <userAccessToken>', 'User Access Token (beta)')
.option('--token-mode <tokenMode>', 'Token Mode, auto or user_access_token or tenant_access_token (default: "auto")')
.option('-m, --mode <mode>', 'Transport Mode, stdio or sse (default: "stdio")')
.option('--host <host>', 'Host to listen (default: "localhost")')
.option('-p, --port <port>', 'Port to listen in sse mode (default: "3000")')
.option('--config <configPath>', 'Config file path (JSON)')
.action(async (options) => {
let fileOptions = {};
if (options.config) {
try {
const configContent = fs.readFileSync(options.config, 'utf-8');
fileOptions = JSON.parse(configContent);
} catch (err) {
console.error('Failed to read config file:', err);
process.exit(1);
}
}
const mergedOptions = { ...OAPI_MCP_DEFAULT_ARGS, ...OAPI_MCP_ENV_ARGS, ...fileOptions, ...options };
const { mcpServer } = initMcpServer(mergedOptions);
if (mergedOptions.mode === 'stdio') {
initStdioServer(mcpServer);
} else if (mergedOptions.mode === 'sse') {
initSSEServer(mcpServer, mergedOptions);
} else {
console.error('Invalid mode:', mergedOptions.mode);
process.exit(1);
}
});
program
.command('recall-developer-documents')
.description('Start Feishu/Lark Open Platform Recall MCP Service')
.option('-d, --domain <domain>', 'Feishu Open Platform Domain', 'https://open.feishu.cn')
.option('-m, --mode <mode>', 'Transport Mode, stdio or sse', 'stdio')
.option('--host <host>', 'Host to listen', 'localhost')
.option('-p, --port <port>', 'Port to listen in sse mode', '3001')
.action((options) => {
const server = new McpServer({
id: 'lark-recall-mcp-server',
name: 'Lark Recall MCP Service',
version: currentVersion,
});
server.tool(RecallTool.name, RecallTool.description, RecallTool.schema, (params) =>
RecallTool.handler(params, options),
);
if (options.mode === 'stdio') {
initStdioServer(server);
} else if (options.mode === 'sse') {
initSSEServer(server, options);
} else {
console.error('Invalid mode:', options.mode);
process.exit(1);
}
});
if (process.argv.length === 2) {
program.help();
}
program.parse(process.argv);
export { program };
