From 296c71290ceb7afa4e23123218d2393824e08750 Mon Sep 17 00:00:00 2001
From: pixelmaxQM <flipped_aurora@163.com>
Date: Sat, 25 Oct 2025 12:05:25 +0800
Subject: [PATCH] =?UTF-8?q?fix(autocode):=20=E4=BF=AE=E5=A4=8D=E6=8F=92?=
 =?UTF-8?q?=E4=BB=B6=E5=8C=85=E6=A3=80=E6=B5=8B=E9=80=BB=E8=BE=91=E5=B9=B6?=
 =?UTF-8?q?=E6=B8=85=E7=90=86=E6=97=A0=E6=95=88=E6=95=B0=E6=8D=AE=E5=BA=93?=
 =?UTF-8?q?=E8=AE=B0=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

改进插件包目录结构检测逻辑，当缺少必要目录时生成警告描述
添加清理数据库中不存在对应实体文件的包记录功能
---
 server/service/system/auto_code_package.go | 63 ++++++++++++++++++----
 1 file changed, 54 insertions(+), 9 deletions(-)

diff --git a/server/service/system/auto_code_package.go b/server/service/system/auto_code_package.go
index e8abcb795..10fbb1183 100644
--- a/server/service/system/auto_code_package.go
+++ b/server/service/system/auto_code_package.go
@@ -3,6 +3,12 @@ package system
 import (
 	"context"
 	"fmt"
+	"go/token"
+	"os"
+	"path/filepath"
+	"strings"
+	"text/template"
+
 	"github.com/flipped-aurora/gin-vue-admin/server/global"
 	common "github.com/flipped-aurora/gin-vue-admin/server/model/common/request"
 	model "github.com/flipped-aurora/gin-vue-admin/server/model/system"
@@ -11,12 +17,7 @@ import (
 	"github.com/flipped-aurora/gin-vue-admin/server/utils/ast"
 	"github.com/flipped-aurora/gin-vue-admin/server/utils/autocode"
 	"github.com/pkg/errors"
-	"go/token"
 	"gorm.io/gorm"
-	"os"
-	"path/filepath"
-	"strings"
-	"text/template"
 )
 
 var AutoCodePackage = new(autoCodePackage)
@@ -162,7 +163,6 @@ func (s *autoCodePackage) All(ctx context.Context) (entities []model.SysAutoCode
 				"api":        true,
 				"config":     true,
 				"initialize": true,
-				"model":      true,
 				"plugin":     true,
 				"router":     true,
 				"service":    true,
@@ -179,14 +179,25 @@ func (s *autoCodePackage) All(ctx context.Context) (entities []model.SysAutoCode
 					}
 				}
 			}
-			if len(dirNameMap) != 0 {
-				continue
+
+			var desc string
+			if len(dirNameMap) == 0 {
+				// 完全符合标准结构
+				desc = "系统自动读取" + pluginDir[i].Name() + "插件，使用前请确认是否为v2版本插件"
+			} else {
+				// 缺少某些结构，生成警告描述
+				var missingDirs []string
+				for dirName := range dirNameMap {
+					missingDirs = append(missingDirs, dirName)
+				}
+				desc = fmt.Sprintf("系统自动读取，但是缺少 %s 结构，不建议自动化和mcp使用", strings.Join(missingDirs, "、"))
 			}
+
 			pluginPackage := model.SysAutoCodePackage{
 				PackageName: pluginDir[i].Name(),
 				Template:    "plugin",
 				Label:       pluginDir[i].Name() + "插件",
-				Desc:        "系统自动读取" + pluginDir[i].Name() + "插件，使用前请确认是否为v2版本插件",
+				Desc:        desc,
 				Module:      global.GVA_CONFIG.AutoCode.Module,
 			}
 			plugin = append(plugin, pluginPackage)
@@ -225,6 +236,40 @@ func (s *autoCodePackage) All(ctx context.Context) (entities []model.SysAutoCode
 		entities = append(entities, createEntity...)
 	}
 
+	// 处理数据库存在但实体文件不存在的情况 - 删除数据库中对应的数据
+	existingPackageNames := make(map[string]bool)
+	// 收集所有存在的包名
+	for i := 0; i < len(server); i++ {
+		existingPackageNames[server[i].PackageName] = true
+	}
+	for i := 0; i < len(plugin); i++ {
+		existingPackageNames[plugin[i].PackageName] = true
+	}
+
+	// 找出需要删除的数据库记录
+	deleteEntityIDs := []uint{}
+	for i := 0; i < len(entities); i++ {
+		if !existingPackageNames[entities[i].PackageName] {
+			deleteEntityIDs = append(deleteEntityIDs, entities[i].ID)
+		}
+	}
+
+	// 删除数据库中不存在文件的记录
+	if len(deleteEntityIDs) > 0 {
+		err = global.GVA_DB.WithContext(ctx).Delete(&model.SysAutoCodePackage{}, deleteEntityIDs).Error
+		if err != nil {
+			return nil, errors.Wrap(err, "删除不存在的包记录失败!")
+		}
+		// 从返回结果中移除已删除的记录
+		filteredEntities := []model.SysAutoCodePackage{}
+		for i := 0; i < len(entities); i++ {
+			if existingPackageNames[entities[i].PackageName] {
+				filteredEntities = append(filteredEntities, entities[i])
+			}
+		}
+		entities = filteredEntities
+	}
+
 	return entities, nil
 }