mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-10 17:30:29 +08:00
116 lines
2.7 KiB
Go
116 lines
2.7 KiB
Go
package rewrite
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/dao"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
|
|
"github.com/iwind/TeaGo/actions"
|
|
"github.com/iwind/TeaGo/types"
|
|
"regexp"
|
|
)
|
|
|
|
type UpdatePopupAction struct {
|
|
actionutils.ParentAction
|
|
}
|
|
|
|
func (this *UpdatePopupAction) Init() {
|
|
}
|
|
|
|
func (this *UpdatePopupAction) RunGet(params struct {
|
|
WebId int64
|
|
RewriteRuleId int64
|
|
}) {
|
|
this.Data["webId"] = params.WebId
|
|
|
|
webConfig, err := dao.SharedHTTPWebDAO.FindWebConfigWithId(this.AdminContext(), params.WebId)
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
isFound := false
|
|
for _, rewriteRule := range webConfig.RewriteRules {
|
|
if rewriteRule.Id == params.RewriteRuleId {
|
|
this.Data["rewriteRule"] = rewriteRule
|
|
isFound = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !isFound {
|
|
this.WriteString("找不到要修改的重写规则")
|
|
return
|
|
}
|
|
|
|
this.Show()
|
|
}
|
|
|
|
func (this *UpdatePopupAction) RunPost(params struct {
|
|
WebId int64
|
|
RewriteRuleId int64
|
|
Pattern string
|
|
Replace string
|
|
Mode string
|
|
RedirectStatus int
|
|
ProxyHost string
|
|
WithQuery bool
|
|
IsBreak bool
|
|
IsOn bool
|
|
CondsJSON []byte
|
|
|
|
Must *actions.Must
|
|
}) {
|
|
defer this.CreateLogInfo("修改Web %d 中的重写规则 %d", params.WebId, params.RewriteRuleId)
|
|
|
|
params.Must.
|
|
Field("pattern", params.Pattern).
|
|
Require("请输入匹配规则").
|
|
Expect(func() (message string, success bool) {
|
|
_, err := regexp.Compile(params.Pattern)
|
|
if err != nil {
|
|
return "匹配规则错误：" + err.Error(), false
|
|
}
|
|
return "", true
|
|
})
|
|
|
|
params.Must.
|
|
Field("replace", params.Replace).
|
|
Require("请输入目标URL")
|
|
|
|
// 校验匹配条件
|
|
if len(params.CondsJSON) > 0 {
|
|
conds := &shared.HTTPRequestCondsConfig{}
|
|
err := json.Unmarshal(params.CondsJSON, conds)
|
|
if err != nil {
|
|
this.Fail("匹配条件校验失败：" + err.Error())
|
|
}
|
|
|
|
err = conds.Init()
|
|
if err != nil {
|
|
this.Fail("匹配条件校验失败：" + err.Error())
|
|
}
|
|
}
|
|
|
|
// 修改
|
|
_, err := this.RPC().HTTPRewriteRuleRPC().UpdateHTTPRewriteRule(this.AdminContext(), &pb.UpdateHTTPRewriteRuleRequest{
|
|
RewriteRuleId: params.RewriteRuleId,
|
|
Pattern: params.Pattern,
|
|
Replace: params.Replace,
|
|
Mode: params.Mode,
|
|
RedirectStatus: types.Int32(params.RedirectStatus),
|
|
ProxyHost: params.ProxyHost,
|
|
WithQuery: params.WithQuery,
|
|
IsBreak: params.IsBreak,
|
|
IsOn: params.IsOn,
|
|
CondsJSON: params.CondsJSON,
|
|
})
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
this.Success()
|
|
}
|