import json
import logging
from typing import Any, Dict, List

from fastapi import APIRouter, HTTPException
from fastapi.responses import StreamingResponse

from models.studio_models import (
    AdvanceRunRequest,
    CreateStudioProjectRequest,
    PipelineDefinition,
    RenameRunRequest,
    RunMessageRequest,
    RunRerollRequest,
    SaveRunRequest,
    StudioProject,
    StudioProjectSummary,
    StudioRun,
    StudioRunSummary,
    SwitchRunNodeRequest,
    UpdateStudioProjectRequest,
    WorkflowTemplateSummary,
    WorkflowVariablesResponse,
)
from services.studio_project_service import studio_project_service
from services.studio_run_service import studio_run_service

logger = logging.getLogger(__name__)

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


@router.get("/projects", response_model=List[StudioProjectSummary])
async def list_studio_projects():
    try:
        return studio_project_service.list_projects()
    except Exception as e:
        logger.error("Failed to list studio projects: %s", e)
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/projects/{project_id}", response_model=StudioProject)
async def get_studio_project(project_id: str):
    try:
        return studio_project_service.get_project(project_id)
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except Exception as e:
        logger.error("Failed to get studio project %s: %s", project_id, e)
        raise HTTPException(status_code=500, detail=str(e))


@router.patch("/projects/{project_id}", response_model=StudioProject)
async def update_studio_project(project_id: str, req: UpdateStudioProjectRequest):
    try:
        return studio_project_service.update_project_meta(
            project_id,
            name=req.name,
            description=req.description,
        )
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except Exception as e:
        logger.error("Failed to update studio project %s: %s", project_id, e)
        raise HTTPException(status_code=500, detail=str(e))


@router.put("/projects/{project_id}/pipeline", response_model=StudioProject)
async def save_studio_pipeline(project_id: str, pipeline: PipelineDefinition):
    try:
        return studio_project_service.save_pipeline(project_id, pipeline)
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except Exception as e:
        logger.error("Failed to save pipeline for %s: %s", project_id, e)
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/templates", response_model=List[WorkflowTemplateSummary])
async def list_workflow_templates():
    try:
        return studio_project_service.list_workflow_templates()
    except Exception as e:
        logger.error("Failed to list workflow templates: %s", e)
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/variables", response_model=WorkflowVariablesResponse)
async def get_workflow_variables(projectId: str | None = None):
    try:
        return studio_project_service.get_workflow_variables(projectId)
    except Exception as e:
        logger.error("Failed to load workflow variables: %s", e)
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/skill-templates")
async def get_skill_templates() -> Dict[str, Any]:
    try:
        return studio_project_service.get_skill_templates()
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except Exception as e:
        logger.error("Failed to load skill templates: %s", e)
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/niches")
async def get_niches() -> Dict[str, Any]:
    try:
        return studio_project_service.get_niches()
    except Exception as e:
        logger.error("Failed to load niches: %s", e)
        raise HTTPException(status_code=500, detail=str(e))


@router.delete("/projects/{project_id}")
async def delete_studio_project(project_id: str):
    try:
        studio_project_service.delete_project(project_id)
        return {"ok": True, "id": project_id}
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except Exception as e:
        logger.error("Failed to delete studio project %s: %s", project_id, e)
        raise HTTPException(status_code=500, detail=str(e))


@router.post("/projects", response_model=StudioProject)
async def create_studio_project(req: CreateStudioProjectRequest):
    try:
        return studio_project_service.create_project(req)
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except Exception as e:
        logger.error("Failed to create studio project: %s", e)
        raise HTTPException(status_code=500, detail=str(e))


@router.post("/projects/{project_id}/runs", response_model=StudioRun)
async def create_studio_run(project_id: str):
    try:
        return studio_run_service.create_run(project_id)
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except Exception as e:
        logger.error("Failed to create studio run for %s: %s", project_id, e)
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/projects/{project_id}/runs", response_model=List[StudioRunSummary])
async def list_studio_runs(project_id: str):
    try:
        studio_project_service.get_project(project_id)
        return studio_run_service.list_runs(project_id)
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except Exception as e:
        logger.error("Failed to list studio runs for %s: %s", project_id, e)
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/projects/{project_id}/runs/{run_id}", response_model=StudioRun)
async def get_studio_run(project_id: str, run_id: str):
    try:
        return studio_run_service.get_run(project_id, run_id)
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except Exception as e:
        logger.error("Failed to get studio run %s/%s: %s", project_id, run_id, e)
        raise HTTPException(status_code=500, detail=str(e))


@router.post("/projects/{project_id}/runs/{run_id}/advance", response_model=StudioRun)
async def advance_studio_run(
    project_id: str, run_id: str, req: AdvanceRunRequest
):
    try:
        return studio_run_service.advance_run(
            project_id, run_id, display_params=req.displayParams, save_mode=req.saveMode
        )
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except NotImplementedError as e:
        raise HTTPException(status_code=501, detail=str(e))
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))
    except Exception as e:
        logger.error(
            "Failed to advance studio run %s/%s: %s", project_id, run_id, e
        )
        raise HTTPException(status_code=500, detail=str(e))


