修改项目树问题 绘制稳态治理分析页面

src/components/tree/govern/treeCommonUtils.ts

@@ -0,0 +1,58 @@
+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 }))
+}