Function discoverTypesFromFile

go/fory/codegen/parser.go:32–67

discoverTypesFromFile scans the source file for //fory:generate comments

(pkg *packages.Package, sourceFile string)

// discoverTypesFromFile scans the source file for //fory:generate comments
func discoverTypesFromFile(pkg *packages.Package, sourceFile string) ([]string, error) {
	var discoveredTypes []string

	// Find the syntax tree for the specific file
	for _, file := range pkg.Syntax {
		// Get the file's position and convert to absolute path for comparison
		filename := pkg.Fset.Position(file.Pos()).Filename
		absFilename, err := filepath.Abs(filename)
		if err != nil {
			continue
		}

		if absFilename != sourceFile {
			continue
		}

		// Scan for type declarations with //fory:generate comments
		for _, decl := range file.Decls {
			if genDecl, ok := decl.(*ast.GenDecl); ok && genDecl.Tok == token.TYPE {
				for _, spec := range genDecl.Specs {
					if typeSpec, ok := spec.(*ast.TypeSpec); ok {
						// Check if it's a struct type
						if _, ok := typeSpec.Type.(*ast.StructType); ok {
							// Look for //fory:generate comment
							if hasGenerateComment(genDecl.Doc) || hasGenerateComment(typeSpec.Doc) {
								discoveredTypes = append(discoveredTypes, typeSpec.Name.Name)
							}
						}
					}
				}
			}
		}
	}

	return discoveredTypes, nil
}

// hasGenerateComment checks if comment group contains //fory:generate
func hasGenerateComment(commentGroup *ast.CommentGroup) bool {