AI: add output field

Author: simuleite

internal/cmd/cli/get_file_structure.go

@@ -67,6 +67,7 @@ Returns a list of functions, types, and variables defined in the file.`,
 				Name      string `json:"name"`
 				Line     int    `json:"line"`
 				Signature string `json:"signature,omitempty"`
+				TypeKind  string `json:"typeKind,omitempty"` // class, typedef, struct, enum, interface
 			}
 
 			var nodes []Node
@@ -78,6 +79,10 @@ Returns a list of functions, types, and variables defined in the file.`,
 				if sig, ok := sym["Signature"].(string); ok {
 					n.Signature = sig
 				}
+				// 添加 TypeKind (class, typedef, struct, enum, interface)
+				if tk, ok := sym["TypeKind"].(string); ok && tk != "" {
+					n.TypeKind = tk
+				}
 				nodes = append(nodes, n)
 			}
 

internal/cmd/cli/get_file_symbol.go

@@ -93,7 +93,8 @@ Returns the symbol's code, type, line number, and relationship with other symbol
 				// 通过 ModPath + PkgPath + Name 反向查找 FilePath
 				filePath := findSymbolFile(data, r["mod_path"], r["pkg_path"], r["name"])
 
-				if r["kind"] == "Dependency" {
+				// Dependency, Inherit, Implement 都视为依赖关系
+			if r["kind"] == "Dependency" || r["kind"] == "Inherit" || r["kind"] == "Implement" {
 					depMap[filePath] = append(depMap[filePath], r["name"])
 				} else {
 					refMap[filePath] = append(refMap[filePath], r["name"])

internal/cmd/cli/search_symbol.go

@@ -161,10 +161,19 @@ abcoder cli search_symbol myrepo "Graph" --path "src/main/java/com/uniast/parser
 					}
 					for name := range nameToLocs {
 						if matchName(name, query) {
+							// 支持两种格式：
+							// 1. 数组格式: {"GetUser": ["src/user.rs"]}
+							// 2. 对象格式: {"GetUser": {"Files": ["src/user.rs"]}}
 							locVal := nameToLocsVal.Get(name)
+							var files []interface{}
 							filesVal := locVal.Get("Files")
 							if filesVal.Exists() {
-								files, _ := filesVal.Array()
+								files, _ = filesVal.Array()
+							} else {
+								// 尝试数组格式
+								files, _ = locVal.Array()
+							}
+							if len(files) > 0 {
 								for _, f := range files {
 									fileStr, _ := f.(string)
 									// path 前缀过滤
@@ -184,17 +193,15 @@ abcoder cli search_symbol myrepo "Graph" --path "src/main/java/com/uniast/parser
 						}
 					}
 
-					// 如果有结果，直接返回
-					if len(results) > 0 {
-						output := SearchResult{
-							RepoName: repoName,
-							Query:    query,
-							Results:  results,
-						}
-						b, _ := json.MarshalIndent(output, "", "  ")
-						fmt.Fprintf(os.Stdout, "%s\n", b)
-						return nil
+					// 无论是否有结果都直接返回
+					output := SearchResult{
+						RepoName: repoName,
+						Query:    query,
+						Results:  results,
 					}
+					b, _ := json.MarshalIndent(output, "", "  ")
+					fmt.Fprintf(os.Stdout, "%s\n", b)
+					return nil
 				}
 			}