feat: store file path

Author: AsterDY

src/compress/golang/plugin/parse/parser.go

@@ -139,6 +139,17 @@ func (p *GoParser) ParseRepo() (Repository, error) {
 }
 
 func (p *GoParser) ParseModule(mod *Module, dir string) (err error) {
+	filepath.Walk(dir, func(path string, info fs.FileInfo, e error) error {
+		if info != nil && info.IsDir() && filepath.Base(path) == ".git" {
+			return filepath.SkipDir
+		}
+		if e != nil || info.IsDir() {
+			return nil
+		}
+		rel, _ := filepath.Rel(p.homePageDir, path)
+		mod.Files[rel] = NewFile(path)
+		return nil
+	})
 	return p.loadPackages(mod, dir, "./...")
 }
 

src/compress/golang/plugin/parse/pkg.go

@@ -200,8 +200,11 @@ func (p *GoParser) loadPackages(mod *Module, dir string, pkgPath PkgPath) (err e
 			}
 			ctx.imports = imports
 			relpath, _ := filepath.Rel(p.homePageDir, filePath)
-			f := NewFile(relpath)
-			mod.Files[relpath] = f
+			f := mod.Files[relpath]
+			if f == nil {
+				f = NewFile(relpath)
+				mod.Files[relpath] = f
+			}
 			f.Imports = imports.Origins
 			if err := p.parseFile(ctx, file); err != nil {
 				return err

src/lang/collect/export.go

@@ -141,7 +141,7 @@ func (c *Collector) exportSymbol(repo *uniast.Repository, symbol *DocumentSymbol
 	fileLine := c.fileLine(symbol.Location)
 	// collect files
 	if module.Files[relfile] == nil {
-		module.Files[relfile] = uniast.NewFile(relfile)
+		module.Files[relfile] = uniast.NewFile(file)
 	}
 
 	content := symbol.Text

src/lang/rust/patch.go

@@ -12,22 +12,6 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-/**
- * Copyright 2024 ByteDance Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     https://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
 package rust
 
 import (
@@ -48,20 +32,22 @@ func (p *RustModulePatcher) Patch(ast *uniast.Module) {
 			return err
 		}
 		if info.IsDir() {
+			if filepath.Base(path) == "target" || filepath.Base(path) == ".git" {
+				return filepath.SkipDir
+			}
 			return nil
 		}
-		if filepath.Ext(path) != ".rs" {
-			return nil
-		}
-		// 找到对应文件package并存放
 		relpath, err := filepath.Rel(p.Root, path)
 		if err != nil {
 			log.Error("get relative path of %s failed: %v", path, err)
 			return nil
 		}
 		file := ast.Files[relpath]
 		if file == nil {
-			log.Error("get file %s failed", relpath)
+			file = uniast.NewFile(path)
+			ast.Files[relpath] = file
+		}
+		if filepath.Ext(path) != ".rs" {
 			return nil
 		}
 		// 解析use语句

src/uniast/ast.go

@@ -49,11 +49,14 @@ func NewRepository(name string) Repository {
 type File struct {
 	Name    string
 	Imports []string
+	Path    string
 }
 
 func NewFile(path string) *File {
+	abs, _ := filepath.Abs(path)
 	ret := File{
 		Name: filepath.Base(path),
+		Path: abs,
 	}
 	return &ret
 }