markdown code render fix

nbonamy · May 1, 2024 · 74e368d · 74e368d
1 parent 65ffb0e
commit 74e368d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "witsy",
   "productName": "Witsy",
-  "version": "1.4.2",
+  "version": "1.4.3",
   "description": "Witsy: desktop AI assistant",
   "repository": {
     "type": "git",

diff --git a/src/main/markdown.ts b/src/main/markdown.ts
@@ -8,17 +8,19 @@ import hljs from 'highlight.js'
 const mdOptions: MarkdownIt.Options = {
   html: true,
   highlight: function (str: string, lang: string) {
-    if (lang && hljs.getLanguage(lang)) {
-      try {
-        let code = '<pre class="hljs"><code class="hljs">';
+    try {
+      let code = '<pre class="hljs"><code class="hljs">';
+      if (lang && hljs.getLanguage(lang)) {
         code += hljs.highlight(str, { language: lang, ignoreIllegals: true }).value;
-        code += '</code></pre>';
-        code += '<p><a href="#" onclick="navigator.clipboard.writeText(window.api.base64.decode(\'' + Buffer.from(str).toString('base64') + '\'));';
-        code += 'this.innerHTML = \'Copied!\'; setTimeout(() => this.innerHTML = \'Copy code\', 1000); return false;" class="copy">Copy code</a></p>';
-        return code;
-      } catch (error) {
-        console.log(error)
+      } else {
+        code += hljs.highlightAuto(str).value;
       }
+      code += '</code></pre>';
+      code += '<p><a href="#" onclick="navigator.clipboard.writeText(window.api.base64.decode(\'' + Buffer.from(str).toString('base64') + '\'));';
+      code += 'this.innerHTML = \'Copied!\'; setTimeout(() => this.innerHTML = \'Copy code\', 1000); return false;" class="copy">Copy code</a></p>';
+      return code;
+    } catch (error) {
+      console.log(error)
     }
     return '' // use external default escaping
   }