53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
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)
|
|
}
|