package table

import (
	"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"
	nftCommand "git.kor-elf.net/kor-elf-shield/go-nftables-client/internal/pkg/nft"
)

type table struct {
	command contract.Command
}

func New(command contract.Command) nft.Table {
	return &table{
		command: command,
	}
}

func (t *table) Add(family family.Type, tableName string) error {
	args := nftCommand.TableAdd(family, tableName)
	return t.command.Run(args...)
}

func (t *table) Delete(family family.Type, tableName string) error {
	args := nftCommand.TableDelete(family, tableName)
	return t.command.Run(args...)
}

func (t *table) Clear(family family.Type, tableName string) error {
	args := nftCommand.TableClear(family, tableName)
	return t.command.Run(args...)
}