go_gameshield/internal/service/webforwarding.go

package service

import (
	"context"
	"fmt"
	v1 "github.com/go-nunu/nunu-layout-advanced/api/v1"
	"github.com/go-nunu/nunu-layout-advanced/internal/model"
	"github.com/go-nunu/nunu-layout-advanced/internal/repository"
	"strconv"
)

type WebForwardingService interface {
	GetWebForwarding(ctx context.Context, id int64) (*model.WebForwarding, error)
	AddWebForwarding(ctx context.Context, req *v1.WebForwardingRequest) (string, error)
	EditWebForwarding(ctx context.Context, req *v1.WebForwardingRequest) (string, error)
	DeleteWebForwarding(ctx context.Context, wafWebId int) (string, error)
}

func NewWebForwardingService(
	service *Service,
	required RequiredService,
	webForwardingRepository repository.WebForwardingRepository,
	crawler CrawlerService,
	parser ParserService,
) WebForwardingService {
	return &webForwardingService{
		Service:                 service,
		webForwardingRepository: webForwardingRepository,
		required:                required,
		parser:                  parser,
		crawler:                 crawler,
	}
}

type webForwardingService struct {
	*Service
	webForwardingRepository repository.WebForwardingRepository
	required                RequiredService
	parser                  ParserService
	crawler                 CrawlerService
}

func (s *webForwardingService) GetWebForwarding(ctx context.Context, id int64) (*model.WebForwarding, error) {
	return s.webForwardingRepository.GetWebForwarding(ctx, id)
}

func (s *webForwardingService) AddWebForwarding(ctx context.Context, req *v1.WebForwardingRequest) (string, error) {
	formData := map[string]interface{}{
		"tag":                   req.Tag,
		"port":                  req.Port,
		"waf_gateway_group_id":  req.WafGatewayGroupId,
		"backend":               req.Backend,
		"is_https":              req.IsHttps,
		"bps":                   req.QpsCount,
		"max_bytes_month":       req.MaxBytesMonth,
		"qps_count":             req.QpsCount,
		"qps_duration":          req.QpsDuration,
		"cc_count":              req.CcCount,
		"cc_duration":           req.CcDuration,
		"cc_block_count":        req.CcBlockCount,
		"cc_block_duration":     req.CcBlockDuration,
		"cc_4xx_count":          req.Cc4xxCount,
		"cc_4xx_duration":       req.Cc4xxDuration,
		"cc_4xx_block_count":    req.Cc4xxBlockCount,
		"cc_4xx_block_duration": req.Cc4xxBlockDuration,
		"cc_5xx_count":          req.Cc5xxCount,
		"cc_5xx_duration":       req.Cc5xxDuration,
		"cc_5xx_block_count":    req.Cc5xxBlockCount,
		"cc_5xx_block_duration": req.Cc5xxBlockDuration,
	}
	respBody, err := s.required.SendForm(ctx, "admin/info/waf_web/new", "admin/new/waf_web", formData)
	if err != nil {
		return "", err
	}
	// 解析响应内容中的 alert 消息
	res, err := s.parser.ParseAlert(string(respBody))
	if err != nil {
		return "", err
	}
	if res != "" {
		return "", fmt.Errorf(res)
	}
	wafWebId, err := s.parser.GetRuleId(ctx, respBody)
	if err != nil {
		return "", err
	}
	fmt.Println("=================================================", wafWebId)
	return "", nil
}

func (s *webForwardingService) EditWebForwarding(ctx context.Context, req *v1.WebForwardingRequest) (string, error) {
	formData := map[string]interface{}{
		"waf_web_id":            req.WafWebId,
		"tag":                   req.Tag,
		"port":                  req.Port,
		"waf_gateway_group_id":  req.WafGatewayGroupId,
		"backend":               req.Backend,
		"is_https":              req.IsHttps,
		"bps":                   req.QpsCount,
		"max_bytes_month":       req.MaxBytesMonth,
		"qps_count":             req.QpsCount,
		"qps_duration":          req.QpsDuration,
		"cc_count":              req.CcCount,
		"cc_duration":           req.CcDuration,
		"cc_block_count":        req.CcBlockCount,
		"cc_block_duration":     req.CcBlockDuration,
		"cc_4xx_count":          req.Cc4xxCount,
		"cc_4xx_duration":       req.Cc4xxDuration,
		"cc_4xx_block_count":    req.Cc4xxBlockCount,
		"cc_4xx_block_duration": req.Cc4xxBlockDuration,
		"cc_5xx_count":          req.Cc5xxCount,
		"cc_5xx_duration":       req.Cc5xxDuration,
		"cc_5xx_block_count":    req.Cc5xxBlockCount,
		"cc_5xx_block_duration": req.Cc5xxBlockDuration,
	}
	respBody, err := s.required.SendForm(ctx, "admin/info/waf_web/edit?&__goadmin_edit_pk="+strconv.Itoa(req.WafWebId), "admin/edit/waf_web", formData)
	if err != nil {
		return "", err
	}
	// 解析响应内容中的 alert 消息
	res, err := s.parser.ParseAlert(string(respBody))
	if err != nil {
		return "", err
	}
	if res != "" {
		return "", fmt.Errorf(res)
	}
	return "", nil
}

func (s *webForwardingService) DeleteWebForwarding(ctx context.Context, wafWebId int) (string, error) {
	res, err := s.crawler.DeleteRule(ctx, wafWebId, "admin/delete/waf_web?page=1&__pageSize=10&__sort=waf_web_id&__sort_type=desc")
	if err != nil {
		return "", err
	}
	return res, nil
}