import flet as ft
from views import main_view, login_view, reg_view, user_view

PUBLIC_ROUTES = {"/login", "/reg"}


def resolve_view(page: ft.Page, route: str) -> list[ft.View] | None:
    """
    Возвращает список вью для заданного маршрута или None если маршрут неизвестен.
    Flet использует page.views как стек: последний элемент — активный экран.
    """
    if route == "/login":
        return [login_view.LoginView(page)]
    elif route == "/reg":
        return [reg_view.RegView(page)]
    elif route == "/main":
        return [main_view.MainView()]
    elif route.startswith("/user/"):
        return [user_view.UserView(page)]
    return None