Repomix

import nodepath from 'node:path'; export interface TreeNode { name: string; children: TreeNode[]; isDirectory: boolean; } const createTreeNode = (name: string, isDirectory: boolean): TreeNode => ({ name, children: [], isDirectory }); export const generateFileTree = (files: string[], emptyDirPaths: string[] = []): TreeNode => { const root: TreeNode = createTreeNode('root', true); for (const file of files) { addPathToTree(root, file, false); } // Add empty directories for (const dir of emptyDirPaths) { addPathToTree(root, dir, true); } return root; }; const addPathToTree = (root: TreeNode, path: string, isDirectory: boolean): void => { const parts = path.split(nodepath.sep); let currentNode = root; for (let i = 0; i < parts.length; i++) { const part = parts[i]; const isLastPart = i === parts.length - 1; let child = currentNode.children.find((c) => c.name === part); if (!child) { child = createTreeNode(part, !isLastPart || isDirectory); currentNode.children.push(child); } currentNode = child; } }; const sortTreeNodes = (node: TreeNode) => { node.children.sort((a, b) => { if (a.isDirectory === b.isDirectory) { return a.name.localeCompare(b.name); } return a.isDirectory ? -1 : 1; }); for (const child of node.children) { sortTreeNodes(child); } }; export const treeToString = (node: TreeNode, prefix = ''): string => { sortTreeNodes(node); let result = ''; for (const child of node.children) { result += `${prefix}${child.name}${child.isDirectory ? '/' : ''}\n`; if (child.isDirectory) { result += treeToString(child, `${prefix} `); } } return result; }; /** * Converts a tree to string with line counts for files. * @param node The tree node to convert * @param lineCounts Map of file paths to line counts * @param prefix Current indentation prefix * @param currentPath Current path being built (for looking up line counts) */ export const treeToStringWithLineCounts = ( node: TreeNode, lineCounts: Record<string, number>, prefix = '', currentPath = '', ): string => { sortTreeNodes(node); let result = ''; for (const child of node.children) { const childPath = currentPath ? `${currentPath}/${child.name}` : child.name; if (child.isDirectory) { result += `${prefix}${child.name}/\n`; result += treeToStringWithLineCounts(child, lineCounts, `${prefix} `, childPath); } else { const lineCount = lineCounts[childPath]; const lineCountSuffix = lineCount !== undefined ? ` (${lineCount} lines)` : ''; result += `${prefix}${child.name}${lineCountSuffix}\n`; } } return result; }; export const generateTreeString = (files: string[], emptyDirPaths: string[] = []): string => { const tree = generateFileTree(files, emptyDirPaths); return treeToString(tree).trim(); }; export const generateTreeStringWithLineCounts = ( files: string[], lineCounts: Record<string, number>, emptyDirPaths: string[] = [], ): string => { const tree = generateFileTree(files, emptyDirPaths); return treeToStringWithLineCounts(tree, lineCounts).trim(); };