0.3.4 小图标功能增加

wolai-frontend/convex/_utils/documentTree.ts

@@ -0,0 +1,44 @@
+export type ParentLinkedRow = {
+  id: string;
+  parent_id?: string | null;
+};
+
+/**
+ * 收集以 rootId 为根的整棵子树（包含根节点本身）。
+ *
+ * 说明：
+ * - 仅依赖 (id, parent_id) 字段，便于在 Convex 与前端共用同一套遍历逻辑。
+ * - 会做去重与环检测，避免异常数据导致死循环。
+ */
+export function collectSubtree<T extends ParentLinkedRow>(rows: readonly T[], rootId: string): T[] {
+  const byParentId = new Map<string | null, T[]>();
+  for (const row of rows) {
+    const parentId = (row.parent_id ?? null) as string | null;
+    const bucket = byParentId.get(parentId);
+    if (bucket) bucket.push(row);
+    else byParentId.set(parentId, [row]);
+  }
+
+  const root = rows.find((r) => r.id === rootId);
+  if (!root) return [];
+
+  const visited = new Set<string>();
+  const stack: T[] = [root];
+  const result: T[] = [];
+
+  while (stack.length > 0) {
+    const current = stack.pop()!;
+    if (visited.has(current.id)) continue;
+    visited.add(current.id);
+    result.push(current);
+
+    const children = byParentId.get(current.id) ?? [];
+    // 倒序入栈，保持更接近“原列表顺序”的遍历结果（不影响正确性）
+    for (let i = children.length - 1; i >= 0; i -= 1) {
+      stack.push(children[i]!);
+    }
+  }
+
+  return result;
+}
+