From 0998bc355e6425e6b2bdf3d2da7124aa7b63b2a2 Mon Sep 17 00:00:00 2001
From: Jacky Zhao <j.zhao2k19@gmail.com>
Date: Thu, 17 Aug 2023 08:58:11 +0000
Subject: [PATCH] fix rebuild debouncing

---
 quartz/components/scripts/search.inline.ts |  153 +++++++++++++++++++++++++++++---------------------
 1 files changed, 88 insertions(+), 65 deletions(-)

diff --git a/quartz/components/scripts/search.inline.ts b/quartz/components/scripts/search.inline.ts
index 78517fe..b94cdfb 100644
--- a/quartz/components/scripts/search.inline.ts
+++ b/quartz/components/scripts/search.inline.ts
@@ -1,25 +1,32 @@
 import { Document } from "flexsearch"
 import { ContentDetails } from "../../plugins/emitters/contentIndex"
-import { registerEscapeHandler, relative, removeAllChildren } from "./util"
+import { registerEscapeHandler, removeAllChildren } from "./util"
+import { CanonicalSlug, getClientSlug, resolveRelative } from "../../util/path"
 
 interface Item {
-  slug: string,
-  title: string,
-  content: string,
+  id: number
+  slug: CanonicalSlug
+  title: string
+  content: string
 }
+
 let index: Document<Item> | undefined = undefined
 
 const contextWindowWords = 30
