tablizer/cfg/config.go

/*
Copyright © 2022-2025 Thomas von Dein

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package cfg

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

	"github.com/gookit/color"
	"github.com/hashicorp/hcl/v2/hclsimple"
)

const (
	Version  = "v1.5.12"
	MAXPARTS = 2
)

var (
	DefaultConfigfile = os.Getenv("HOME") + "/.config/tablizer/config"
	VERSION           string // maintained by -x

	SeparatorTemplates = map[string]string{
		":tab:":      `\s*\t\s*`,                               // tab but eats spaces around
		":spaces:":   `\s{2,}`,                                 // 2 or more spaces
		":pipe:":     `\s*\|\s*`,                               // one pipe eating spaces around
		":default:":  `(\s\s+|\t)`,                             // 2 or more spaces or tab
		":nonword:":  `\W`,                                     // word boundary
		":nondigit:": `\D`,                                     // same for numbers
		":special:":  `[\*\+\-_\(\)\[\]\{\}?\\/<>=&$§"':,\^]+`, // match any special char
		":nonprint:": `[[:^print:]]+`,                          // non printables
	}
)

// public config, set via config file or using defaults
type Settings struct {
	FG             string `hcl:"FG"`
	BG             string `hcl:"BG"`
	HighlightFG    string `hcl:"HighlightFG"`
	HighlightBG    string `hcl:"HighlightBG"`
	NoHighlightFG  string `hcl:"NoHighlightFG"`
	NoHighlightBG  string `hcl:"NoHighlightBG"`
	HighlightHdrFG string `hcl:"HighlightHdrFG"`
	HighlightHdrBG string `hcl:"HighlightHdrBG"`
}

type Transposer struct {
	Search  regexp.Regexp
	Replace string
}

type Pattern struct {
	Pattern   string
	PatternRe *regexp.Regexp
	Negate    bool
}

type Filter struct {
	Regex  *regexp.Regexp
	Negate bool
}

// internal config
type Config struct {
	Debug          bool
	Numbering      bool
	NoHeaders      bool
	Columns        string
	UseColumns     []int
	YankColumns    string
	UseYankColumns []int
	Separator      string
	OutputMode     int
	InvertMatch    bool
	Patterns       []*Pattern
	UseFuzzySearch bool
	UseHighlight   bool
	Interactive    bool
	InputJSON      bool
	AutoHeaders    bool
	CustomHeaders  []string
	Template       string

	SortMode        string
	SortDescending  bool
	SortByColumn    string // 1,2
	UseSortByColumn []int  // []int{1,2}

	TransposeColumns    string       // 1,2
	UseTransposeColumns []int        // []int{1,2}
	Transposers         []string     // []string{"/ /-/", "/foo/bar/"}
	UseTransposers      []Transposer // {Search: re, Replace: string}

	/*
	 FIXME: make configurable somehow, config file or ENV
	 see https://github.com/gookit/color.
	*/
	ColorStyle        color.Style
	HighlightStyle    color.Style
	NoHighlightStyle  color.Style
	HighlightHdrStyle color.Style

	NoColor bool

	// config file, optional
	Configfile string

	Settings Settings

	// used for field filtering
	Rawfilters []string
	Filters    map[string]Filter //map[string]*regexp.Regexp

	// -r <file>
	InputFile string

	OFS string
}

// maps outputmode short flags to output mode, ie. -O => -o orgtbl
type Modeflag struct {
	X bool
	O bool
	M bool
	S bool
	Y bool
	A bool
	C bool
	J bool
	P bool // template
}

// used for switching printers
const (
	Extended = iota + 1
	Orgtbl
	Markdown
	Shell
	Yaml
	CSV
	ASCII
	Json
	Template
)

// various sort types
type Sortmode struct {
	Numeric bool
	Time    bool
	Age     bool
}

