import unittest
from unittest import mock

import gi
gi.require_version("Atspi", "2.0")
from gi.repository import Atspi

from cthulhu import ax_object
from cthulhu import cthulhu_state
from cthulhu import focus_manager
from cthulhu.scripts import default
from cthulhu.scripts.toolkits.Chromium import script as chromium_script
from cthulhu.scripts.toolkits.Chromium import script_utilities as chromium_utilities


class ChromiumOmniboxRegressionTests(unittest.TestCase):
    def tearDown(self):
        cthulhu_state.activeWindow = None
        cthulhu_state.locusOfFocus = None

    def test_detects_chromium_popup_item_with_broken_ancestry(self):
        popupItem = object()

        with (
            mock.patch.object(ax_object.AXObject, "is_dead", return_value=False),
            mock.patch.object(ax_object.AXObject, "get_toolkit_name", return_value="chromium"),
            mock.patch.object(ax_object.AXObject, "get_role", return_value=Atspi.Role.LIST_ITEM),
            mock.patch.object(ax_object.AXObject, "get_parent", return_value=None),
            mock.patch.object(ax_object.debug, "print_tokens"),
        ):
            self.assertTrue(ax_object.AXObject.has_broken_popup_ancestry(popupItem))

    def test_ignores_chromium_popup_item_with_valid_ancestry(self):
        popupItem = object()
        parent = object()

        def get_role(obj):
            if obj is popupItem:
                return Atspi.Role.LIST_ITEM
            return Atspi.Role.APPLICATION

        with (
            mock.patch.object(ax_object.AXObject, "is_dead", return_value=False),
            mock.patch.object(ax_object.AXObject, "get_toolkit_name", return_value="chromium"),
            mock.patch.object(ax_object.AXObject, "get_role", side_effect=get_role),
            mock.patch.object(ax_object.AXObject, "get_parent", side_effect=[parent, None]),
        ):
            self.assertFalse(ax_object.AXObject.has_broken_popup_ancestry(popupItem))

    def test_preserves_active_window_when_chromium_popup_ancestry_is_broken(self):
        activeWindow = object()
        popupItem = object()
        cthulhu_state.activeWindow = activeWindow
        cthulhu_state.locusOfFocus = popupItem

        controller = mock.Mock()
        app = mock.Mock()
        manager = None
        with (
            mock.patch.object(focus_manager.dbus_service, "get_remote_controller", return_value=controller),
            mock.patch.object(focus_manager.AXObject, "has_broken_popup_ancestry", return_value=True),
        ):
            manager = focus_manager.FocusManager(app)
            manager.set_active_window(None)

        self.assertIs(manager.get_active_window(), activeWindow)
        self.assertIs(cthulhu_state.activeWindow, activeWindow)

    def test_detects_nameless_frame_with_listbox_as_autocomplete_popup(self):
        utilities = chromium_utilities.Utilities.__new__(chromium_utilities.Utilities)
        frame = object()
        listbox = object()

        with (
            mock.patch.object(chromium_utilities.AXUtilities, "is_frame", return_value=True),
            mock.patch.object(chromium_utilities.AXObject, "get_name", return_value=""),
            mock.patch.object(chromium_utilities.AXObject, "get_child_count", return_value=1),
            mock.patch.object(chromium_utilities.AXObject, "get_child", return_value=listbox),
            mock.patch.object(chromium_utilities.AXUtilities, "is_list_box", return_value=True),
        ):
            result = utilities.autocompletePopupForFrame(frame)

        self.assertIs(result, listbox)

    def test_does_not_treat_nested_listbox_as_autocomplete_popup(self):
        utilities = chromium_utilities.Utilities.__new__(chromium_utilities.Utilities)
        frame = object()

        with (
            mock.patch.object(chromium_utilities.AXUtilities, "is_frame", return_value=True),
            mock.patch.object(chromium_utilities.AXObject, "get_name", return_value=""),
            mock.patch.object(chromium_utilities.AXObject, "get_child_count", return_value=2),
        ):
            result = utilities.autocompletePopupForFrame(frame)

        self.assertIsNone(result)

    @staticmethod
    def _make_chromium_script():
        testScript = chromium_script.Script.__new__(chromium_script.Script)
        testScript.utilities = mock.Mock()
        testScript.utilities.canBeActiveWindow.return_value = True
        testScript.utilities.autocompletePopupForFrame.return_value = None
        testScript.utilities.popupMenuForFrame.return_value = None
        testScript._lastAutocompletePopupItem = None
        testScript.presentObject = mock.Mock()
        return testScript

    def test_autocomplete_popup_activation_presents_item_without_frame_fallback(self):
        testScript = self._make_chromium_script()
        frame = object()
        listbox = object()
        item = object()
        event = mock.Mock(source=frame)
        testScript.utilities.autocompletePopupForFrame.return_value = listbox
        testScript.utilities.selectedChildren.return_value = [item]

        with (
            mock.patch.object(chromium_script.AXUtilities, "get_focused_object", return_value=item),
            mock.patch.object(chromium_script.cthulhu, "setActiveWindow") as setActiveWindow,
            mock.patch.object(chromium_script.cthulhu, "setLocusOfFocus") as setLocusOfFocus,
            mock.patch.object(default.Script, "onWindowActivated") as defaultHandler,
        ):
            chromium_script.Script.onWindowActivated(testScript, event)

        setActiveWindow.assert_not_called()
        setLocusOfFocus.assert_not_called()
        testScript.presentObject.assert_called_once_with(item, interrupt=True)
        self.assertIs(testScript._lastAutocompletePopupItem, item)
        defaultHandler.assert_not_called()

    def test_autocomplete_popup_activation_uses_focused_item_when_none_selected(self):
        testScript = self._make_chromium_script()
        frame = object()
        listbox = object()
        item = object()
        event = mock.Mock(source=frame)
        testScript.utilities.autocompletePopupForFrame.return_value = listbox
        testScript.utilities.selectedChildren.return_value = []

        with mock.patch.object(
            chromium_script.AXUtilities,
            "get_focused_object",
            return_value=item,
        ):
            chromium_script.Script.onWindowActivated(testScript, event)

        testScript.presentObject.assert_called_once_with(item, interrupt=True)
        self.assertIs(testScript._lastAutocompletePopupItem, item)

    def test_autocomplete_popup_selection_is_presented_without_changing_document_focus(self):
        testScript = self._make_chromium_script()
        comboBox = object()
        listbox = object()
        frame = object()
        item = object()
        event = mock.Mock(source=item, detail1=True)
        cthulhu_state.locusOfFocus = comboBox
        testScript.utilities.inDocumentContent.return_value = False

        with (
            mock.patch.object(chromium_script.web.Script, "onSelectedChanged", return_value=False),
            mock.patch.object(
                chromium_script.AXObject,
                "find_ancestor_inclusive",
                return_value=listbox,
            ),
            mock.patch.object(chromium_script.AXObject, "get_parent", return_value=frame),
            mock.patch.object(chromium_script.AXObject, "get_name", return_value=""),
            mock.patch.object(chromium_script.AXUtilities, "is_frame", return_value=True),
        ):
            chromium_script.Script.onSelectedChanged(testScript, event)
            chromium_script.Script.onSelectedChanged(testScript, event)

        self.assertIs(cthulhu_state.locusOfFocus, comboBox)
        testScript.presentObject.assert_called_once_with(item, interrupt=True)

    def test_main_window_deactivation_for_autocomplete_popup_preserves_state(self):
        testScript = self._make_chromium_script()
        mainFrame = object()
        comboBox = object()
        event = mock.Mock(source=mainFrame)
        cthulhu_state.locusOfFocus = comboBox
        testScript.utilities.lastKeyAndModifiers.return_value = ("Down", 8)

        with (
            mock.patch.object(chromium_script.AXUtilities, "is_combo_box", return_value=True),
            mock.patch.object(
                chromium_script.AXUtilities,
                "supports_autocompletion",
                return_value=True,
            ),
            mock.patch.object(default.Script, "onWindowDeactivated") as defaultHandler,
        ):
            chromium_script.Script.onWindowDeactivated(testScript, event)

        defaultHandler.assert_not_called()

    def test_unrelated_autocomplete_window_deactivation_reaches_default(self):
        testScript = self._make_chromium_script()
        mainFrame = object()
        comboBox = object()
        app = object()
        event = mock.Mock(source=mainFrame)
        cthulhu_state.locusOfFocus = comboBox
        testScript.utilities.lastKeyAndModifiers.return_value = ("w", 4)

        with (
            mock.patch.object(chromium_script.AXUtilities, "is_combo_box", return_value=True),
            mock.patch.object(
                chromium_script.AXUtilities,
                "supports_autocompletion",
                return_value=True,
            ),
            mock.patch.object(chromium_script.AXObject, "get_application", return_value=app),
            mock.patch.object(chromium_script.AXObject, "iter_children", return_value=iter(())),
            mock.patch.object(chromium_script.web.Script, "onWindowDeactivated", return_value=False),
            mock.patch.object(default.Script, "onWindowDeactivated") as defaultHandler,
        ):
            chromium_script.Script.onWindowDeactivated(testScript, event)

        defaultHandler.assert_called_once_with(testScript, event)

    def test_unmodified_arrow_deactivation_reaches_default(self):
        testScript = self._make_chromium_script()
        mainFrame = object()
        comboBox = object()
        app = object()
        event = mock.Mock(source=mainFrame)
        cthulhu_state.locusOfFocus = comboBox
        testScript.utilities.lastKeyAndModifiers.return_value = ("Down", 0)

        with (
            mock.patch.object(chromium_script.AXUtilities, "is_combo_box", return_value=True),
            mock.patch.object(
                chromium_script.AXUtilities,
                "supports_autocompletion",
                return_value=True,
            ),
            mock.patch.object(chromium_script.AXObject, "get_application", return_value=app),
            mock.patch.object(chromium_script.AXObject, "iter_children", return_value=iter(())),
            mock.patch.object(chromium_script.web.Script, "onWindowDeactivated", return_value=False),
            mock.patch.object(default.Script, "onWindowDeactivated") as defaultHandler,
        ):
            chromium_script.Script.onWindowDeactivated(testScript, event)

        defaultHandler.assert_called_once_with(testScript, event)


if __name__ == "__main__":
    unittest.main()