from typing import Annotated
from fastapi import APIRouter, Depends
from pydantic import BaseModel

from .agents import auth_agent, Agent

router = APIRouter()


class RenameBody(BaseModel):
    name: str


# TODO depend on ship instead of auth_agent
@router.post("/customize_ship/{ship_symbol}/rename")
async def rename(ship_symbol: str, rename_body: RenameBody, agent: Annotated[Agent, Depends(auth_agent)]):
    for current_ship in agent.ships.values():
        if current_ship.symbol == ship_symbol:
            current_ship.rename(rename_body.name)
            return current_ship.get_data()

    return {"error": "Unknown ship symbol."}