@mintlify/cli 4.0.1434 → 4.0.1438

package/src/mintTestTree.ts

@@ -0,0 +1,134 @@
+import type { DocsPage } from './agent-harness/index.js';
+
+export type ScopeSelectionState = 'all' | 'some' | 'none';
+
+export interface ScopeTreeRow {
+  id: string;
+  kind: 'directory' | 'file';
+  label: string;
+  depth: number;
+  file?: string;
+  blockCount: number;
+  selectableFiles: string[];
+  selection: ScopeSelectionState;
+  expanded?: boolean;
+}
+
+interface TreeNode {
+  id: string;
+  kind: 'directory' | 'file';
+  label: string;
+  file?: string;
+  blockCount: number;
+  children: TreeNode[];
+}
+
+function buildTree(pages: DocsPage[]): TreeNode {
+  const root: TreeNode = { id: 'dir:', kind: 'directory', label: '.', blockCount: 0, children: [] };
+  const directories = new Map<string, TreeNode>([['', root]]);
+
+  for (const page of pages) {
+    const parts = page.file.split('/');
+    let parent = root;
+    let directoryPath = '';
+    for (const part of parts.slice(0, -1)) {
+      directoryPath = directoryPath ? `${directoryPath}/${part}` : part;
+      let directory = directories.get(directoryPath);
+      if (!directory) {
+        directory = {
+          id: `dir:${directoryPath}`,
+          kind: 'directory',
+          label: part,
+          blockCount: 0,
+          children: [],
+        };
+        directories.set(directoryPath, directory);
+        parent.children.push(directory);
+      }
+      parent = directory;
+    }
+
+    parent.children.push({
+      id: `file:${page.file}`,
+      kind: 'file',
+      label: parts[parts.length - 1] ?? page.file,
+      file: page.file,
+      blockCount: page.blockCount,
+      children: [],
+    });
+  }
+
+  const sumBlocks = (node: TreeNode): number => {
+    if (node.kind === 'file') return node.blockCount;
+    node.blockCount = node.children.reduce((total, child) => total + sumBlocks(child), 0);
+    return node.blockCount;
+  };
+  sumBlocks(root);
+
+  const sortChildren = (node: TreeNode): void => {
+    node.children.sort((left, right) => {
+      if (left.kind !== right.kind) return left.kind === 'directory' ? -1 : 1;
+      return left.label.localeCompare(right.label);
+    });
+    for (const child of node.children) sortChildren(child);
+  };
+  sortChildren(root);
+  return root;
+}
+
+function descendantFiles(node: TreeNode): string[] {
+  if (node.kind === 'file') return node.file ? [node.file] : [];
+  return node.children.flatMap(descendantFiles);
+}
+
+function selectionState(files: string[], selectedFiles: ReadonlySet<string>): ScopeSelectionState {
+  if (files.length === 0) return 'none';
+  const selectedCount = files.filter((file) => selectedFiles.has(file)).length;
+  if (selectedCount === 0) return 'none';
+  return selectedCount === files.length ? 'all' : 'some';
+}
+
+export function visibleScopeRows(
+  pages: DocsPage[],
+  expandedDirectories: ReadonlySet<string>,
+  selectedFiles: ReadonlySet<string>
+): ScopeTreeRow[] {
+  const root = buildTree(pages);
+  const rows: ScopeTreeRow[] = [];
+
+  const visit = (node: TreeNode, depth: number) => {
+    const files = descendantFiles(node);
+    const expanded = node.kind === 'directory' ? expandedDirectories.has(node.id) : undefined;
+    rows.push({
+      id: node.id,
+      kind: node.kind,
+      label: node.label,
+      depth,
+      ...(node.file ? { file: node.file } : {}),
+      blockCount: node.blockCount,
+      selectableFiles: files,
+      selection: selectionState(files, selectedFiles),
+      ...(expanded !== undefined ? { expanded } : {}),
+    });
+    if (node.kind === 'directory' && expanded) {
+      for (const child of node.children) visit(child, depth + 1);
+    }
+  };
+
+  for (const child of root.children) visit(child, 0);
+  return rows;
+}
+
+export function toggleScopeSelection(
+  selectedFiles: ReadonlySet<string>,
+  row: ScopeTreeRow
+): Set<string> {
+  const next = new Set(selectedFiles);
+  if (row.selectableFiles.length === 0) return next;
+  const shouldSelect = row.selectableFiles.some((file) => !next.has(file));
+  for (const file of row.selectableFiles) {
+    if (shouldSelect) next.add(file);
+    else next.delete(file);
+  }
+  return next;
+}