from functools import lru_cache
from typing import Any, cast

from app.core.advisor_allocation_contracts import AssetAllocationQuery, CustomerProfileQuery
from app.core.fund_contracts import FundQuoteQuery
from app.core.investment_goal_contracts import InvestmentGoalQuery
from app.core.nl2sql_contracts import FinancialNL2SQLInput
from app.core.portfolio_analysis_contracts import PortfolioAnalysisQuery
from app.core.portfolio_rebalance_contracts import PortfolioRebalanceQuery
from app.core.product_recommendation_contracts import ProductRecommendationQuery
from app.service.agent.advisor_agent import AdvisorAgent
from app.service.agent.factory import AgentFactory
from app.service.agent.offsite_fund_agent import OffsiteFundAgent
from app.service.asset_allocation_service import asset_allocation_tool
from app.service.customer_profile_service import customer_profile_query_tool
from app.service.financial_nl2sql_service import query_financial_data_tool
from app.service.fund_quote_service import query_fund_quote_tool
from app.service.intent_classifier import IntentClassifier
from app.service.investment_goal_service import investment_goal_query_tool
from app.service.model_gateway import (
    DatabaseModelEndpointResolver,
    DatabaseModelGateway,
    ModelDispatchService,
    ModelGenerationService,
)
from app.service.portfolio_analysis_service import portfolio_analysis_tool
from app.service.portfolio_rebalance_service import portfolio_rebalance_tool
from app.service.product_recommendation_service import product_recommendation_tool
from app.service.suitability_service import SuitabilityToolInput, suitability_tool_handler
from app.service.tool_executor import ToolDefinition, ToolExecutor, ToolRegistry


@lru_cache(maxsize=1)
def get_agent_factory() -> AgentFactory:
    """HTTP 与 Worker 共用的唯一底座依赖组装入口。"""
    registry = ToolRegistry()
    registry.register(ToolDefinition(
        name="check_suitability",
        input_model=SuitabilityToolInput,
        handler=cast(Any, suitability_tool_handler),
        required_permission="suitability:read",
        allowed_roles=("customer", "advisor", "operator", "admin"),
    ))
    registry.register(ToolDefinition(
        name="query_fund_quote",
        input_model=FundQuoteQuery,
        handler=cast(Any, query_fund_quote_tool),
        required_permission="fund:quote:read",
        allowed_roles=("customer", "advisor", "operator", "risk_operator", "admin"),
        timeout_seconds=5,
    ))
    registry.register(ToolDefinition(
        name="query_financial_data",
        input_model=FinancialNL2SQLInput,
        handler=cast(Any, query_financial_data_tool),
        required_permission="financial:nl2sql:read",
        allowed_roles=("advisor", "operator", "admin", "super_admin"),
        timeout_seconds=10,
    ))
    registry.register(ToolDefinition(
        name="query_investment_goal",
        input_model=InvestmentGoalQuery,
        handler=cast(Any, investment_goal_query_tool),
        required_permission="investment-goal:read:self",
        allowed_roles=("customer", "advisor", "operator", "admin"),
    ))
    registry.register(ToolDefinition(
        name="query_customer_profile",
        input_model=CustomerProfileQuery,
        handler=cast(Any, customer_profile_query_tool),
        required_permission="customer-profile:read:self",
        allowed_roles=("customer",),
    ))
    registry.register(ToolDefinition(
        name="generate_asset_allocation",
        input_model=AssetAllocationQuery,
        handler=cast(Any, asset_allocation_tool),
        required_permission="asset-allocation:generate:self",
        allowed_roles=("customer",),
    ))
    registry.register(ToolDefinition(
        name="recommend_products",
        input_model=ProductRecommendationQuery,
        handler=cast(Any, product_recommendation_tool),
        required_permission="product-recommendation:read:self",
        allowed_roles=("customer",),
        timeout_seconds=10,
    ))
    registry.register(ToolDefinition(
        name="analyze_portfolio",
        input_model=PortfolioAnalysisQuery,
        handler=cast(Any, portfolio_analysis_tool),
        required_permission="portfolio-analysis:read:self",
        allowed_roles=("customer",),
        timeout_seconds=10,
    ))
    registry.register(ToolDefinition(
        name="simulate_portfolio_rebalance",
        input_model=PortfolioRebalanceQuery,
        handler=cast(Any, portfolio_rebalance_tool),
        required_permission="portfolio-analysis:read:self",
        allowed_roles=("customer",),
        timeout_seconds=10,
    ))
    model_service = ModelGenerationService(ModelDispatchService(DatabaseModelGateway()))
    endpoint_resolver = DatabaseModelEndpointResolver()
    factory = AgentFactory(
        model_service=model_service,
        tool_executor=ToolExecutor(registry),
        intent_classifier=IntentClassifier(model_service),
        intent_endpoint_resolver=endpoint_resolver,
    )
    factory.register(
        OffsiteFundAgent.definition,
        lambda _context: OffsiteFundAgent(OffsiteFundAgent.definition),
    )
    factory.register(
        AdvisorAgent.definition,
        lambda _context: AdvisorAgent(AdvisorAgent.definition),
    )
    return factory