Remove unused methods

Signed-off-by: Guillaume Tardif <guillaume.tardif@gmail.com>

Author: gtardif

pkg/fsx/collect.go

@@ -58,7 +58,7 @@ func CollectFiles(ctx context.Context, paths []string, shouldIgnore func(path st
 
 			if info.IsDir() {
 				// Use DirectoryTree to collect files from directory
-				tree, err := DirectoryTreeWithContext(ctx, normalized, func(string) error { return nil }, shouldIgnore, 0)
+				tree, err := DirectoryTree(ctx, normalized, func(string) error { return nil }, shouldIgnore, 0)
 				if err != nil {
 					return nil, fmt.Errorf("failed to read directory %s: %w", normalized, err)
 				}

pkg/fsx/collect_cancellation_test.go

@@ -70,7 +70,7 @@ func TestDirectoryTree_ContextCancellation(t *testing.T) {
 		// Cancel context immediately
 		cancel()
 
-		_, err := DirectoryTreeWithContext(ctx, tmpDir, func(string) error { return nil }, nil, 0)
+		_, err := DirectoryTree(ctx, tmpDir, func(string) error { return nil }, nil, 0)
 		assert.ErrorIs(t, err, context.Canceled)
 	})
 
@@ -81,7 +81,7 @@ func TestDirectoryTree_ContextCancellation(t *testing.T) {
 		// Give time for timeout to trigger
 		time.Sleep(10 * time.Millisecond)
 
-		_, err := DirectoryTreeWithContext(ctx, tmpDir, func(string) error { return nil }, nil, 0)
+		_, err := DirectoryTree(ctx, tmpDir, func(string) error { return nil }, nil, 0)
 		assert.ErrorIs(t, err, context.DeadlineExceeded)
 	})
 }

pkg/fsx/fs.go

@@ -15,13 +15,7 @@ type TreeNode struct {
 	Children []*TreeNode `json:"children,omitempty"`
 }
 
-func DirectoryTree(path string, isPathAllowed func(string) error, shouldIgnore func(string) bool, maxItems int) (*TreeNode, error) {
-	itemCount := 0
-	return directoryTree(context.Background(), path, isPathAllowed, shouldIgnore, maxItems, &itemCount)
-}
-
-// DirectoryTreeWithContext is a context-aware version of DirectoryTree.
-func DirectoryTreeWithContext(ctx context.Context, path string, isPathAllowed func(string) error, shouldIgnore func(string) bool, maxItems int) (*TreeNode, error) {
+func DirectoryTree(ctx context.Context, path string, isPathAllowed func(string) error, shouldIgnore func(string) bool, maxItems int) (*TreeNode, error) {
 	itemCount := 0
 	return directoryTree(ctx, path, isPathAllowed, shouldIgnore, maxItems, &itemCount)
 }
@@ -87,17 +81,6 @@ func directoryTree(ctx context.Context, path string, isPathAllowed func(string)
 	return node, nil
 }
 
-func ListDirectory(path string, shouldIgnore func(string) bool) ([]string, error) {
-	tree, err := DirectoryTree(path, func(string) error { return nil }, shouldIgnore, 0)
-	if err != nil {
-		return nil, err
-	}
-
-	var files []string
-	CollectFilesFromTree(tree, "", &files)
-	return files, nil
-}
-
 // CollectFilesFromTree recursively collects file paths from a DirectoryTree.
 // Pass basePath="" for relative paths, or a parent directory for absolute paths.
 func CollectFilesFromTree(node *TreeNode, basePath string, files *[]string) {

pkg/tools/builtin/filesystem.go

@@ -391,7 +391,7 @@ func (t *FilesystemTool) handleDirectoryTree(ctx context.Context, args Directory
 		return nil
 	}
 
-	tree, err := fsx.DirectoryTreeWithContext(ctx, resolvedPath, isPathAllowed, t.shouldIgnorePath, maxFiles)
+	tree, err := fsx.DirectoryTree(ctx, resolvedPath, isPathAllowed, t.shouldIgnorePath, maxFiles)
 	if err != nil {
 		return tools.ResultError(fmt.Sprintf("Error building directory tree: %s", err)), nil
 	}