alihan/crush-code-agent-ide
1
0
Fork 0
mirror of https://github.com/charmbracelet/crush.git synced 2025-08-02 05:20:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat: make "list" tool preview cleaner (#335)

Browse Source
This commit is contained in:
Kira Kawai
2025-07-30 23:07:06 +09:00
committed by GitHub
parent 10500051bf
commit 4ae839a023
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
internal/llm/tools/ls.go
View File

@@ -197,7 +197,7 @@ func ListDirectoryTree(searchPath string, ignore []string) (string, error) {
return "", fmt.Errorf("error listing directory: %w", err)
}
tree := createFileTree(files)
tree := createFileTree(files, searchPath)
output := printTree(tree, searchPath)
if truncated {
@@ -207,12 +207,13 @@ func ListDirectoryTree(searchPath string, ignore []string) (string, error) {
return output, nil
}
func createFileTree(sortedPaths []string) []*TreeNode {
func createFileTree(sortedPaths []string, rootPath string) []*TreeNode {
root := []*TreeNode{}
pathMap := make(map[string]*TreeNode)
for _, path := range sortedPaths {
parts := strings.Split(path, string(filepath.Separator))
relativePath := strings.TrimPrefix(path, rootPath)
parts := strings.Split(relativePath, string(filepath.Separator))
currentPath := ""
var parentPath string
@@ -241,7 +242,7 @@ func createFileTree(sortedPaths []string) []*TreeNode {
}
isLastPart := i == len(parts)-1
isDir := !isLastPart || strings.HasSuffix(path, string(filepath.Separator))
isDir := !isLastPart || strings.HasSuffix(relativePath, string(filepath.Separator))
nodeType := "file"
if isDir {
nodeType = "directory"
@@ -273,7 +274,12 @@ func createFileTree(sortedPaths []string) []*TreeNode {
func printTree(tree []*TreeNode, rootPath string) string {
var result strings.Builder
result.WriteString(fmt.Sprintf("- %s%s\n", rootPath, string(filepath.Separator)))
result.WriteString("- ")
result.WriteString(rootPath)
if rootPath[len(rootPath)-1] != '/' {
result.WriteByte(filepath.Separator)
}
result.WriteByte('\n')
for _, node := range tree {
printNode(&result, node, 1)