kra-oa/internal/server/system_config_routes.go

package server

import (
	"runtime"
	"syscall"

	"kra/internal/biz"
	"kra/internal/conf"
	"kra/internal/service"

	"github.com/gin-gonic/gin"
)

func registerSystemConfigRoutes(group *gin.RouterGroup, config *conf.AdminBackend, settings *service.SettingsService) {
	security := group.Group("/securityConfig")
	security.GET("/getSecurityConfig", func(c *gin.Context) {
		value, err := settings.Security(c.Request.Context())
		if err != nil {
			fail(c, "获取安全配置失败")
			return
		}
		writeResult(c, codeSuccess, value, "获取成功")
	})
	security.POST("/setSecurityConfig", func(c *gin.Context) {
		var value biz.SecurityConfig
		if c.ShouldBindJSON(&value) != nil {
			fail(c, "参数错误")
			return
		}
		result, err := settings.SaveSecurity(c.Request.Context(), &value)
		if err != nil {
			fail(c, err.Error())
			return
		}
		writeResult(c, codeSuccess, result, "设置成功")
	})

	system := group.Group("/system")
	system.POST("/getSystemConfig", func(c *gin.Context) {
		admin := gin.H{"routerPrefix": ""}
		email := gin.H{}
		if config != nil {
			admin["routerPrefix"] = config.RouterPrefix
			if config.Jwt != nil {
				admin["jwt"] = gin.H{"signingKey": "******", "expiresTime": config.Jwt.ExpiresTime.AsDuration().String(), "bufferTime": config.Jwt.BufferTime.AsDuration().String(), "issuer": config.Jwt.Issuer}
			}
			if config.Captcha != nil {
				admin["captcha"] = gin.H{"keyLong": config.Captcha.KeyLong, "imgWidth": config.Captcha.ImgWidth, "imgHeight": config.Captcha.ImgHeight, "storeExpiration": config.Captcha.StoreExpiration.AsDuration().String()}
			}
			if config.Local != nil {
				admin["local"] = gin.H{"storePath": config.Local.StorePath, "pathPrefix": config.Local.PathPrefix}
			}
			if config.Email != nil {
				email = gin.H{"to": config.Email.To, "from": config.Email.From, "host": config.Email.Host, "secret": "******", "nickname": config.Email.Nickname, "port": config.Email.Port, "is-ssl": config.Email.IsSsl, "is-loginauth": config.Email.IsLoginAuth}
			}
		}
		writeResult(c, codeSuccess, gin.H{"config": gin.H{"admin": admin, "email": email}}, "获取成功")
	})
	system.POST("/setSystemConfig", func(c *gin.Context) {
		var req struct {
			Config struct {
				Admin struct {
					RouterPrefix string `json:"routerPrefix"`
					JWT          struct {
						SigningKey string `json:"signingKey"`
						Issuer     string `json:"issuer"`
					} `json:"jwt"`
					Local struct {
						StorePath  string `json:"storePath"`
						PathPrefix string `json:"pathPrefix"`
					} `json:"local"`
				} `json:"admin"`
				Email *struct {
					To          string `json:"to"`
					From        string `json:"from"`
					Host        string `json:"host"`
					Secret      string `json:"secret"`
					Nickname    string `json:"nickname"`
					Port        int32  `json:"port"`
					IsSSL       bool   `json:"is-ssl"`
					IsLoginAuth bool   `json:"is-loginauth"`
				} `json:"email"`
			} `json:"config"`
		}
		if c.ShouldBindJSON(&req) != nil {
			fail(c, "参数错误")
			return
		}
		if config != nil {
			config.RouterPrefix = req.Config.Admin.RouterPrefix
			if config.Jwt != nil {
				if req.Config.Admin.JWT.SigningKey != "" && req.Config.Admin.JWT.SigningKey != "******" {
					config.Jwt.SigningKey = req.Config.Admin.JWT.SigningKey
				}
				if req.Config.Admin.JWT.Issuer != "" {
					config.Jwt.Issuer = req.Config.Admin.JWT.Issuer
				}
			}
			if config.Local != nil {
				if req.Config.Admin.Local.StorePath != "" {
					config.Local.StorePath = req.Config.Admin.Local.StorePath
				}
				if req.Config.Admin.Local.PathPrefix != "" {
					config.Local.PathPrefix = req.Config.Admin.Local.PathPrefix
				}
			}
			if config.Email != nil && req.Config.Email != nil {
				config.Email.To = req.Config.Email.To
				config.Email.From = req.Config.Email.From
				config.Email.Host = req.Config.Email.Host
				config.Email.Nickname = req.Config.Email.Nickname
				config.Email.Port = req.Config.Email.Port
				config.Email.IsSsl = req.Config.Email.IsSSL
				config.Email.IsLoginAuth = req.Config.Email.IsLoginAuth
				if req.Config.Email.Secret != "" && req.Config.Email.Secret != "******" {
					config.Email.Secret = req.Config.Email.Secret
				}
			}
		}
		ok(c)
	})
	system.POST("/reloadSystem", func(c *gin.Context) { writeResult(c, codeSuccess, gin.H{"reload": true}, "配置已热更新") })
	system.POST("/getServerInfo", func(c *gin.Context) {
		var mem runtime.MemStats
		runtime.ReadMemStats(&mem)
		disk := []gin.H{}
		var stat syscall.Statfs_t
		if syscall.Statfs(".", &stat) == nil {
			total := uint64(stat.Blocks) * uint64(stat.Bsize)
			free := uint64(stat.Bavail) * uint64(stat.Bsize)
			used := total - free
			percent := 0
			if total > 0 {
				percent = int(used * 100 / total)
			}
			disk = append(disk, gin.H{"mountPoint": ".", "usedMb": used / 1024 / 1024, "usedGb": used / 1024 / 1024 / 1024, "totalMb": total / 1024 / 1024, "totalGb": total / 1024 / 1024 / 1024, "usedPercent": percent})
		}
		server := gin.H{"os": gin.H{"goos": runtime.GOOS, "numCpu": runtime.NumCPU(), "compiler": runtime.Compiler, "goVersion": runtime.Version(), "numGoroutine": runtime.NumGoroutine()}, "cpu": gin.H{"cpus": []float64{}, "cores": runtime.NumCPU()}, "ram": gin.H{"usedMb": mem.Alloc / 1024 / 1024, "totalMb": mem.Sys / 1024 / 1024, "usedPercent": 0}, "disk": disk}
		writeResult(c, codeSuccess, gin.H{"server": server}, "获取成功")
	})
}