pacnpal/Roo-Code
1
0
Fork 0
mirror of https://github.com/pacnpal/Roo-Code.git synced 2025-12-20 12:21:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

Improve recursive file list sorting for best output

Browse Source
This commit is contained in:
Saoud Rizwan
2024-08-31 00:41:29 -04:00
parent adb000aed0
commit 5a05279a4d
1 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/ClaudeDev.ts
View File

@@ -1028,16 +1028,27 @@ export class ClaudeDev {
const relativePath = path.relative(absolutePath, file) const relativePath = path.relative(absolutePath, file)
return file.endsWith("/") ? relativePath + "/" : relativePath return file.endsWith("/") ? relativePath + "/" : relativePath
}) })
// Sort so files are listed under their respective directories to make it clear what files are children of what directories. Since we build file list top down, even if file list is truncated it will show directories that claude can then explore further.
.sort((a, b) => { .sort((a, b) => {
// sort directories before files const aParts = a.split("/")
const aIsDir = a.endsWith("/") const bParts = b.split("/")
const bIsDir = b.endsWith("/") for (let i = 0; i < Math.min(aParts.length, bParts.length); i++) {
if (aIsDir !== bIsDir) { if (aParts[i] !== bParts[i]) {
return aIsDir ? -1 : 1 // If one is a directory and the other isn't at this level, sort the directory first
if (i + 1 === aParts.length && i + 1 < bParts.length) {
return -1
}
if (i + 1 === bParts.length && i + 1 < aParts.length) {
return 1
}
// Otherwise, sort alphabetically
return aParts[i].localeCompare(bParts[i], undefined, { numeric: true, sensitivity: "base" })
}
} }
return a.localeCompare(b, undefined, { numeric: true, sensitivity: "base" }) // If all parts are the same up to the length of the shorter path,
// the shorter one comes first
return aParts.length - bParts.length
}) })
if (sorted.length >= LIST_FILES_LIMIT) { if (sorted.length >= LIST_FILES_LIMIT) {
const truncatedList = sorted.slice(0, LIST_FILES_LIMIT).join("\n") const truncatedList = sorted.slice(0, LIST_FILES_LIMIT).join("\n")
return `${truncatedList}\n\n(Truncated at ${LIST_FILES_LIMIT} results. Try listing files in subdirectories if you need to explore further.)` return `${truncatedList}\n\n(Truncated at ${LIST_FILES_LIMIT} results. Try listing files in subdirectories if you need to explore further.)`