/*
 * SPDX-FileCopyrightText: Copyright (c) 2023-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */


// get the search term from the url query string
async function getSearchTerm(term, tokenSeparator) {
  const normalizedTerm = term.replace(/"/g, "'");  // normalized quotes
  const escapedTerm = normalizedTerm.replace(/([\\\*\+\-\:])/g, "\\$1");  // escape lunr special characters
  const quoteSplitter = /'(.*?)'/g;  // when splitting regexes have a group, the group is included in results

  // separate out quoted phrases
  const searchString = Array.from(escapedTerm.split(quoteSplitter))
    .map(function (e, i) {
      return e.split(tokenSeparator)
        .map(d => d.trim())
        .filter(d => d.length > 0)
        // `+` in lunr means: "must have this term", so use it for quoted terms
        // add `*` prefix/suffix so users searching `wor` will get `world` results when unquoted
        // also retain un-wildcarded entry to bump results with exact matches
        .map(d => i % 2 == 1 ? `+${d}` : `${d} *${d}*`)
        .join(" ");
    })
    .join(" ");

  return searchString;
}


// build the lunr search index based on the project data
async function buildIndex(searchData, tokenSeparator) {
  // split the text on any non-word characters
  lunr.tokenizer.separator = tokenSeparator;

  var idx = lunr(function () {
    this.ref('id');
    this.field('display_name', { boost: 10 });
    this.field('keywords', { boost: 5 });
    this.field('content');
    this.metadataWhitelist = ['position'];

    searchData.data.forEach(function (doc) {
      let attributes = {};
      if (doc.type == "doc" ) {
        // boost results that are pages
        attributes = { boost: 3 };
      }
      else if (doc.type == "section") {
        // boost results that are sections
        attributes = { boost: 2 };
      }
      this.add(doc, attributes);
    }, this);
  });

  return idx;
}


// build the text surrounding the search match to be displayed in the results
function buildSearchResultContext(ob) {
  const match = Object.values(ob.matchData.metadata)[0]['content'];
  const offset = 100;

  // get the positions of the first match
  const [matchStart, matchLength] = match !== undefined ? match.position[0] : [0, 0];
  const matchEnd = matchStart + matchLength;

  // clamp to string length
  const contextStartUnclamped = matchStart - offset;
  const contextStart = Math.max(0, contextStartUnclamped);
  const isStartTrimmed = contextStart === contextStartUnclamped && matchLength !== 0;
  const contextEndUnclamped = matchEnd + offset;
  const contextEnd = Math.min(ob.data.content.length, contextEndUnclamped);
  const isEndTrimmed = contextEnd === contextEndUnclamped;

  // short circuit if there is not text context (maybe page title match)
  if (matchLength === 0) {
    return null;
  }

  // trim the context to match desired bounds
  let context = ob.data.content.substring(0, contextEnd);

  const startText = `${isStartTrimmed ? '...' : ''}${context.slice(0, matchStart).substring(contextStart)}`;
  const highlightText = context.slice(matchStart, matchEnd);
  const endText = `${context.slice(matchEnd)}${isEndTrimmed ? '...' : ''}`;

  return { "start": startText, "highlight": highlightText, "end": endText }
}


// build the list item element for each search result
function buildSearchResultListItem(ob) {
  let item = document.createElement("li");

  if (ob.sitename) {
    let siteName = document.createElement("div");
    let siteNameText = document.createTextNode(ob.sitename);
    siteName.appendChild(siteNameText);
    item.appendChild(siteName);
  }

  let link = document.createElement("a")
  link.href = encodeURI(ob.uri);
  let linkText = document.createTextNode(`${ob.parent_name}${ob.parent_name ? ' — ' : ''}${ob.name}`);
  link.appendChild(linkText);
  item.appendChild(link);

  let displayType = document.createElement("span");
  let displayTypeText = document.createTextNode(ob.display_type);
  displayType.appendChild(displayTypeText);
  item.appendChild(displayType);

  if (ob.context) {
    // create the container element for the context
    let contextElement = document.createElement("p");
    contextElement.classList.add("context");

    let startText = document.createTextNode(ob.context.start);
    contextElement.appendChild(startText);

    let highlightElement = document.createElement("span");
    highlightElement.classList.add("highlighted");
    let highlightElementText = document.createTextNode(ob.context.highlight);
    highlightElement.appendChild(highlightElementText);
    contextElement.appendChild(highlightElement);

    let endText = document.createTextNode(ob.context.end);
    contextElement.appendChild(endText);

    item.appendChild(contextElement);
  }

  return item;
}


// show the resulting object, its type, and the context around the result
async function displayResults(highlightText, searchData, searchResults) {
  const relativeRoot = document.documentElement.getAttribute("data-content_root");
  let resultsdiv = document.getElementById('search-results');

  let heading = document.createElement("h2");
  heading.textContent = 'Search results';
  resultsdiv.appendChild(heading);

  let summary = document.createElement("p");
  summary.classList.add("search-summary");
  summary.textContent = `Found ${searchResults.length} match${searchResults.length === 1 ? '' : 'es'}.`;
  resultsdiv.appendChild(summary);

  let results = document.createElement("ul")
  results.classList.add("search")

  searchResults
    .map(ele => ({ data: searchData.data[ele.ref], matchData: ele.matchData }))
    .map(ob => ({
      uri: `${relativeRoot}${ob.data.filename}?highlight=${highlightText}#${ob.data.anchor}`,
      name: `${ob.data.display_name}`,
      sitename: Object.hasOwn(ob.data, 'sitename') ? ob.data.sitename : null,
      parent_name: `${ob.data.type !== 'doc' ? `${searchData.data[ob.data.doc_id].display_name}` : ''}`,
      display_type: `${ob.data.display_type}`,
      context: buildSearchResultContext(ob),
    }))
    .forEach(ob => results.appendChild(buildSearchResultListItem(ob)));

  resultsdiv.appendChild(results);
};


async function main() {
  const params = (new URL(document.location)).searchParams;
  const term = params.get('q');
  // set the search box content, .value is a safe sink so we can set it directly
  // see: https://cheatsheetseries.owasp.org/cheatsheets/Cross_Site_Scripting_Prevention_Cheat_Sheet.html#safe-sinks
  document.querySelectorAll('input[name="q"]').forEach(input => input.value = term);
  const tokenSeparator = /[\s\-]/g;
  const searchString = await getSearchTerm(term, tokenSeparator);
  const idx = await buildIndex(searchData, tokenSeparator);
  const searchResults = idx.search(searchString);
  const highlightText = encodeURIComponent(term.replace(/\W+/g, " ").trim());
  displayResults(highlightText, searchData, searchResults);
}


document.addEventListener('DOMContentLoaded', main);