Function buildImportMap

codegen/import_cleanup.go:24–58

buildImportMap creates a map of package names to import information. It handles standard imports, named imports, and excludes blank/dot imports.

(file *ast.File)

// buildImportMap creates a map of package names to import information.
// It handles standard imports, named imports, and excludes blank/dot imports.
func buildImportMap(file *ast.File) map[string]*importInfo {
	imports := make(map[string]*importInfo)

	for _, impDecl := range file.Imports {
		path := strings.Trim(impDecl.Path.Value, `"`)

		// Handle imports with explicit names
		if impDecl.Name != nil {
			switch impDecl.Name.Name {
			case "_", ".":
				// Blank imports (for side effects) and dot imports are always kept
				continue
			default:
				// Named import: use the alias as the local name
				imports[impDecl.Name.Name] = &importInfo{
					spec:      impDecl,
					path:      path,
					localName: impDecl.Name.Name,
					used:      false,
				}
			}
		} else {
			// Standard import: infer package name from path
			localName := inferPackageName(path)
			imports[localName] = &importInfo{
				spec:      impDecl,
				path:      path,
				localName: localName,
				used:      false,
			}
		}
	}

	return imports
}

// detectUsedImports performs a single AST walk to mark which imports are used.
// It looks for qualified identifiers (pkg.Name).