实现路径规则部分功能

2026-04-05 13:55:18 +08:00 · 2020-09-21 19:51:50 +08:00
parent 47bf305df6
commit a4b806fa7e
50 changed files with 630 additions and 114 deletions
--- a/internal/web/actions/default/servers/server/settings/locations/create.go
+++ b/internal/web/actions/default/servers/server/settings/locations/create.go
@@ -0,0 +1,116 @@
+package locations
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/servers/server/settings/webutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/actions"
+	"regexp"
+	"strings"
+)
+
+type CreateAction struct {
+	actionutils.ParentAction
+}
+
+func (this *CreateAction) Init() {
+	this.Nav("", "setting", "create")
+	this.SecondMenu("locations")
+}
+
+func (this *CreateAction) RunGet(params struct {
+	ServerId int64
+}) {
+	webConfig, err := webutils.FindWebConfigWithServerId(this.Parent(), params.ServerId)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Data["webId"] = webConfig.Id
+	this.Data["patternTypes"] = serverconfigs.AllLocationPatternTypes()
+
+	this.Show()
+}
+
+func (this *CreateAction) RunPost(params struct {
+	WebId int64
+
+	Name        string
+	Pattern     string
+	PatternType int
+	Description string
+
+	IsBreak           bool
+	IsCaseInsensitive bool
+	IsReverse         bool
+
+	Must *actions.Must
+}) {
+	params.Must.
+		Field("pattern", params.Pattern).
+		Require("请输入路径匹配规则")
+
+	// 校验正则
+	if params.PatternType == serverconfigs.HTTPLocationPatternTypeRegexp {
+		_, err := regexp.Compile(params.Pattern)
+		if err != nil {
+			this.Fail("正则表达式校验失败：" + err.Error())
+		}
+	}
+
+	// 自动加上前缀斜杠
+	if params.PatternType == serverconfigs.HTTPLocationPatternTypePrefix ||
+		params.PatternType == serverconfigs.HTTPLocationPatternTypeExact {
+		params.Pattern = "/" + strings.TrimLeft(params.Pattern, "/")
+	}
+
+	location := &serverconfigs.HTTPLocationConfig{}
+	location.SetPattern(params.Pattern, params.PatternType, params.IsCaseInsensitive, params.IsReverse)
+	resultPattern := location.Pattern
+
+	locationResp, err := this.RPC().HTTPLocationRPC().CreateHTTPLocation(this.AdminContext(), &pb.CreateHTTPLocationRequest{
+		ParentId:    0, // TODO 需要实现
+		Name:        params.Name,
+		Description: params.Description,
+		Pattern:     resultPattern,
+		IsBreak:     params.IsBreak,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	locationId := locationResp.LocationId
+
+	// Web中Location
+	webConfig, err := webutils.FindWebConfigWithId(this.Parent(), params.WebId)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	// TODO 支持Location嵌套
+	webConfig.LocationRefs = append(webConfig.LocationRefs, &serverconfigs.HTTPLocationRef{
+		IsOn:       true,
+		LocationId: locationId,
+		Children:   nil,
+	})
+
+	refJSON, err := json.Marshal(webConfig.LocationRefs)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	_, err = this.RPC().HTTPWebRPC().UpdateHTTPWebLocations(this.AdminContext(), &pb.UpdateHTTPWebLocationsRequest{
+		WebId:         params.WebId,
+		LocationsJSON: refJSON,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}
--- a/internal/web/actions/default/servers/server/settings/locations/delete.go
+++ b/internal/web/actions/default/servers/server/settings/locations/delete.go
@@ -0,0 +1,40 @@
+package locations
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/servers/server/settings/webutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+type DeleteAction struct {
+	actionutils.ParentAction
+}
+
+func (this *DeleteAction) RunPost(params struct {
+	WebId      int64
+	LocationId int64
+}) {
+	webConfig, err := webutils.FindWebConfigWithId(this.Parent(), params.WebId)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	webConfig.RemoveLocationRef(params.LocationId)
+
+	refJSON, err := json.Marshal(webConfig.LocationRefs)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	_, err = this.RPC().HTTPWebRPC().UpdateHTTPWebLocations(this.AdminContext(), &pb.UpdateHTTPWebLocationsRequest{
+		WebId:         params.WebId,
+		LocationsJSON: refJSON,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}
--- a/internal/web/actions/default/servers/server/settings/locations/index.go
+++ b/internal/web/actions/default/servers/server/settings/locations/index.go
@@ -2,6 +2,7 @@ package locations

 import (
 	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/servers/server/settings/webutils"
 )

 type IndexAction struct {
@@ -16,7 +17,18 @@ func (this *IndexAction) Init() {
 func (this *IndexAction) RunGet(params struct {
 	ServerId int64
 }) {
-	// TODO
+	webConfig, err := webutils.FindWebConfigWithServerId(this.Parent(), params.ServerId)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["webId"] = webConfig.Id
+
+	if webConfig.Locations != nil {
+		this.Data["locations"] = webConfig.Locations
+	} else {
+		this.Data["locations"] = []interface{}{}
+	}

 	this.Show()
 }
--- a/internal/web/actions/default/servers/server/settings/locations/init.go
+++ b/internal/web/actions/default/servers/server/settings/locations/init.go
@@ -13,6 +13,8 @@ func init() {
 			Helper(serverutils.NewServerHelper()).
 			Prefix("/servers/server/settings/locations").
 			Get("", new(IndexAction)).
+			GetPost("/create", new(CreateAction)).
+			Post("/delete", new(DeleteAction)).
 			EndAll()
 	})
 }
--- a/internal/web/actions/default/servers/server/settings/locations/location/index.go
+++ b/internal/web/actions/default/servers/server/settings/locations/location/index.go
@@ -0,0 +1,38 @@
+package location
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+)
+
+// 路径规则详情
+type IndexAction struct {
+	actionutils.ParentAction
+}
+
+func (this *IndexAction) Init() {
+	this.TinyMenu("basic")
+}
+
+func (this *IndexAction) RunGet(params struct {
+	LocationId int64
+}) {
+	locationConfigResp, err := this.RPC().HTTPLocationRPC().FindEnabledHTTPLocationConfig(this.AdminContext(), &pb.FindEnabledHTTPLocationConfigRequest{LocationId: params.LocationId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	locationConfig := &serverconfigs.HTTPLocationConfig{}
+	err = json.Unmarshal(locationConfigResp.LocationJSON, locationConfig)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Data["locationId"] = locationConfig.Id
+	this.Data["locationConfig"] = locationConfig
+
+	this.Show()
+}
--- a/internal/web/actions/default/servers/server/settings/locations/location/init.go
+++ b/internal/web/actions/default/servers/server/settings/locations/location/init.go
@@ -0,0 +1,20 @@
+package location
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/servers/server/settings/locations/locationutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/servers/serverutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/helpers"
+	"github.com/iwind/TeaGo"
+)
+
+func init() {
+	TeaGo.BeforeStart(func(server *TeaGo.Server) {
+		server.
+			Helper(helpers.NewUserMustAuth()).
+			Helper(locationutils.NewLocationHelper()).
+			Helper(serverutils.NewServerHelper()).
+			Prefix("/servers/server/settings/locations/location").
+			GetPost("", new(IndexAction)).
+			EndAll()
+	})
+}
--- a/internal/web/actions/default/servers/server/settings/locations/locationutils/location_helper.go
+++ b/internal/web/actions/default/servers/server/settings/locations/locationutils/location_helper.go
@@ -0,0 +1,113 @@
+package locationutils
+
+import (
+	"github.com/iwind/TeaGo/actions"
+	"github.com/iwind/TeaGo/maps"
+	"net/http"
+)
+
+type LocationHelper struct {
+}
+
+func NewLocationHelper() *LocationHelper {
+	return &LocationHelper{}
+}
+
+func (this *LocationHelper) BeforeAction(action *actions.ActionObject) {
+	if action.Request.Method != http.MethodGet {
+		return
+	}
+
+	serverIdString := action.ParamString("serverId")
+	locationIdString := action.ParamString("locationId")
+
+	action.Data["leftMenuItemIsDisabled"] = true
+	action.Data["mainMenu"] = "server"
+	action.Data["mainTab"] = "setting"
+	action.Data["secondMenuItem"] = "locations"
+	action.Data["tinyLeftMenuItems"] = this.createMenus(serverIdString, locationIdString, action.Data.GetString("tinyMenuItem"))
+}
+
+func (this *LocationHelper) createMenus(serverIdString string, locationIdString string, secondMenuItem string) []maps.Map {
+	menuItems := []maps.Map{
+		{
+			"name":     "基本信息",
+			"url":      "/servers/server/settings/locations/location?serverId=" + serverIdString + "&locationId=" + locationIdString,
+			"isActive": secondMenuItem == "basic",
+		},
+	}
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "Web设置",
+		"url":      "/servers/server/settings/locations/web?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "web",
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "反向代理",
+		"url":      "/servers/server/settings/locations/reverseProxy?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "reverseProxy",
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "访问控制",
+		"url":      "/servers/server/settings/locations/access?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "access",
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "WAF",
+		"url":      "/servers/server/settings/locations/waf?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "waf",
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "缓存",
+		"url":      "/servers/server/settings/locations/cache?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "cache",
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "-",
+		"url":      "",
+		"isActive": false,
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "字符编码",
+		"url":      "/servers/server/settings/locations/charset?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "charset",
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "访问日志",
+		"url":      "/servers/server/settings/locations/accessLog?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "accessLog",
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "统计",
+		"url":      "/servers/server/settings/locations/stat?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "stat",
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "Gzip压缩",
+		"url":      "/servers/server/settings/locations/gzip?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "gzip",
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "特殊页面",
+		"url":      "/servers/server/settings/locations/pages?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "pages",
+	})
+
+	menuItems = append(menuItems, maps.Map{
+		"name":     "HTTP Header",
+		"url":      "/servers/server/settings/locations/headers?serverId=" + serverIdString + "&locationId=" + locationIdString,
+		"isActive": secondMenuItem == "header",
+	})
+
+	return menuItems
+}