nfsense/internal/api/firewall/destination_nat_rules.go

package firewall

import (
	"context"
	"fmt"

	"nfsense.net/nfsense/internal/definitions"
)

type GetDestinationNATRulesResult struct {
	DestinationNATRules []definitions.DestinationNATRule `json:"destination_nat_rules"`
}

func (f *Firewall) GetDestinationNATRules(ctx context.Context, params struct{}) (GetDestinationNATRulesResult, error) {
	return GetDestinationNATRulesResult{
		DestinationNATRules: f.ConfigManager.GetPendingConfig().Firewall.DestinationNATRules,
	}, nil
}

type CreateDestinationNATRuleParameters struct {
	DestinationNATRule definitions.DestinationNATRule `json:"destination_nat_rule"`
}

func (f *Firewall) CreateDestinationNATRule(ctx context.Context, params CreateDestinationNATRuleParameters) (struct{}, error) {
	t, conf := f.ConfigManager.StartTransaction()
	defer t.Discard()

	conf.Firewall.DestinationNATRules = append(conf.Firewall.DestinationNATRules, params.DestinationNATRule)
	return struct{}{}, t.Commit()
}

type UpdateDestinationNATRuleParameters struct {
	Index              uint64                         `json:"index"`
	DestinationNATRule definitions.DestinationNATRule `json:"destination_nat_rule"`
}

func (f *Firewall) UpdateDestinationNATRule(ctx context.Context, params UpdateDestinationNATRuleParameters) (struct{}, error) {
	if int(params.Index) >= len(f.ConfigManager.GetPendingConfig().Firewall.DestinationNATRules) {
		return struct{}{}, fmt.Errorf("DestinationNATRule does not Exist")
	}

	t, conf := f.ConfigManager.StartTransaction()
	defer t.Discard()

	conf.Firewall.DestinationNATRules[params.Index] = params.DestinationNATRule
	return struct{}{}, t.Commit()
}

type MoveDestinationNATRuleParameters struct {
	Index   uint64 `json:"index"`
	ToIndex uint64 `json:"to_index"`
}

func (f *Firewall) MoveDestinationNATRule(ctx context.Context, params DeleteDestinationNATRuleParameters) (struct{}, error) {
	if int(params.Index) >= len(f.ConfigManager.GetPendingConfig().Firewall.DestinationNATRules) {
		return struct{}{}, fmt.Errorf("DestinationNATRule does not Exist")
	}

	t, conf := f.ConfigManager.StartTransaction()
	defer t.Discard()

	conf.Firewall.DestinationNATRules = append(conf.Firewall.DestinationNATRules[:params.Index], conf.Firewall.DestinationNATRules[params.Index+1:]...)
	return struct{}{}, t.Commit()
}

type DeleteDestinationNATRuleParameters struct {
	Index uint64 `json:"index"`
}

func (f *Firewall) DeleteDestinationNATRule(ctx context.Context, params DeleteDestinationNATRuleParameters) (struct{}, error) {
	if int(params.Index) >= len(f.ConfigManager.GetPendingConfig().Firewall.DestinationNATRules) {
		return struct{}{}, fmt.Errorf("DestinationNATRule does not Exist")
	}

	t, conf := f.ConfigManager.StartTransaction()
	defer t.Discard()

	conf.Firewall.DestinationNATRules = append(conf.Firewall.DestinationNATRules[:params.Index], conf.Firewall.DestinationNATRules[params.Index+1:]...)
	return struct{}{}, t.Commit()
}