anki/qt/aqt/toolbar.py

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

from __future__ import annotations

from typing import Dict

import aqt
from anki.lang import _
from aqt import gui_hooks
from aqt.qt import *
from aqt.webview import AnkiWebView


# wrapper class for set_bridge_command()
class TopToolbar:
    def __init__(self, toolbar: Toolbar):
        self.toolbar = toolbar


# wrapper class for set_bridge_command()
class BottomToolbar:
    def __init__(self, toolbar: Toolbar):
        self.toolbar = toolbar


class Toolbar:
    def __init__(self, mw: aqt.AnkiQt, web: AnkiWebView) -> None:
        self.mw = mw
        self.web = web
        self.link_handlers: Dict[str, Callable] = {
            "decks": self._deckLinkHandler,
            "study": self._studyLinkHandler,
            "add": self._addLinkHandler,
            "browse": self._browseLinkHandler,
            "stats": self._statsLinkHandler,
            "sync": self._syncLinkHandler,
        }
        self.web.setFixedHeight(30)
        self.web.requiresCol = False

    def draw(self):
        self.web.set_bridge_command(self._linkHandler, TopToolbar(self))
        self.web.stdHtml(self._body % self._centerLinks(), css=["toolbar.css"])
        self.web.adjustHeightToFit()

    # Available links
    ######################################################################

    def addLink(
        self, name: str, cmd: str, func: Callable, tip: str = "",
    ):
        self.link_handlers[cmd] = func

        return (cmd, name, tip)

    def _centerLinks(self):
        links = [
            ("decks", _("Decks"), _("Shortcut key: %s") % "D"),
            ("add", _("Add"), _("Shortcut key: %s") % "A"),
            ("browse", _("Browse"), _("Shortcut key: %s") % "B"),
            ("stats", _("Stats"), _("Shortcut key: %s") % "T"),
            ("sync", _("Sync"), _("Shortcut key: %s") % "Y"),
        ]
        gui_hooks.top_toolbar_did_init_links(links, self)
        return self._linkHTML(links)

    def _linkHTML(self, links):
        buf = ""
        for ln, name, title in links:
            buf += """
            <a class=hitem tabindex="-1" aria-label="%s" title="%s" href=# onclick="return pycmd('%s')">%s</a>""" % (
                name,
                title,
                ln,
                name,
            )
        return buf

    # Link handling
    ######################################################################

    def _linkHandler(self, link):
        if link in self.link_handlers:
            self.link_handlers[link]()
        return False

    def _deckLinkHandler(self):
        self.mw.moveToState("deckBrowser")

    def _studyLinkHandler(self):
        # if overview already shown, switch to review
        if self.mw.state == "overview":
            self.mw.col.startTimebox()
            self.mw.moveToState("review")
        else:
            self.mw.onOverview()

    def _addLinkHandler(self):
        self.mw.onAddCard()

    def _browseLinkHandler(self):
        self.mw.onBrowse()

    def _statsLinkHandler(self):
        self.mw.onStats()

    def _syncLinkHandler(self):
        self.mw.onSync()

    # HTML & CSS
    ######################################################################

    _body = """
<center id=outer>
<table id=header width=100%%>
<tr>
<td class=tdcenter align=center>%s</td>
</tr></table>
</center>
"""


# Bottom bar
######################################################################


class BottomBar(Toolbar):

    _centerBody = """
<center id=outer><table width=100%% id=header><tr><td align=center>
%s</td></tr></table></center>
"""

    def draw(self, buf):
        # note: some screens may override this
        self.web.set_bridge_command(self._linkHandler, BottomToolbar(self))
        self.web.stdHtml(
            self._centerBody % buf, css=["toolbar.css", "toolbar-bottom.css"]
        )
        self.web.adjustHeightToFit()