From fcd5d2807d2bab68c6776e031e85d65fe88a6f7a Mon Sep 17 00:00:00 2001
From: Jacky Zhao <j.zhao2k19@gmail.com>
Date: Wed, 16 Feb 2022 00:39:14 +0000
Subject: [PATCH] feat: dynamically fetch indices

---
 layouts/partials/search.html |  323 +++++++++++++++++++++++++++--------------------------
 1 files changed, 164 insertions(+), 159 deletions(-)

diff --git a/layouts/partials/search.html b/layouts/partials/search.html
index 0c853a0..c2ff77f 100644
--- a/layouts/partials/search.html
+++ b/layouts/partials/search.html
@@ -67,189 +67,194 @@
     };
 </script>
 <script>
-    const contentIndex = new FlexSearch.Document({
-        cache: true,
-        charset: "latin:extra",
-        optimize: true,
-        worker: true,
-        document: {
-            index: [{
-                field: "content",
-                tokenize: "strict",
-                context: {
-                    resolution: 5,
-                    depth: 3,
-                    bidirectional: true
-                },
-                suggest: true,
-            }, {
-                field: "title",
-                tokenize: "forward",
-            }]
-        }
+async function run() {
+  const contentIndex = new FlexSearch.Document({
+    cache: true,
+    charset: "latin:extra",
+    optimize: true,
+    worker: true,
+    document: {
+      index: [{
+        field: "content",
+        tokenize: "strict",
+        context: {
+          resolution: 5,
+          depth: 3,
+          bidirectional: true
+        },
+        suggest: true,
+      }, {
+        field: "title",
+        tokenize: "forward",
+      }]
+    }
+  })
+
+  const { content } = await fetchData()
+  for (const [key, value] of Object.entries(content)) {
+    contentIndex.add({
+      id: key,
+      title: value.title,
+      content: removeMarkdown(value.content),
     })
+  }
 
-    for (const [key, value] of Object.entries(content)) {
-        contentIndex.add({
-            id: key,
-            title: value.title,
-            content: removeMarkdown(value.content),
-        })
+  const highlight = (content, term) => {
+    const highlightWindow = 20
+    const tokenizedTerm = term.split(/\s+/).filter(t => t !== "")
+    const splitText = content.split(/\s+/).filter(t => t !== "")
+    const includesCheck = (token) => tokenizedTerm.some(term => token.toLowerCase().startsWith(term.toLowerCase()))
+
+    const occurrencesIndices = splitText
+      .map(includesCheck)
+
+    // calculate best index
+    let bestSum = 0
+    let bestIndex = 0
+    for (let i = 0; i < Math.max(occurrencesIndices.length - highlightWindow, 0); i++) {
+      const window = occurrencesIndices.slice(i, i + highlightWindow)
+      const windowSum = window.reduce((total, cur) => total + cur, 0)
+      if (windowSum >= bestSum) {
+        bestSum = windowSum
+        bestIndex = i
+      }
     }
 
-    const highlight = (content, term) => {
-        const highlightWindow = 20
-        const tokenizedTerm = term.split(/\s+/).filter(t => t !== "")
-        const splitText = content.split(/\s+/).filter(t => t !== "")
-        const includesCheck = (token) => tokenizedTerm.some(term => token.toLowerCase().startsWith(term.toLowerCase()))
-
-        const occurrencesIndices = splitText
-            .map(includesCheck)
-
-        // calculate best index
-        let bestSum = 0
-        let bestIndex = 0
-        for (let i = 0; i < Math.max(occurrencesIndices.length - highlightWindow, 0); i++) {
-            const window = occurrencesIndices.slice(i, i + highlightWindow)
-            const windowSum = window.reduce((total, cur) => total + cur, 0)
-            if (windowSum >= bestSum) {
-                bestSum = windowSum
-                bestIndex = i
-            }
+    const startIndex = Math.max(bestIndex - highlightWindow, 0)
+    const endIndex = Math.min(startIndex + 2 * highlightWindow, splitText.length)
+    const mappedText = splitText
+      .slice(startIndex, endIndex)
+      .map(token => {
+        if (includesCheck(token)) {
+          return `<span class="search-highlight">${token}</span>`
         }
+        return token
+      })
+      .join(" ")
+      .replaceAll('</span> <span class="search-highlight">', " ")
+    return `${startIndex === 0 ? "" : "..."}${mappedText}${endIndex === splitText.length ? "" : "..."}`
+  }
 
-        const startIndex = Math.max(bestIndex - highlightWindow, 0)
-        const endIndex = Math.min(startIndex + 2 * highlightWindow, splitText.length)
-        const mappedText = splitText
-            .slice(startIndex, endIndex)
-            .map(token => {
-                if (includesCheck(token)) {
-                    return `<span class="search-highlight">${token}</span>`
-                }
-                return token
-            })
-            .join(" ")
-            .replaceAll('</span> <span class="search-highlight">', " ")
-        return `${startIndex === 0 ? "" : "..."}${mappedText}${endIndex === splitText.length ? "" : "..."}`
-    }
-
-    const resultToHTML = ({url, title, content, term}) => {
-        const text = removeMarkdown(content)
-        const resultTitle = highlight(title, term)
-        const resultText = highlight(text, term)
-        return `<button class="result-card" id="${url}">
+  const resultToHTML = ({url, title, content, term}) => {
+    const text = removeMarkdown(content)
+    const resultTitle = highlight(title, term)
+    const resultText = highlight(text, term)
+    return `<button class="result-card" id="${url}">
         <h3>${resultTitle}</h3>
         <p>${resultText}</p>
     </button>`
+  }
+
+  const redir = (id, term) => {
+    window.location.href = "{{.Site.BaseURL}}" + `${id}#:~:text=${encodeURIComponent(term)}`
+  }
+
+  const formatForDisplay = id => ({
+    id,
+    url: id,
+    title: content[id].title,
+    content: content[id].content
+  })
+
+  const source = document.getElementById('search-bar')
+  const results = document.getElementById("results-container")
+  let term
+  source.addEventListener("keyup", (e) => {
+    if (e.key === "Enter") {
+      const anchor = document.getElementsByClassName("result-card")[0]
+      redir(anchor.id, term)
     }
-
-    const redir = (id, term) => {
-        window.location.href = "{{.Site.BaseURL}}" + `${id}#:~:text=${encodeURIComponent(term)}`
-    }
-
-    const fetch = id => ({
-        id,
-        url: id,
-        title: content[id].title,
-        content: content[id].content
-    })
-
-    const source = document.getElementById('search-bar')
-    const results = document.getElementById("results-container")
-    let term
-    source.addEventListener("keyup", (e) => {
-        if (e.key === "Enter") {
-            const anchor = document.getElementsByClassName("result-card")[0]
-            redir(anchor.id, term)
+  })
+  source.addEventListener('input', (e) => {
+    term = e.target.value
+    contentIndex.search(term, [
+      {
+        field: "content",
+        limit: 10,
+        suggest: true,
+      },
+      {
+        field: "title",
+        limit: 5,
+      }
+    ]).then(searchResults => {
+      const getByField = field => {
+        const results = searchResults.filter(x => x.field === field)
+        if (results.length === 0) {
+          return []
+        } else {
+          return [...results[0].result]
         }
-    })
-    source.addEventListener('input', (e) => {
-        term = e.target.value
-        contentIndex.search(term, [
-            {
-                field: "content",
-                limit: 10,
-                suggest: true,
-            },
-            {
-                field: "title",
-                limit: 5,
-            }
-        ]).then(searchResults => {
-            const getByField = field => {
-                const results = searchResults.filter(x => x.field === field)
-                if (results.length === 0) {
-                    return []
-                } else {
-                    return [...results[0].result]
-                }
-            }
-            const allIds = new Set([...getByField('title'), ...getByField('content')])
-            const finalResults = [...allIds].map(fetch)
+      }
+      const allIds = new Set([...getByField('title'), ...getByField('content')])
+      const finalResults = [...allIds].map(formatForDisplay)
 
-            // display
-            if (finalResults.length === 0) {
-                results.innerHTML = `<button class="result-card">
+      // display
+      if (finalResults.length === 0) {
+        results.innerHTML = `<button class="result-card">
                     <h3>No results.</h3>
                     <p>Try another search term?</p>
                 </button>`
-            } else {
-                results.innerHTML = finalResults
-                    .map(result => resultToHTML({
-                        ...result,
-                        term,
-                    }))
-                    .join("\n")
-                const anchors = document.getElementsByClassName("result-card");
-                [...anchors].forEach(anchor => {
-                    anchor.onclick = () => redir(anchor.id, term)
-                })
-            }
+      } else {
+        results.innerHTML = finalResults
+          .map(result => resultToHTML({
+            ...result,
+            term,
+          }))
+          .join("\n")
+        const anchors = document.getElementsByClassName("result-card");
+        [...anchors].forEach(anchor => {
+          anchor.onclick = () => redir(anchor.id, term)
         })
+      }
     })
+  })
 
 
-    const searchContainer = document.getElementById("search-container")
-    function openSearch() {
-        if (searchContainer.style.display === "none" || searchContainer.style.display === "") {
-            source.value = ""
-            results.innerHTML = ""
-            searchContainer.style.display = "block"
-            source.focus()
-        } else {
-            searchContainer.style.display = "none"
-        }
+  const searchContainer = document.getElementById("search-container")
+
+  function openSearch() {
+    if (searchContainer.style.display === "none" || searchContainer.style.display === "") {
+      source.value = ""
+      results.innerHTML = ""
+      searchContainer.style.display = "block"
+      source.focus()
+    } else {
+      searchContainer.style.display = "none"
     }
+  }
 
-    function closeSearch() {
-        searchContainer.style.display = "none"
+  function closeSearch() {
+    searchContainer.style.display = "none"
+  }
+
+  document.addEventListener('keydown', (event) => {
+    if (event.key === "/") {
+      event.preventDefault()
+      openSearch()
     }
+    if (event.key === "Escape") {
+      event.preventDefault()
+      closeSearch()
+    }
+  })
 
-    document.addEventListener('keydown', (event) => {
-        if (event.key === "/") {
-            event.preventDefault()
-            openSearch()
-        }
-        if (event.key === "Escape") {
-            event.preventDefault()
-            closeSearch()
-        }
+  window.addEventListener('DOMContentLoaded', () => {
+    const searchButton = document.getElementById("search-icon")
+    searchButton.addEventListener('click', (evt) => {
+      openSearch()
     })
-
-    window.addEventListener('DOMContentLoaded', () => {
-        const searchButton = document.getElementById("search-icon")
-        searchButton.addEventListener('click', (evt) => {
-            openSearch()
-        })
-        searchButton.addEventListener('keydown', (evt) => {
-            openSearch()
-        })
-        searchContainer.addEventListener('click', (evt) => {
-            closeSearch()
-        })
-        document.getElementById("search-space").addEventListener('click', (evt) => {
-            evt.stopPropagation()
-        })
+    searchButton.addEventListener('keydown', (evt) => {
+      openSearch()
     })
+    searchContainer.addEventListener('click', (evt) => {
+      closeSearch()
+    })
+    document.getElementById("search-space").addEventListener('click', (evt) => {
+      evt.stopPropagation()
+    })
+  })
+}
 
+run()
 </script>

--
Gitblit v1.10.0