enable to update AI Drawing configuarations in admin console page

api/handler/admin/config_handler.go

@@ -12,6 +12,8 @@ import (
 	"geekai/core/types"
 	"geekai/handler"
 	"geekai/service"
+	"geekai/service/mj"
+	"geekai/service/sd"
 	"geekai/store"
 	"geekai/store/model"
 	"geekai/utils"
@@ -26,10 +28,18 @@ type ConfigHandler struct {
 	handler.BaseHandler
 	levelDB        *store.LevelDB
 	licenseService *service.LicenseService
+	mjServicePool *mj.ServicePool
+	sdServicePool *sd.ServicePool
 }
 
-func NewConfigHandler(app *core.AppServer, db *gorm.DB, levelDB *store.LevelDB, licenseService *service.LicenseService) *ConfigHandler {
-	return &ConfigHandler{BaseHandler: handler.BaseHandler{App: app, DB: db}, levelDB: levelDB, licenseService: licenseService}
+func NewConfigHandler(app *core.AppServer, db *gorm.DB, levelDB *store.LevelDB, licenseService *service.LicenseService, mjPool *mj.ServicePool, sdPool *sd.ServicePool) *ConfigHandler {
+	return &ConfigHandler{
+		BaseHandler:    handler.BaseHandler{App: app, DB: db},
+		levelDB:        levelDB,
+		mjServicePool:  mjPool,
+		sdServicePool:  sdPool,
+		licenseService: licenseService,
+	}
 }
 
 func (h *ConfigHandler) Update(c *gin.Context) {
@@ -138,3 +148,49 @@ func (h *ConfigHandler) GetDrawingConfig(c *gin.Context) {
 		"sd":       h.App.Config.SdConfigs,
 	})
 }
+
+// SaveDrawingConfig 保存AI绘画配置
+func (h *ConfigHandler) SaveDrawingConfig(c *gin.Context) {
+	var data struct {
+		Sd      []types.StableDiffusionConfig `json:"sd"`
+		MjPlus  []types.MjPlusConfig          `json:"mj_plus"`
+		MjProxy []types.MjProxyConfig         `json:"mj_proxy"`
+	}
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	changed := false
+	if configChanged(data.Sd, h.App.Config.SdConfigs) {
+		logger.Debugf("SD 配置变动了")
+		h.App.Config.SdConfigs = data.Sd
+		h.sdServicePool.InitServices(data.Sd)
+		changed = true
+	}
+
+	if configChanged(data.MjPlus, h.App.Config.MjPlusConfigs) || configChanged(data.MjProxy, h.App.Config.MjProxyConfigs) {
+		logger.Debugf("MidJourney 配置变动了")
+		h.App.Config.MjPlusConfigs = data.MjPlus
+		h.App.Config.MjProxyConfigs = data.MjProxy
+		h.mjServicePool.InitServices(data.MjPlus, data.MjProxy)
+		changed = true
+	}
+
+	if changed {
+		err := core.SaveConfig(h.App.Config)
+		if err != nil {
+			resp.ERROR(c, "更新配置文档失败！")
+			return
+		}
+	}
+
+	resp.SUCCESS(c)
+
+}
+
+func configChanged(c1 interface{}, c2 interface{}) bool {
+	encode1 := utils.JsonEncode(c1)
+	encode2 := utils.JsonEncode(c2)
+	return utils.Md5(encode1) != utils.Md5(encode2)
+}

api/handler/admin/user_handler.go

@@ -8,6 +8,7 @@ package admin
 // * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 
 import (
+	"fmt"
 	"geekai/core"
 	"geekai/core/types"
 	"geekai/handler"
@@ -16,7 +17,6 @@ import (
 	"geekai/store/vo"
 	"geekai/utils"
 	"geekai/utils/resp"
-	"fmt"
 	"time"
 
 	"github.com/gin-gonic/gin"
@@ -87,7 +87,7 @@ func (h *UserHandler) Save(c *gin.Context) {
 	// 检测最大注册人数
 	var totalUser int64
 	h.DB.Model(&model.User{}).Count(&totalUser)
-	if int(totalUser) >= h.licenseService.GetLicense().UserNum {
+	if h.licenseService.GetLicense().UserNum > 0 && int(totalUser) >= h.licenseService.GetLicense().UserNum {
 		resp.ERROR(c, "当前注册用户数已达上限，请请升级 License")
 		return
 	}

api/handler/user_handler.go

@@ -8,6 +8,7 @@ package handler
 // * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 
 import (
+	"fmt"
 	"geekai/core"
 	"geekai/core/types"
 	"geekai/service"
@@ -15,7 +16,6 @@ import (
 	"geekai/store/vo"
 	"geekai/utils"
 	"geekai/utils/resp"
-	"fmt"
 	"strings"
 	"time"
 
@@ -71,7 +71,7 @@ func (h *UserHandler) Register(c *gin.Context) {
 	// 检测最大注册人数
 	var totalUser int64
 	h.DB.Model(&model.User{}).Count(&totalUser)
-	if int(totalUser) >= h.licenseService.GetLicense().UserNum {
+	if h.licenseService.GetLicense().UserNum > 0 && int(totalUser) >= h.licenseService.GetLicense().UserNum {
 		resp.ERROR(c, "当前注册用户数已达上限，请请升级 License")
 		return
 	}