import datetime
from enum import Enum
from typing import Annotated

from pydantic import BaseModel, BeforeValidator

ObjectId = Annotated[str, BeforeValidator(str)]


class FlightModel(BaseModel):
    user: ObjectId

    date: datetime.date
    aircraft: str = ""
    waypoint_from: str = ""
    waypoint_to: str = ""
    route: str = ""

    hobbs_start: float | None = None
    hobbs_end: float | None = None
    tach_start: float | None = None
    tach_end: float | None = None

    time_start: datetime.datetime | None = None
    time_end: datetime.datetime | None = None
    time_down: datetime.datetime | None = None
    time_stop: datetime.datetime | None = None

    time_total: float = 0.
    time_pic: float = 0.
    time_sic: float = 0.
    time_night: float = 0.
    time_solo: float = 0.

    time_xc: float = 0.
    dist_xc: float = 0.

    takeoffs_day: int = 0
    landings_day: int = 0
    takeoffs_night: int = 0
    landings_all: int = 0

    time_instrument: float = 0
    time_sim_instrument: float = 0
    holds_instrument: float = 0

    dual_given: float = 0
    dual_recvd: float = 0
    time_sim: float = 0
    time_ground: float = 0

    tags: list[str] = []

    pax: list[str] = []
    crew: list[str] = []

    comments: str = ""


class AuthLevel(Enum):
    GUEST = 0
    USER = 1
    ADMIN = 2

    def __lt__(self, other):
        if self.__class__ is other.__class__:
            return self.value < other.value
        return NotImplemented

    def __gt__(self, other):
        if self.__class__ is other.__class__:
            return self.value > other.value
        return NotImplemented

    def __eq__(self, other):
        if self.__class__ is other.__class__:
            return self.value == other.value
        return NotImplemented


class LoginUserSchema(BaseModel):
    username: str
    password: str


class CreateUserSchema(BaseModel):
    username: str
    password: str
    level: AuthLevel = AuthLevel.USER


class UpdateUserSchema(BaseModel):
    username: str | None = None
    password: str | None = None
    level: AuthLevel | None = None


class GetUserSchema(BaseModel):
    id: str
    username: str
    level: AuthLevel = AuthLevel.USER


class GetSystemUserSchema(GetUserSchema):
    password: str


class TokenSchema(BaseModel):
    access_token: str
    refresh_token: str


class TokenPayload(BaseModel):
    sub: str = None
    exp: int = None