from asyncio import sleep
from contextlib import asynccontextmanager

from nicegui import ui
from nicegui.elements.button import Button
from nicegui.elements.mixins.disableable_element import DisableableElement

from ..documentation_tools import text_demo


def main_demo() -> None:
    ui.button('Click me!', on_click=lambda: ui.notify(f'You clicked me!'))


def more() -> None:
    @text_demo('Icons', '''
        You can also add an icon to a button.
    ''')
    async def icons() -> None:
        with ui.row():
            ui.button('demo', icon='history')
            ui.button(icon='thumb_up')
            with ui.button():
                ui.label('sub-elements')
                ui.image('https://picsum.photos/id/377/640/360') \
                    .classes('rounded-full w-16 h-16 ml-4')

    @text_demo('Await button click', '''
        Sometimes it is convenient to wait for a button click before continuing the execution.
    ''')
    async def await_button_click() -> None:
        # @ui.page('/')
        # async def index():
            b = ui.button('Step')
            await b.clicked()
            ui.label('One')
            await b.clicked()
            ui.label('Two')
            await b.clicked()
            ui.label('Three')

    @text_demo('Disable button with a context manager', '''
        This showcases a async context manager that can be used to disable a button for the duration of an async process.
    ''')
    async def disable_context_manager() -> None:
        @asynccontextmanager
        async def disable(element: DisableableElement) -> None:
            element.disable()
            try:
                yield
            finally:
                element.enable()

        async def disable_and_sleep_3(button: Button) -> None:
            async with disable(button):
                await sleep(3)

        b = ui.button("Disable for 3 seconds", on_click=lambda: disable_and_sleep_3(b))