#  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
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

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: str

@client_router.post("")
async def create_message(message: Message,
                         token_data: TokenData = Depends(Accessor.get_current_user)):
    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 == "任务已创建":
        return {"message": event.trace_id}
    elif message == "未知相应类型":
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail="模型回复错误")
    else:
        return {"message": message}