package config

import (
	"log/slog"
	"net/url"
	"os"
	"searchix/internal/file"

	"github.com/pelletier/go-toml/v2"
	"github.com/pkg/errors"
)

type URL struct {
	*url.URL
}

func (u *URL) UnmarshalText(text []byte) (err error) {
	u.URL, err = url.Parse(string(text))

	if err != nil {
		return errors.WithMessage(err, "could not parse URL")
	}

	return nil
}

type Config struct {
	DataPath string `toml:"data_path"`
	CSP      CSP    `toml:"content-security-policy"`
	Headers  map[string]string
}

var defaultConfig = Config{
	DataPath: "./data",
	CSP: CSP{
		DefaultSrc: []string{"'self'"},
	},
	Headers: map[string]string{
		"x-content-type-options": "nosniff",
	},
}

func GetConfig() (*Config, error) {
	config := defaultConfig
	slog.Debug("reading config.toml")
	f, err := os.Open("config.toml")
	if err := file.NeedNotExist(err); err != nil {
		return nil, errors.Wrap(err, "reading config.toml failed")
	}
	if f != nil {
		defer f.Close()
		dec := toml.NewDecoder(f)
		err = dec.Decode(&config)
		if err != nil {
			var tomlError toml.DecodeError
			if errors.As(err, &tomlError) {
				return nil, errors.WithMessage(err, tomlError.Error())
			}

			return nil, errors.Wrap(err, "config error")
		}
	}

	return &config, nil
}