package config

import (
	"github.com/spf13/viper"
)

var config *Config

type Config struct {
	ChatGpt ChatGptConfig `json:"chatgpt"`
	Slack   SlackConfig   `json:"slack"`
}

type ChatGptConfig struct {
	Keyword string `json:"keyword,omitempty"`
	Token   string `json:"token,omitempty" json:"token,omitempty"`
}

type SlackConfig struct {
	Token string `json:"token"`
}

func LoadConfig() error {
	viper.SetConfigName("config")
	viper.SetConfigType("yaml")
	viper.AddConfigPath("./local")
	viper.AddConfigPath("./config")

	if err := viper.ReadInConfig(); err != nil {
		return err
	}
	if err := viper.Unmarshal(&config); err != nil {
		return err
	}
	return nil
}

func GetConfig() *Config {
	return config
}