阶段性提交

internal/web/actions/default/servers/server/settings/gzip/index.go

@@ -1,7 +1,13 @@
 package gzip
 
 import (
+	"encoding/json"
+	"errors"
 	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/servers/serverutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/actions"
 )
 
 type IndexAction struct {
@@ -16,7 +22,119 @@ func (this *IndexAction) Init() {
 func (this *IndexAction) RunGet(params struct {
 	ServerId int64
 }) {
-	// TODO
+	server, _, isOk := serverutils.FindServer(&this.ParentAction, params.ServerId)
+	if !isOk {
+		return
+	}
+
+	webId := server.WebId
+	if webId <= 0 {
+		resp, err := this.RPC().ServerRPC().InitServerWeb(this.AdminContext(), &pb.InitServerWebRequest{ServerId: params.ServerId})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		webId = resp.WebId
+	}
+
+	webResp, err := this.RPC().HTTPWebRPC().FindEnabledHTTPWeb(this.AdminContext(), &pb.FindEnabledHTTPWebRequest{WebId: webId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	web := webResp.Web
+	if web == nil {
+		this.ErrorPage(errors.New("web should not be nil"))
+		return
+	}
+	this.Data["webId"] = web.Id
+
+	gzipId := web.GzipId
+	gzipConfig := &serverconfigs.HTTPGzipConfig{
+		Id:   0,
+		IsOn: true,
+	}
+	if gzipId > 0 {
+		resp, err := this.RPC().HTTPGzipRPC().FindEnabledHTTPGzipConfig(this.AdminContext(), &pb.FindEnabledGzipConfigRequest{GzipId: gzipId})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		err = json.Unmarshal(resp.Config, gzipConfig)
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+	}
+
+	this.Data["gzipConfig"] = gzipConfig
 
 	this.Show()
 }
+
+func (this *IndexAction) RunPost(params struct {
+	WebId     int64
+	GzipId    int64
+	Level     int
+	MinLength string
+	MaxLength string
+
+	Must *actions.Must
+}) {
+	if params.Level < 0 || params.Level > 9 {
+		this.Fail("请选择正确的压缩级别")
+	}
+
+	minLength := &pb.SizeCapacity{Count: -1}
+	if len(params.MinLength) > 0 {
+		err := json.Unmarshal([]byte(params.MinLength), minLength)
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+	}
+
+	maxLength := &pb.SizeCapacity{Count: -1}
+	if len(params.MaxLength) > 0 {
+		err := json.Unmarshal([]byte(params.MaxLength), maxLength)
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+	}
+
+
+	if params.GzipId > 0 {
+		_, err := this.RPC().HTTPGzipRPC().UpdateHTTPGzip(this.AdminContext(), &pb.UpdateHTTPGzipRequest{
+			GzipId:    params.GzipId,
+			Level:     int32(params.Level),
+			MinLength: minLength,
+			MaxLength: maxLength,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+	} else {
+		resp, err := this.RPC().HTTPGzipRPC().CreateHTTPGzip(this.AdminContext(), &pb.CreateHTTPGzipRequest{
+			Level:     0,
+			MinLength: nil,
+			MaxLength: nil,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		gzipId := resp.GzipId
+
+		_, err = this.RPC().HTTPWebRPC().UpdateHTTPWebGzip(this.AdminContext(), &pb.UpdateHTTPWebGzipRequest{
+			WebId:  params.WebId,
+			GzipId: gzipId,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+		}
+	}
+
+	this.Success()
+}