logranger/rule.go

// SPDX-FileCopyrightText: 2023 Winni Neessen <wn@neessen.dev>
//
// SPDX-License-Identifier: MIT

package logranger

import (
	"fmt"
	"os"
	"path/filepath"
	"regexp"
	"strings"

	"github.com/kkyr/fig"
)

// Ruleset represents a collection of rules.
type Ruleset struct {
	Rule []Rule `fig:"rule"`
}

// Rule represents a rule with its properties.
type Rule struct {
	ID        string         `fig:"id" validate:"required"`
	Regexp    *regexp.Regexp `fig:"regexp" validate:"required"`
	HostMatch *regexp.Regexp `fig:"host_match"`
	Actions   map[string]any `fig:"actions"`
}

// NewRuleset initializes a new Ruleset based on the provided Config.
// It reads the rule file specified in the Config, validates the file's
// existence, and loads the Ruleset using the fig library.
// It checks for duplicate rules and returns an error if any duplicates are found.
// If all operations are successful, it returns the created Ruleset and no error.
func NewRuleset(c *Config) (*Ruleset, error) {
	rs := &Ruleset{}
	p := filepath.Dir(c.Server.RuleFile)
	f := filepath.Base(c.Server.RuleFile)
	_, err := os.Stat(fmt.Sprintf("%s/%s", p, f))
	if err != nil {
		return rs, fmt.Errorf("failed to read config: %w", err)
	}

	if err = fig.Load(rs, fig.Dirs(p), fig.File(f), fig.UseStrict()); err != nil {
		return rs, fmt.Errorf("failed to load ruleset: %w", err)
	}

	rna := make([]string, 0)
	for _, r := range rs.Rule {
		for _, rn := range rna {
			if strings.EqualFold(r.ID, rn) {
				return nil, fmt.Errorf("duplicate rule found: %s", r.ID)
			}
		}
		rna = append(rna, r.ID)
	}

	return rs, nil
}