ship customization: renaming

1 files changed, 21 insertions, 0 deletions

diff --git a/apirouters/customize_ship.py b/apirouters/customize_ship.py
new file mode 100644
index 0000000..7be3279
--- /dev/null
+++ b/apirouters/customize_ship.py
@@ -0,0 +1,21 @@
+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
+
+
+@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."}