@zhongguo168a/yxeditor-common 0.0.78 → 0.0.79

package/dist/index.d.ts

@@ -454,10 +454,18 @@ declare class TreeNode<T = any> {
      */
     walkRemove(f: (node: TreeNode) => boolean): void;
     private _walkRemove;
+    flat(): TreeNode[];
+    /**
+     * 广度优先遍历（BFS）：从当前节点开始，逐层遍历子树
+     * @param f - 节点处理函数，返回 false 表示停止遍历
+     * @param inChild - 决定是否将子节点加入队列：若提供，调用 inChild(node)，返回 true 才加入其子节点
+     */
+    walkBFS(f: (node: TreeNode) => boolean, inChild?: (node: TreeNode) => boolean): boolean;
     /**
      * 如果返回true则继续，包含当前节点
+     * 深度优先遍历(DFS)
      * @param f
-     * @param inChild
+     * @param inChild 如果返回true，则继续遍历， 否则不进入节点的子节点
      */
     walk(f: (node: TreeNode) => boolean, inChild?: (node: TreeNode) => boolean): void;
     private _walk;

package/dist/index.esm.js

@@ -2036,10 +2036,53 @@ class TreeNode {
         }
         return true;
     }
+    flat() {
+        let result = [];
+        this.walkBFS((node) => {
+            result.push(node);
+            return true;
+        });
+        return result;
+    }
+    /**
+     * 广度优先遍历（BFS）：从当前节点开始，逐层遍历子树
+     * @param f - 节点处理函数，返回 false 表示停止遍历
+     * @param inChild - 决定是否将子节点加入队列：若提供，调用 inChild(node)，返回 true 才加入其子节点
+     */
+    walkBFS(f, inChild) {
+        var _a;
+        const queue = [this]; // 初始化队列
+        while (queue.length > 0) {
+            const node = queue.shift(); // 取出队首节点
+            try {
+                // 1. 处理当前节点
+                if (!f(node)) {
+                    return false; // 停止遍历
+                }
+            }
+            catch (e) {
+                console.warn('walkBFS error:', e);
+                return false;
+            }
+            // 2. 决定是否展开子节点
+            const shouldEnqueueChildren = inChild ? inChild(node) : true;
+            if (!shouldEnqueueChildren || node.isLeaf) {
+                continue;
+            }
+            // 3. 将子节点加入队列（从左到右）
+            if ((_a = node.children) === null || _a === void 0 ? void 0 : _a.length) {
+                for (const child of node.children) {
+                    queue.push(child);
+                }
+            }
+        }
+        return true; // 遍历完成
+    }
     /**
      * 如果返回true则继续，包含当前节点
+     * 深度优先遍历(DFS)
      * @param f
-     * @param inChild
+     * @param inChild 如果返回true，则继续遍历， 否则不进入节点的子节点
      */
     walk(f, inChild) {
         this._walk(f, inChild);