TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-25 08:20:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

同步防盗链功能

Browse Source
This commit is contained in:
GoEdgeLab
2022-09-22 16:41:02 +08:00
parent 1ea0acd027
commit e2dd012b49
1 changed files with 19 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
internal/rpc/services/service_http_web.go
View File

@@ -757,11 +757,20 @@ func (this *HTTPWebService) FindHTTPWebRequestScripts(ctx context.Context, req *
// UpdateHTTPWebReferers 修改防盗链设置 // UpdateHTTPWebReferers 修改防盗链设置
func (this *HTTPWebService) UpdateHTTPWebReferers(ctx context.Context, req *pb.UpdateHTTPWebReferersRequest) (*pb.RPCSuccess, error) { func (this *HTTPWebService) UpdateHTTPWebReferers(ctx context.Context, req *pb.UpdateHTTPWebReferersRequest) (*pb.RPCSuccess, error) {
_, err := this.ValidateAdmin(ctx) _, userId, err := this.ValidateAdminAndUser(ctx, true)
if err != nil { if err != nil {
return nil, err return nil, err
} }
var tx = this.NullTx()
if userId > 0 {
err = models.SharedHTTPWebDAO.CheckUserWeb(tx, userId, req.HttpWebId)
if err != nil {
return nil, err
}
}
var config = &serverconfigs.ReferersConfig{} var config = &serverconfigs.ReferersConfig{}
if len(req.ReferersJSON) > 0 { if len(req.ReferersJSON) > 0 {
err = json.Unmarshal(req.ReferersJSON, config) err = json.Unmarshal(req.ReferersJSON, config)
@@ -775,7 +784,6 @@ func (this *HTTPWebService) UpdateHTTPWebReferers(ctx context.Context, req *pb.U
} }
} }
var tx = this.NullTx()
err = models.SharedHTTPWebDAO.UpdateWebReferers(tx, req.HttpWebId, config) err = models.SharedHTTPWebDAO.UpdateWebReferers(tx, req.HttpWebId, config)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -786,12 +794,20 @@ func (this *HTTPWebService) UpdateHTTPWebReferers(ctx context.Context, req *pb.U
// FindHTTPWebReferers 查找防盗链设置 // FindHTTPWebReferers 查找防盗链设置
func (this *HTTPWebService) FindHTTPWebReferers(ctx context.Context, req *pb.FindHTTPWebReferersRequest) (*pb.FindHTTPWebReferersResponse, error) { func (this *HTTPWebService) FindHTTPWebReferers(ctx context.Context, req *pb.FindHTTPWebReferersRequest) (*pb.FindHTTPWebReferersResponse, error) {
_, err := this.ValidateAdmin(ctx) _, userId, err := this.ValidateAdminAndUser(ctx, true)
if err != nil { if err != nil {
return nil, err return nil, err
} }
var tx = this.NullTx() var tx = this.NullTx()
if userId > 0 {
err = models.SharedHTTPWebDAO.CheckUserWeb(tx, userId, req.HttpWebId)
if err != nil {
return nil, err
}
}
config, err := models.SharedHTTPWebDAO.FindWebReferers(tx, req.HttpWebId) config, err := models.SharedHTTPWebDAO.FindWebReferers(tx, req.HttpWebId)
if err != nil { if err != nil {
return nil, err return nil, err