| | |
| | | // @ts-ignore |
| | | import { QuartzPluginData } from "vfile" |
| | | import { QuartzPluginData } from "../plugins/vfile" |
| | | import { resolveRelative } from "../util/path" |
| | | |
| | | export interface Options { |
| | |
| | | folderClickBehavior: "collapse" | "link" |
| | | useSavedState: boolean |
| | | sortFn: (a: FileNode, b: FileNode) => number |
| | | filterFn?: (node: FileNode) => boolean |
| | | mapFn?: (node: FileNode) => void |
| | | } |
| | | |
| | | type DataWrapper = { |
| | |
| | | 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 |
| | | } |
| | | |
| | |
| | | } |
| | | |
| | | /** |
| | | * 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) |