package rule

import (
	"strconv"

	"git.kor-elf.net/kor-elf-shield/go-nftables-client/contract"
	"git.kor-elf.net/kor-elf-shield/go-nftables-client/contract/nft"
	"git.kor-elf.net/kor-elf-shield/go-nftables-client/family"
)

type rule struct {
	command contract.Command
}

func New(command contract.Command) nft.Rule {
	return &rule{
		command: command,
	}
}

func (r *rule) Add(family family.Type, tableName string, chainName string, expr ...string) error {
	args := []string{"add", "rule", family.String(), tableName, chainName}
	args = append(args, expr...)
	return r.command.Run(args...)
}

func (r *rule) Insert(family family.Type, tableName string, chainName string, expr ...string) error {
	args := []string{"insert", "rule", family.String(), tableName, chainName}
	args = append(args, expr...)
	return r.command.Run(args...)
}

func (r *rule) Replace(family family.Type, tableName string, chainName string, handle uint64, expr ...string) error {
	args := []string{"replace", "rule", family.String(), tableName, chainName, "handle", strconv.Itoa(int(handle))}
	args = append(args, expr...)
	return r.command.Run(args...)
}

func (r *rule) Delete(family family.Type, tableName string, chainName string, handle uint64) error {
	args := []string{"delete", "rule", family.String(), tableName, chainName, "handle", strconv.Itoa(int(handle))}
	return r.command.Run(args...)
}