Function buildHierarchy

src/tools/semantic-navigate.ts:185–227

(files: FileInfo[], vectors: number[][], maxClusters: number, depth: number, maxDepth: number)

}

async function buildHierarchy(files: FileInfo[], vectors: number[][], maxClusters: number, depth: number, maxDepth: number): Promise<ClusterNode> {
  if (files.length <= MAX_FILES_PER_LEAF || depth >= maxDepth) {
    return {
      label: "",
      pathPattern: findPathPattern(files.map((f) => f.relativePath)),
      files,
      children: [],
    };
  }

  const clusterResults = spectralCluster(vectors, maxClusters);

  if (clusterResults.length <= 1) {
    return {
      label: "",
      pathPattern: findPathPattern(files.map((f) => f.relativePath)),
      files,
      children: [],
    };
  }

  const childMetas = clusterResults.map((cluster) => ({
    files: cluster.indices.map((i) => files[i]),
    vectors: cluster.indices.map((i) => vectors[i]),
    pathPattern: findPathPattern(cluster.indices.map((i) => files[i].relativePath)),
  }));

  const labels = await labelSiblingClusters(childMetas.map((c) => ({ files: c.files, pathPattern: c.pathPattern })));

  const children: ClusterNode[] = [];
  for (let i = 0; i < childMetas.length; i++) {
    const child = await buildHierarchy(childMetas[i].files, childMetas[i].vectors, maxClusters, depth + 1, maxDepth);
    child.label = labels[i];
    children.push(child);
  }

  return {
    label: "",
    pathPattern: findPathPattern(files.map((f) => f.relativePath)),
    files: [],
    children,
  };
}

function renderClusterTree(node: ClusterNode, indent: number = 0): string {
  const pad = "  ".repeat(indent);