aicodestat 0.0.1__py3-none-any.whl

local_mcp_server.py

@@ -0,0 +1,260 @@
+"""应用入口，使用 FastMCP 框架实现 MCP Server，通过 HTTP 后台服务暴露接口"""
+import argparse
+import sys
+import logging
+import os
+from pathlib import Path
+from typing import Optional, Any, Dict
+from config import get_server_config, get_database_config
+from logging_config import setup_logging
+from storage.db import get_db
+from storage.scheduler import start_scheduler, stop_scheduler
+from storage.models import (
+    save_before_edit,
+    get_before_edit,
+    delete_before_edit,
+    save_session_summary,
+    save_code_diff_lines,
+)
+from compute.diff_engine import extract_diff_lines
+from utils.port_utils import find_available_port
+
+# 设置日志（初始化为INFO级别，main函数中会根据配置调整）
+logger = setup_logging(log_level="INFO", module_name="mcp_server")
+
+
+def _import_fastmcp():
+    """Import FastMCP from the official MCP SDK, avoiding local mcp/ shadowing."""
+    original_sys_path = list(sys.path)
+    try:
+        project_root = os.path.abspath(os.path.dirname(__file__))
+        sys.path = [p for p in sys.path if p not in ("", ".", project_root)]
+        from mcp.server.fastmcp import FastMCP  # type: ignore
+        return FastMCP
+    finally:
+        sys.path = original_sys_path
+
+
+def create_mcp_app(host: str = "127.0.0.1", port: int = 8000):
+    """创建 FastMCP 应用并注册工具"""
+    FastMCP = _import_fastmcp()
+    # FastMCP 构造函数接受 host, port, streamable_http_path 等参数
+    # streamable_http_path 默认为 '/mcp'，但 Cursor 可能请求根路径，所以设置为 '/'
+    mcp_app = FastMCP(
+        "local-code-stat",
+        host=host,
+        port=port,
+        streamable_http_path="/"  # 设置为根路径，兼容 Cursor 的请求
+    )
+
+    @mcp_app.tool()
+    def RecordBeforeEdit(session_id: str, file_path: str, code_before: str) -> Dict[str, Any]:
+        """
+        Record the complete code content before file editing (temporary local storage, will be cleaned up automatically, no redundant retention)
+        
+        Args:
+            session_id: Current session ID, generated by Agent, must match RecordAfterEdit's session_id
+            file_path: Absolute path of the target file
+            code_before: Complete code content before editing (preserve original format, blank lines, ensure accurate line numbers)
+        
+        Returns:
+            Dictionary containing status and message
+        """
+        if not session_id or not session_id.strip():
+            return {"status": "error", "message": "session_id cannot be empty"}
+        if not file_path or not file_path.strip():
+            return {"status": "error", "message": "file_path cannot be empty"}
+        if code_before is None or code_before == "":
+            return {"status": "error", "message": "code_before cannot be empty"}
+
+        success = save_before_edit(session_id=session_id, file_path=file_path, code_before=code_before)
+        if not success:
+            return {"status": "error", "message": "Failed to save code before editing"}
+
+        logger.info(f"RecordBeforeEdit: session={session_id}, file={file_path}")
+        return {"status": "success", "message": "Code before editing recorded successfully"}
+
+    @mcp_app.tool()
+    def RecordAfterEdit(
+        session_id: str,
+        file_path: str,
+        code_after: str,
+        session_info: Optional[str] = None,
+    ) -> Dict[str, Any]:
+        """
+        Record the complete code content after file editing, extract specific diff lines (added/modified) with line numbers, clean up temporary data, and retain only diff information
+        
+        Args:
+            session_id: Current session ID, must match RecordBeforeEdit's session_id to associate the same editing operation
+            file_path: Absolute path of the target file, must match RecordBeforeEdit's file_path
+            code_after: Complete code content after editing (preserve original format, blank lines, ensure accurate line numbers)
+            session_info: Optional session supplementary information (e.g., user instructions, Agent type, operation time) for subsequent statistical filtering
+        
+        Returns:
+            Dictionary containing status, message, and data
+        """
+        if not session_id or not session_id.strip():
+            return {"status": "error", "message": "session_id cannot be empty"}
+        if not file_path or not file_path.strip():
+            return {"status": "error", "message": "file_path cannot be empty"}
+        if code_after is None or code_after == "":
+            return {"status": "error", "message": "code_after cannot be empty"}
+
+        code_before = get_before_edit(session_id, file_path)
+        if code_before is None:
+            return {
+                "status": "error",
+                "message": (
+                    f"Code record before editing not found, please call RecordBeforeEdit Tool first "
+                    f"(session_id={session_id}, file_path={file_path})"
+                ),
+            }
+
+        diff_lines = extract_diff_lines(code_before, code_after)
+        add_lines_count = sum(1 for d in diff_lines if d["diff_type"] == "add")
+        modify_lines_count = sum(1 for d in diff_lines if d["diff_type"] == "modify")
+        total_lines_after = len(code_after.split("\n"))
+
+        success = save_session_summary(
+            session_id=session_id,
+            file_path=file_path,
+            add_lines_count=add_lines_count,
+            modify_lines_count=modify_lines_count,
+            total_lines_after=total_lines_after,
+            session_info=session_info,
+        )
+        if not success:
+            return {"status": "error", "message": "Failed to save session summary"}
+
+        if diff_lines:
+            ok = save_code_diff_lines(session_id=session_id, file_path=file_path, diff_lines=diff_lines)
+            if not ok:
+                logger.warning("Failed to save diff lines details, but session summary has been saved")
+
+        delete_before_edit(session_id, file_path)
+
+        logger.info(
+            f"RecordAfterEdit: session={session_id}, file={file_path}, "
+            f"add={add_lines_count}, modify={modify_lines_count}, total_diff={len(diff_lines)}"
+        )
+        return {
+            "status": "success",
+            "message": "Code after editing recorded successfully, diff lines extracted and stored",
+            "data": {
+                "add_lines_count": add_lines_count,
+                "modify_lines_count": modify_lines_count,
+                "total_diff_lines": len(diff_lines),
+                "total_lines_after": total_lines_after,
+            },
+        }
+
+    return mcp_app
+
+
+def main():
+    """主函数：启动 FastMCP HTTP 服务器"""
+    parser = argparse.ArgumentParser(description="本地MCP Server AI代码统计系统（基于FastMCP）")
+    parser.add_argument(
+        "command",
+        choices=["start"],
+        help="启动服务器"
+    )
+    parser.add_argument(
+        "--host",
+        type=str,
+        help="服务器监听地址（默认从config.json读取）"
+    )
+    parser.add_argument(
+        "--port",
+        type=int,
+        help="服务器监听端口（默认从config.json读取）"
+    )
+    parser.add_argument(
+        "--daemon",
+        action="store_true",
+        help="后台运行（Windows下使用简单后台进程，Unix下建议使用nohup或systemd）"
+    )
+    
+    args = parser.parse_args()
+    
+    if args.command == "start":
+        # 初始化数据库和定时任务
+        try:
+            db = get_db()
+            db.initialize()
+            logger.info("Database initialized successfully")
+            
+            start_scheduler()
+            logger.info("Scheduler started (auto backup and cleanup)")
+        except Exception as e:
+            logger.error(f"Failed to initialize: {e}", exc_info=True)
+            sys.exit(1)
+        
+        # 获取配置
+        server_config = get_server_config()
+        host = args.host or server_config["host"]
+        preferred_port = args.port or server_config["port"]
+        log_level = server_config.get("log_level", "info")
+        log_file = server_config.get("log_file")
+        
+        # 检查端口是否可用，如果被占用则选择随机端口
+        try:
+            port = find_available_port(host, preferred_port)
+            if port != preferred_port:
+                logger.warning(f"Port {preferred_port} is in use, using port {port} instead")
+        except RuntimeError as e:
+            logger.error(f"Failed to find available port: {e}")
+            sys.exit(1)
+        
+        # 设置日志（如果配置了日志文件，添加文件handler）
+        if log_file:
+            # 添加文件handler到现有logger
+            from pathlib import Path
+            log_path = Path(log_file)
+            log_path.parent.mkdir(parents=True, exist_ok=True)
+            file_handler = logging.FileHandler(log_file, encoding='utf-8')
+            file_handler.setLevel(getattr(logging, log_level.upper(), logging.INFO))
+            formatter = logging.Formatter(
+                fmt='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+                datefmt='%Y-%m-%d %H:%M:%S'
+            )
+            file_handler.setFormatter(formatter)
+            logger.addHandler(file_handler)
+            logger.info(f"Logging to file: {log_file}")
+        
+        # 设置日志级别
+        logger.setLevel(getattr(logging, log_level.upper(), logging.INFO))
+        
+        # 创建 FastMCP 应用（传入 host 和 port）
+        mcp_app = create_mcp_app(host=host, port=port)
+        
+        logger.info(f"Starting FastMCP Server on {host}:{port} (transport: streamable-http)")
+        logger.info("MCP Tools available:")
+        logger.info("  - RecordBeforeEdit: Record code before file editing")
+        logger.info("  - RecordAfterEdit: Record code after file editing and extract diff lines")
+        
+        if args.daemon:
+            # 后台运行（简单实现，生产环境建议使用systemd/supervisor等）
+            logger.warning("Daemon mode is a simple implementation. For production, use systemd/supervisor/nohup.")
+            import multiprocessing
+            
+            def run_server():
+                mcp_app.run(transport="streamable-http")
+            
+            process = multiprocessing.Process(target=run_server, daemon=True)
+            process.start()
+            logger.info(f"Server started in background (PID: {process.pid})")
+            logger.info("To stop the server, use: kill <PID>")
+        else:
+            # 前台运行 FastMCP HTTP 服务器
+            try:
+                mcp_app.run(transport="streamable-http")
+            except KeyboardInterrupt:
+                logger.info("Server stopped by user")
+                stop_scheduler()
+                logger.info("Scheduler stopped")
+
+
+if __name__ == "__main__":
+    main()
+

