From 3d8c470c0d298f720614318fb4c14575e72bbd2e Mon Sep 17 00:00:00 2001
From: Ben Schlegel <ben5.schlegel@gmail.com>
Date: Sat, 16 Sep 2023 17:35:27 +0000
Subject: [PATCH] feat(explorer): implement `map` fn argument
---
quartz/components/ExplorerNode.tsx | 11 +++++++++++
quartz/components/Explorer.tsx | 5 +++++
2 files changed, 16 insertions(+), 0 deletions(-)
diff --git a/quartz/components/Explorer.tsx b/quartz/components/Explorer.tsx
index efc9f6a..23c5db2 100644
--- a/quartz/components/Explorer.tsx
+++ b/quartz/components/Explorer.tsx
@@ -40,6 +40,11 @@
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 5cf3f01..b8d8c14 100644
--- a/quartz/components/ExplorerNode.tsx
+++ b/quartz/components/ExplorerNode.tsx
@@ -9,6 +9,7 @@
useSavedState: boolean
sortFn: (a: FileNode, b: FileNode) => number
filterFn?: (node: FileNode) => boolean
+ mapFn?: (node: FileNode) => void
}
type DataWrapper = {
@@ -87,6 +88,16 @@
}
/**
+ * 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)
--
Gitblit v1.10.0