Lark MCP

""" Lark Message Recorder - Main Entry Point Records all received Lark messages to a MySQL database. """ import multiprocessing import sys import asyncio from loguru import logger def start_mcp_server(): """Start the MCP server in a separate process""" logger.info("Starting MCP server...") from app.core.mcp_server import mcp mcp.run(transport="stdio") async def main(): """Main function to run the Lark message recorder""" from app.db.session import init_db from app.api.auth import get_auth from app.api.lark_client import LarkClient from app.core.message_service import MessageService mcp_process = multiprocessing.Process(target=start_mcp_server) mcp_process.daemon = True mcp_process.start() logger.info("初始化数据库...") try: init_db() logger.info("数据库初始化成功.") except Exception as e: logger.error(f"数据库初始化失败: {str(e)}") sys.exit(1) logger.info("初始化认证...") try: auth = get_auth() logger.info("认证初始化成功.") except Exception as e: logger.error(f"认证初始化失败: {str(e)}") sys.exit(1) logger.info("创建 Lark 客户端...") try: lark_client = LarkClient(auth) logger.info("Lark 客户端创建成功.") except Exception as e: logger.error(f"Lark 客户端创建失败: {str(e)}") sys.exit(1) logger.info("创建消息服务...") message_service = MessageService(lark_client) try: logger.info("连接到 Lark WebSocket...") logger.info("开始接收消息...") logger.info('================================================================') await lark_client.connect_websocket(message_service.process_message) except KeyboardInterrupt: logger.info("程序被用户中断.") except Exception as e: logger.error(f"Error: {str(e)}") if __name__ == "__main__": asyncio.run(main())