go_gameshield/internal/service/udpforwarding.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 UdpForWardingService interface {
	GetUdpForWarding(ctx context.Context, id int64) (*model.UdpForWarding, error)
	AddUdpForwarding(ctx context.Context, req *v1.UdpForwardingRequest) (string, error)
	EditUdpForwarding(ctx context.Context, req *v1.UdpForwardingRequest) (string, error)
	DeleteUdpForwarding(ctx context.Context, wafUdpId int) (string, error)
}

func NewUdpForWardingService(
	service *Service,
	udpForWardingRepository repository.UdpForWardingRepository,
	required RequiredService,
	parser ParserService,
	crawler CrawlerService,
) UdpForWardingService {
	return &udpForWardingService{
		Service:                 service,
		udpForWardingRepository: udpForWardingRepository,
		required:                required,
		parser:                  parser,
		crawler:                 crawler,
	}
}

type udpForWardingService struct {
	*Service
	udpForWardingRepository repository.UdpForWardingRepository
	required                RequiredService
	parser                  ParserService
	crawler                 CrawlerService
}

func (s *udpForWardingService) GetUdpForWarding(ctx context.Context, id int64) (*model.UdpForWarding, error) {
	return s.udpForWardingRepository.GetUdpForWarding(ctx, id)
}

func (s *udpForWardingService) AddUdpForwarding(ctx context.Context, req *v1.UdpForwardingRequest) (string, error) {
	formData := map[string]interface{}{
		"tag":                  req.Tag,
		"port":                 req.Port,
		"waf_gateway_group_id": req.WafGatewayGroupId,
		"cc_packet_count":      req.CcPacketCount,
		"cc_packet_duration":   req.CcPacketDuration,
		"cc_count":             req.CcCount,
		"cc_duration":          req.CcDuration,
		"cc_block_count":       req.CcBlockCount,
		"cc_block_duration":    req.CcBlockDuration,
		"qos_packet_count":     req.QosPacketCount,
		"qos_packet_duration":  req.QosPacketDuration,
		"max_bytes_month":      req.MaxBytesMonth,
		"max_conn_count":       req.MaxConnCount,
		"session_timeout":      req.SessionTimeout,
		"bps":                  req.Bps,
		"backend_protocol":     req.BackendProtocol,
		"backend_timeout":      req.BackendTimeout,
		"backend_list":         req.BackendList,
		"allow_ip_list":        req.AllowIpList,
		"deny_ip_list":         req.DenyIpList,
		"access_rule":          req.AccessRule,
		"comment":              req.Comment,
	}
	respBody, err := s.required.SendForm(ctx, "admin/info/waf_udp/new", "admin/new/waf_udp", 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)
	}

	wafUdpId, err := s.parser.GetRuleId(ctx, respBody)
	if err != nil {
		return "", err
	}
	fmt.Println("=================================================", wafUdpId)
	return "", nil
}

func (s *udpForWardingService) EditUdpForwarding(ctx context.Context, req *v1.UdpForwardingRequest) (string, error) {
	formData := map[string]interface{}{
		"waf_udp_id":           req.WafUdpId,
		"tag":                  req.Tag,
		"port":                 req.Port,
		"waf_gateway_group_id": req.WafGatewayGroupId,
		"cc_packet_count":      req.CcPacketCount,
		"cc_packet_duration":   req.CcPacketDuration,
		"cc_count":             req.CcCount,
		"cc_duration":          req.CcDuration,
		"cc_block_count":       req.CcBlockCount,
		"cc_block_duration":    req.CcBlockDuration,
		"qos_packet_count":     req.QosPacketCount,
		"qos_packet_duration":  req.QosPacketDuration,
		"max_bytes_month":      req.MaxBytesMonth,
		"max_conn_count":       req.MaxConnCount,
		"session_timeout":      req.SessionTimeout,
		"bps":                  req.Bps,
		"backend_protocol":     req.BackendProtocol,
		"backend_timeout":      req.BackendTimeout,
		"backend_list":         req.BackendList,
		"allow_ip_list":        req.AllowIpList,
		"deny_ip_list":         req.DenyIpList,
		"access_rule":          req.AccessRule,
		"comment":              req.Comment,
	}
	respBody, err := s.required.SendForm(ctx, "admin/info/waf_udp/edit?&__goadmin_edit_pk="+strconv.Itoa(req.WafUdpId), "admin/edit/waf_udp", 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 *udpForWardingService) DeleteUdpForwarding(ctx context.Context, wafUdpId int) (string, error) {
	res, err := s.crawler.DeleteRule(ctx, wafUdpId, "admin/delete/waf_udp?page=1&__pageSize=10&__sort=waf_udp_id&__sort_type=desc")
	if err != nil {
		return "", err
	}
	return res, nil
}