from fastapi import APIRouter
from typing import List
from fastapi import APIRouter, HTTPException

# from app.routers import get_cached_data
from app.schemas.user import ChapterRelationCreate, ChapterRelationResponse
from app.services.user_service import  get_graph_relations, search_chapters, \
    get_relations_to_level_simple, get_chapter_relations

router = APIRouter(
    prefix="/chapters",
    tags=["Chapters"],
)


@router.get("/relations/", response_model=List[ChapterRelationResponse])
async def list_chapter_relations():
    # cache_key = "all_chapter_relations"
    # cached_data = get_cached_data(cache_key)  # 从缓存获取
    # if cached_data:
    #     return [ChapterRelationResponse(**data) for data in cached_data]  # 缓存命中，返回数据
    """
    查询所有章节关系
    """
    return await get_graph_relations()


@router.get("/search_chapters", response_model=List[ChapterRelationResponse])
async def search_chapters_endpoint(q: str):
    """
    API 端点：根据搜索关键字模糊查询章节及其相关章节
    :param q: 搜索关键字
    :return: 匹配的章节关系列表
    """
    # cache_key = f"search_chapters:{q}"
    # # 尝试从缓存中获取数据
    # cached_data = get_cached_data(cache_key)
    # if cached_data:
    #     # 如果缓存命中，直接返回缓存中的数据
    #     return [ChapterRelationResponse(**data) for data in cached_data]
    if not q:
        raise HTTPException(status_code=400, detail="搜索关键字不能为空")
    results = await search_chapters(q)
    if not results:
        raise HTTPException(status_code=404, detail="未找到匹配的章节关系")
    return results
@router.get("/relations/level/{level}", response_model=List[ChapterRelationResponse])
async def get_relations_by_level_simple(level: int):
    """
    简化版：根据目标层级查询从 Root 到目标层级的所有节点和关系
    :param level: 目标层级（1 -> Root & Subject, 2 -> Root, Subject, Topic, ..., 5 -> Problem）
    :return: 对应层级的节点和关系
    """
    # cache_key = f"level_{level}_relations"
    # cached_data = get_cached_data(cache_key)  # 从缓存获取
    # if cached_data:
    #     return [ChapterRelationResponse(**data) for data in cached_data]  # 缓存命中，返回数据

    if level < 0 or level > 5:
        raise HTTPException(status_code=400, detail="层级参数必须在 0 到 5 之间")

    try:
        relations =  await get_relations_to_level_simple(level)
        if not relations:
            raise HTTPException(status_code=404, detail="未找到相关节点关系")
        return relations
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))
@router.get("/{chapter}/relations", response_model=List[ChapterRelationResponse])
async def get_relations_by_chapter(chapter: str):
    """
    根据章节名称获取该章节及其相关关系，但排除 label 为 Subject 的 next_SB 关系
    :param chapter_name: 章节名称
    :return: 章节关系列表
    """
    # cache_key = f"chapter_relations:{chapter}"  # 不使用分页，直接根据章节名称生成缓存键
    # cached_data = get_cached_data(cache_key)  # 从缓存中获取数据
    # if cached_data:
    #     return [ChapterRelationResponse(**data) for data in cached_data]  # 如果缓存命中，直接返回缓存数据
    relations = await get_chapter_relations(chapter)
    if not relations:
        raise HTTPException(status_code=404, detail=f"No relations found for chapter {chapter}")
    return relations