Ben Schlegel
2023-09-16 fea352849c6972da4b3b8935eb2e86f6cefc76ed
quartz/components/ExplorerNode.tsx
@@ -1,5 +1,5 @@
// @ts-ignore
import { QuartzPluginData } from "vfile"
import { QuartzPluginData } from "../plugins/vfile"
import { resolveRelative } from "../util/path"
export interface Options {
@@ -8,6 +8,8 @@
  folderClickBehavior: "collapse" | "link"
  useSavedState: boolean
  sortFn: (a: FileNode, b: FileNode) => number
  filterFn?: (node: FileNode) => boolean
  mapFn?: (node: FileNode) => void
}
type DataWrapper = {
@@ -30,7 +32,7 @@
  constructor(name: string, file?: QuartzPluginData, depth?: number) {
    this.children = []
    this.name = name
    this.file = file ?? null
    this.file = file ? structuredClone(file) : null
    this.depth = depth ?? 0
  }
@@ -67,6 +69,35 @@
  }
  /**
   * Filter FileNode tree. Behaves similar to `Array.prototype.filter()`, but modifies tree in place
   * @param filterFn function to filter tree with
   */
  filter(filterFn: (node: FileNode) => boolean) {
    const filteredNodes: FileNode[] = []
    const traverse = (node: FileNode) => {
      if (filterFn(node)) {
        filteredNodes.push(node)
      }
      node.children.forEach(traverse)
    }
    traverse(this)
    this.children = filteredNodes
  }
  /**
   * Filter FileNode tree. Behaves similar to `Array.prototype.map()`, but modifies tree in place
   * @param mapFn function to filter tree with
   */
  map(mapFn: (node: FileNode) => void) {
    mapFn(this)
    this.children.forEach((child) => child.map(mapFn))
  }
  /**
   * Get folder representation with state of tree.
   * Intended to only be called on root node before changes to the tree are made
   * @param collapsed default state of folders (collapsed by default or not)