more search improvements

This commit is contained in:
Jacky Zhao 2021-10-30 23:27:33 -07:00
parent 6005a2e0a0
commit df23b99951

View File

@ -67,16 +67,35 @@
}; };
</script> </script>
<script> <script>
const contentIndex = new FlexSearch.Worker({ const contentIndex = new FlexSearch.Document({
tokenize: "reverse", cache: true,
charset: "latin:extra", charset: "latin:extra",
suggest: true, optimize: true,
cache: 10, worker: true,
document: {
index: [{
field: "content",
tokenize: "strict",
context: {
resolution: 5,
depth: 3,
bidirectional: true
},
suggest: true,
}, {
field: "title",
tokenize: "forward",
}]
}
}) })
const scrapedContent = {{$.Site.Data.contentIndex}} const scrapedContent = {{$.Site.Data.contentIndex}}
for (const [key, value] of Object.entries(scrapedContent)) { for (const [key, value] of Object.entries(scrapedContent)) {
contentIndex.add(key, value.content) contentIndex.add({
id: key,
title: value.title,
content: removeMarkdown(value.content),
})
} }
const highlight = (content, term) => { const highlight = (content, term) => {
@ -129,6 +148,13 @@
window.location.href = "{{.Site.BaseURL}}" + `${id}#:~:text=${encodeURIComponent(term)}` window.location.href = "{{.Site.BaseURL}}" + `${id}#:~:text=${encodeURIComponent(term)}`
} }
const fetch = id => ({
id,
url: id,
title: scrapedContent[id].title,
content: scrapedContent[id].content
})
const source = document.getElementById('search-bar') const source = document.getElementById('search-bar')
const results = document.getElementById("results-container") const results = document.getElementById("results-container")
let term let term
@ -140,22 +166,27 @@
}) })
source.addEventListener('input', (e) => { source.addEventListener('input', (e) => {
term = e.target.value term = e.target.value
contentIndex.search(term, { contentIndex.search(term, [
limit: 15, {
suggest: true, field: "content",
}).then(searchResults => { limit: 10,
const resultIds = [...new Set(searchResults)] suggest: true,
const finalResults = resultIds.map(id => ({ },
url: id, {
title: scrapedContent[id].title, field: "title",
content: scrapedContent[id].content limit: 5,
})) }
]).then(searchResults => {
const titleIds = [...searchResults.filter(x => x.field === 'title')?.[0].result] ?? []
const contentIds = [...searchResults.filter(x => x.field === 'content')?.[0].result] ?? []
const allIds = [...titleIds, ...contentIds]
const finalResults = allIds.map(fetch)
// display // display
if (finalResults.length === 0) { if (finalResults.length === 0) {
results.innerHTML = `<div class="result-card"> results.innerHTML = `<div class="result-card">
<p>No results.</p> <p>No results.</p>
</div>` </div>`
} else { } else {
results.innerHTML = finalResults results.innerHTML = finalResults
.map(result => resultToHTML({ .map(result => resultToHTML({