59 lines
1.6 KiB
TypeScript
59 lines
1.6 KiB
TypeScript
|
import { nextTick } from 'vue'
|
||
|
|
import { createLineTreeDecorators, type LineTreeDecorators } from './lineTreeUtils'
|
||
|
|
|
||
|
|
export { createLineTreeDecorators, type LineTreeDecorators }
|
||
|
|
|
||
|
|
export function findNodeById(nodes: any[], id: string): any | null {
|
||
|
|
for (const node of nodes) {
|
||
|
|
if (node.id === id) return node
|
||
|
|
if (node.children?.length) {
|
||
|
|
const found = findNodeById(node.children, id)
|
||
|
|
if (found) return found
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return null
|
||
|
|
}
|
||
|
|
|
||
|
|
export async function waitForSingleTreeRef(treRef: any, maxRetries = 20) {
|
||
|
|
for (let i = 0; i < maxRetries; i++) {
|
||
|
|
await nextTick()
|
||
|
|
if (treRef?.treeRef) return treRef.treeRef
|
||
|
|
await new Promise(resolve => setTimeout(resolve, 50))
|
||
|
|
}
|
||
|
|
return null
|
||
|
|
}
|
||
|
|
|
||
|
|
export async function selectTreeNode(
|
||
|
|
treRef: any,
|
||
|
|
node: any | undefined,
|
||
|
|
options?: { level?: number; onSelect?: (node: any) => void }
|
||
|
|
) {
|
||
|
|
if (!node) return false
|
||
|
|
const treeInstance = await waitForSingleTreeRef(treRef)
|
||
|
|
treeInstance?.setCurrentKey(node.id)
|
||
|
|
options?.onSelect?.(node)
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
export function bootstrapWithTemplate(
|
||
|
|
template: boolean,
|
||
|
|
loadFn: () => void,
|
||
|
|
fetchTemplate: () => Promise<any>,
|
||
|
|
onPolicy: (data: any) => void
|
||
|
|
) {
|
||
|
|
if (template) {
|
||
|
|
fetchTemplate()
|
||
|
|
.then(res => {
|
||
|
|
onPolicy(res.data)
|
||
|
|
loadFn()
|
||
|
|
})
|
||
|
|
.catch(() => loadFn())
|
||
|
|
} else {
|
||
|
|
loadFn()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export function mapUserTreeNodes(data: any[], icon = 'el-icon-User', color = 'royalblue') {
|
||
|
|
return data.map(item => ({ ...item, icon, color }))
|
||
|
|
}
|