package firewall

import (
	"context"
	"errors"
	"fmt"
	"strings"
)

var (
	ErrIP6Tables = errors.New("failed ip6tables command")
)

func (c *configurator) runIP6tablesInstructions(ctx context.Context, instructions []string) error {
	for _, instruction := range instructions {
		if err := c.runIP6tablesInstruction(ctx, instruction); err != nil {
			return err
		}
	}
	return nil
}

func (c *configurator) runIP6tablesInstruction(ctx context.Context, instruction string) error {
	c.ip6tablesMutex.Lock() // only one ip6tables command at once
	defer c.ip6tablesMutex.Unlock()
	if c.debug {
		fmt.Println("ip6tables " + instruction)
	}
	flags := strings.Fields(instruction)
	if output, err := c.commander.Run(ctx, "ip6tables", flags...); err != nil {
		return fmt.Errorf("%w \"ip6tables %s\": %s: %s", ErrIP6Tables, instruction, output, err)
	}
	return nil
}

func (c *configurator) setIPv6AllPolicies(ctx context.Context, policy string) error {
	switch policy {
	case "ACCEPT", "DROP":
	default:
		return fmt.Errorf("policy %q not recognized", policy)
	}
	return c.runIP6tablesInstructions(ctx, []string{
		"--policy INPUT " + policy,
		"--policy OUTPUT " + policy,
		"--policy FORWARD " + policy,
	})
}