#  Copyright 2026 zhaoxi826
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

from fastapi import APIRouter, Depends, HTTPException, status, UploadFile, File
from pydantic import BaseModel
from pretor.utils.access import Accessor, TokenData
from pretor.api.platform.event import PretorEvent
from pretor.utils.ray_hook import ray_actor_hook
import os
import anyio
from pretor.utils.logger import get_logger


logger = get_logger("frontend")
client_router = APIRouter(prefix="/api/v1/adapter/client", tags=["client"])


class Message(BaseModel):
    """Message 核心组件类。
    这是一个领域数据模型或功能封装类，承载了 Message 相关的内聚属性定义与状态维护。它的存在隔离了局部的业务复杂性，并对外提供了类型安全的访问接口。"""

    message: str


@client_router.post("")
async def create_message(
    message: Message, token_data: TokenData = Depends(Accessor.get_current_user)
):
    """处理针对 create message 相关的 HTTP API 请求。
    该接口负责解析前端传入的载荷数据，调用底层核心业务逻辑进行处理，并组装标准化的 JSON 响应。
    Args:        message (Message): 参与 create message 逻辑运算或数据构建的上下文依赖对象。        token_data (TokenData): 从客户端传递过来或由上游组件生成的核心业务数据体，通常需要进一步的清洗和结构化解析。
    Returns:        : 序列化后的标准网络响应模型（如包含业务状态码、成功标志及对应的数据载荷 Data）。"""
    logger.info("收到消息，来源：客户端")
    logger.debug(f"消息内容：{message.message}")
    event = PretorEvent(
        platform="client",
        user_id=str(token_data.user_id),
        user_name=token_data.username,
        message=message.message,
    )
    supervisory_node = ray_actor_hook("supervisory_node").supervisory_node
    message = await supervisory_node.working.remote(event)
    if message.startswith("任务已创建"):
        return {"message": f"{event.trace_id}\n\n{message}"}
    elif message == "未知相应类型":
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="模型回复错误"
        )
    else:
        return {"message": message}


@client_router.post("/upload")
async def upload_file(
    file: UploadFile = File(...),
    token_data: TokenData = Depends(Accessor.get_current_user),
):
    """处理针对 upload file 相关的 HTTP API 请求。
    该接口负责解析前端传入的载荷数据，调用底层核心业务逻辑进行处理，并组装标准化的 JSON 响应。
    Args:        file (UploadFile): 参与 upload file 逻辑运算或数据构建的上下文依赖对象。        token_data (TokenData): 从客户端传递过来或由上游组件生成的核心业务数据体，通常需要进一步的清洗和结构化解析。
    Returns:        : 序列化后的标准网络响应模型（如包含业务状态码、成功标志及对应的数据载荷 Data）。"""
    try:
        upload_dir = "uploads"
        os.makedirs(upload_dir, exist_ok=True)
        file_path = os.path.join(upload_dir, file.filename)
        async with await anyio.open_file(file_path, "wb") as buffer:
            while chunk := await file.read(64 * 1024):  # 64KB chunks
                await buffer.write(chunk)
        logger.info(f"用户 {token_data.username} 上传了文件: {file.filename}")
        return {
            "filename": file.filename,
            "message": f"File {file.filename} uploaded successfully",
        }
    except Exception as e:
        logger.error(f"文件上传失败: {e}")
        raise HTTPException(status_code=500, detail="文件上传失败")