Function collectCode

src/wiki-engine/code-knowledge/code-collector.ts:50–106

(options: CollectCodeOptions)

}

export async function collectCode(options: CollectCodeOptions): Promise<{ manifest: CodeCollectionManifest; files: CodeCollectedFile[] }> {
  const root = path.resolve(options.root);
  const filePaths: string[] = [];
  await walk(root, filePaths, options.includeTests ?? false);

  // Sort: key files first, then by directory depth (shallow first)
  let filtered = filePaths.sort((a, b) => {
    const relA = toPosix(path.relative(root, a));
    const relB = toPosix(path.relative(root, b));
    const langA = languageFor(a);
    const langB = languageFor(b);
    const keyA = isKeyFile(relA, langA) ? 0 : 1;
    const keyB = isKeyFile(relB, langB) ? 0 : 1;
    if (keyA !== keyB) return keyA - keyB;
    const depthA = relA.split('/').length;
    const depthB = relB.split('/').length;
    if (depthA !== depthB) return depthA - depthB;
    return relA.localeCompare(relB);
  });

  // Filter to only changed files if specified
  if (options.changedFiles && options.changedFiles.length > 0) {
    const changedSet = new Set(options.changedFiles.map((f) => toPosix(f)));
    filtered = filtered.filter((fp) => {
      const relativePath = toPosix(path.relative(root, fp));
      return changedSet.has(relativePath);
    });
  }

  const limited = filtered.slice(0, options.maxFiles ?? 200);
  const files: CodeCollectedFile[] = [];

  for (const filePath of limited) {
    const content = await readFile(filePath, "utf8");
    const relativePath = toPosix(path.relative(root, filePath));
    const language = languageFor(filePath);
    files.push({
      path: filePath,
      relativePath,
      language,
      sha256: createHash("sha256").update(content).digest("hex"),
      content,
      isKeyFile: isKeyFile(relativePath, language)
    });
  }

  return {
    manifest: {
      schemaVersion: "team-wiki.code-collection.v1",
      root,
      commit: await gitCommit(root),
      collectedAt: new Date().toISOString(),
      files: files.map(({ content: _content, ...file }) => file)
    },
    files
  };
}