Go AST 怎样列出导入路径及显式别名

我在做 Go 源码索引时，最先想要的通常不是完整 AST，而是每个 import 到底引入了什么、在文件里用了哪个名字。这个问题不需要类型检查：直接读取 ast.File.Imports，对每个 *ast.ImportSpec 取 Path.Value，再看 Name 是否为空即可。

核心规则是：Path 保存带引号的导入路径，Name 只在显式别名、点导入或下划线导入时出现；默认导入时 Name == nil。要把路径变成普通字符串，再用 strconv.Unquote 解码。

先把导入声明拆成两个字段

Go AST 的 ast.File 有一个 Imports 字段，类型是 []*ast.ImportSpec。它已经把单行导入和括号包裹的多行导入统一成切片，因此列清单时不必先区分两种语法。

ImportSpec.Path 是 *ast.BasicLit，例如 "fmt" 在 AST 中仍然带着引号；ImportSpec.Name 是 *ast.Ident，普通别名时是 myfmt，点导入时是 .，下划线导入时是 _。没有显式名称时它为 nil。

解析并列出路径与别名

下面这段代码把源码解析成 AST，再输出路径、名称类别和源码位置。这里没有调用类型检查器，所以它回答的是“源文件写了哪些导入声明”，不是“最终依赖图里有哪些模块”。

package main

import (
    "fmt"       // 用于打印导入清单
    "go/ast"    // 访问 ImportSpec 节点
    "go/parser" // 把源码解析成 AST
    "go/token"  // 把节点位置还原成行列号
    "strconv"   // 去掉 Path.Value 外层的 Go 字符串引号
)

func main() {
    src := `package demo

import (
    "fmt"
    myfmt "fmt"
    . "strings"
    _ "net/http/pprof"
)`

    fset := token.NewFileSet()
    file, err := parser.ParseFile(fset, "demo.go", src, parser.SkipObjectResolution)
    if err != nil {
        // 源码不完整时先返回解析错误，不把残缺 AST 当成完整清单。
        fmt.Println("parse error:", err)
        return
    }

    for _, spec := range file.Imports {
        path, err := strconv.Unquote(spec.Path.Value)
        if err != nil {
            // Path 是 BasicLit；解码失败时保留错误边界，避免输出错误路径。
            fmt.Println("path error:", err)
            continue
        }

        alias := "(default)"
        if spec.Name != nil {
            // Name 不为空表示源码显式写了别名、点导入或下划线导入。
            alias = spec.Name.Name
        }

        pos := fset.Position(spec.Pos())
        fmt.Printf("%s:%d:%d path=%q name=%s\\n",
            pos.Filename, pos.Line, pos.Column, path, alias)
    }
}

输出字段可以按自己的索引模型改名。对源代码工具来说，建议保留三类信息：原始路径、显式名称和 FileSet 映射出的文件位置。这样后续展示“哪个文件的哪一行引入了什么”时，不需要再次扫描文本。

别把默认导入和特殊名称混为一谈

我第一次写这类索引时，把空名称和下划线名称都当成“没有别名”，结果丢失了导入语义。实际可以按下面的方式处理：

因此不要用 alias == "" 来判断所有情况，而要先判断指针是否为 nil。只有 Name == nil 才代表源码没有写显式名称；一旦非空，就应原样保留 Name.Name。

把位置和输入边界留在报告里

parser.ParseFile 负责产生语法树，token.FileSet 负责把 token.Pos 解释为文件名、行号和列号。两者最好从一开始就一起创建并传给解析器；否则后面只能得到难以直接展示的抽象位置。

还要注意三个边界：第一，解析成功不等于依赖已经解析，AST 不会替你判断模块版本；第二，Path.Value 是 Go 字符串字面量，不能直接把包含引号的值当作路径；第三，外部输入可能是半截文件，解析报错时应返回错误并停止生成清单，而不是静默吞掉错误。

最后的速查结论

• 遍历 file.Imports，每项就是一个 *ast.ImportSpec。
• 用 strconv.Unquote(spec.Path.Value) 得到干净路径。
• 用 spec.Name == nil 区分默认导入，否则原样记录 .、_ 或普通别名。
• 用 fset.Position(spec.Pos()) 记录来源文件和行列号。

这套做法适合依赖索引、源码导航和导入规范检查。若任务进一步要求判断真实包对象、类型或模块依赖，再把 AST 结果交给对应的类型检查或模块解析流程，不要在这一步越界。

相关问题

为什么不用遍历所有 AST 节点找 BasicLit？

因为 ast.File.Imports 已经准确限定了导入声明，直接遍历更清楚，也不会把普通字符串误判成导入路径。

显式别名和包名是一回事吗？

不是。AST 只能告诉你源码写了什么名称；默认导入时 Name 为空，实际包名还涉及被导入包的声明和解析结果。

官方资料：https://pkg.go.dev/go/ast#ImportSpec、https://pkg.go.dev/go/parser、https://pkg.go.dev/go/token#FileSet。