from urllib.parse import unquote
from cishu.cishubase import cishubase
from myutils.utils import simplehtmlparser_all, simplehtmlparser, localcachehelper
import re, threading


class weblio(cishubase):
    backgroundparser = """
            document.querySelectorAll('.lunawb').forEach((ele) => {
                ele.style.backgroundColor = {color}
                ele.querySelectorAll('#base').forEach((ele) => {
                    ele.style.backgroundColor = {color}
                })
            });
"""

    def init(self):
        self.cache = localcachehelper("cishucss/weblio_1")
        self.klass = "lunawb"

    def search(self, word):
        url = self.getUrl(word)
        html = self.proxysession.get(url).text
        head = simplehtmlparser_all(html, "div", '<div class="pbarT">')
        content = simplehtmlparser_all(html, "div", '<div class="kijiWrp">')
        if not content:
            return
        collect = []
        for i, xx in enumerate(head):
            xx = re.sub('src="//(.*?)"', 'src="https://\\1"', xx + content[i])
            collect.append(xx)
        join = '<div ID="base" style="overflow-x: hidden; min-width: 0; margin:0;">{}</div>'.format(
            "".join(collect)
        )
        for shit in simplehtmlparser_all(join, "script", "<script"):
            join = join.replace(shit, "")

        def removeklass(join, klass):
            while True:
                sig = "class=" + klass
                fnd = join.find(sig)
                if fnd == -1:
                    break
                start = join.rfind("<img", None, fnd)
                if start == -1:
                    break
                end = join.find(">", fnd)
                if end == -1:
                    break
                join = join[:start] + join[end + 1 :]
            return join

        join = removeklass(join, "lgDictLg")
        join = removeklass(join, "lgDictSp")
        join = join.replace('href="//', 'href="https://')

        def __(match):
            word = unquote(match.groups()[0])
            return '''href="javascript:safe_weblio_search_word('{}')"'''.format(word)

        join = re.sub('href="https://www.weblio.jp/content/(.*?)"', __, join)
        join += r"""
<script>
function safe_weblio_search_word(word){
    if(window.LUNAJSObject)
        LUNAJSObject.luna_search_word(word)
}</script>
"""
        links = []
        style = self.parse_stylesheet(
            simplehtmlparser(html, "style", "<style>")[7:-8], self.klass
        )
        for link in simplehtmlparser_all(html, "link", '<link rel="stylesheet"'):
            for _ in re.findall('href="(.*?)"', link):
                links.append(_)
        ts = []
        for _ in links:
            ts.append(threading.Thread(target=self.makelink, args=(_,), daemon=True))
            ts[-1].start()
        for t in ts:
            t.join()
        style += "".join(self.cache.get(link) for link in links)
        style = re.sub(r"([^-])width:\d+px", "\\1", style)
        return '<style>{}</style><div class="{}">{}</div>'.format(
            style, self.klass, join
        )

    def makelink(self, link):
        if not self.cache.get(link):
            req = self.proxysession.get(link)
            html = req.text if req.status_code == 200 else ""
            self.cache[link] = self.parse_stylesheet(html, self.klass)

    def getUrl(self, word):
        return "https://www.weblio.jp/content/{}".format(word)
