from typing import cast

from mdbx.mdbx import TXN

import utils.netex_monkeypatching  # noqa: F401
from domain.netex.model import (
    AvailabilityCondition,
    UicOperatingPeriod,
    OperatingDay,
    DayTypeAssignment,
    DayType,
    OperatingDayRef,
    DayTypeRef,
    DayOfWeekEnumeration,
    UicOperatingPeriodRef,
)
from domain.netex.services.refs import getRef
from storage.mdbx.core.implementation import MdbxStorage

# from netexio.database import Database
# from netexio.dbaccess import load_local
# from utils.refs import getRef
from utils.utils import project


def datetime_weekday_to_dow(wd: int) -> DayOfWeekEnumeration:
    match wd:
        case 0:
            return DayOfWeekEnumeration.MONDAY
        case 1:
            return DayOfWeekEnumeration.TUESDAY
        case 2:
            return DayOfWeekEnumeration.WEDNESDAY
        case 3:
            return DayOfWeekEnumeration.THURSDAY
        case 4:
            return DayOfWeekEnumeration.FRIDAY
        case 5:
            return DayOfWeekEnumeration.SATURDAY
        case 6:
            return DayOfWeekEnumeration.SUNDAY

    return DayOfWeekEnumeration.NONE


def get_day_type_from_availability_condition(
    db_read: MdbxStorage, txn: TXN, availability_condition: AvailabilityCondition
) -> tuple[DayType, list[DayTypeAssignment], list[OperatingDay], UicOperatingPeriod]:
    day_type: DayType
    day_type_assignments: list[DayTypeAssignment] = []
    my_operating_days: list[OperatingDay] = []
    uic_operating_period: UicOperatingPeriod

    if availability_condition.day_types is not None:
        if len(availability_condition.day_types.day_type_ref_or_day_type_dummy) == 1:
            if isinstance(availability_condition.day_types.day_type_ref_or_day_type_dummy[0], DayType):
                day_type = availability_condition.day_types.day_type_ref_or_day_type_dummy[0]
            elif isinstance(availability_condition.day_types.day_type_ref_or_day_type_dummy[0], DayTypeRef):
                day_type = db_read.load_object_by_reference(txn, availability_condition.day_types.day_type_ref_or_day_type_dummy[0])
        else:
            day_type = cast(DayType, project(availability_condition, DayType))
    else:
        day_type = cast(DayType, project(availability_condition, DayType))

    if availability_condition.valid_day_bits is not None:
        uic_operating_period = cast(UicOperatingPeriod, project(availability_condition, UicOperatingPeriod))
        if uic_operating_period:
            uic_operating_period.from_operating_day_ref_or_from_date = availability_condition.from_date
            uic_operating_period.to_operating_day_ref_or_to_date = availability_condition.to_date

        day_type_assignment: DayTypeAssignment = cast(DayTypeAssignment, project(availability_condition, DayTypeAssignment))
        if day_type_assignment:
            day_type_assignment.uic_operating_period_ref_or_operating_period_ref_or_operating_day_ref_or_date = cast(
                UicOperatingPeriodRef, getRef(uic_operating_period)
            )
            day_type_assignment.day_type_ref = cast(DayTypeRef, getRef(day_type))
            day_type_assignment.order = 1
            day_type_assignments.append(day_type_assignment)

    elif availability_condition.operating_days is not None:
        for operating_day in availability_condition.operating_days.operating_day_ref_or_operating_day:
            if isinstance(operating_day, OperatingDayRef):
                operating_day_ref = operating_day
                operating_day = db_read.load_object_by_reference(txn, operating_day)
            else:
                operating_day_ref = cast(OperatingDayRef, getRef(operating_day))

            my_operating_days.append(operating_day)
            day_type_assignment = cast(DayTypeAssignment, project(operating_day, DayTypeAssignment))
            day_type_assignment.is_available = availability_condition.is_available
            day_type_assignment.uic_operating_period_ref_or_operating_period_ref_or_operating_day_ref_or_date = operating_day_ref
            day_type_assignment.day_type_ref = cast(DayTypeRef, getRef(day_type))
            day_type_assignment.order = 1
            day_type_assignments.append(day_type_assignment)

    return day_type, day_type_assignments, my_operating_days, uic_operating_period