Function parseStructsFromPackage

go/fory/codegen/parser.go:122–177

parseStructsFromPackage finds and parses structs from a package

(pkg *packages.Package, targetTypes []string)

// parseStructsFromPackage finds and parses structs from a package
func parseStructsFromPackage(pkg *packages.Package, targetTypes []string) ([]*StructInfo, error) {
	var structs []*StructInfo

	// Check if package has types
	if pkg.Types == nil {
		return nil, fmt.Errorf("package %s has no type information", pkg.PkgPath)
	}

	// Iterate through all types in the package
	scope := pkg.Types.Scope()
	allNames := scope.Names()

	for _, name := range allNames {
		obj := scope.Lookup(name)
		if obj == nil {
			continue
		}

		// Check if it's a named type
		named, ok := obj.Type().(*types.Named)
		if !ok {
			continue
		}

		// Check if underlying type is struct
		structType, ok := named.Underlying().(*types.Struct)
		if !ok {
			continue
		}

		// Check if we should generate code for this type
		shouldGenerate := false
		if len(targetTypes) > 0 {
			for _, t := range targetTypes {
				if strings.TrimSpace(t) == name {
					shouldGenerate = true
					break
				}
			}
		}

		if !shouldGenerate {
			continue
		}

		// Extract struct information
		structInfo, err := extractStructInfo(name, structType)
		if err != nil {
			return nil, fmt.Errorf("extracting struct info for %s: %w", name, err)
		}

		structs = append(structs, structInfo)
	}

	return structs, nil
}