feat: enhance list_code_definition_names to support files

Fix 'cwd option must be a path to a directory' error.
Add support for analyzing individual source files.
Update tool description to clarify file and directory usage.

Signed-off-by: Eric Wheeler <[email protected]>

Author: Eric Wheeler

src/core/Cline.ts

@@ -2505,10 +2505,10 @@ export class Cline extends EventEmitter<ClineEvents> {
 						}
 					}
 					case "list_code_definition_names": {
-						const relDirPath: string | undefined = block.params.path
+						const relPath: string | undefined = block.params.path
 						const sharedMessageProps: ClineSayTool = {
 							tool: "listCodeDefinitionNames",
-							path: getReadablePath(this.cwd, removeClosingTag("path", relDirPath)),
+							path: getReadablePath(this.cwd, removeClosingTag("path", relPath)),
 						}
 						try {
 							if (block.partial) {
@@ -2519,19 +2519,35 @@ export class Cline extends EventEmitter<ClineEvents> {
 								await this.ask("tool", partialMessage, block.partial).catch(() => {})
 								break
 							} else {
-								if (!relDirPath) {
+								if (!relPath) {
 									this.consecutiveMistakeCount++
 									pushToolResult(
 										await this.sayAndCreateMissingParamError("list_code_definition_names", "path"),
 									)
 									break
 								}
 								this.consecutiveMistakeCount = 0
-								const absolutePath = path.resolve(this.cwd, relDirPath)
-								const result = await parseSourceCodeForDefinitionsTopLevel(
-									absolutePath,
-									this.rooIgnoreController,
-								)
+								const absolutePath = path.resolve(this.cwd, relPath)
+								let result: string
+								try {
+									const stats = await fs.stat(absolutePath)
+									if (stats.isFile()) {
+										const fileResult = await parseSourceCodeDefinitionsForFile(
+											absolutePath,
+											this.rooIgnoreController,
+										)
+										result = fileResult ?? "No source code definitions found in this file."
+									} else if (stats.isDirectory()) {
+										result = await parseSourceCodeForDefinitionsTopLevel(
+											absolutePath,
+											this.rooIgnoreController,
+										)
+									} else {
+										result = "The specified path is neither a file nor a directory."
+									}
+								} catch {
+									result = `${absolutePath}: does not exist or cannot be accessed.`
+								}
 								const completeMessage = JSON.stringify({
 									...sharedMessageProps,
 									content: result,

src/core/prompts/tools/list-code-definition-names.ts

@@ -2,16 +2,23 @@ import { ToolArgs } from "./types"
 
 export function getListCodeDefinitionNamesDescription(args: ToolArgs): string {
 	return `## list_code_definition_names
-Description: Request to list definition names (classes, functions, methods, etc.) used in source code files at the top level of the specified directory. This tool provides insights into the codebase structure and important constructs, encapsulating high-level concepts and relationships that are crucial for understanding the overall architecture.
+Description: Request to list definition names (classes, functions, methods, etc.) from source code. This tool can analyze either a single file or all files at the top level of a specified directory. It provides insights into the codebase structure and important constructs, encapsulating high-level concepts and relationships that are crucial for understanding the overall architecture.
 Parameters:
-- path: (required) The path of the directory (relative to the current working directory ${args.cwd}) to list top level source code definitions for.
+- path: (required) The path of the file or directory (relative to the current working directory ${args.cwd}) to analyze. When given a directory, it lists definitions from all top-level source files.
 Usage:
 <list_code_definition_names>
 <path>Directory path here</path>
 </list_code_definition_names>
 
-Example: Requesting to list all top level source code definitions in the current directory
+Examples:
+
+1. List definitions from a specific file:
+<list_code_definition_names>
+<path>src/main.ts</path>
+</list_code_definition_names>
+
+2. List definitions from all files in a directory:
 <list_code_definition_names>
-<path>.</path>
+<path>src/</path>
 </list_code_definition_names>`
 }