feat:(go) support mock identity for unloaded symobols

Author: AsterDY

lang/golang/parser/ctx.go

@@ -252,10 +252,41 @@ func (ctx *fileContext) GetTypeId(typ ast.Expr) (x Identity, isPointer bool, isS
 	if tinfo, ok := ctx.pkgTypeInfo.Types[typ]; ok {
 		return ctx.getIdFromType(tinfo.Type)
 	} else {
-		panic("cannot find type info for " + string(ctx.GetRawContent(typ)))
+		// NOTICE: for unloaded type, we only mock the type name
+		fmt.Fprintf(os.Stderr, "cannot find type info for %s\n", ctx.GetRawContent(typ))
+		return ctx.mockType(typ)
 	}
 }
 
+func (ctx *fileContext) mockType(typ ast.Expr) (x Identity, isPointer bool, isStdOrBuiltin bool) {
+	switch ty := typ.(type) {
+	case *ast.StarExpr:
+		id, _, std := ctx.mockType(ty.X)
+		return id, true, std
+	case *ast.CallExpr:
+		// try get func type
+		id, _, std := ctx.mockType(ty.Fun)
+		return id, false, std
+	case *ast.SelectorExpr:
+		// try get import path
+		switch xx := ty.X.(type) {
+		case *ast.Ident:
+			impt, mod, err := ctx.imports.GetImportPath(xx.Name, "")
+			if err != nil {
+				goto fallback
+			}
+			return NewIdentity(mod, PkgPath(impt), ty.Sel.Name), false, false
+		case *ast.SelectorExpr:
+			// recurse
+			id, _, std := ctx.mockType(xx)
+			return NewIdentity(id.ModPath, id.PkgPath, ty.Sel.Name), false, std
+		}
+	}
+
+fallback:
+	return NewIdentity("UNLOADED", ctx.pkgPath, string(ctx.GetRawContent(typ))), false, true
+}
+
 func (ctx *fileContext) collectFields(fields []*ast.Field, m *[]Dependency) {
 	for _, fieldDecl := range fields {
 		id, _, isStdOrBuiltin := ctx.GetTypeId(fieldDecl.Type)

lang/golang/parser/parser.go

@@ -143,16 +143,16 @@ 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 {
-		// run go mod tidy before parse
-		cmd := exec.Command("go", "mod", "tidy")
-		cmd.Dir = dir
-		buf := bytes.NewBuffer(nil)
-		cmd.Stderr = buf
-		if err := cmd.Run(); err != nil {
-			return fmt.Errorf("run go mod tidy failed in %s: %v", dir, buf.String())
-		}
+	// run go mod tidy before parse
+	cmd := exec.Command("go", "mod", "tidy")
+	cmd.Dir = dir
+	buf := bytes.NewBuffer(nil)
+	cmd.Stderr = buf
+	if err := cmd.Run(); err != nil {
+		fmt.Fprintf(os.Stderr, "run go mod tidy failed in %s: %v\n", dir, buf.String())
+	}
 
+	filepath.Walk(dir, func(path string, info fs.FileInfo, e error) error {
 		if info != nil && info.IsDir() && filepath.Base(path) == ".git" {
 			return filepath.SkipDir
 		}
@@ -163,6 +163,7 @@ func (p *GoParser) ParseModule(mod *Module, dir string) (err error) {
 		mod.Files[rel] = NewFile(rel)
 		return nil
 	})
+
 	return p.loadPackages(mod, dir, "./...")
 }
 

lang/uniast/ast.go

@@ -279,7 +279,7 @@ type Identity struct {
 
 func NewIdentity(mod, pkg, name string) Identity {
 	if mod == "" {
-		fmt.Fprintf(os.Stderr, "module name cannot be empty: %s.%s\n", pkg, name)
+		fmt.Fprintf(os.Stderr, "module name cannot be empty: %s#%s\n", pkg, name)
 		// panic(fmt.Sprintf("module name cannot be empty: %s.%s", pkg, name))
 	}
 	return Identity{ModPath: mod, PkgPath: pkg, Name: name}