Ben Schlegel
2023-09-16 58aea1cb0791e18cd092d88de5374431eba7f1d3
quartz/components/ExplorerNode.tsx
@@ -7,6 +7,7 @@
  folderDefaultState: "collapsed" | "open"
  folderClickBehavior: "collapse" | "link"
  useSavedState: boolean
  sortFn: (a: FileNode, b: FileNode) => number
}
type DataWrapper = {
@@ -65,6 +66,21 @@
    this.children.forEach((e) => e.print(depth + 1))
  }
  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
  }
  /**
   * Get folder representation with state of tree.
   * Intended to only be called on root node before changes to the tree are made
@@ -90,19 +106,13 @@
  }
  // Sort order: folders first, then files. Sort folders and files alphabetically
  sort() {
    this.children = this.children.sort((a, b) => {
      if ((!a.file && !b.file) || (a.file && b.file)) {
        return a.name.localeCompare(b.name)
      }
      if (a.file && !b.file) {
        return 1
      } else {
        return -1
      }
    })
    this.children.forEach((e) => e.sort())
  /**
   * Sorts tree according to sort/compare function
   * @param sortFn compare function used for `.sort()`, also used recursively for children
   */
  sort(sortFn: (a: FileNode, b: FileNode) => number) {
    this.children = this.children.sort(sortFn)
    this.children.forEach((e) => e.sort(sortFn))
  }
}