From c35cd422c65a58f1069302aad0cf9eef7f93d987 Mon Sep 17 00:00:00 2001
From: Ben Schlegel <31989404+benschlegel@users.noreply.github.com>
Date: Mon, 28 Aug 2023 17:00:49 +0000
Subject: [PATCH] fix: correct graph labels for `index.md` nodes (#431)

---
 quartz/plugins/transformers/description.ts |   28 +++++++++++++++++++---------
 1 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/quartz/plugins/transformers/description.ts b/quartz/plugins/transformers/description.ts
index 0e41c5b..08af5c7 100644
--- a/quartz/plugins/transformers/description.ts
+++ b/quartz/plugins/transformers/description.ts
@@ -1,4 +1,4 @@
-import { Root as HTMLRoot } from 'hast'
+import { Root as HTMLRoot } from "hast"
 import { toString } from "hast-util-to-string"
 import { QuartzTransformerPlugin } from "../types"
 
@@ -7,7 +7,16 @@
 }
 
 const defaultOptions: Options = {
-  descriptionLength: 150
+  descriptionLength: 150,
+}
+
+const escapeHTML = (unsafe: string) => {
+  return unsafe
+    .replaceAll("&", "&amp;")
+    .replaceAll("<", "&lt;")
+    .replaceAll(">", "&gt;")
+    .replaceAll('"', "&quot;")
+    .replaceAll("'", "&#039;")
 }
 
 export const Description: QuartzTransformerPlugin<Partial<Options> | undefined> = (userOpts) => {
@@ -19,31 +28,32 @@
         () => {
           return async (tree: HTMLRoot, file) => {
             const frontMatterDescription = file.data.frontmatter?.description
-            const text = toString(tree)
+            const text = escapeHTML(toString(tree))
 
             const desc = frontMatterDescription ?? text
-            const sentences = desc.replace(/\s+/g, ' ').split('.')
+            const sentences = desc.replace(/\s+/g, " ").split(".")
             let finalDesc = ""
             let sentenceIdx = 0
             const len = opts.descriptionLength
             while (finalDesc.length < len) {
-              finalDesc += sentences[sentenceIdx] + '.'
+              const sentence = sentences[sentenceIdx]
+              if (!sentence) break
+              finalDesc += sentence + "."
               sentenceIdx++
             }
 
             file.data.description = finalDesc
             file.data.text = text
           }
-        }
+        },
       ]
-    }
+    },
   }
 }
 
-declare module 'vfile' {
+declare module "vfile" {
   interface DataMap {
     description: string
     text: string
   }
 }
-

--
Gitblit v1.10.0