Real-time Stock MCP Service

"""
股票数据 MCP Server（托管友好入口）
"""

import logging
import os
from datetime import datetime

from mcp.server.fastmcp import FastMCP

from stock_mcp.data_source_interface import FinancialDataInterface
from stock_mcp.stock_data_source import WebCrawlerDataSource
from stock_mcp.utils.utils import setup_logging

from stock_mcp.mcp_tools.search import register_search_tools
from stock_mcp.mcp_tools.kline_data import register_kline_tools
from stock_mcp.mcp_tools.real_time_data import register_real_time_data_tools
from stock_mcp.mcp_tools.fundamental import register_fundamental_tools
from stock_mcp.mcp_tools.valuation import register_valuation_tools
from stock_mcp.mcp_tools.financial_analysis import register_financial_analysis_tools
from stock_mcp.mcp_tools.market import register_market_tools
from stock_mcp.mcp_tools.smart_review import register_smart_review_tools


def build_app(active_data_source: FinancialDataInterface) -> FastMCP:
    """
    构建 FastMCP app（只做“创建 + 注册工具”）
    """
    current_date = datetime.now().strftime("%Y-%m-%d")

    app = FastMCP(
        name="real-time-stock-mcp-service",
        instructions=f"""📊 一个获取实时股票数据服务和分析的MCP服务器

**今天日期**: {current_date}

📈 主要功能:
- 查找股票名称，代码
- 实时股票数据
- K线数据（日线、周线、月线）
- 计算技术指标
- 基本面数据（主营构成、经营范围、经营评述等）
- 估值分析数据（市盈率、市净率等）
- 板块行情数据
- 智能点评和评分
""",
    )

    # ✅ 注册所有工具
    register_search_tools(app, active_data_source)
    register_real_time_data_tools(app, active_data_source)
    register_kline_tools(app, active_data_source)
    register_fundamental_tools(app, active_data_source)
    register_valuation_tools(app, active_data_source)
    register_financial_analysis_tools(app, active_data_source)
    register_market_tools(app, active_data_source)
    register_smart_review_tools(app, active_data_source)

    return app


def main() -> None:
    """
    程序主入口：
    1) 配日志
    2) 创建数据源（依赖注入）
    3) 构建 app + 注册工具
    4) 初始化数据源
    5) app.run() 启动（stdio）
    6) finally 清理资源
    """
    # ✅ 托管环境常用环境变量控制日志级别，方便排障
    log_level = os.getenv("LOG_LEVEL", "INFO").upper()
    setup_logging(level=getattr(logging, log_level, logging.INFO))
    logger = logging.getLogger(__name__)

    # 1) 依赖注入：后续切换数据源只改这里
    active_data_source: FinancialDataInterface = WebCrawlerDataSource()
    logger.info("数据源: %s", active_data_source.__class__.__name__)

    # 2) 构建 app（注册工具）
    app = build_app(active_data_source)
    logger.info("工具模块注册完成")

    # 3) 初始化数据源
    try:
        if active_data_source.initialize():
            logger.info("✅ 数据源初始化成功")
        else:
            logger.warning("⚠️ 数据源初始化失败，某些功能可能不可用")
    except Exception:
        logger.exception("💥 数据源初始化异常：将继续启动（功能可能受限）")

    # 4) 运行服务（托管通常走 stdio，保持默认）
    try:
        logger.info("🚀 启动 MCP Server（stdio）")
        app.run()
    except KeyboardInterrupt:
        logger.info("🛑 服务被中断")
    except Exception:
        logger.exception("💥 服务运行出错")
        raise
    finally:
        # 5) 清理资源
        try:
            active_data_source.cleanup()
            logger.info("🧹 资源清理完成")
        except Exception:
            logger.exception("💥 资源清理异常")


if __name__ == "__main__":
    main()