openide/docs/plugin-graph/src/GraphTextSearch.js

// Copyright 2000-2021 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
import {GraphHighlighter} from "./GraphHighlighter"
import {wrap} from "comlink"

// https://bugs.webkit.org/show_bug.cgi?id=164860 - inline worker as workaround for Safari (module web worker is not supported)
// noinspection ES6CheckImport
import SearchWorker from "./SearchWorker?worker&inline"

export class GraphTextSearch {
  constructor(graph, cy) {
    this.cy = cy
    this.selectedNodes = new Set()
    this.totalUnion = cy.collection()

    this.index = wrap(new SearchWorker())
    this.index.add(graph
      .filter(it => it.group === "nodes" && "type" in it.data /* not a compound node */)
      .map(it => it.data))
  }

  async searchNodes(text) {
    const {selectedNodes, cy, index} = this

    const newNodes = []
    if (text.length !== 0) {
      const originalText = text
      const extraPrefix = "com.intellij."
      // if (text.startsWith(extraPrefix) && text.length !== extraPrefix.length) {
      //   // remove useless prefix to avoid a lot of results
      //   text = text.substring(extraPrefix.length)
      // }

      const result = await index.search(text)
      console.log(`Search result for "${text}`, result)
      for (const id of result) {
        const node = cy.getElementById(id)
        if (node == null) {
          console.error(`Cannot find node by id ${id}`)
        }
        selectedNodes.delete(node)
        newNodes.push(node)
      }
    }

    cy.elements().removeClass("semiTransparent")
    for (const prevNode of selectedNodes) {
      prevNode.removeClass(["highlight", "found"])
      prevNode.outgoers().union(prevNode.incomers()).removeClass("highlight")
    }

    selectedNodes.clear()
    if (newNodes.length === 0) {
      this.totalUnion = cy.collection()
      return
    }

    let totalUnion = null

    for (const newNode of newNodes) {
      selectedNodes.add(newNode)

      const union = GraphHighlighter.computeToHighlight(newNode)
      totalUnion = totalUnion == null ? union : totalUnion.union(union)

      newNode.addClass("found")
    }

    cy.elements().difference(totalUnion).addClass("semiTransparent")
    totalUnion.addClass("highlight")
    this.totalUnion = totalUnion

    cy.animate({
      // pan: totalUnion.boundingBox(),
      // center: {eles: totalUnion},
      fit: {eles: totalUnion},
    })
  }
}