✨ init project

utils/ast/plugin_initialize_v2.go

@@ -0,0 +1,52 @@
+package ast
+
+import (
+	"fmt"
+	"go/ast"
+	"io"
+)
+
+type PluginInitializeV2 struct {
+	Base
+	Type         Type   // 类型
+	Path         string // 文件路径
+	PluginPath   string // 插件路径
+	RelativePath string // 相对路径
+	ImportPath   string // 导包路径
+	StructName   string // 结构体名称
+	PackageName  string // 包名
+}
+
+func (a *PluginInitializeV2) Parse(filename string, writer io.Writer) (file *ast.File, err error) {
+	if filename == "" {
+		if a.RelativePath == "" {
+			filename = a.PluginPath
+			a.RelativePath = a.Base.RelativePath(a.PluginPath)
+			return a.Base.Parse(filename, writer)
+		}
+		a.PluginPath = a.Base.AbsolutePath(a.RelativePath)
+		filename = a.PluginPath
+	}
+	return a.Base.Parse(filename, writer)
+}
+
+func (a *PluginInitializeV2) Injection(file *ast.File) error {
+	if !CheckImport(file, a.ImportPath) {
+		NewImport(a.ImportPath).Injection(file)
+		funcDecl := FindFunction(file, "bizPluginV2")
+		stmt := CreateStmt(fmt.Sprintf("PluginInitV2(engine, %s.Plugin)", a.PackageName))
+		funcDecl.Body.List = append(funcDecl.Body.List, stmt)
+	}
+	return nil
+}
+
+func (a *PluginInitializeV2) Rollback(file *ast.File) error {
+	return nil
+}
+
+func (a *PluginInitializeV2) Format(filename string, writer io.Writer, file *ast.File) error {
+	if filename == "" {
+		filename = a.PluginPath
+	}
+	return a.Base.Format(filename, writer, file)
+}