@router.post("/projects/{project_id}/runs/{run_id}/save", response_model=StudioRun)
async def save_studio_run(
    project_id: str, run_id: str, req: SaveRunRequest
):
    try:
        return studio_run_service.save_run(project_id, run_id, req.mode)
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))
    except Exception as e:
        logger.error(
            "Failed to save studio run %s/%s: %s", project_id, run_id, e
        )
        raise HTTPException(status_code=500, detail=str(e))


@router.post("/projects/{project_id}/runs/{run_id}/switch-node", response_model=StudioRun)
async def switch_studio_run_node(
    project_id: str, run_id: str, req: SwitchRunNodeRequest
):
    try:
        return studio_run_service.switch_run_node(project_id, run_id, req.nodeId)
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))
    except Exception as e:
        logger.error(
            "Failed to switch studio run node %s/%s: %s", project_id, run_id, e
        )
        raise HTTPException(status_code=500, detail=str(e))


@router.post("/projects/{project_id}/runs/{run_id}/message")
async def send_studio_run_message(
    project_id: str, run_id: str, req: RunMessageRequest
):
    if req.stream:
        async def ndjson_stream():
            try:
                async for event in studio_run_service.send_run_message_stream(
                    project_id,
                    run_id,
                    req.content,
                    profile_id=req.profileId,
                    api_config=req.apiConfig,
                ):
                    yield json.dumps(event, ensure_ascii=False) + "\n"
            except FileNotFoundError as e:
                yield json.dumps(
                    {"type": "error", "detail": str(e)}, ensure_ascii=False
                ) + "\n"
            except ValueError as e:
                yield json.dumps(
                    {"type": "error", "detail": str(e)}, ensure_ascii=False
                ) + "\n"
            except Exception as e:
                logger.error(
                    "Failed to stream studio run message %s/%s: %s",
                    project_id,
                    run_id,
                    e,
                )
                yield json.dumps(
                    {"type": "error", "detail": f"消息处理失败：{e}"},
                    ensure_ascii=False,
                ) + "\n"

        return StreamingResponse(
            ndjson_stream(),
            media_type="application/x-ndjson",
        )

    try:
        return await studio_run_service.send_run_message(
            project_id,
            run_id,
            req.content,
            stream=req.stream,
            profile_id=req.profileId,
            api_config=req.apiConfig,
        )
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))
    except Exception as e:
        logger.error(
            "Failed to send studio run message %s/%s: %s", project_id, run_id, e
        )
        raise HTTPException(status_code=500, detail=f"消息处理失败：{e}")


@router.post("/projects/{project_id}/runs/{run_id}/undo", response_model=StudioRun)
async def undo_studio_run(project_id: str, run_id: str):
    try:
        return studio_run_service.undo_run(project_id, run_id)
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))
    except Exception as e:
        logger.error(
            "Failed to undo studio run %s/%s: %s", project_id, run_id, e
        )
        raise HTTPException(status_code=500, detail=f"回退失败：{e}")


@router.post("/projects/{project_id}/runs/{run_id}/reroll")
async def reroll_studio_run(
    project_id: str, run_id: str, req: RunRerollRequest
):
    if req.stream:
        async def ndjson_stream():
            try:
                async for event in studio_run_service.reroll_run_stream(
                    project_id,
                    run_id,
                    profile_id=req.profileId,
                    api_config=req.apiConfig,
                ):
                    yield json.dumps(event, ensure_ascii=False) + "\n"
            except FileNotFoundError as e:
                yield json.dumps(
                    {"type": "error", "detail": str(e)}, ensure_ascii=False
                ) + "\n"
            except ValueError as e:
                yield json.dumps(
                    {"type": "error", "detail": str(e)}, ensure_ascii=False
                ) + "\n"
            except Exception as e:
                logger.error(
                    "Failed to stream studio run reroll %s/%s: %s",
                    project_id,
                    run_id,
                    e,
                )
                yield json.dumps(
                    {"type": "error", "detail": f"重 roll 失败：{e}"},
                    ensure_ascii=False,
                ) + "\n"

        return StreamingResponse(
            ndjson_stream(),
            media_type="application/x-ndjson",
        )

    try:
        return await studio_run_service.reroll_run(
            project_id,
            run_id,
            stream=req.stream,
            profile_id=req.profileId,
            api_config=req.apiConfig,
        )
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))
    except Exception as e:
        logger.error(
            "Failed to reroll studio run %s/%s: %s", project_id, run_id, e
        )
        raise HTTPException(status_code=500, detail=f"重 roll 失败：{e}")


@router.delete("/projects/{project_id}/runs/{run_id}")
async def delete_studio_run(project_id: str, run_id: str):
    try:
        studio_run_service.delete_run(project_id, run_id)
        return {"ok": True, "id": run_id}
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except Exception as e:
        logger.error(
            "Failed to delete studio run %s/%s: %s", project_id, run_id, e
        )
        raise HTTPException(status_code=500, detail=str(e))


@router.patch("/projects/{project_id}/runs/{run_id}", response_model=StudioRun)
async def rename_studio_run(
    project_id: str, run_id: str, req: RenameRunRequest
):
    try:
        return studio_run_service.rename_run(project_id, run_id, req.title)
    except FileNotFoundError as e:
        raise HTTPException(status_code=404, detail=str(e))
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))
    except Exception as e:
        logger.error(
            "Failed to rename studio run %s/%s: %s", project_id, run_id, e
        )
        raise HTTPException(status_code=500, detail=str(e))