Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-02-09 01:20:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: extract common shiftNodeWindow helper function

Browse Source 
Apply DRY principle suggestion from coderabbitai
This commit is contained in:
Yourz
2025-12-15 20:01:29 +08:00
parent 8656cc8468
commit 0738773b15
1 changed files with 18 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
src/components/common/TreeExplorer.vue
View File

@@ -221,8 +221,12 @@ const handleTreeScroll = useThrottleFn(() => {
}
}, 100)
// Shift window forward for a single node (recursive)
const shiftNodeWindowForward = (node: RenderedTreeExplorerNode) => {
// Shift window for a single node in given direction (recursive)
type ShiftDirection = 'forward' | 'backward'
const shiftNodeWindow = (
node: RenderedTreeExplorerNode,
direction: ShiftDirection
) => {
if (!node.children || node.leaf) return
const isExpanded = expandedKeys.value?.[node.key] ?? false
if (!isExpanded) return
@@ -231,61 +235,31 @@ const shiftNodeWindowForward = (node: RenderedTreeExplorerNode) => {
const range =
parentWindowRanges.value[node.key] ??
createInitialWindowRange(totalChildren, WINDOW_SIZE)
const newRange = shiftWindowForwardUtil(
range,
totalChildren,
BUFFER_SIZE,
WINDOW_SIZE
)
const shiftFn =
direction === 'forward' ? shiftWindowForwardUtil : shiftWindowBackwardUtil
const newRange = shiftFn(range, totalChildren, BUFFER_SIZE, WINDOW_SIZE)
if (newRange) {
parentWindowRanges.value[node.key] = newRange
}
// Recursively process children in current window
for (let i = range.start; i < range.end && i < node.children.length; i++) {
shiftNodeWindowForward(node.children[i])
shiftNodeWindow(node.children[i], direction)
}
}
// Shift all windows forward (load more at end)
const shiftWindowsForward = () => {
// Shift all windows in given direction
const shiftWindows = (direction: ShiftDirection) => {
for (const parent of renderedRoot.value.children || []) {
shiftNodeWindowForward(parent)
shiftNodeWindow(parent, direction)
}
}
// Shift window backward for a single node (recursive)
const shiftNodeWindowBackward = (node: RenderedTreeExplorerNode) => {
if (!node.children || node.leaf) return
const isExpanded = expandedKeys.value?.[node.key] ?? false
if (!isExpanded) return
const totalChildren = node.children.length
const range =
parentWindowRanges.value[node.key] ??
createInitialWindowRange(totalChildren, WINDOW_SIZE)
const newRange = shiftWindowBackwardUtil(
range,
totalChildren,
BUFFER_SIZE,
WINDOW_SIZE
)
if (newRange) {
parentWindowRanges.value[node.key] = newRange
}
// Recursively process children in current window
for (let i = range.start; i < range.end && i < node.children.length; i++) {
shiftNodeWindowBackward(node.children[i])
}
}
// Shift all windows backward (load more at start)
const shiftWindowsBackward = () => {
for (const parent of renderedRoot.value.children || []) {
shiftNodeWindowBackward(parent)
}
}
// Convenience functions for forward/backward
const shiftWindowsForward = () => shiftWindows('forward')
const shiftWindowsBackward = () => shiftWindows('backward')
const renderedRoot = computed<RenderedTreeExplorerNode>(() => {
const renderedRoot = fillNodeInfo(props.root)