package configStructs

import (
	"errors"
	"fmt"
	"regexp"
	"strings"

	"github.com/up9inc/mizu/shared/units"
)

const (
	GuiPortTapName                = "gui-port"
	NamespacesTapName             = "namespaces"
	AnalysisTapName               = "analysis"
	AllNamespacesTapName          = "all-namespaces"
	PlainTextFilterRegexesTapName = "regex-masking"
	DisableRedactionTapName       = "no-redact"
	HumanMaxEntriesDBSizeTapName  = "max-entries-db-size"
	DirectionTapName              = "direction"
	DryRunTapName                 = "dry-run"
	EnforcePolicyFile             = "test-rules"
)

type TapConfig struct {
	AnalysisDestination          string    `yaml:"dest" default:"up9.app"`
	SleepIntervalSec             int       `yaml:"upload-interval" default:"10"`
	PodRegexStr                  string    `yaml:"regex" default:".*"`
	GuiPort                      uint16    `yaml:"gui-port" default:"8899"`
	Namespaces                   []string  `yaml:"namespaces"`
	Analysis                     bool      `yaml:"analysis" default:"false"`
	AllNamespaces                bool      `yaml:"all-namespaces" default:"false"`
	PlainTextFilterRegexes       []string  `yaml:"regex-masking"`
	HealthChecksUserAgentHeaders []string  `yaml:"ignored-user-agents"`
	DisableRedaction             bool      `yaml:"no-redact" default:"false"`
	HumanMaxEntriesDBSize        string    `yaml:"max-entries-db-size" default:"200MB"`
	Direction                    string    `yaml:"direction" default:"in"`
	DryRun                       bool      `yaml:"dry-run" default:"false"`
	EnforcePolicyFile            string    `yaml:"test-rules"`
	ApiServerResources           Resources `yaml:"api-server-resources"`
	TapperResources              Resources `yaml:"tapper-resources"`
}

type Resources struct {
	CpuLimit       string `yaml:"cpu-limit" default:"750m"`
	MemoryLimit    string `yaml:"memory-limit" default:"1Gi"`
	CpuRequests    string `yaml:"cpu-requests" default:"50m"`
	MemoryRequests string `yaml:"memory-requests" default:"50Mi"`
}

func (config *TapConfig) PodRegex() *regexp.Regexp {
	podRegex, _ := regexp.Compile(config.PodRegexStr)
	return podRegex
}

func (config *TapConfig) TapOutgoing() bool {
	directionLowerCase := strings.ToLower(config.Direction)
	if directionLowerCase == "any" {
		return true
	}

	return false
}

func (config *TapConfig) MaxEntriesDBSizeBytes() int64 {
	maxEntriesDBSizeBytes, _ := units.HumanReadableToBytes(config.HumanMaxEntriesDBSize)
	return maxEntriesDBSizeBytes
}

func (config *TapConfig) Validate() error {
	_, compileErr := regexp.Compile(config.PodRegexStr)
	if compileErr != nil {
		return errors.New(fmt.Sprintf("%s is not a valid regex %s", config.PodRegexStr, compileErr))
	}

	_, parseHumanDataSizeErr := units.HumanReadableToBytes(config.HumanMaxEntriesDBSize)
	if parseHumanDataSizeErr != nil {
		return errors.New(fmt.Sprintf("Could not parse --%s value %s", HumanMaxEntriesDBSizeTapName, config.HumanMaxEntriesDBSize))
	}

	directionLowerCase := strings.ToLower(config.Direction)
	if directionLowerCase != "any" && directionLowerCase != "in" {
		return errors.New(fmt.Sprintf("%s is not a valid value for flag --%s. Acceptable values are in/any.", config.Direction, DirectionTapName))
	}

	return nil
}