From 0ed17307cf5c9c686348952ebb8932822b351b7f Mon Sep 17 00:00:00 2001
From: Ben Schlegel <ben5.schlegel@gmail.com>
Date: Sat, 16 Sep 2023 19:35:27 +0200
Subject: [PATCH] feat(explorer): implement `map` fn argument

Add a function for mapping over all FileNodes as an option for `Explorer`
---
 quartz/components/Explorer.tsx     |  5 +++++
 quartz/components/ExplorerNode.tsx | 11 +++++++++++
 2 files changed, 16 insertions(+)

diff --git a/quartz/components/Explorer.tsx b/quartz/components/Explorer.tsx
index efc9f6aa..23c5db26 100644
--- a/quartz/components/Explorer.tsx
+++ b/quartz/components/Explorer.tsx
@@ -40,6 +40,11 @@ export default ((userOpts?: Partial<Options>) => {
       fileTree.filter(opts.filterFn)
     }
 
+    // If provided, apply map function to fileTree
+    if (opts.mapFn) {
+      fileTree.map(opts.mapFn)
+    }
+
     // Get all folders of tree. Initialize with collapsed state
     const folders = fileTree.getFolderPaths(opts.folderDefaultState === "collapsed")
 
diff --git a/quartz/components/ExplorerNode.tsx b/quartz/components/ExplorerNode.tsx
index 5cf3f015..b8d8c140 100644
--- a/quartz/components/ExplorerNode.tsx
+++ b/quartz/components/ExplorerNode.tsx
@@ -9,6 +9,7 @@ export interface Options {
   useSavedState: boolean
   sortFn: (a: FileNode, b: FileNode) => number
   filterFn?: (node: FileNode) => boolean
+  mapFn?: (node: FileNode) => void
 }
 
 type DataWrapper = {
@@ -86,6 +87,16 @@ export class FileNode {
     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