Add UniAST Index for search_symbol & get_file_structure

Author: simuleite

lang/golang/parser/parser.go

@@ -295,6 +295,7 @@ func (p *GoParser) ParseRepo() (Repository, error) {
 	}
 	p.associateStructWithMethods()
 	p.associateImplements()
+	p.buildNameToLocations()
 	fmt.Fprintf(os.Stderr, "total call packages.Load %d times\n", loadCount)
 	return p.getRepo(), nil
 }
@@ -362,6 +363,74 @@ func (p *GoParser) getRepo() Repository {
 	return p.repo
 }
 
+// buildNameToLocations 构建 name → files 反向索引
+// 解析完成后调用，一次遍历 Package 填充
+func (p *GoParser) buildNameToLocations() {
+	if p.repo.NameToLocations == nil {
+		p.repo.NameToLocations = make(map[string]NameLocations)
+	}
+
+	for _, mod := range p.repo.Modules {
+		for _, pkg := range mod.Packages {
+			// Functions
+			for name, fn := range pkg.Functions {
+				if fn.File == "" {
+					continue
+				}
+				loc := p.repo.NameToLocations[name]
+				// 去重
+				exists := false
+				for _, f := range loc.Files {
+					if f == fn.File {
+						exists = true
+						break
+					}
+				}
+				if !exists {
+					loc.Files = append(loc.Files, fn.File)
+				}
+				p.repo.NameToLocations[name] = loc
+			}
+			// Types
+			for name, t := range pkg.Types {
+				if t.FileLine.File == "" {
+					continue
+				}
+				loc := p.repo.NameToLocations[name]
+				exists := false
+				for _, f := range loc.Files {
+					if f == t.FileLine.File {
+						exists = true
+						break
+					}
+				}
+				if !exists {
+					loc.Files = append(loc.Files, t.FileLine.File)
+				}
+				p.repo.NameToLocations[name] = loc
+			}
+			// Vars
+			for name, v := range pkg.Vars {
+				if v.FileLine.File == "" {
+					continue
+				}
+				loc := p.repo.NameToLocations[name]
+				exists := false
+				for _, f := range loc.Files {
+					if f == v.FileLine.File {
+						exists = true
+						break
+					}
+				}
+				if !exists {
+					loc.Files = append(loc.Files, v.FileLine.File)
+				}
+				p.repo.NameToLocations[name] = loc
+			}
+		}
+	}
+}
+
 // ToABS converts a local package path to absolute path
 // If the path is not a local package, return empty string
 // func (p *goParser) pkgPathToABS(path PkgPath) string {

lang/golang/parser/pkg.go

@@ -276,6 +276,9 @@ func (p *GoParser) loadPackages(mod *Module, dir string, pkgPath PkgPath) (err e
 			if f.Package == "" {
 				f.Package = pkg.ID
 				f.Imports = imports.Origins
+				// [新增] 填充 ModPath 和 PkgPath
+				f.ModPath = mod.Name
+				f.PkgPath = pkg.ID
 			}
 			if err := p.parseFile(ctx, file); err != nil {
 				return err

lang/uniast/ast.go

@@ -95,6 +95,16 @@ type Repository struct {
 	Path        string             // repo absolute path
 	Modules     map[string]*Module // module name => module
 	Graph       NodeGraph          // node id => node
+
+	// [新增] name → files 反向索引
+	// 加速 search_symbol API，无需独立 .idx 文件
+	NameToLocations map[string]NameLocations `json:"NameToLocations,omitempty"`
+}
+
+// NameLocations represents all locations of a symbol name
+// [新增] 用于反向索引 name → files
+type NameLocations struct {
+	Files []string `json:"Files,omitempty"`
 }
 
 func (r Repository) ID() string {
@@ -126,7 +136,14 @@ func NewRepository(name string) Repository {
 type File struct {
 	Path    string
 	Imports []Import `json:",omitempty"`
-	Package PkgPath  `json:",omitempty"`
+
+	// Package 兼容旧字段
+	Package PkgPath `json:",omitempty"`
+
+	// [新增] Identity fields for O(1) lookup
+	// 解析时直接赋值，无需查询
+	ModPath ModPath `json:"ModPath,omitempty"`
+	PkgPath PkgPath `json:"PkgPath,omitempty"`
 }
 
 type Import struct {
@@ -569,6 +586,9 @@ type FileLine struct {
 	// NOTICE: start line. line number start from 1
 	Line int
 
+	// [新增] end line number (1-based)
+	EndLine int `json:"EndLine,omitempty"`
+
 	// start offset in file
 	StartOffset int