from typing import Optional

from fastapi import APIRouter
from pydantic import BaseModel

from ..rotation_service import rotate as rotate_impl, status as status_impl


router = APIRouter()


class RotateRequest(BaseModel):
    cityhash: Optional[str] = None
    num: Optional[int] = None


@router.post("/rotate")
def rotate(req: RotateRequest):
    result = rotate_impl(cityhash=req.cityhash, num=req.num)
    return result


@router.get("/status")
def get_status():
    return status_impl()