from nicegui import ui
from montydb import MontyClient

def main(db: MontyClient):
    def choose(sede, livello):
        if int(livello) < 0:
            page.refresh(sede)
        else:
            ui.open(f"/manager/{sede}/{livello}")
    @ui.refreshable
    def page(sede):
        if int(sede) <0:
            results = db.sedi.find()
            ui.button(on_click=lambda: (app.storage.user.clear(), ui.open('/login')), icon='logout').props('outline round').style("positon:absolute")
            with ui.column().classes("w-full h-screen place-items-center items-center").style(""):
                ui.row().classes("grow")
                for result in results:
                    ui.button(text=str(result['nome']).capitalize(),on_click=lambda id=result['id']:choose(id, -1))
                ui.row().classes("grow")
        else:
            results = db.livelli.find()

            with ui.row().classes("w-full").style("positon:absolute"):
                ui.icon("chevron_left", size="lg").on("click", lambda: page.refresh(-1))
            with ui.column().classes("w-full h-screen place-items-center items-center"):
                ui.row().classes("grow")
                for result in results:
                    ui.button(text=str(result['nome']).capitalize(),on_click=lambda id=result['id']:choose(sede, id))
                ui.row().classes("grow")

    ## FRONTEND
    page(-1)