logging_config.py

@@ -0,0 +1,68 @@
+"""统一日志配置模块"""
+import logging
+import sys
+from pathlib import Path
+from typing import Optional
+
+
+def setup_logging(
+    log_level: str = "INFO",
+    log_file: Optional[str] = None,
+    module_name: Optional[str] = None,
+    stream: Optional[object] = None
+) -> logging.Logger:
+    """
+    设置日志配置
+    
+    Args:
+        log_level: 日志级别 (DEBUG, INFO, WARNING, ERROR, CRITICAL)
+        log_file: 日志文件路径（可选）
+        module_name: 模块名称（用于logger命名）
+    
+    Returns:
+        配置好的logger实例
+    """
+    # 转换日志级别
+    numeric_level = getattr(logging, log_level.upper(), logging.INFO)
+    
+    # 创建formatter
+    formatter = logging.Formatter(
+        fmt='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S'
+    )
+    
+    # 创建logger
+    logger_name = module_name if module_name else __name__
+    logger = logging.getLogger(logger_name)
+    logger.setLevel(numeric_level)
+    
+    # 清除已有的handlers
+    logger.handlers.clear()
+    
+    # 控制台handler
+    # IMPORTANT:
+    # - MCP stdio mode uses stdout for JSON-RPC. Writing logs to stdout will break the protocol.
+    # - Default to stderr to be safe for both CLI and MCP.
+    if stream is None:
+        stream = sys.stderr
+    console_handler = logging.StreamHandler(stream)
+    console_handler.setLevel(numeric_level)
+    console_handler.setFormatter(formatter)
+    logger.addHandler(console_handler)
+    
+    # 文件handler（如果指定）
+    if log_file:
+        log_path = Path(log_file)
+        log_path.parent.mkdir(parents=True, exist_ok=True)
+        file_handler = logging.FileHandler(log_file, encoding='utf-8')
+        file_handler.setLevel(numeric_level)
+        file_handler.setFormatter(formatter)
+        logger.addHandler(file_handler)
+    
+    return logger
+
+
+def get_logger(name: str) -> logging.Logger:
+    """获取指定名称的logger"""
+    return logging.getLogger(name)
+

