package config

import (
	"os"

	"github.com/sirupsen/logrus"
	"gopkg.in/yaml.v3"
)

type SqliteConfig struct {
	Dbname string `yaml:"dbname"`
	Debug  bool   `yaml:"debug"`
}

type ServerConfig struct {
	ServerName   string       `yaml:"server-name"`
	ServerListen string       `yaml:"server-listen"`
	Environment  string       `yaml:"environment"`
	DBConfig     SqliteConfig `yaml:"db"`
}

type ServerConf struct {
	ConfigFile   string
	ServerConfig `yaml:",inline"`
}

var conf *ServerConf

func New(fileName string) {
	conf = new(ServerConf)
	conf.ConfigFile = fileName
	if err := conf.Load(); err != nil {
		logrus.Fatalf("read config file failed: %v", err)
	}
}

func Get() *ServerConf {
	if conf == nil {
		panic("config file not initialized")
	}
	return conf
}

func (receiver *ServerConf) Load() error {
	data, err := os.ReadFile(receiver.ConfigFile)
	if err != nil && !os.IsNotExist(err) {
		return err
	}
	if err := yaml.Unmarshal(data, receiver); err != nil {
		return err
	}
	return nil
}

func (receiver *ServerConf) Rewrite() error {
	data, err := yaml.Marshal(receiver)
	if err != nil {
		return err
	}
	if err := os.WriteFile(receiver.ConfigFile, data, os.ModePerm); err != nil {
		return err
	}
	return nil
}