| | |
| | | // @ts-ignore |
| | | import { QuartzPluginData } from "vfile" |
| | | import { QuartzPluginData } from "../plugins/vfile" |
| | | import { resolveRelative } from "../util/path" |
| | | |
| | | export interface Options { |
| | |
| | | folderDefaultState: "collapsed" | "open" |
| | | 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) |
| | |
| | | } |
| | | |
| | | // 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)) |
| | | } |
| | | } |
| | | |