themes/common.py - master - githubmirror/gpt_academic

themes/common.py
76 строк3 KB
binary-husky
change some open fn encoding to utf-8
19 ноя 2024, 18:53
19 ноя 2024, 18:53e62deca
from functools import lru_cache
from toolbox import get_conf
CODE_HIGHLIGHT, ADD_WAIFU, LAYOUT = get_conf("CODE_HIGHLIGHT", "ADD_WAIFU", "LAYOUT")

def inject_mutex_button_code(js_content):
    from crazy_functional import get_multiplex_button_functions
    fns = get_multiplex_button_functions()

    template = """
    if (multiplex_sel === "{x}") {
        let _align_name_in_crazy_function_py = "{y}";
        call_plugin_via_name(_align_name_in_crazy_function_py);
        return;
    }
    """

    replacement = ""
    for fn in fns.keys():
        if fn == "常规对话": continue
        replacement += template.replace("{x}", fn).replace("{y}", fns[fn])
    js_content = js_content.replace("// REPLACE_EXTENDED_MULTIPLEX_FUNCTIONS_HERE", replacement)
    return js_content

def minimize_js(common_js_path):
    try:
        import rjsmin, hashlib, glob, os
        # clean up old minimized js files, matching `common_js_path + '.min.*'`
        for old_min_js in glob.glob(common_js_path + '.min.*.js'):
            os.remove(old_min_js)
        # use rjsmin to minimize `common_js_path`
        c_jsmin = rjsmin.jsmin
        with open(common_js_path, "r", encoding='utf-8') as f:
            js_content = f.read()
        if common_js_path == "themes/common.js":
            js_content = inject_mutex_button_code(js_content)
        minimized_js_content = c_jsmin(js_content)
        # compute sha256 hash of minimized js content
        sha_hash = hashlib.sha256(minimized_js_content.encode()).hexdigest()[:8]
        minimized_js_path = common_js_path + '.min.' + sha_hash + '.js'
        # save to minimized js file
        with open(minimized_js_path, "w", encoding='utf-8') as f:
            f.write(minimized_js_content)
        # return minimized js file path
        return minimized_js_path
    except:
        return common_js_path

@lru_cache
def get_common_html_javascript_code():
    js = "\n"
    common_js_path_list = [
        "themes/common.js",
        "themes/theme.js",
        "themes/tts.js",
        "themes/init.js",
        "themes/welcome.js",
    ]

    if ADD_WAIFU: # 添加Live2D
        common_js_path_list += [
            "themes/waifu_plugin/jquery.min.js",
            "themes/waifu_plugin/jquery-ui.min.js",
        ]

    for common_js_path in common_js_path_list:
        if '.min.' not in common_js_path:
            minimized_js_path = minimize_js(common_js_path)
        else:
            minimized_js_path = common_js_path
        jsf = f"file={minimized_js_path}"
        js += f"""<script src="{jsf}"></script>\n"""

    if not ADD_WAIFU:
        js += """<script>window.loadLive2D = function(){};</script>\n"""

    return js