better badges for empty/loading model library folders (#953)

* better badges for empty/loading model library folders

for #945

* fix total count on loaded nodes

* fix test break

* additional test fix

* use a null prop fallback instead of having to explicitly calc

* patch
This commit is contained in:
Alex "mcmonkey" Goodwin
2024-09-24 16:37:30 +09:00
committed by Chenlei Hu
parent 4e5bcd6a3b
commit d2d645f74b
5 changed files with 38 additions and 15 deletions

View File

@@ -137,6 +137,22 @@ const renderedRoot = computed<TreeExplorerNode<ComfyModelDef>>(() => {
return 'pi pi-file'
}
},
getBadgeText: (node: TreeExplorerNode<ComfyModelDef>) => {
if (node.leaf) {
return null
}
if (node.children?.length === 1) {
const onlyChild = node.children[0]
if (onlyChild.data?.is_fake_object) {
if (onlyChild.data.name === '(No Content)') {
return '0'
} else if (onlyChild.data.name === 'Loading') {
return '?'
}
}
}
return null
},
children,
draggable: node.leaf,
handleClick: (