init

server/service/system/auto_code_mcp.go

@@ -0,0 +1,45 @@
+package system
+
+import (
+	"context"
+	"github.com/flipped-aurora/gin-vue-admin/server/global"
+	"github.com/flipped-aurora/gin-vue-admin/server/model/system/request"
+	"github.com/flipped-aurora/gin-vue-admin/server/utils"
+	"github.com/flipped-aurora/gin-vue-admin/server/utils/autocode"
+	"os"
+	"path/filepath"
+	"text/template"
+)
+
+func (s *autoCodeTemplate) CreateMcp(ctx context.Context, info request.AutoMcpTool) (toolFilePath string, err error) {
+	mcpTemplatePath := filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "resource", "mcp", "tools.tpl")
+	mcpToolPath := filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server, "mcp")
+
+	var files *template.Template
+
+	templateName := filepath.Base(mcpTemplatePath)
+
+	files, err = template.New(templateName).Funcs(autocode.GetTemplateFuncMap()).ParseFiles(mcpTemplatePath)
+	if err != nil {
+		return
+	}
+
+	fileName := utils.HumpToUnderscore(info.Name)
+
+	toolFilePath = filepath.Join(mcpToolPath, fileName+".go")
+
+	f, err := os.Create(toolFilePath)
+	if err != nil {
+		return
+	}
+	defer f.Close()
+
+	// 执行模板，将内容写入文件
+	err = files.Execute(f, info)
+	if err != nil {
+		return
+	}
+
+	return
+
+}