| | |
| | | import FlexSearch from "flexsearch" |
| | | import FlexSearch, { DefaultDocumentSearchResults } from "flexsearch" |
| | | import { ContentDetails } from "../../plugins/emitters/contentIndex" |
| | | import { registerEscapeHandler, removeAllChildren } from "./util" |
| | | import { FullSlug, normalizeRelativeURLs, resolveRelative } from "../../util/path" |
| | |
| | | title: string |
| | | content: string |
| | | tags: string[] |
| | | [key: string]: any |
| | | } |
| | | |
| | | // Can be expanded with things like "term" in the future |
| | | type SearchType = "basic" | "tags" |
| | | |
| | | // Current searchType |
| | | let searchType: SearchType = "basic" |
| | | // Current search term // TODO: exact match |
| | | let currentSearchTerm: string = "" |
| | | // index for search |
| | | let index: FlexSearch.Document<Item> | undefined = undefined |
| | | const encoder = (str: string): string[] => { |
| | | const tokens: string[] = [] |
| | | let bufferStart = -1 |
| | | let bufferEnd = -1 |
| | | const lower = str.toLowerCase() |
| | | |
| | | let i = 0 |
| | | for (const char of lower) { |
| | | const code = char.codePointAt(0)! |
| | | |
| | | const isCJK = |
| | | (code >= 0x3040 && code <= 0x309f) || |
| | | (code >= 0x30a0 && code <= 0x30ff) || |
| | | (code >= 0x4e00 && code <= 0x9fff) || |
| | | (code >= 0xac00 && code <= 0xd7af) || |
| | | (code >= 0x20000 && code <= 0x2a6df) |
| | | |
| | | const isWhitespace = code === 32 || code === 9 || code === 10 || code === 13 |
| | | |
| | | if (isCJK) { |
| | | if (bufferStart !== -1) { |
| | | tokens.push(lower.slice(bufferStart, bufferEnd)) |
| | | bufferStart = -1 |
| | | } |
| | | tokens.push(char) |
| | | } else if (isWhitespace) { |
| | | if (bufferStart !== -1) { |
| | | tokens.push(lower.slice(bufferStart, bufferEnd)) |
| | | bufferStart = -1 |
| | | } |
| | | } else { |
| | | if (bufferStart === -1) bufferStart = i |
| | | bufferEnd = i + char.length |
| | | } |
| | | |
| | | i += char.length |
| | | } |
| | | |
| | | if (bufferStart !== -1) { |
| | | tokens.push(lower.slice(bufferStart)) |
| | | } |
| | | |
| | | return tokens |
| | | } |
| | | |
| | | let index = new FlexSearch.Document<Item>({ |
| | | encode: encoder, |
| | | document: { |
| | | id: "id", |
| | | tag: "tags", |
| | | index: [ |
| | | { |
| | | field: "title", |
| | | tokenize: "forward", |
| | | }, |
| | | { |
| | | field: "content", |
| | | tokenize: "forward", |
| | | }, |
| | | { |
| | | field: "tags", |
| | | tokenize: "forward", |
| | | }, |
| | | ], |
| | | }, |
| | | }) |
| | | |
| | | const p = new DOMParser() |
| | | const fetchContentCache: Map<FullSlug, Element[]> = new Map() |
| | | const contextWindowWords = 30 |
| | | const numSearchResults = 8 |
| | | const numTagResults = 5 |
| | | |
| | | const tokenizeTerm = (term: string) => |
| | | term |
| | | .split(/\s+/) |
| | | .filter((t) => t !== "") |
| | | .sort((a, b) => b.length - a.length) |
| | | const tokenizeTerm = (term: string) => { |
| | | const tokens = term.split(/\s+/).filter((t) => t.trim() !== "") |
| | | const tokenLen = tokens.length |
| | | if (tokenLen > 1) { |
| | | for (let i = 1; i < tokenLen; i++) { |
| | | tokens.push(tokens.slice(0, i + 1).join(" ")) |
| | | } |
| | | } |
| | | |
| | | return tokens.sort((a, b) => b.length - a.length) // always highlight longest terms first |
| | | } |
| | | |
| | | function highlight(searchTerm: string, text: string, trim?: boolean) { |
| | | // try to highlight longest tokens first |
| | | const tokenizedTerms = tokenizeTerm(searchTerm) |
| | | let tokenizedText = text.split(/\s+/).filter((t) => t !== "") |
| | | |
| | |
| | | } |
| | | return tok |
| | | }) |
| | | .slice(startIndex, endIndex + 1) |
| | | .join(" ") |
| | | |
| | | return `${startIndex === 0 ? "" : "..."}${slice}${ |
| | |
| | | } |
| | | |
| | | function highlightHTML(searchTerm: string, el: HTMLElement) { |
| | | // try to highlight longest tokens first |
| | | const p = new DOMParser() |
| | | const tokenizedTerms = tokenizeTerm(searchTerm) |
| | | const html = p.parseFromString(el.innerHTML, "text/html") |
| | |
| | | return span |
| | | } |
| | | |
| | | const highlightTextNodes = (node: Node) => { |
| | | const highlightTextNodes = (node: Node, term: string) => { |
| | | if (node.nodeType === Node.TEXT_NODE) { |
| | | let nodeText = node.nodeValue || "" |
| | | tokenizedTerms.forEach((term) => { |
| | | const regex = new RegExp(term.toLowerCase(), "gi") |
| | | const matches = nodeText.match(regex) |
| | | const spanContainer = document.createElement("span") |
| | | let lastIndex = 0 |
| | | matches?.forEach((match) => { |
| | | const matchIndex = nodeText.indexOf(match, lastIndex) |
| | | spanContainer.appendChild(document.createTextNode(nodeText.slice(lastIndex, matchIndex))) |
| | | spanContainer.appendChild(createHighlightSpan(match)) |
| | | lastIndex = matchIndex + match.length |
| | | }) |
| | | spanContainer.appendChild(document.createTextNode(nodeText.slice(lastIndex))) |
| | | node.parentNode?.replaceChild(spanContainer, node) |
| | | }) |
| | | const nodeText = node.nodeValue ?? "" |
| | | const regex = new RegExp(term.toLowerCase(), "gi") |
| | | const matches = nodeText.match(regex) |
| | | if (!matches || matches.length === 0) return |
| | | const spanContainer = document.createElement("span") |
| | | let lastIndex = 0 |
| | | for (const match of matches) { |
| | | const matchIndex = nodeText.indexOf(match, lastIndex) |
| | | spanContainer.appendChild(document.createTextNode(nodeText.slice(lastIndex, matchIndex))) |
| | | spanContainer.appendChild(createHighlightSpan(match)) |
| | | lastIndex = matchIndex + match.length |
| | | } |
| | | spanContainer.appendChild(document.createTextNode(nodeText.slice(lastIndex))) |
| | | node.parentNode?.replaceChild(spanContainer, node) |
| | | } else if (node.nodeType === Node.ELEMENT_NODE) { |
| | | Array.from(node.childNodes).forEach(highlightTextNodes) |
| | | if ((node as HTMLElement).classList.contains("highlight")) return |
| | | Array.from(node.childNodes).forEach((child) => highlightTextNodes(child, term)) |
| | | } |
| | | } |
| | | |
| | | highlightTextNodes(html.body) |
| | | for (const term of tokenizedTerms) { |
| | | highlightTextNodes(html.body, term) |
| | | } |
| | | |
| | | return html.body |
| | | } |
| | | |
| | | const p = new DOMParser() |
| | | const encoder = (str: string) => str.toLowerCase().split(/([^a-z]|[^\x00-\x7F])/) |
| | | let prevShortcutHandler: ((e: HTMLElementEventMap["keydown"]) => void) | undefined = undefined |
| | | async function setupSearch(searchElement: Element, currentSlug: FullSlug, data: ContentIndex) { |
| | | const container = searchElement.querySelector(".search-container") as HTMLElement |
| | | if (!container) return |
| | | |
| | | const fetchContentCache: Map<FullSlug, Element[]> = new Map() |
| | | const sidebar = container.closest(".sidebar") as HTMLElement | null |
| | | |
| | | document.addEventListener("nav", async (e: CustomEventMap["nav"]) => { |
| | | const currentSlug = e.detail.url |
| | | const searchButton = searchElement.querySelector(".search-button") as HTMLButtonElement |
| | | if (!searchButton) return |
| | | |
| | | const data = await fetchData |
| | | 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 searchLayout = document.getElementById("search-layout") |
| | | const searchBar = searchElement.querySelector(".search-bar") as HTMLInputElement |
| | | if (!searchBar) return |
| | | |
| | | const searchLayout = searchElement.querySelector(".search-layout") as HTMLElement |
| | | if (!searchLayout) return |
| | | |
| | | const idDataMap = Object.keys(data) as FullSlug[] |
| | | |
| | | const appendLayout = (el: HTMLElement) => { |
| | | if (searchLayout?.querySelector(`#${el.id}`) === null) { |
| | | searchLayout?.appendChild(el) |
| | | } |
| | | searchLayout.appendChild(el) |
| | | } |
| | | |
| | | const enablePreview = searchLayout?.dataset?.preview === "true" |
| | | const enablePreview = searchLayout.dataset.preview === "true" |
| | | let preview: HTMLDivElement | undefined = undefined |
| | | let previewInner: HTMLDivElement | undefined = undefined |
| | | const results = document.createElement("div") |
| | | results.id = "results-container" |
| | | results.style.flexBasis = enablePreview ? "30%" : "100%" |
| | | results.className = "results-container" |
| | | appendLayout(results) |
| | | |
| | | if (enablePreview) { |
| | | preview = document.createElement("div") |
| | | preview.id = "preview-container" |
| | | preview.style.flexBasis = "70%" |
| | | preview.className = "preview-container" |
| | | appendLayout(preview) |
| | | } |
| | | |
| | | 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) |
| | | } |
| | | container.classList.remove("active") |
| | | searchBar.value = "" // clear the input when we dismiss the search |
| | | if (sidebar) sidebar.style.zIndex = "" |
| | | removeAllChildren(results) |
| | | if (preview) { |
| | | removeAllChildren(preview) |
| | | } |
| | | if (searchLayout) { |
| | | searchLayout.style.opacity = "0" |
| | | } |
| | | |
| | | searchLayout.classList.remove("display-results") |
| | | searchType = "basic" // reset search type after closing |
| | | searchButton.focus() |
| | | } |
| | | |
| | | function showSearch(searchTypeNew: SearchType) { |
| | | searchType = searchTypeNew |
| | | if (sidebar) { |
| | | sidebar.style.zIndex = "1" |
| | | } |
| | | container?.classList.add("active") |
| | | searchBar?.focus() |
| | | if (sidebar) sidebar.style.zIndex = "1" |
| | | container.classList.add("active") |
| | | searchBar.focus() |
| | | } |
| | | |
| | | let currentHover: HTMLInputElement | null = null |
| | | |
| | | async function shortcutHandler(e: HTMLElementEventMap["keydown"]) { |
| | | if (e.key === "k" && (e.ctrlKey || e.metaKey) && !e.shiftKey) { |
| | | e.preventDefault() |
| | | const searchBarOpen = container?.classList.contains("active") |
| | | const searchBarOpen = container.classList.contains("active") |
| | | searchBarOpen ? hideSearch() : showSearch("basic") |
| | | return |
| | | } else if (e.shiftKey && (e.ctrlKey || e.metaKey) && e.key.toLowerCase() === "k") { |
| | | // Hotkey to open tag search |
| | | e.preventDefault() |
| | | const searchBarOpen = container?.classList.contains("active") |
| | | const searchBarOpen = container.classList.contains("active") |
| | | searchBarOpen ? hideSearch() : showSearch("tags") |
| | | |
| | | // add "#" prefix for tag search |
| | | if (searchBar) searchBar.value = "#" |
| | | searchBar.value = "#" |
| | | return |
| | | } |
| | | |
| | | if (currentHover) { |
| | |
| | | } |
| | | |
| | | // If search is active, then we will render the first result and display accordingly |
| | | if (!container?.classList.contains("active")) return |
| | | else if (e.key === "Enter") { |
| | | if (!container.classList.contains("active")) return |
| | | if (e.key === "Enter" && !e.isComposing) { |
| | | // If result has focus, navigate to that one, otherwise pick first result |
| | | if (results?.contains(document.activeElement)) { |
| | | if (results.contains(document.activeElement)) { |
| | | const active = document.activeElement as HTMLInputElement |
| | | if (active.classList.contains("no-match")) return |
| | | await displayPreview(active) |
| | | active.click() |
| | | } else { |
| | | const anchor = document.getElementsByClassName("result-card")[0] as HTMLInputElement | null |
| | | if (!anchor || anchor?.classList.contains("no-match")) return |
| | | if (!anchor || anchor.classList.contains("no-match")) return |
| | | await displayPreview(anchor) |
| | | anchor.click() |
| | | } |
| | | } else if (e.key === "ArrowUp" || (e.shiftKey && e.key === "Tab")) { |
| | | e.preventDefault() |
| | | if (results?.contains(document.activeElement)) { |
| | | if (results.contains(document.activeElement)) { |
| | | // If an element in results-container already has focus, focus previous one |
| | | const currentResult = currentHover |
| | | ? currentHover |
| | | : (document.activeElement as HTMLInputElement | null) |
| | | const prevResult = currentResult?.previousElementSibling as HTMLInputElement | null |
| | | currentResult?.classList.remove("focus") |
| | | await displayPreview(prevResult) |
| | | prevResult?.focus() |
| | | currentHover = prevResult |
| | | if (prevResult) currentHover = prevResult |
| | | await displayPreview(prevResult) |
| | | } |
| | | } else if (e.key === "ArrowDown" || e.key === "Tab") { |
| | | e.preventDefault() |
| | | // The results should already been focused, so we need to find the next one. |
| | | // The activeElement is the search bar, so we need to find the first result and focus it. |
| | | if (!results?.contains(document.activeElement)) { |
| | | if (document.activeElement === searchBar || currentHover !== null) { |
| | | const firstResult = currentHover |
| | | ? currentHover |
| | | : (document.getElementsByClassName("result-card")[0] as HTMLInputElement | null) |
| | | const secondResult = firstResult?.nextElementSibling as HTMLInputElement | null |
| | | firstResult?.classList.remove("focus") |
| | | await displayPreview(secondResult) |
| | | secondResult?.focus() |
| | | currentHover = secondResult |
| | | } else { |
| | | // If an element in results-container already has focus, focus next one |
| | | const active = currentHover |
| | | ? currentHover |
| | | : (document.activeElement as HTMLInputElement | null) |
| | | active?.classList.remove("focus") |
| | | const nextResult = active?.nextElementSibling as HTMLInputElement | null |
| | | await displayPreview(nextResult) |
| | | nextResult?.focus() |
| | | currentHover = nextResult |
| | | if (secondResult) currentHover = secondResult |
| | | await displayPreview(secondResult) |
| | | } |
| | | } |
| | | } |
| | | |
| | | function trimContent(content: string) { |
| | | // works without escaping html like in `description.ts` |
| | | const sentences = content.replace(/\s+/g, " ").split(".") |
| | | let finalDesc = "" |
| | | let sentenceIdx = 0 |
| | | |
| | | // Roughly estimate characters by (words * 5). Matches description length in `description.ts`. |
| | | const len = contextWindowWords * 5 |
| | | while (finalDesc.length < len) { |
| | | const sentence = sentences[sentenceIdx] |
| | | if (!sentence) break |
| | | finalDesc += sentence + "." |
| | | sentenceIdx++ |
| | | } |
| | | |
| | | // If more content would be available, indicate it by finishing with "..." |
| | | if (finalDesc.length < content.length) { |
| | | finalDesc += ".." |
| | | } |
| | | |
| | | return finalDesc |
| | | } |
| | | |
| | | const formatForDisplay = (term: string, id: number) => { |
| | | const slug = idDataMap[id] |
| | | return { |
| | | id, |
| | | slug, |
| | | title: searchType === "tags" ? data[slug].title : highlight(term, data[slug].title ?? ""), |
| | | // if searchType is tag, display context from start of file and trim, otherwise use regular highlight |
| | | content: |
| | | searchType === "tags" |
| | | ? trimContent(data[slug].content) |
| | | : highlight(term, data[slug].content ?? "", true), |
| | | tags: highlightTags(term, data[slug].tags), |
| | | content: highlight(term, data[slug].content ?? "", true), |
| | | tags: highlightTags(term.substring(1), data[slug].tags), |
| | | } |
| | | } |
| | | |
| | | function highlightTags(term: string, tags: string[]) { |
| | | if (tags && searchType === "tags") { |
| | | // Find matching tags |
| | | const termLower = term.toLowerCase() |
| | | let matching = tags.filter((str) => str.includes(termLower)) |
| | | |
| | | // Subtract matching from original tags, then push difference |
| | | if (matching.length > 0) { |
| | | let difference = tags.filter((x) => !matching.includes(x)) |
| | | |
| | | // Convert to html (cant be done later as matches/term dont get passed to `resultToHTML`) |
| | | matching = matching.map((tag) => `<li><p class="match-tag">#${tag}</p></li>`) |
| | | difference = difference.map((tag) => `<li><p>#${tag}</p></li>`) |
| | | matching.push(...difference) |
| | | } |
| | | |
| | | // Only allow max of `numTagResults` in preview |
| | | if (tags.length > numTagResults) { |
| | | matching.splice(numTagResults) |
| | | } |
| | | |
| | | return matching |
| | | } else { |
| | | if (!tags || searchType !== "tags") { |
| | | return [] |
| | | } |
| | | |
| | | return tags |
| | | .map((tag) => { |
| | | if (tag.toLowerCase().includes(term.toLowerCase())) { |
| | | return `<li><p class="match-tag">#${tag}</p></li>` |
| | | } else { |
| | | return `<li><p>#${tag}</p></li>` |
| | | } |
| | | }) |
| | | .slice(0, numTagResults) |
| | | } |
| | | |
| | | function resolveUrl(slug: FullSlug): URL { |
| | |
| | | } |
| | | |
| | | const resultToHTML = ({ slug, title, content, tags }: Item) => { |
| | | const htmlTags = tags.length > 0 ? `<ul>${tags.join("")}</ul>` : `` |
| | | const resultContent = enablePreview && window.innerWidth > 600 ? "" : `<p>${content}</p>` |
| | | |
| | | const htmlTags = tags.length > 0 ? `<ul class="tags">${tags.join("")}</ul>` : `` |
| | | const itemTile = document.createElement("a") |
| | | itemTile.classList.add("result-card") |
| | | Object.assign(itemTile, { |
| | | id: slug, |
| | | href: resolveUrl(slug).toString(), |
| | | innerHTML: `<h3>${title}</h3>${htmlTags}${resultContent}`, |
| | | itemTile.id = slug |
| | | itemTile.href = resolveUrl(slug).toString() |
| | | itemTile.innerHTML = ` |
| | | <h3 class="card-title">${title}</h3> |
| | | ${htmlTags} |
| | | <p class="card-description">${content}</p> |
| | | ` |
| | | itemTile.addEventListener("click", (event) => { |
| | | if (event.altKey || event.ctrlKey || event.metaKey || event.shiftKey) return |
| | | hideSearch() |
| | | }) |
| | | |
| | | const handler = (event: MouseEvent) => { |
| | | if (event.altKey || event.ctrlKey || event.metaKey || event.shiftKey) return |
| | | hideSearch() |
| | | } |
| | | |
| | | async function onMouseEnter(ev: MouseEvent) { |
| | | // Actually when we hover, we need to clean all highlights within the result childs |
| | | if (!ev.target) return |
| | | for (const el of document.getElementsByClassName( |
| | | "result-card", |
| | | ) as HTMLCollectionOf<HTMLElement>) { |
| | | el.classList.remove("focus") |
| | | el.blur() |
| | | } |
| | | const target = ev.target as HTMLInputElement |
| | | await displayPreview(target) |
| | | currentHover = target |
| | | currentHover.classList.remove("focus") |
| | | } |
| | | |
| | | async function onMouseLeave(ev: MouseEvent) { |
| | | const target = ev.target as HTMLAnchorElement |
| | | target.classList.remove("focus") |
| | | } |
| | | |
| | | const events = [ |
| | | ["mouseenter", onMouseEnter], |
| | | ["mouseleave", onMouseLeave], |
| | | [ |
| | | "click", |
| | | (event: MouseEvent) => { |
| | | if (event.altKey || event.ctrlKey || event.metaKey || event.shiftKey) return |
| | | hideSearch() |
| | | }, |
| | | ], |
| | | ] as [keyof HTMLElementEventMap, (this: HTMLElement) => void][] |
| | | |
| | | events.forEach(([event, handler]) => itemTile.addEventListener(event, handler)) |
| | | itemTile.addEventListener("mouseenter", onMouseEnter) |
| | | window.addCleanup(() => itemTile.removeEventListener("mouseenter", onMouseEnter)) |
| | | itemTile.addEventListener("click", handler) |
| | | window.addCleanup(() => itemTile.removeEventListener("click", handler)) |
| | | |
| | | return itemTile |
| | | } |
| | | |
| | | async function displayResults(finalResults: Item[]) { |
| | | if (!results) return |
| | | |
| | | removeAllChildren(results) |
| | | if (finalResults.length === 0) { |
| | | results.innerHTML = `<a class="result-card no-match"> |
| | | <h3>No results.</h3> |
| | | <p>Try another search term?</p> |
| | | </a>` |
| | | <h3>No results.</h3> |
| | | <p>Try another search term?</p> |
| | | </a>` |
| | | } else { |
| | | results.append(...finalResults.map(resultToHTML)) |
| | | } |
| | | // focus on first result, then also dispatch preview immediately |
| | | if (results?.firstElementChild) { |
| | | |
| | | if (finalResults.length === 0 && preview) { |
| | | // no results, clear previous preview |
| | | removeAllChildren(preview) |
| | | } else { |
| | | // focus on first result, then also dispatch preview immediately |
| | | const firstChild = results.firstElementChild as HTMLElement |
| | | if (firstChild.classList.contains("no-match")) { |
| | | removeAllChildren(preview as HTMLElement) |
| | | } else { |
| | | firstChild.classList.add("focus") |
| | | await displayPreview(firstChild) |
| | | } |
| | | firstChild.classList.add("focus") |
| | | currentHover = firstChild as HTMLInputElement |
| | | await displayPreview(firstChild) |
| | | } |
| | | } |
| | | |
| | |
| | | } |
| | | |
| | | async function displayPreview(el: HTMLElement | null) { |
| | | if (!searchLayout || !enablePreview || !el) return |
| | | |
| | | if (!searchLayout || !enablePreview || !el || !preview) return |
| | | const slug = el.id as FullSlug |
| | | el.classList.add("focus") |
| | | |
| | | removeAllChildren(preview as HTMLElement) |
| | | |
| | | const innerDiv = await fetchContent(slug).then((contents) => |
| | | contents.flatMap((el) => [...highlightHTML(currentSearchTerm, el as HTMLElement).children]), |
| | | ) |
| | | previewInner = document.createElement("div") |
| | | previewInner.classList.add("preview-inner") |
| | | preview?.appendChild(previewInner) |
| | | |
| | | const innerDiv = await fetchContent(slug).then((contents) => |
| | | contents.map((el) => highlightHTML(currentSearchTerm, el as HTMLElement)), |
| | | ) |
| | | previewInner.append(...innerDiv) |
| | | preview.replaceChildren(previewInner) |
| | | |
| | | // scroll to longest |
| | | const highlights = [...preview.getElementsByClassName("highlight")].sort( |
| | | (a, b) => b.innerHTML.length - a.innerHTML.length, |
| | | ) |
| | | highlights[0]?.scrollIntoView({ block: "start" }) |
| | | } |
| | | |
| | | async function onType(e: HTMLElementEventMap["input"]) { |
| | | let term = (e.target as HTMLInputElement).value |
| | | let searchResults: FlexSearch.SimpleDocumentSearchResultSetUnit[] |
| | | if (!searchLayout || !index) return |
| | | currentSearchTerm = (e.target as HTMLInputElement).value |
| | | searchLayout.classList.toggle("display-results", currentSearchTerm !== "") |
| | | searchType = currentSearchTerm.startsWith("#") ? "tags" : "basic" |
| | | |
| | | if (searchLayout) { |
| | | searchLayout.style.opacity = "1" |
| | | } |
| | | |
| | | if (term === "" && searchLayout) { |
| | | searchLayout.style.opacity = "0" |
| | | } |
| | | |
| | | if (term.toLowerCase().startsWith("#")) { |
| | | searchType = "tags" |
| | | } else { |
| | | searchType = "basic" |
| | | } |
| | | |
| | | switch (searchType) { |
| | | case "tags": { |
| | | term = term.substring(1) |
| | | searchResults = |
| | | (await index?.searchAsync({ query: term, limit: numSearchResults, index: ["tags"] })) ?? |
| | | [] |
| | | break |
| | | let searchResults: DefaultDocumentSearchResults<Item> |
| | | if (searchType === "tags") { |
| | | currentSearchTerm = currentSearchTerm.substring(1).trim() |
| | | const separatorIndex = currentSearchTerm.indexOf(" ") |
| | | if (separatorIndex != -1) { |
| | | // search by title and content index and then filter by tag (implemented in flexsearch) |
| | | const tag = currentSearchTerm.substring(0, separatorIndex) |
| | | const query = currentSearchTerm.substring(separatorIndex + 1).trim() |
| | | searchResults = await index.searchAsync({ |
| | | query: query, |
| | | // return at least 10000 documents, so it is enough to filter them by tag (implemented in flexsearch) |
| | | limit: Math.max(numSearchResults, 10000), |
| | | index: ["title", "content"], |
| | | tag: { tags: tag }, |
| | | }) |
| | | for (let searchResult of searchResults) { |
| | | searchResult.result = searchResult.result.slice(0, numSearchResults) |
| | | } |
| | | // set search type to basic and remove tag from term for proper highlightning and scroll |
| | | searchType = "basic" |
| | | currentSearchTerm = query |
| | | } else { |
| | | // default search by tags index |
| | | searchResults = await index.searchAsync({ |
| | | query: currentSearchTerm, |
| | | limit: numSearchResults, |
| | | index: ["tags"], |
| | | }) |
| | | } |
| | | case "basic": |
| | | default: { |
| | | searchResults = |
| | | (await index?.searchAsync({ |
| | | query: term, |
| | | limit: numSearchResults, |
| | | index: ["title", "content"], |
| | | })) ?? [] |
| | | } |
| | | } else if (searchType === "basic") { |
| | | searchResults = await index.searchAsync({ |
| | | query: currentSearchTerm, |
| | | limit: numSearchResults, |
| | | index: ["title", "content"], |
| | | }) |
| | | } |
| | | |
| | | const getByField = (field: string): number[] => { |
| | |
| | | ...getByField("content"), |
| | | ...getByField("tags"), |
| | | ]) |
| | | const finalResults = [...allIds].map((id) => formatForDisplay(term, id)) |
| | | const finalResults = [...allIds].map((id) => formatForDisplay(currentSearchTerm, id)) |
| | | await displayResults(finalResults) |
| | | } |
| | | |
| | | if (prevShortcutHandler) { |
| | | document.removeEventListener("keydown", prevShortcutHandler) |
| | | } |
| | | |
| | | document.addEventListener("keydown", shortcutHandler) |
| | | prevShortcutHandler = shortcutHandler |
| | | searchIcon?.removeEventListener("click", () => showSearch("basic")) |
| | | searchIcon?.addEventListener("click", () => showSearch("basic")) |
| | | searchBar?.removeEventListener("input", onType) |
| | | searchBar?.addEventListener("input", onType) |
| | | window.addCleanup(() => document.removeEventListener("keydown", shortcutHandler)) |
| | | searchButton.addEventListener("click", () => showSearch("basic")) |
| | | window.addCleanup(() => searchButton.removeEventListener("click", () => showSearch("basic"))) |
| | | searchBar.addEventListener("input", onType) |
| | | window.addCleanup(() => searchBar.removeEventListener("input", onType)) |
| | | |
| | | // setup index if it hasn't been already |
| | | if (!index) { |
| | | index = new FlexSearch.Document({ |
| | | charset: "latin:extra", |
| | | encode: encoder, |
| | | document: { |
| | | id: "id", |
| | | index: [ |
| | | { |
| | | field: "title", |
| | | tokenize: "forward", |
| | | }, |
| | | { |
| | | field: "content", |
| | | tokenize: "forward", |
| | | }, |
| | | { |
| | | field: "tags", |
| | | tokenize: "forward", |
| | | }, |
| | | ], |
| | | }, |
| | | }) |
| | | |
| | | fillDocument(index, data) |
| | | } |
| | | |
| | | // register handlers |
| | | registerEscapeHandler(container, hideSearch) |
| | | }) |
| | | await fillDocument(data) |
| | | } |
| | | |
| | | /** |
| | | * Fills flexsearch document with data |
| | | * @param index index to fill |
| | | * @param data data to fill index with |
| | | */ |
| | | async function fillDocument(index: FlexSearch.Document<Item, false>, data: any) { |
| | | let indexPopulated = false |
| | | async function fillDocument(data: ContentIndex) { |
| | | if (indexPopulated) return |
| | | let id = 0 |
| | | const promises: Array<Promise<unknown>> = [] |
| | | for (const [slug, fileData] of Object.entries<ContentDetails>(data)) { |
| | | await index.addAsync(id, { |
| | | id, |
| | | slug: slug as FullSlug, |
| | | title: fileData.title, |
| | | content: fileData.content, |
| | | tags: fileData.tags, |
| | | }) |
| | | id++ |
| | | promises.push( |
| | | index.addAsync(id++, { |
| | | id, |
| | | slug: slug as FullSlug, |
| | | title: fileData.title, |
| | | content: fileData.content, |
| | | tags: fileData.tags, |
| | | }), |
| | | ) |
| | | } |
| | | |
| | | await Promise.all(promises) |
| | | indexPopulated = true |
| | | } |
| | | |
| | | document.addEventListener("nav", async (e: CustomEventMap["nav"]) => { |
| | | const currentSlug = e.detail.url |
| | | const data = await fetchData |
| | | const searchElement = document.getElementsByClassName("search") |
| | | for (const element of searchElement) { |
| | | await setupSearch(element, currentSlug, data) |
| | | } |
| | | }) |