from __future__ import annotations

from textual import on
from textual.app import App, ComposeResult
from textual.widgets import Collapsible, Label
from textual.widgets._collapsible import CollapsibleTitle

COLLAPSED_CLASS = "-collapsed"


def get_title(collapsible: Collapsible) -> CollapsibleTitle:
    return collapsible.get_child_by_type(CollapsibleTitle)


def get_contents(collapsible: Collapsible) -> Collapsible.Contents:
    return collapsible.get_child_by_type(Collapsible.Contents)


async def test_collapsible():
    """It should be possible to access title and collapsed."""
    collapsible = Collapsible(title="Pilot", collapsed=True)
    assert collapsible._title.label == "Pilot"
    assert collapsible.collapsed


async def test_compose_default_collapsible():
    """Test default settings of Collapsible with 1 widget in contents."""

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(Label("Some Contents"))

    async with CollapsibleApp().run_test() as pilot:
        collapsible = pilot.app.query_one(Collapsible)
        assert get_title(collapsible).label == "Toggle"
        assert len(get_contents(collapsible).children) == 1


async def test_compose_empty_collapsible():
    """It should be possible to create an empty Collapsible."""

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible()

    async with CollapsibleApp().run_test() as pilot:
        collapsible = pilot.app.query_one(Collapsible)
        assert len(get_contents(collapsible).children) == 0


async def test_compose_nested_collapsible():
    """Children Collapsibles are independent from parents Collapsibles."""

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            with Collapsible(Label("Outer"), id="outer", collapsed=False):
                yield Collapsible(Label("Inner"), id="inner", collapsed=False)

    async with CollapsibleApp().run_test() as pilot:
        outer: Collapsible = pilot.app.get_child_by_id("outer", Collapsible)
        inner: Collapsible = get_contents(outer).get_child_by_id("inner", Collapsible)
        outer.collapsed = True
        assert not inner.collapsed


async def test_compose_expanded_collapsible():
    """It should be possible to create a Collapsible with expanded contents."""

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(collapsed=False)

    async with CollapsibleApp().run_test() as pilot:
        collapsible = pilot.app.query_one(Collapsible)
        assert not get_title(collapsible).has_class(COLLAPSED_CLASS)
        assert not get_contents(collapsible).has_class(COLLAPSED_CLASS)


async def test_collapsible_collapsed_contents_display_false():
    """Test default settings of Collapsible with 1 widget in contents."""

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(Label("Some Contents"), collapsed=True)

    async with CollapsibleApp().run_test() as pilot:
        collapsible = pilot.app.query_one(Collapsible)
        assert not get_contents(collapsible).display


async def test_collapsible_expanded_contents_display_true():
    """Test default settings of Collapsible with 1 widget in contents."""

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(Label("Some Contents"), collapsed=False)

    async with CollapsibleApp().run_test() as pilot:
        collapsible = pilot.app.query_one(Collapsible)
        assert get_contents(collapsible).display


async def test_toggle_title():
    """Clicking title should update ``collapsed``."""

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(collapsed=False)

    async with CollapsibleApp().run_test() as pilot:
        collapsible = pilot.app.query_one(Collapsible)
        assert not collapsible.collapsed

        await pilot.click(CollapsibleTitle)
        assert collapsible.collapsed

        await pilot.click(CollapsibleTitle)
        assert not collapsible.collapsed


async def test_toggle_message():
    """Toggling should post a message."""

    hits = []

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(collapsed=True)

        @on(Collapsible.Toggled)
        def catch_collapsible_events(self) -> None:
            hits.append("toggled")

    async with CollapsibleApp().run_test() as pilot:
        assert pilot.app.query_one(Collapsible).collapsed

        await pilot.click(CollapsibleTitle)
        await pilot.pause()

        assert not pilot.app.query_one(Collapsible).collapsed
        assert len(hits) == 1

        await pilot.click(CollapsibleTitle)
        await pilot.pause()

        assert pilot.app.query_one(Collapsible).collapsed
        assert len(hits) == 2


async def test_expand_message():
    """Clicking to expand should post a message."""

    hits = []

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(collapsed=True)

        def on_collapsible_expanded(self) -> None:
            hits.append("expanded")

    async with CollapsibleApp().run_test() as pilot:
        assert pilot.app.query_one(Collapsible).collapsed

        await pilot.click(CollapsibleTitle)
        await pilot.pause()

        assert not pilot.app.query_one(Collapsible).collapsed
        assert len(hits) == 1


async def test_expand_via_watcher_message():
    """Setting `collapsed` to `False` should post a message."""

    hits = []

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(collapsed=True)

        def on_collapsible_expanded(self) -> None:
            hits.append("expanded")

    async with CollapsibleApp().run_test() as pilot:
        assert pilot.app.query_one(Collapsible).collapsed

        pilot.app.query_one(Collapsible).collapsed = False
        await pilot.pause()

        assert not pilot.app.query_one(Collapsible).collapsed
        assert len(hits) == 1


async def test_collapse_message():
    """Clicking on collapsed should post a message."""

    hits = []

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(collapsed=False)

        def on_collapsible_collapsed(self) -> None:
            hits.append("collapsed")

    async with CollapsibleApp().run_test() as pilot:
        assert not pilot.app.query_one(Collapsible).collapsed

        await pilot.click(CollapsibleTitle)
        await pilot.pause()

        assert pilot.app.query_one(Collapsible).collapsed
        assert len(hits) == 1


async def test_collapse_via_watcher_message():
    """Setting `collapsed` to `True` should post a message."""

    hits = []

    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(collapsed=False)

        def on_collapsible_collapsed(self) -> None:
            hits.append("collapsed")

    async with CollapsibleApp().run_test() as pilot:
        assert not pilot.app.query_one(Collapsible).collapsed

        pilot.app.query_one(Collapsible).collapsed = True
        await pilot.pause()

        assert pilot.app.query_one(Collapsible).collapsed
        assert len(hits) == 1


async def test_collapsible_title_reactive_change():
    class CollapsibleApp(App[None]):
        def compose(self) -> ComposeResult:
            yield Collapsible(title="Old title")

    async with CollapsibleApp().run_test() as pilot:
        collapsible = pilot.app.query_one(Collapsible)
        assert get_title(collapsible).label == "Old title"
        collapsible.title = "New title"
        assert get_title(collapsible).label == "New title"