package chain

import (
	chain2 "git.kor-elf.net/kor-elf-shield/go-nftables-client/chain"
	"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 chain struct {
	command contract.Command
}

func New(command contract.Command) nft.Chain {
	return &chain{
		command: command,
	}
}

func (c *chain) Add(family family.Type, tableName string, chainName string, baseChain chain2.ChainOptions) error {
	args := []string{"add", "chain", family.String(), tableName, chainName, baseChain.String()}
	return c.command.Run(args...)
}

func (c *chain) Create(family family.Type, tableName string, chainName string, baseChain chain2.ChainOptions) error {
	args := []string{"create", "chain", family.String(), tableName, chainName, baseChain.String()}
	return c.command.Run(args...)
}

func (c *chain) Delete(family family.Type, tableName string, chainName string) error {
	args := []string{"delete", "chain", family.String(), tableName, chainName}
	return c.command.Run(args...)
}

func (c *chain) Clear(family family.Type, tableName string, chainName string) error {
	args := []string{"flush", "chain", family.String(), tableName, chainName}
	return c.command.Run(args...)
}

func (c *chain) Rename(family family.Type, tableName string, oldChainName string, newChainName string) error {
	args := []string{"rename", "chain", family.String(), tableName, oldChainName, newChainName}
	return c.command.Run(args...)
}