+const numSearchResults = 5
 function highlight(searchTerm: string, text: string, trim?: boolean) {
-  const tokenizedTerms = searchTerm.split(/\s+/).filter(t => t !== "")
-  let tokenizedText = text
+  // try to highlight longest tokens first
+  const tokenizedTerms = searchTerm
     .split(/\s+/)
-    .filter(t => t !== "")
+    .filter((t) => t !== "")
+    .sort((a, b) => b.length - a.length)
+  let tokenizedText = text.split(/\s+/).filter((t) => t !== "")
 
   let startIndex = 0
   let endIndex = tokenizedText.length - 1
   if (trim) {
-    const includesCheck = (tok: string) => tokenizedTerms.some((term) => tok.toLowerCase().startsWith(term.toLowerCase()))
+    const includesCheck = (tok: string) =>
+      tokenizedTerms.some((term) => tok.toLowerCase().startsWith(term.toLowerCase()))
     const occurencesIndices = tokenizedText.map(includesCheck)
 
     let bestSum = 0
@@ -38,73 +45,53 @@
     tokenizedText = tokenizedText.slice(startIndex, endIndex)
   }
 
-  const slice = tokenizedText.map(tok => {
-    // see if this tok is prefixed by any search terms 
-    for (const searchTok of tokenizedTerms) {
-      if (tok.toLowerCase().includes(searchTok.toLowerCase())) {
-        const regex = new RegExp(searchTok, "gi")
-        return tok.replace(regex, `<span class="highlight">$&</span>`)
+  const slice = tokenizedText
+    .map((tok) => {
+      // see if this tok is prefixed by any search terms
+      for (const searchTok of tokenizedTerms) {
+        if (tok.toLowerCase().includes(searchTok.toLowerCase())) {
+          const regex = new RegExp(searchTok.toLowerCase(), "gi")
+          return tok.replace(regex, `<span class="highlight">$&</span>`)
+        }
       }
-    }
-    return tok
-  })
+      return tok
+    })
     .join(" ")
 
-  return `${startIndex === 0 ? "" : "..."}${slice}${endIndex === tokenizedText.length - 1 ? "" : "..."}`
+  return `${startIndex === 0 ? "" : "..."}${slice}${
+    endIndex === tokenizedText.length - 1 ? "" : "..."
+  }`
 }
 
 const encoder = (str: string) => str.toLowerCase().split(/([^a-z]|[^\x00-\x7F])/)
 document.addEventListener("nav", async (e: unknown) => {
   const currentSlug = (e as CustomEventMap["nav"]).detail.url
 
-  // setup index if it hasn't been already
   const data = await fetchData
-  if (!index) {
-    index = new Document({
-      cache: true,
-      charset: 'latin:extra',
-      optimize: true,
-      encode: encoder,
-      document: {
-        id: "slug",
-        index: [
-          {
-            field: "title",
-            tokenize: "forward",
-          },
-          {
-            field: "content",
-            tokenize: "reverse",
-          },
-        ]
-      },
-    })
-
-    for (const [slug, fileData] of Object.entries<ContentDetails>(data)) {
-      index.add({
-        slug,
-        title: fileData.title,
-        content: fileData.content
-      })
-    }
-  }
-
   const container = document.getElementById("search-container")
+  const sidebar = container?.closest(".sidebar") as HTMLElement
   const searchIcon = document.getElementById("search-icon")
   const searchBar = document.getElementById("search-bar") as HTMLInputElement | null
   const results = document.getElementById("results-container")
+  const idDataMap = Object.keys(data) as CanonicalSlug[]
 
   function hideSearch() {
     container?.classList.remove("active")
     if (searchBar) {
       searchBar.value = "" // clear the input when we dismiss the search
     }
+    if (sidebar) {
+      sidebar.style.zIndex = "unset"
+    }
     if (results) {
       removeAllChildren(results)
     }
   }
 
   function showSearch() {
+    if (sidebar) {
+      sidebar.style.zIndex = "1"
+    }
     container?.classList.add("active")
     searchBar?.focus()
   }
@@ -122,20 +109,24 @@
     }
   }
 
-  const formatForDisplay = (term: string, slug: string) => ({
-    slug,
-    title: highlight(term, data[slug].title ?? ""),
-    content: highlight(term, data[slug].content ?? "", true),
-  })
+  const formatForDisplay = (term: string, id: number) => {
+    const slug = idDataMap[id]
+    return {
+      id,
+      slug,
+      title: highlight(term, data[slug].title ?? ""),
+      content: highlight(term, data[slug].content ?? "", true),
+    }
+  }
 
   const resultToHTML = ({ slug, title, content }: Item) => {
     const button = document.createElement("button")
     button.classList.add("result-card")
     button.id = slug
     button.innerHTML = `<h3>${title}</h3><p>${content}</p>`
-    button.addEventListener('click', () => {
-      const targ = relative(currentSlug, slug)
-      window.spaNavigate(new URL(targ))
+    button.addEventListener("click", () => {
+      const targ = resolveRelative(currentSlug, slug)
+      window.spaNavigate(new URL(targ, getClientSlug(window)))
     })
     return button
   }
@@ -152,24 +143,22 @@
     } else {
       results.append(...finalResults.map(resultToHTML))
     }
-
   }
 
-  function onType(e: HTMLElementEventMap["input"]) {
+  async function onType(e: HTMLElementEventMap["input"]) {
     const term = (e.target as HTMLInputElement).value
-    const searchResults = index?.search(term, 5) ?? []
-    const getByField = (field: string): string[] => {
+    const searchResults = (await index?.searchAsync(term, numSearchResults)) ?? []
+    const getByField = (field: string): number[] => {
       const results = searchResults.filter((x) => x.field === field)
-      return results.length === 0 ? [] : [...results[0].result] as string[]
+      return results.length === 0 ? [] : ([...results[0].result] as number[])
     }
 
     // order titles ahead of content
-    const allIds: Set<string> = new Set([...getByField("title"), ...getByField("content")])
-    const finalResults = [...allIds].map(id => formatForDisplay(term, id))
+    const allIds: Set<number> = new Set([...getByField("title"), ...getByField("content")])
+    const finalResults = [...allIds].map((id) => formatForDisplay(term, id))
     displayResults(finalResults)
   }
 
-
   document.removeEventListener("keydown", shortcutHandler)
   document.addEventListener("keydown", shortcutHandler)
   searchIcon?.removeEventListener("click", showSearch)
@@ -177,6 +166,40 @@
   searchBar?.removeEventListener("input", onType)
   searchBar?.addEventListener("input", onType)
 
+  // setup index if it hasn't been already
+  if (!index) {
+    index = new Document({
+      cache: true,
+      charset: "latin:extra",
+      optimize: true,
+      encode: encoder,
+      document: {
+        id: "id",
+        index: [
+          {
+            field: "title",
+            tokenize: "reverse",
+          },
+          {
+            field: "content",
+            tokenize: "reverse",
+          },
+        ],
+      },
+    })
+
+    let id = 0
+    for (const [slug, fileData] of Object.entries<ContentDetails>(data)) {
+      await index.addAsync(id, {
+        id,
+        slug: slug as CanonicalSlug,
+        title: fileData.title,
+        content: fileData.content,
+      })
+      id++
+    }
+  }
+
   // register handlers
   registerEscapeHandler(container, hideSearch)
 })

--
Gitblit v1.10.0