__init__.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# anki-webbrowse
# Copyright (C) 2020 Tom Z.

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.

# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.


import sys
import re

from aqt import mw, gui_hooks
from aqt.utils import showInfo
from aqt.qt import *
from aqt.editor import Editor
from anki.hooks import wrap, addHook
from aqt.addcards import AddCards
from anki.utils import isMac

sys.path.insert(0, os.path.dirname(__file__))

from .browser import AddCardsTabbedBrowser

browser_displayed   = False
browser             = None
menu                = None
shortcut_set        = False

nightmode           = False




def display_browser(web):
    global browser_displayed, browser, menu


    # first init
    if browser is None:
        config = mw.addonManager.getConfig(__name__)

        addcards = web.parent().parent()
        if not isinstance(addcards, AddCards):
            addcards = mw.app.activeWindow()
        assert(isinstance(addcards, AddCards))

        layout = addcards.layout()
        hbox = QHBoxLayout()
        vbox = QVBoxLayout()
        while layout.count() > 0:
            item = layout.takeAt(0)
            if item.widget() is None or isinstance(item, QBoxLayout):
                vbox.addLayout(item.layout())
            else:
                vbox.addWidget(item.widget())
        temp = QWidget()
        temp.setLayout(layout)

        flds = QWidget()
        flds.setLayout(vbox)
        hbox.setSpacing(0)
        hbox.setContentsMargins(0, 0, 0, 0)
       
        menu = QWidget()
        menu_vbox = QVBoxLayout()

        # build bookmarks list
        btree = QTreeWidget()
        btree.itemClicked.connect(bookmark_clicked)
        bookmarks = config["bookmarks"]
        expand = config["expand_bookmark_folders_by_default"]
        for k, blist in bookmarks.items():
            folder = QTreeWidgetItem(btree, [k])
            folder.setExpanded(expand)
            for b in blist:
                b = re.sub("(https?://)?(www2?\\.)?", "", b)
                if b.endswith("/"):
                    b = b[:-1]
                item = QTreeWidgetItem([b])
                folder.addChild(item)
        btree.setHeaderLabels(["Bookmarks"])
        btree.setIconSize(QSize(0,0))
        menu_vbox.addWidget(btree, 2)
        menu.setLayout(menu_vbox)

        menu.setVisible(False)

        browser = AddCardsTabbedBrowser(flds, menu, web, nightmode)

        flds_width = config["fields_area_width_in_percent"]
        flds_width = min(95, max(1, flds_width))
        bookmarks_width = config["bookmarks_area_width_in_percent"]
 
        if not config["switch_fields_area_to_right"]:
            hbox.addWidget(flds, flds_width)
            hbox.addWidget(browser, 100-flds_width-bookmarks_width)
            hbox.addWidget(menu, bookmarks_width) 
        else:
            hbox.addWidget(menu, bookmarks_width) 
            hbox.addWidget(browser, 100-flds_width-bookmarks_width)
            hbox.addWidget(flds, flds_width)

        addcards.setLayout(hbox)

        shortcut = QShortcut(QKeySequence(config["toggle_fields_shortcut"]), browser)
        shortcut.activated.connect(toggle_sidebar)

        shortcut = QShortcut(QKeySequence(config["zoom_in_shortcut"]), browser)
        shortcut.activated.connect(zoom_in)

        shortcut = QShortcut(QKeySequence(config["zoom_out_shortcut"]), browser)
        shortcut.activated.connect(zoom_out)

        addcards.form.deckArea.setMinimumWidth(35)
        addcards.form.modelArea.setMinimumWidth(35)

        # addcards.form.deckArea.setMaximumWidth(125)
        # addcards.form.modelArea.setMaximumWidth(125)
        addcards.helpButton.setVisible(False)

    else:
        browser.setVisible(True)
    
    web.eval("""
        document.getElementById('toggleBrowserBtn').style.color = `#2496dc`;
        if (document.getElementById('switchBtn')) {
            document.getElementById('switchBtn').setAttribute('style', 'display:none !important');
        }
    """)
        
    browser_displayed = True

def hide_browser(web):
    global browser_displayed, browser, menu
    browser.setVisible(False)
    menu.setVisible(False)
    browser_displayed = False
    addcards = web.parent().parent().parent()
    # bad but works
    addcards.form.deckArea.setMaximumWidth(3000)
    addcards.form.modelArea.setMaximumWidth(3000)
    
    web.eval("""
        document.getElementById('toggleBrowserBtn').style.color = `inherit`;
        if (document.getElementById('switchBtn') ) {
            document.getElementById('switchBtn').setAttribute('style', 'display:inline-block');
        }
    """)


def bookmark_clicked(item):
    if item.parent() is None:
        return
    browser.load(item.text(0))

def on_editor_did_init(editor):
    global shortcut_set

    if editor.addMode:
        if not shortcut_set:
            config = mw.addonManager.getConfig(__name__)
            shortcut = QShortcut(QKeySequence(config["toggle_browser_shortcut"]), mw.app.activeWindow())
            shortcut.activated.connect(toggle_browser)
            shortcut_set = True
    
        def cb(info):
            global browser, nightmode
            rendered = info[0]
            nightmode = info[1]
            if not rendered:
                browser = None
                if browser_displayed:
                    display_browser(editor.web)

        if isMac:
            btn_text = "www"
        else:
            btn_text = "&#127760;"

        editor.web.evalWithCallback("""
                    (() => {
                        let rendered = false;
                        if (!document.getElementById('toggleBrowserBtn')) {
                            document.getElementById('topbutsleft').innerHTML += `<button id='toggleBrowserBtn' onclick='pycmd("toggle_browser")'>%s</button> `; 
                        } else {
                            rendered = true;
                        }
                        let nightmode = document.body.classList.contains("nightMode");
                        return [rendered, nightmode];
                    })();
                """ % btn_text, cb)

def toggle_sidebar():
    if not browser_displayed:
        return
    browser.toggle()

def toggle_browser():
    win = mw.app.activeWindow()
    if hasattr(win, "editor"):
        web = win.editor.web
        if browser_displayed:
            hide_browser(web)
        else:
            display_browser(web)
    

def zoom_out():
    if not browser_displayed:
        return
    browser.zoom_out()

def zoom_in():
    if not browser_displayed:
        return
    browser.zoom_in()

def expanded_on_bridge(self, cmd, _old):
    if cmd == "toggle_browser":
        if browser_displayed:
            hide_browser(self.web)
        else:
            display_browser(self.web)
    else:
        return _old(self, cmd)


def on_keypress(addcards, evt, _old):
    if evt.key() in (Qt.Key_Enter, Qt.Key_Return) and browser.url.hasFocus():
        return
    return _old(addcards, evt)

def init_addon():
    gui_hooks.editor_did_load_note.append(on_editor_did_init)
    Editor.onBridgeCmd = wrap(Editor.onBridgeCmd, expanded_on_bridge, "around")
   

init_addon()