package cmd

import (
	"fmt"

	"github.com/jedib0t/go-pretty/v6/table"
	"github.com/spf13/cobra"
)

// ListArgs tracks the resolved flag configuration for the list command.
type ListArgs struct {
	header  bool
	key     bool
	value   bool
	ttl     bool
	binary  bool
	secrets bool
	render  func(table.Writer)
}

// formatEnum implements pflag.Value for format selection.
type formatEnum string

func (e *formatEnum) String() string {
	return string(*e)
}

func (e *formatEnum) Set(v string) error {
	switch v {
	case "table", "csv", "html", "markdown":
		*e = formatEnum(v)
		return nil
	default:
		return fmt.Errorf("must be one of \"table\", \"csv\", \"html\", or \"markdown\"")
	}
}

func (e *formatEnum) Type() string {
	return "format"
}

var (
	binary   bool       = false
	secret   bool       = false
	noKeys   bool       = false
	noValues bool       = false
	ttl      bool       = false
	noHeader bool       = false
	format   formatEnum = "table"
)

func parseFlags(cmd *cobra.Command) (ListArgs, error) {
	var renderFunc func(tw table.Writer)
	switch format.String() {
	case "csv":
		renderFunc = func(tw table.Writer) { tw.RenderCSV() }
	case "html":
		renderFunc = func(tw table.Writer) { tw.RenderHTML() }
	case "markdown":
		renderFunc = func(tw table.Writer) { tw.RenderMarkdown() }
	default:
		renderFunc = func(tw table.Writer) { tw.Render() }
	}

	if noKeys && noValues && !ttl {
		return ListArgs{}, fmt.Errorf("no columns selected; disable --no-keys/--no-values or pass --ttl")
	}

	return ListArgs{
		header:  !noHeader,
		key:     !noKeys,
		value:   !noValues,
		ttl:     ttl,
		binary:  binary,
		render:  renderFunc,
		secrets: secret,
	}, nil
}