from __future__ import annotations

from typing import Optional

from rio import Component, Column, Spacer, page, GuardEvent, Rectangle, Text, FlowContainer, Color, ProgressCircle
from rio.event import on_populate

from elm.services import UserService
from elm.types import UserSession
from elm.types.CateringTypes import CateringOrder, CateringOrderStatus


def my_orders_page_guard(event: GuardEvent) -> Optional[str]:
    try:
        _ = event.session[UserSession].user_name
        return None
    except KeyError:
        return "/"

@page(name="My Orders", url_segment="my-orders", guard=my_orders_page_guard)
class MyOrdersPage(Component):
    orders: list[CateringOrder] = []
    is_loading: bool = True

    @on_populate
    async def on_populate(self) -> None:
        user = await self.session[UserService].get_user(self.session[UserSession].user_name)
        if not user:
            self.is_loading = False
            return

        self.orders = await CateringOrder.find_many(CateringOrder.customer_id == user.id).to_list()
        self.is_loading = False

    def get_status_color(self, status: CateringOrderStatus) -> Color:
        color = self.session.theme.warning_color
        if status == CateringOrderStatus.DELAYED or status == CateringOrderStatus.CANCELED:
            color = self.session.theme.danger_color
        elif status == CateringOrderStatus.COMPLETED:
            color = self.session.theme.success_color
        return color

    def build(self) -> Component:
        if self.is_loading:
            return ProgressCircle(margin=self.session.screen_width // 5)
        return Column(
            FlowContainer(
                *[Rectangle(
                    content=Column(
                        Rectangle(
                            content=Rectangle(
                                content=Text(f"Bestellung\n\n{str(order.id)[-5:]}", margin=0.5, selectable=False, overflow="wrap", justify="center"),
                                fill=self.session.theme.header_box_background_color,
                                margin=0.4
                            ),
                            stroke_width=0.1,
                            stroke_color=self.session.theme.box_border_color,
                        ),
                        Column(
                            *[Text(item.name, overflow="wrap") for item in order.items],
                            spacing=1,
                            margin=1
                        ),
                        Spacer(),
                        Text(CateringOrder.translate_order_status(order.status), fill=self.get_status_color(order.status), margin=1, font_weight="bold", font_size=1, justify="center"),
                    ),
                    fill=self.session.theme.box_color,
                    min_width=18
                ) for order in self.orders],
                spacing=1
            ),
            Spacer(),
            spacing=1,
            margin=1
        )