feat: 调整自动化插件的注册逻辑

Author: piexlMax(奇淼

server/service/system/auto_code_package.go

@@ -337,7 +337,7 @@ func (s *autoCodePackage) templates(ctx context.Context, entity model.SysAutoCod
 						pluginInitialize := &ast.PluginInitializeV2{
 							Type:        ast.TypePluginInitializeV2,
 							Path:        filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "plugin", entity.PackageName, name),
-							PluginPath:  filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "initialize", "plugin_biz_v2.go"),
+							PluginPath:  filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "plugin", "register.go"),
 							ImportPath:  fmt.Sprintf(`"%s/plugin/%s"`, global.GVA_CONFIG.AutoCode.Module, entity.PackageName),
 							PackageName: entity.PackageName,
 						}

server/utils/ast/ast_type.go

@@ -43,7 +43,7 @@ const (
 	TypePluginGen                 = "PluginGen"                 // server/plugin/{package}/gen/main.go
 	TypePluginApiEnter            = "PluginApiEnter"            // server/plugin/{package}/enter.go
 	TypePluginInitializeV1        = "PluginInitializeV1"        // server/initialize/plugin_biz_v1.go
-	TypePluginInitializeV2        = "PluginInitializeV2"        // server/initialize/plugin_biz_v2.go
+	TypePluginInitializeV2        = "PluginInitializeV2"        // server/plugin/register.go
 	TypePluginRouterEnter         = "PluginRouterEnter"         // server/plugin/{package}/enter.go
 	TypePluginServiceEnter        = "PluginServiceEnter"        // server/plugin/{package}/enter.go
 	TypePluginInitializeApi       = "PluginInitializeApi"       // server/plugin/{package}/initialize/api.go

server/utils/ast/plugin_initialize_v2.go

@@ -2,7 +2,10 @@ package ast
 
 import (
 	"go/ast"
+	"go/token"
 	"io"
+	"strconv"
+	"strings"
 )
 
 type PluginInitializeV2 struct {
@@ -30,6 +33,40 @@ func (a *PluginInitializeV2) Parse(filename string, writer io.Writer) (file *ast
 }
 
 func (a *PluginInitializeV2) Injection(file *ast.File) error {
+	importPath := strings.TrimSpace(a.ImportPath)
+	if importPath == "" {
+		return nil
+	}
+	importPath = strings.Trim(importPath, "\"")
+	if importPath == "" || CheckImport(file, importPath) {
+		return nil
+	}
+
+	importSpec := &ast.ImportSpec{
+		Name: ast.NewIdent("_"),
+		Path: &ast.BasicLit{Kind: token.STRING, Value: strconv.Quote(importPath)},
+	}
+	var importDecl *ast.GenDecl
+	for _, decl := range file.Decls {
+		genDecl, ok := decl.(*ast.GenDecl)
+		if !ok {
+			continue
+		}
+		if genDecl.Tok == token.IMPORT {
+			importDecl = genDecl
+			break
+		}
+	}
+	if importDecl == nil {
+		file.Decls = append([]ast.Decl{
+			&ast.GenDecl{
+				Tok:   token.IMPORT,
+				Specs: []ast.Spec{importSpec},
+			},
+		}, file.Decls...)
+		return nil
+	}
+	importDecl.Specs = append(importDecl.Specs, importSpec)
 	return nil
 }
 

server/utils/ast/plugin_initialize_v2_test.go

@@ -22,8 +22,8 @@ func TestPluginInitialize_Injection(t *testing.T) {
 			name: "测试 Gva插件 注册注入",
 			fields: fields{
 				Type:       TypePluginInitializeV2,
-				Path:       filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "initialize", "plugin_biz_v2.go"),
-				PluginPath: filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "plugin", "gva", "plugin.go"),
+				Path:       filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "plugin", "gva", "plugin.go"),
+				PluginPath: filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "plugin", "register.go"),
 				ImportPath: `"github.com/flipped-aurora/gin-vue-admin/server/plugin/gva"`,
 			},
 			wantErr: false,
@@ -37,12 +37,12 @@ func TestPluginInitialize_Injection(t *testing.T) {
 				PluginPath: tt.fields.PluginPath,
 				ImportPath: tt.fields.ImportPath,
 			}
-			file, err := a.Parse(a.Path, nil)
+			file, err := a.Parse("", nil)
 			if err != nil {
 				t.Errorf("Parse() error = %v, wantErr %v", err, tt.wantErr)
 			}
 			a.Injection(file)
-			err = a.Format(a.Path, nil, file)
+			err = a.Format("", nil, file)
 			if (err != nil) != tt.wantErr {
 				t.Errorf("Injection() error = %v, wantErr %v", err, tt.wantErr)
 			}
@@ -69,8 +69,8 @@ func TestPluginInitialize_Rollback(t *testing.T) {
 			name: "测试 Gva插件 回滚",
 			fields: fields{
 				Type:       TypePluginInitializeV2,
-				Path:       filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "initialize", "plugin_biz_v2.go"),
-				PluginPath: filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "plugin", "gva", "plugin.go"),
+				Path:       filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "plugin", "gva", "plugin.go"),
+				PluginPath: filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "plugin", "register.go"),
 				ImportPath: `"github.com/flipped-aurora/gin-vue-admin/server/plugin/gva"`,
 			},
 			wantErr: false,
@@ -86,12 +86,12 @@ func TestPluginInitialize_Rollback(t *testing.T) {
 				StructName:  "Plugin",
 				PackageName: "gva",
 			}
-			file, err := a.Parse(a.Path, nil)
+			file, err := a.Parse("", nil)
 			if err != nil {
 				t.Errorf("Parse() error = %v, wantErr %v", err, tt.wantErr)
 			}
 			a.Rollback(file)
-			err = a.Format(a.Path, nil, file)
+			err = a.Format("", nil, file)
 			if (err != nil) != tt.wantErr {
 				t.Errorf("Rollback() error = %v, wantErr %v", err, tt.wantErr)
 			}