from datetime import datetime
from fastapi import APIRouter, HTTPException, Path, Query
from typing import Any
from app.services.tjnetwork import get_scheme_schema, get_scheme, get_all_schemes
from app.services.scheme_management import query_scheme_detail
from app.services.time_api import extract_date

router = APIRouter()

@router.get("/getschemeschema/", summary="获取方案模式", description="获取指定网络的方案模式定义")
async def fastapi_get_scheme_schema(network: str = Query(..., description="管网名称（或数据库名称）")) -> dict[str, dict[Any, Any]]:
    """
    获取方案模式定义
    
    返回指定网络的方案模式结构定义
    """
    return get_scheme_schema(network)

@router.get("/getscheme/", summary="获取单个方案", description="根据名称获取指定的方案信息")
async def fastapi_get_scheme(network: str = Query(..., description="管网名称（或数据库名称）"), schema_name: str = Query(..., description="方案名称")) -> dict[Any, Any]:
    """
    获取单个方案详情
    
    返回指定网络中指定名称的方案详细信息
    """
    return get_scheme(network, schema_name)

@router.get("/schemes", summary="获取所有方案", description="获取指定网络的所有方案信息")
@router.get("/getallschemes/", summary="获取所有方案（旧路径）", description="获取指定网络的所有方案信息", deprecated=True)
async def fastapi_get_all_schemes(
    network: str = Query(..., description="管网名称（或数据库名称）"),
    scheme_type: str | None = Query(None, description="方案类型；为空时返回全部类型"),
    query_date: datetime | None = Query(None, description="查询日期（可选）"),
) -> list[dict[Any, Any]]:
    """
    获取所有方案列表
    
    返回指定网络中所有可用的方案
    """
    parsed_date = (
        extract_date(query_date, field_name="query_date")
        if query_date is not None
        else None
    )
    return get_all_schemes(network, scheme_type=scheme_type, query_date=parsed_date)


@router.get("/schemes/{scheme_name}", summary="获取方案详情", description="按方案类型获取指定方案详情")
async def fastapi_get_scheme_detail(
    scheme_name: str = Path(..., description="方案名称"),
    network: str = Query(..., description="管网名称（或数据库名称）"),
    scheme_type: str | None = Query(None, description="方案类型；为空时返回通用方案详情"),
) -> dict[Any, Any]:
    result = query_scheme_detail(
        name=network,
        scheme_name=scheme_name,
        scheme_type=scheme_type,
    )
    if not result:
        raise HTTPException(status_code=404, detail=f"Scheme {scheme_name} not found")
    return result