from typing import Optional

from .base import FlightDatabaseBase
from models.flight import Flight, FlightState, Waypoint
from models.results import FrameResult
from models.chunks import ChunkHandle
from models.core import Pose


class FlightDatabase(FlightDatabaseBase):
    async def create_flight(self, flight: Flight) -> str:
        raise NotImplementedError

    async def get_flight(self, flight_id: str) -> Optional[Flight]:
        raise NotImplementedError

    async def delete_flight(self, flight_id: str) -> bool:
        raise NotImplementedError

    async def update_flight_state(self, state: FlightState) -> bool:
        raise NotImplementedError

    async def get_flight_state(self, flight_id: str) -> Optional[FlightState]:
        raise NotImplementedError

    async def save_frame_result(self, flight_id: str, result: FrameResult) -> bool:
        raise NotImplementedError

    async def get_frame_result(
        self, flight_id: str, frame_id: int
    ) -> Optional[FrameResult]:
        raise NotImplementedError

    async def get_all_frame_results(self, flight_id: str) -> list[FrameResult]:
        raise NotImplementedError

    async def save_chunk(self, chunk: ChunkHandle) -> bool:
        raise NotImplementedError

    async def get_chunk(self, chunk_id: str) -> Optional[ChunkHandle]:
        raise NotImplementedError

    async def get_flight_chunks(self, flight_id: str) -> list[ChunkHandle]:
        raise NotImplementedError

    async def save_pose(self, flight_id: str, pose: Pose) -> bool:
        raise NotImplementedError

    async def get_poses(self, flight_id: str) -> list[Pose]:
        raise NotImplementedError

    async def update_waypoints(
        self, flight_id: str, waypoints: list[Waypoint]
    ) -> bool:
        raise NotImplementedError