Merge pull request #171 from lgc2333/main

KroMiose · Jun 14, 2023 · 4c172d7 · 4c172d7
2 parents 8e9d8d9 + 1308a1d
commit 4c172d7
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 16 deletions.
diff --git a/nonebot_plugin_naturel_gpt/res/additional.css b/nonebot_plugin_naturel_gpt/res/additional.css
@@ -10,9 +10,10 @@
 .markdown-body kbd,
 .markdown-body pre,
 .markdown-body samp {
-  font-family: 'JetBrains Mono', 'Cascadia Mono', 'Segoe UI Mono',
-    'Liberation Mono', 'Menlo', 'Monaco', 'Consolas', 'Roboto Mono',
-    'Courier New', 'Courier', 'Microsoft YaHei UI', monospace !important;
+  font-family: 'VictorMono Nerd Font', 'Victor Mono', 'JetBrainsMono Nerd Font',
+    'JetBrains Mono', 'Cascadia Mono', 'Segoe UI Mono', 'Liberation Mono',
+    'Menlo', 'Monaco', 'Consolas', 'Roboto Mono', 'Courier New', 'Courier',
+    'Microsoft YaHei UI', monospace !important;
 }
 
 .markdown-body code {

diff --git a/nonebot_plugin_naturel_gpt/res/additional.html b/nonebot_plugin_naturel_gpt/res/additional.html
@@ -0,0 +1,9 @@
+<script>
+  (() => {
+    const anchors = document.getElementsByTagName('a');
+    for (let i = 0; i < anchors.length; i += 1) {
+      const elem = anchors.item(i);
+      elem.innerText = `${elem.innerText} (${elem.href})`;
+    }
+  })();
+</script>
diff --git a/nonebot_plugin_naturel_gpt/text_to_image.py b/nonebot_plugin_naturel_gpt/text_to_image.py
@@ -13,9 +13,10 @@
     read_tpl,
 )
 
-ADDITIONAL_CSS = (Path(__file__).parent / "res" / "additional.css").read_text(
-    encoding="u8"
-)
+
+RES_PATH = Path(__file__).parent / "res"
+ADDITIONAL_CSS = (RES_PATH / "additional.css").read_text(encoding="u8")
+ADDITIONAL_HTML = (RES_PATH / "additional.html").read_text(encoding="u8")
 
 
 async def text_to_img(text: str) -> bytes:
@@ -55,16 +56,16 @@ async def md_to_img(md: str) -> bytes:
     )
     logger.debug(md)
 
-    extra = ""
-    if "math/tex" in md:
-        katex_css = await read_tpl("katex/katex.min.b64_fonts.css")
-        katex_js = await read_tpl("katex/katex.min.js")
-        mathtex_js = await read_tpl("katex/mathtex-script-type.min.js")
-        extra = (
-            f'<style type="text/css">{katex_css}</style>'
-            f"<script defer>{katex_js}</script>"
-            f"<script defer>{mathtex_js}</script>"
-        )
+    # if "math/tex" in md:
+    katex_css = await read_tpl("katex/katex.min.b64_fonts.css")
+    katex_js = await read_tpl("katex/katex.min.js")
+    mathtex_js = await read_tpl("katex/mathtex-script-type.min.js")
+    extra = (
+        f"{ADDITIONAL_HTML}\n"
+        f'<style type="text/css">{katex_css}</style>\n'
+        f"<script defer>{katex_js}</script>\n"
+        f"<script defer>{mathtex_js}</script>\n"
+    )
 
     css = "\n".join(
         [