# Copyright: Ankitects Pty Ltd and contributors
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html

from __future__ import annotations

import aqt
import aqt.browser
import aqt.gui_hooks
from aqt import colors
from aqt.qt import *
from aqt.theme import theme_manager


class SidebarSearchBar(QLineEdit):
    def __init__(self, sidebar: aqt.browser.sidebar.SidebarTreeView) -> None:
        QLineEdit.__init__(self, sidebar)
        self.setPlaceholderText(sidebar.col.tr.browsing_sidebar_filter())
        self.sidebar = sidebar
        self.timer = QTimer(self)
        self.timer.setInterval(600)
        self.timer.setSingleShot(True)
        self.setFrame(False)
        self.setup_style()

        qconnect(self.timer.timeout, self.onSearch)
        qconnect(self.textChanged, self.onTextChanged)

        aqt.gui_hooks.theme_did_change.append(self.setup_style)

    def setup_style(self) -> None:
        styles = [
            "padding: 2px",
            f"border: 1px solid {theme_manager.color(colors.BORDER)}",
            "border-radius: 5px",
        ]

        self.setStyleSheet(
            "QLineEdit { %s }" % ";".join(styles)
            + f"""
QLineEdit:focus {{
    border: 1px solid {theme_manager.color(colors.FOCUS_BORDER)};
}}
            """
        )

    def onTextChanged(self, text: str) -> None:
        if not self.timer.isActive():
            self.timer.start()

    def onSearch(self) -> None:
        self.sidebar.search_for(self.text())

    def keyPressEvent(self, evt: QKeyEvent) -> None:
        if evt.key() in (Qt.Key.Key_Up, Qt.Key.Key_Down):
            self.sidebar.setFocus()
        elif evt.key() in (Qt.Key.Key_Enter, Qt.Key.Key_Return):
            self.onSearch()
        else:
            QLineEdit.keyPressEvent(self, evt)

    def cleanup(self) -> None:
        aqt.gui_hooks.theme_did_change.remove(self.setup_style)