// default color schemes
func (conf *Config) Colors() map[color.Level]map[string]color.Color {
	colors := map[color.Level]map[string]color.Color{
		color.Level16: {
			"bg": color.BgGreen, "fg": color.FgWhite,
			"hlbg": color.BgGray, "hlfg": color.FgWhite,
		},
		color.Level256: {
			"bg": color.BgLightGreen, "fg": color.FgWhite,
			"hlbg": color.BgLightBlue, "hlfg": color.FgWhite,
		},
		color.LevelRgb: {
			"bg": color.BgLightGreen, "fg": color.FgWhite,
			"hlbg": color.BgHiGreen, "hlfg": color.FgWhite,
			"nohlbg": color.BgWhite, "nohlfg": color.FgLightGreen,
			"hdrbg": color.BgBlue, "hdrfg": color.FgWhite,
		},
	}

	if len(conf.Settings.BG) > 0 {
		colors[color.Level16]["bg"] = ColorStringToBGColor(conf.Settings.BG)
		colors[color.Level256]["bg"] = ColorStringToBGColor(conf.Settings.BG)
		colors[color.LevelRgb]["bg"] = ColorStringToBGColor(conf.Settings.BG)
	}

	if len(conf.Settings.FG) > 0 {
		colors[color.Level16]["fg"] = ColorStringToColor(conf.Settings.FG)
		colors[color.Level256]["fg"] = ColorStringToColor(conf.Settings.FG)
		colors[color.LevelRgb]["fg"] = ColorStringToColor(conf.Settings.FG)
	}

	if len(conf.Settings.HighlightBG) > 0 {
		colors[color.Level16]["hlbg"] = ColorStringToBGColor(conf.Settings.HighlightBG)
		colors[color.Level256]["hlbg"] = ColorStringToBGColor(conf.Settings.HighlightBG)
		colors[color.LevelRgb]["hlbg"] = ColorStringToBGColor(conf.Settings.HighlightBG)
	}

	if len(conf.Settings.HighlightFG) > 0 {
		colors[color.Level16]["hlfg"] = ColorStringToColor(conf.Settings.HighlightFG)
		colors[color.Level256]["hlfg"] = ColorStringToColor(conf.Settings.HighlightFG)
		colors[color.LevelRgb]["hlfg"] = ColorStringToColor(conf.Settings.HighlightFG)
	}

	if len(conf.Settings.NoHighlightBG) > 0 {
		colors[color.Level16]["nohlbg"] = ColorStringToBGColor(conf.Settings.NoHighlightBG)
		colors[color.Level256]["nohlbg"] = ColorStringToBGColor(conf.Settings.NoHighlightBG)
		colors[color.LevelRgb]["nohlbg"] = ColorStringToBGColor(conf.Settings.NoHighlightBG)
	}

	if len(conf.Settings.NoHighlightFG) > 0 {
		colors[color.Level16]["nohlfg"] = ColorStringToColor(conf.Settings.NoHighlightFG)
		colors[color.Level256]["nohlfg"] = ColorStringToColor(conf.Settings.NoHighlightFG)
		colors[color.LevelRgb]["nohlfg"] = ColorStringToColor(conf.Settings.NoHighlightFG)
	}

	if len(conf.Settings.HighlightHdrBG) > 0 {
		colors[color.Level16]["hdrbg"] = ColorStringToBGColor(conf.Settings.HighlightHdrBG)
		colors[color.Level256]["hdrbg"] = ColorStringToBGColor(conf.Settings.HighlightHdrBG)
		colors[color.LevelRgb]["hdrbg"] = ColorStringToBGColor(conf.Settings.HighlightHdrBG)
	}

	if len(conf.Settings.HighlightHdrFG) > 0 {
		colors[color.Level16]["hdrfg"] = ColorStringToColor(conf.Settings.HighlightHdrFG)
		colors[color.Level256]["hdrfg"] = ColorStringToColor(conf.Settings.HighlightHdrFG)
		colors[color.LevelRgb]["hdrfg"] = ColorStringToColor(conf.Settings.HighlightHdrFG)
	}

	return colors
}

// find supported color mode, modifies config based on constants
func (conf *Config) DetermineColormode() {
	if !isTerminal(os.Stdout) {
		color.Disable()
	} else {
		level := color.TermColorLevel()
		colors := conf.Colors()

		conf.ColorStyle = color.New(colors[level]["bg"], colors[level]["fg"])
		conf.HighlightStyle = color.New(colors[level]["hlbg"], colors[level]["hlfg"])
		conf.NoHighlightStyle = color.New(colors[level]["nohlbg"], colors[level]["nohlfg"])
		conf.HighlightHdrStyle = color.New(colors[level]["hdrbg"], colors[level]["hdrfg"])
	}
}

// Return true if current terminal is interactive
func isTerminal(f *os.File) bool {
	o, _ := f.Stat()

	return (o.Mode() & os.ModeCharDevice) == os.ModeCharDevice
}

// main program version
// generated  version string, used  by -v contains  lib.Version on
//
//	main branch, and lib.Version-$branch-$lastcommit-$date on
//
// development branch
func Getversion() string {
	return fmt.Sprintf("This is tablizer version %s", VERSION)
}

func (conf *Config) PrepareSortFlags(flag Sortmode) {
	switch {
	case flag.Numeric:
		conf.SortMode = "numeric"
	case flag.Age:
		conf.SortMode = "duration"
	case flag.Time:
		conf.SortMode = "time"
	default:
		conf.SortMode = "string"
	}
}

func (conf *Config) PrepareModeFlags(flag Modeflag) {
	switch {
	case flag.X:
		conf.OutputMode = Extended
	case flag.O:
		conf.OutputMode = Orgtbl
	case flag.M:
		conf.OutputMode = Markdown
	case flag.S:
		conf.OutputMode = Shell
	case flag.Y:
		conf.OutputMode = Yaml
	case flag.C:
		conf.OutputMode = CSV
	case flag.J:
		conf.OutputMode = Json
	case conf.Template != "":
		conf.OutputMode = Template
	default:
		conf.OutputMode = ASCII
	}
}

