70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/perfect-panel/ppanel-server/pkg/rules"
|
|
|
|
"github.com/perfect-panel/ppanel-server/internal/model/server"
|
|
"github.com/perfect-panel/ppanel-server/internal/svc"
|
|
"github.com/perfect-panel/ppanel-server/internal/types"
|
|
"github.com/perfect-panel/ppanel-server/pkg/logger"
|
|
"github.com/perfect-panel/ppanel-server/pkg/tool"
|
|
"github.com/perfect-panel/ppanel-server/pkg/xerr"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type CreateRuleGroupLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// Create rule group
|
|
func NewCreateRuleGroupLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateRuleGroupLogic {
|
|
return &CreateRuleGroupLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
func parseAndValidateRules(ruleText, ruleName string) ([]string, error) {
|
|
var rs []string
|
|
ruleArr := strings.Split(ruleText, "\n")
|
|
if len(ruleArr) == 0 {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.InvalidParams), "rules is empty")
|
|
}
|
|
|
|
for _, s := range ruleArr {
|
|
r := rules.NewRule(s, ruleName)
|
|
if r == nil {
|
|
continue
|
|
}
|
|
if err := r.Validate(); err != nil {
|
|
continue
|
|
}
|
|
rs = append(rs, r.String())
|
|
}
|
|
return rs, nil
|
|
}
|
|
func (l *CreateRuleGroupLogic) CreateRuleGroup(req *types.CreateRuleGroupRequest) error {
|
|
rs, err := parseAndValidateRules(req.Rules, req.Name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = l.svcCtx.ServerModel.InsertRuleGroup(l.ctx, &server.RuleGroup{
|
|
Name: req.Name,
|
|
Icon: req.Icon,
|
|
Tags: tool.StringSliceToString(req.Tags),
|
|
Rules: strings.Join(rs, "\n"),
|
|
Enable: req.Enable,
|
|
})
|
|
if err != nil {
|
|
l.Errorw("[CreateRuleGroup] Insert Database Error: ", logger.Field("error", err.Error()))
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseInsertError), "create server rule group error: %v", err)
|
|
}
|
|
return nil
|
|
}
|