main.py

@@ -0,0 +1,164 @@
+"""统一入口：支持启动服务器或CLI"""
+import argparse
+import sys
+
+
+def main():
+    """主入口函数"""
+    parser = argparse.ArgumentParser(
+        description="本地MCP Server AI代码统计系统",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+示例:
+  %(prog)s server start          # 启动MCP服务器
+  %(prog)s server start --port 8080  # 指定端口启动服务器
+  %(prog)s cli                    # 启动交互式CLI
+  %(prog)s cli --session <id>     # 快捷查询会话
+  %(prog)s cli --file <path>      # 快捷查询文件
+        """
+    )
+    
+    subparsers = parser.add_subparsers(dest='command', help='可用命令')
+    
+    # Server子命令
+    server_parser = subparsers.add_parser('server', help='启动MCP服务器')
+    server_parser.add_argument(
+        'action',
+        choices=['start'],
+        help='启动服务器'
+    )
+    server_parser.add_argument(
+        '--host',
+        type=str,
+        help='服务器监听地址（默认从config.json读取）'
+    )
+    server_parser.add_argument(
+        '--port',
+        type=int,
+        help='服务器监听端口（默认从config.json读取）'
+    )
+    server_parser.add_argument(
+        '--daemon',
+        action='store_true',
+        help='后台运行'
+    )
+    
+    # CLI子命令
+    cli_parser = subparsers.add_parser('cli', help='启动交互式CLI')
+    cli_parser.add_argument(
+        '--session',
+        type=str,
+        help='按会话ID查询（快捷命令）'
+    )
+    cli_parser.add_argument(
+        '--file',
+        type=str,
+        help='按文件路径查询（快捷命令）'
+    )
+    cli_parser.add_argument(
+        '--project',
+        type=str,
+        help='按项目根目录查询（快捷命令）'
+    )
+    cli_parser.add_argument(
+        '--export',
+        type=str,
+        choices=['json', 'csv'],
+        help='导出格式（需配合--session/--file/--project使用）'
+    )
+    cli_parser.add_argument(
+        '--output',
+        type=str,
+        help='导出文件路径（需配合--export使用）'
+    )
+    
+    args = parser.parse_args()
+    
+    if not args.command:
+        parser.print_help()
+        sys.exit(1)
+    
+    if args.command == 'server':
+        # 启动服务器
+        if args.action == 'start':
+            import uvicorn
+            import logging
+            from config import get_server_config
+            from logging_config import setup_logging
+            from storage.db import get_db
+            from storage.scheduler import start_scheduler
+            from local_mcp_server import app
+            
+            logger = setup_logging(log_level="INFO", module_name="mcp_server")
+            
+            # 初始化数据库
+            try:
+                db = get_db()
+                db.initialize()
+                logger.info("Database initialized successfully")
+                start_scheduler()
+                logger.info("Scheduler started (auto backup and cleanup)")
+            except Exception as e:
+                logger.error(f"Failed to initialize: {e}")
+                sys.exit(1)
+            
+            # 获取配置
+            server_config = get_server_config()
+            host = args.host or server_config["host"]
+            preferred_port = args.port or server_config["port"]
+            log_level = server_config.get("log_level", "info")
+            
+            # 检查端口是否可用，如果被占用则选择随机端口
+            from utils.port_utils import find_available_port
+            try:
+                port = find_available_port(host, preferred_port)
+                if port != preferred_port:
+                    logger.warning(f"Port {preferred_port} is in use, using port {port} instead")
+            except RuntimeError as e:
+                logger.error(f"Failed to find available port: {e}")
+                sys.exit(1)
+            
+            logger.setLevel(getattr(logging, log_level.upper(), logging.INFO))
+            logger.info(f"Starting MCP Server on {host}:{port}")
+            logger.info("MCP Tools available at:")
+            logger.info(f"  GET  http://{host}:{port}/mcp/tools (Tool discovery)")
+            logger.info(f"  POST http://{host}:{port}/mcp/record_before")
+            logger.info(f"  POST http://{host}:{port}/mcp/record_after")
+            
+            if args.daemon:
+                logger.warning("Daemon mode is a simple implementation. For production, use systemd/supervisor/nohup.")
+                import multiprocessing
+                
+                def run_server():
+                    uvicorn.run(app, host=host, port=port, log_level=log_level)
+                
+                process = multiprocessing.Process(target=run_server, daemon=True)
+                process.start()
+                logger.info(f"Server started in background (PID: {process.pid})")
+                logger.info("To stop the server, use: kill <PID>")
+            else:
+                uvicorn.run(app, host=host, port=port, log_level=log_level)
+    
+    elif args.command == 'cli':
+        # 启动CLI
+        from cli.main import main as cli_main
+        # 设置快捷命令参数
+        if args.session or args.file or args.project:
+            # 快捷命令模式，需要重新设置sys.argv
+            sys.argv = ['cli/main.py']
+            if args.session:
+                sys.argv.extend(['--session', args.session])
+            if args.file:
+                sys.argv.extend(['--file', args.file])
+            if args.project:
+                sys.argv.extend(['--project', args.project])
+            if args.export:
+                sys.argv.extend(['--export', args.export])
+            if args.output:
+                sys.argv.extend(['--output', args.output])
+        cli_main()
+
+
+if __name__ == '__main__':
+    main()
+

mcp/__init__.py

@@ -0,0 +1,2 @@
+"""MCP Tool实现模块"""
+

mcp/agent_adapter.py

@@ -0,0 +1,69 @@
+"""可选，封装不同Agent字段名差异的映射"""
+from typing import Dict, Any, Optional
+
+# Agent参数映射表
+AGENT_FIELD_MAPPINGS = {
+    "cursor": {
+        "file_path": "file_path",
+        "target_file": "file_path",  # 别名
+    },
+    "claude": {
+        "file_path": "file_path",
+        "target_file": "file_path",
+    },
+    "trea": {
+        "file_path": "file_path",
+    },
+    "qoder": {
+        "file_path": "file_path",
+    }
+}
+
+
+def normalize_request_params(agent_type: str, params: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    标准化不同Agent的请求参数
+    
+    Args:
+        agent_type: Agent类型（cursor/claude/trea/qoder）
+        params: 原始请求参数
+    
+    Returns:
+        标准化后的参数
+    """
+    agent_type_lower = agent_type.lower()
+    mapping = AGENT_FIELD_MAPPINGS.get(agent_type_lower, {})
+    
+    normalized = params.copy()
+    
+    # 字段映射
+    for alias, target_field in mapping.items():
+        if alias in normalized and alias != target_field:
+            if target_field not in normalized:
+                normalized[target_field] = normalized[alias]
+            del normalized[alias]
+    
+    return normalized
+
+
+def detect_agent_type(session_info: Optional[str]) -> str:
+    """
+    从session_info中检测Agent类型
+    
+    Args:
+        session_info: 会话信息字符串
+    
+    Returns:
+        Agent类型（默认返回"unknown"）
+    """
+    if not session_info:
+        return "unknown"
+    
+    session_info_lower = session_info.lower()
+    
+    for agent_type in ["cursor", "claude", "trea", "qoder"]:
+        if agent_type in session_info_lower:
+            return agent_type
+    
+    return "unknown"
+

mcp/api_schemas.py

@@ -0,0 +1,26 @@
+"""请求/响应 Pydantic 模型"""
+from pydantic import BaseModel, Field
+from typing import Optional
+
+
+class RecordBeforeEditRequest(BaseModel):
+    """RecordBeforeEdit Tool请求模型"""
+    session_id: str = Field(..., description="当前会话ID，由Agent生成，需与RecordAfterEdit的session_id一致")
+    file_path: str = Field(..., description="目标文件的绝对路径")
+    code_before: str = Field(..., description="文件编辑前的完整代码内容")
+
+
+class RecordAfterEditRequest(BaseModel):
+    """RecordAfterEdit Tool请求模型"""
+    session_id: str = Field(..., description="当前会话ID，需与RecordBeforeEdit的session_id一致")
+    file_path: str = Field(..., description="目标文件的绝对路径，需与RecordBeforeEdit的file_path一致")
+    code_after: str = Field(..., description="文件编辑后的完整代码内容")
+    session_info: Optional[str] = Field(None, description="会话补充信息（如用户指令、Agent类型、操作时间）")
+
+
+class MCPResponse(BaseModel):
+    """MCP Tool统一响应模型"""
+    status: str = Field(..., description="状态：success 或 error")
+    message: str = Field(..., description="响应消息")
+    data: Optional[dict] = Field(None, description="响应数据（可选）")
+