Initial commit (code only without large binaries)

EdgeAdmin/internal/web/actions/default/settings/user-ui/index.go

@@ -0,0 +1,261 @@
+//go:build plus
+
+package userui
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/configloaders"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/systemconfigs"
+	"github.com/iwind/TeaGo/actions"
+	"io"
+	"strings"
+)
+
+type IndexAction struct {
+	actionutils.ParentAction
+}
+
+func (this *IndexAction) Init() {
+	this.Nav("", "", "index")
+}
+
+func (this *IndexAction) RunGet(params struct{}) {
+	config, err := configloaders.LoadUserUIConfig()
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["config"] = config
+
+	// 时区
+	this.Data["timeZoneGroups"] = nodeconfigs.FindAllTimeZoneGroups()
+	this.Data["timeZoneLocations"] = nodeconfigs.FindAllTimeZoneLocations()
+
+	if len(config.TimeZone) == 0 {
+		config.TimeZone = nodeconfigs.DefaultTimeZoneLocation
+	}
+	this.Data["timeZoneLocation"] = nodeconfigs.FindTimeZoneLocation(config.TimeZone)
+
+	// 风格样式
+	this.Data["themeBackgroundColors"] = systemconfigs.DefaultThemeBackgroundColors()
+
+	this.Show()
+}
+
+func (this *IndexAction) RunPost(params struct {
+	ProductName          string
+	UserSystemName       string
+	ShowPageFooter       bool
+	PageFooterHTML       string
+	ThemeBackgroundColor string
+	ShowVersion          bool
+	Version              string
+	ShowFinance          bool
+	FaviconFile          *actions.File
+	LogoFile             *actions.File
+
+	TimeZone            string
+	DnsResolverType     string
+	ClientIPHeaderNames string
+
+	ServerCheckCNAME bool
+
+	ShowTrafficCharts            bool
+	ShowCacheInfoInTrafficCharts bool
+	ShowBandwidthCharts          bool
+	BandwidthUnit                string
+
+	PortalIsOn     bool
+	PortalLogoFile *actions.File
+
+	Must *actions.Must
+	CSRF *actionutils.CSRF
+}) {
+	defer this.CreateLogInfo(codes.AdminUserUI_LogUpdateUISettings)
+
+	params.Must.
+		Field("productName", params.ProductName).
+		Require("请输入产品名称").
+		Field("userSystemName", params.UserSystemName).
+		Require("请输入管理员系统名称")
+
+	config, err := configloaders.LoadUserUIConfig()
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	config.ProductName = params.ProductName
+	config.UserSystemName = params.UserSystemName
+	config.ShowPageFooter = params.ShowPageFooter
+	config.PageFooterHTML = params.PageFooterHTML
+	config.Theme.BackgroundColor = strings.TrimLeft(params.ThemeBackgroundColor, "#")
+	config.ShowVersion = params.ShowVersion
+	config.Version = params.Version
+	config.ShowFinance = params.ShowFinance
+
+	config.Server.CheckCNAME = params.ServerCheckCNAME
+
+	config.ShowTrafficCharts = params.ShowTrafficCharts
+	config.ShowCacheInfoInTrafficCharts = params.ShowCacheInfoInTrafficCharts
+	config.ShowBandwidthCharts = params.ShowBandwidthCharts
+	config.BandwidthUnit = params.BandwidthUnit
+
+	config.TimeZone = params.TimeZone
+	config.DNSResolver.Type = params.DnsResolverType
+	config.ClientIPHeaderNames = params.ClientIPHeaderNames
+
+	// 上传Favicon文件
+	if params.FaviconFile != nil {
+		createResp, err := this.RPC().FileRPC().CreateFile(this.AdminContext(), &pb.CreateFileRequest{
+			Filename: params.FaviconFile.Filename,
+			Size:     params.FaviconFile.Size,
+			IsPublic: true,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		var fileId = createResp.FileId
+
+		// 上传内容
+		var buf = make([]byte, 512*1024)
+		reader, err := params.FaviconFile.OriginFile.Open()
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		for {
+			n, err := reader.Read(buf)
+			if n > 0 {
+				_, err = this.RPC().FileChunkRPC().CreateFileChunk(this.AdminContext(), &pb.CreateFileChunkRequest{
+					FileId: fileId,
+					Data:   buf[:n],
+				})
+				if err != nil {
+					this.Fail("上传失败：" + err.Error())
+				}
+			}
+			if err != nil {
+				if err == io.EOF {
+					break
+				}
+				this.Fail("上传失败：" + err.Error())
+			}
+		}
+
+		// 置为已完成
+		_, err = this.RPC().FileRPC().UpdateFileFinished(this.AdminContext(), &pb.UpdateFileFinishedRequest{FileId: fileId})
+		if err != nil {
+			this.ErrorPage(err)
+		}
+		config.FaviconFileId = fileId
+	}
+
+	// 上传Logo文件
+	if params.LogoFile != nil {
+		createResp, err := this.RPC().FileRPC().CreateFile(this.AdminContext(), &pb.CreateFileRequest{
+			Filename: params.LogoFile.Filename,
+			Size:     params.LogoFile.Size,
+			IsPublic: true,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		var fileId = createResp.FileId
+
+		// 上传内容
+		var buf = make([]byte, 512*1024)
+		reader, err := params.LogoFile.OriginFile.Open()
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		for {
+			n, err := reader.Read(buf)
+			if n > 0 {
+				_, err = this.RPC().FileChunkRPC().CreateFileChunk(this.AdminContext(), &pb.CreateFileChunkRequest{
+					FileId: fileId,
+					Data:   buf[:n],
+				})
+				if err != nil {
+					this.Fail("上传失败：" + err.Error())
+				}
+			}
+			if err != nil {
+				if err == io.EOF {
+					break
+				}
+				this.Fail("上传失败：" + err.Error())
+			}
+		}
+
+		// 置为已完成
+		_, err = this.RPC().FileRPC().UpdateFileFinished(this.AdminContext(), &pb.UpdateFileFinishedRequest{FileId: fileId})
+		if err != nil {
+			this.ErrorPage(err)
+		}
+		config.LogoFileId = fileId
+	}
+
+	// portal
+	config.Portal.IsOn = params.PortalIsOn
+
+	// 上传Portal Logo文件
+	if params.PortalLogoFile != nil {
+		createResp, err := this.RPC().FileRPC().CreateFile(this.AdminContext(), &pb.CreateFileRequest{
+			Filename: params.PortalLogoFile.Filename,
+			Size:     params.PortalLogoFile.Size,
+			IsPublic: true,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		var fileId = createResp.FileId
+
+		// 上传内容
+		var buf = make([]byte, 512*1024)
+		reader, err := params.PortalLogoFile.OriginFile.Open()
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		for {
+			n, err := reader.Read(buf)
+			if n > 0 {
+				_, err = this.RPC().FileChunkRPC().CreateFileChunk(this.AdminContext(), &pb.CreateFileChunkRequest{
+					FileId: fileId,
+					Data:   buf[:n],
+				})
+				if err != nil {
+					this.Fail("上传失败：" + err.Error())
+				}
+			}
+			if err != nil {
+				if err == io.EOF {
+					break
+				}
+				this.Fail("上传失败：" + err.Error())
+			}
+		}
+
+		// 置为已完成
+		_, err = this.RPC().FileRPC().UpdateFileFinished(this.AdminContext(), &pb.UpdateFileFinishedRequest{FileId: fileId})
+		if err != nil {
+			this.ErrorPage(err)
+		}
+		config.Portal.LogoFileId = fileId
+	}
+
+	err = configloaders.UpdateUserUIConfig(config)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}