mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-10 09:20:25 +08:00
124 lines
2.9 KiB
Go
124 lines
2.9 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"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
|
|
"github.com/iwind/TeaGo/actions"
|
|
"github.com/iwind/TeaGo/types"
|
|
"regexp"
|
|
)
|
|
|
|
type CreatePopupAction struct {
|
|
actionutils.ParentAction
|
|
}
|
|
|
|
func (this *CreatePopupAction) Init() {
|
|
}
|
|
|
|
func (this *CreatePopupAction) RunGet(params struct {
|
|
WebId int64
|
|
}) {
|
|
this.Data["webId"] = params.WebId
|
|
|
|
this.Show()
|
|
}
|
|
|
|
func (this *CreatePopupAction) RunPost(params struct {
|
|
WebId int64
|
|
Pattern string
|
|
Replace string
|
|
Mode string
|
|
RedirectStatus int
|
|
ProxyHost string
|
|
WithQuery bool
|
|
IsBreak bool
|
|
IsOn bool
|
|
|
|
CondsJSON []byte
|
|
|
|
Must *actions.Must
|
|
}) {
|
|
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())
|
|
}
|
|
}
|
|
|
|
// web配置
|
|
webConfig, err := dao.SharedHTTPWebDAO.FindWebConfigWithId(this.AdminContext(), params.WebId)
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
// 创建
|
|
createResp, err := this.RPC().HTTPRewriteRuleRPC().CreateHTTPRewriteRule(this.AdminContext(), &pb.CreateHTTPRewriteRuleRequest{
|
|
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
|
|
}
|
|
|
|
ref := &serverconfigs.HTTPRewriteRef{
|
|
IsOn: true,
|
|
RewriteRuleId: createResp.RewriteRuleId,
|
|
}
|
|
webConfig.RewriteRefs = append(webConfig.RewriteRefs, ref)
|
|
refsJSON, err := json.Marshal(webConfig.RewriteRefs)
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
// 设置Web中的重写规则
|
|
_, err = this.RPC().HTTPWebRPC().UpdateHTTPWebRewriteRules(this.AdminContext(), &pb.UpdateHTTPWebRewriteRulesRequest{
|
|
HttpWebId: params.WebId,
|
|
RewriteRulesJSON: refsJSON,
|
|
})
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
|
|
// 日志
|
|
defer this.CreateLogInfo("在Web %d 中创建重写规则 %d", params.WebId, createResp.RewriteRuleId)
|
|
|
|
this.Success()
|
|
}
|