package config

import (
	"bytes"

	"github.com/BurntSushi/toml"
)

type Config struct {
	DebugMode             bool             `toml:"debug"`
	Slomo                 bool             `toml:"slomo"`
	ColourScheme          ColourScheme     `toml:"colours"`
	DPIScale              float32          `toml:"dpi-scale"`
	Shell                 string           `toml:"shell"`
	KeyMapping            KeyMappingConfig `toml:"keys"`
	SearchURL             string           `toml:"search_url"`
	MaxLines              uint64           `toml:"max_lines"`
	CopyAndPasteWithMouse bool             `toml:"copy_and_paste_with_mouse"`
	Prof                  string           `toml:"prof"`
	Threads               int              `toml:"threads"`
}

type KeyMappingConfig map[string]string

func Parse(data []byte) (*Config, error) {
	c := DefaultConfig
	err := toml.Unmarshal(data, &c)
	if c.KeyMapping == nil {
		c.KeyMapping = KeyMappingConfig(map[string]string{})
	}
	return &c, err
}

func (c *Config) Encode() ([]byte, error) {
	var buf bytes.Buffer
	e := toml.NewEncoder(&buf)
	err := e.Encode(c)
	if err != nil {
		return nil, err
	}
	return buf.Bytes(), nil
}