func (conf *Config) PrepareFilters() error {
	conf.Filters = make(map[string]Filter, len(conf.Rawfilters))

	for _, rawfilter := range conf.Rawfilters {
		filter := Filter{}

		parts := strings.Split(rawfilter, "!=")
		if len(parts) != MAXPARTS {
			parts = strings.Split(rawfilter, "=")

			if len(parts) != MAXPARTS {
				return errors.New("filter field and value must be separated by '=' or '!='")
			}
		} else {
			filter.Negate = true
		}

		reg, err := regexp.Compile(parts[1])
		if err != nil {
			return fmt.Errorf("failed to compile filter regex for field %s: %w",
				parts[0], err)
		}

		filter.Regex = reg
		conf.Filters[strings.ToLower(parts[0])] = filter
	}

	return nil
}

// check if transposers match transposer columns and prepare transposer structs
func (conf *Config) PrepareTransposers() error {
	if len(conf.Transposers) != len(conf.UseTransposeColumns) {
		return fmt.Errorf("the number of transposers needs to correspond to the number of transpose columns: %d != %d",
			len(conf.Transposers), len(conf.UseTransposeColumns))
	}

	for _, transposer := range conf.Transposers {
		parts := strings.Split(transposer, string(transposer[0]))
		if len(parts) != 4 {
			return fmt.Errorf("transposer function must have the format /regexp/replace-string/")
		}

		conf.UseTransposers = append(conf.UseTransposers,
			Transposer{
				Search:  *regexp.MustCompile(parts[1]),
				Replace: parts[2]},
		)
	}

	return nil
}

func (conf *Config) CheckEnv() {
	// check for environment vars, command line flags have precedence,
	// NO_COLOR is being checked by the color module itself.
	if !conf.Numbering {
		_, set := os.LookupEnv("T_HEADER_NUMBERING")
		if set {
			conf.Numbering = true
		}
	}

	if len(conf.Columns) == 0 {
		cols := os.Getenv("T_COLUMNS")
		if len(cols) > 1 {
			conf.Columns = cols
		}
	}
}

func (conf *Config) ApplyDefaults() {
	// mode specific defaults
	if conf.OutputMode == Yaml || conf.OutputMode == CSV {
		conf.Numbering = false
	}

	if conf.Separator[0] == ':' && conf.Separator[len(conf.Separator)-1] == ':' {
		separator, ok := SeparatorTemplates[conf.Separator]
		if ok {
			conf.Separator = separator
		}
	}
}

func (conf *Config) PreparePattern(patterns []*Pattern) error {
	// regex checks if a pattern looks like /$pattern/[i!]
	flagre := regexp.MustCompile(`^/(.*)/([i!]*)$`)

	for _, pattern := range patterns {
		matches := flagre.FindAllStringSubmatch(pattern.Pattern, -1)

		// we have a regex with flags
		for _, match := range matches {
			pattern.Pattern = match[1] // the inner part is our actual pattern
			flags := match[2]          // the flags

			for _, flag := range flags {
				switch flag {
				case 'i':
					pattern.Pattern = `(?i)` + pattern.Pattern
				case '!':
					pattern.Negate = true
				}
			}
		}

		PatternRe, err := regexp.Compile(pattern.Pattern)
		if err != nil {
			return fmt.Errorf("regexp pattern %s is invalid: %w", pattern.Pattern, err)
		}

		pattern.PatternRe = PatternRe
	}

	conf.Patterns = patterns

	return nil
}

func (conf *Config) PrepareCustomHeaders(custom string) {
	if len(custom) > 0 {
		conf.CustomHeaders = strings.Split(custom, ",")
	}
}

// Parse config file.  Ignore if the file doesn't exist  but return an
// error if it exists but fails to read or parse
func (conf *Config) ParseConfigfile() error {
	path, err := os.Stat(conf.Configfile)

	if err != nil {
		if os.IsNotExist(err) {
			// ignore non-existent files
			return nil
		}

		return fmt.Errorf("failed to stat config file: %w", err)
	}

	if path.IsDir() {
		// ignore non-existent or dirs
		return nil
	}

	configstring, err := os.ReadFile(path.Name())
	if err != nil {
		return fmt.Errorf("failed to read config file %s: %w", path.Name(), err)
	}

	err = hclsimple.Decode(
		path.Name(),
		configstring,
		nil,
		&conf.Settings)
	if err != nil {
		return fmt.Errorf("failed to load configuration file %s: %w",
			path.Name(), err)
	}

	return nil
}

// translate color string to internal color value
func ColorStringToColor(colorname string) color.Color {
	for name, color := range color.FgColors {
		if name == colorname {
			return color
		}
	}

	for name, color := range color.ExFgColors {
		if name == colorname {
			return color
		}
	}

	return color.Normal
}

// same, for background colors
func ColorStringToBGColor(colorname string) color.Color {
	for name, color := range color.BgColors {
		if name == colorname {
			return color
		}
	}

	for name, color := range color.ExBgColors {
		if name == colorname {
			return color
		}
	}

	return color.Normal
}