Go文件解析

实际上，ent的generate指的是entc，即ent codegen.

解析Go包，获取实体（即schema）

这里使用了 golang.org/x/tools/go/packages用来校验和分析Go包。

// 从包路径中解析出schema的实体名称
func Generate(cmd *cobra.Command, args []string)  {
	cfg := &packages.Config{Mode: packages.NeedName | packages.NeedTypes | packages.NeedTypesInfo | packages.NeedModule,}
	// 加载要解析的Go包
    pkgs, err := packages.Load(cfg, args[0], entInterface.PkgPath())
	if err != nil {
		fmt.Fprintf(os.Stderr, "load: %v\n", err)
		os.Exit(1)
	}
	entPkg, pkg := pkgs[0], pkgs[1]
	if len(pkg.Errors) != 0 {
		log.Println(pkg.Errors[0])
	}
	if pkgs[0].PkgPath != entInterface.PkgPath() {
		entPkg, pkg = pkgs[1], pkgs[0]
	}
	var names []string
	//entInterface.Name() 反射类型名称
	//entPkg.Types.Scope().Lookup(entInterface.Name()) 从包的类型中找到给定名称的类型，返回其对象
	//.Type().Underlying() 对象类型的基础类型
	//(*types.Interface) 转为接口类型
	iface := entPkg.Types.Scope().Lookup(entInterface.Name()).Type().Underlying().(*types.Interface)
	//pkg.TypesInfo提供有关包的语法树的类型信息
    //pkg.TypesInfo.Defs 对象类型的定义信息
	//k ast.ident 
    //type Ident struct {
    //    NamePos token.Pos // 标识符的位置
    //    Name    string    // 标识符的名称
    //    Obj     *Object   // 所表示的对象; or nil
    //}
    //v obj
	for k, v := range pkg.TypesInfo.Defs {
        //做一些必要的校验
		typ, ok := v.(*types.TypeName)
		//IsExported id是否是大写开头
		//types.Implements 第一参数是否实现了第二参数
		if !ok || !k.IsExported() || !types.Implements(typ.Type(), iface) {
			continue
		}
		//一个类型的声明
		spec, ok := k.Obj.Decl.(*ast.TypeSpec)
		if !ok {
			fmt.Errorf("invalid declaration %T for %s", k.Obj.Decl, k.Name)
		}
		// ast.StructType 结构体类型声明
		if _, ok := spec.Type.(*ast.StructType); !ok {
			fmt.Errorf("invalid spec type %T for %s", spec.Type, k.Name)
		}
        //获取到schema的名称
		names = append(names, k.Name)
	}

	fmt.Println(names)
	fmt.Println(pkg.PkgPath)
}

//输出：
//[Link User]
//gitee.com/wennmu/gint.git/gint/schema

参考资料

go/packages - 用来校验和分析Go包。
go/ast - 用来表示Go包的语法树的类型。