67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/pelletier/go-toml/v2"
|
|
)
|
|
|
|
type (
|
|
TwitchConfig struct {
|
|
ChannelName string `toml:"channel_name"`
|
|
ClientID string `toml:"client_id"`
|
|
ClientSecret string `toml:"client_secret"`
|
|
}
|
|
|
|
Config struct {
|
|
Host string `toml:"host"`
|
|
Port int16 `toml:"port"`
|
|
Twitch TwitchConfig `toml:"twitch"`
|
|
}
|
|
)
|
|
|
|
var CONFIG_DIR = func() string {
|
|
baseDir, _ := os.UserConfigDir()
|
|
dir := path.Join(baseDir, "ari-stream-tools")
|
|
if err := os.MkdirAll(dir, 0750); err != nil { panic(err) }
|
|
return dir
|
|
}()
|
|
var configFilePath = path.Join(CONFIG_DIR, "config.toml")
|
|
|
|
var DEFAULT_HOST string = "127.0.0.1"
|
|
var DEFAULT_PORT int16 = 8080
|
|
|
|
func DefaultConfig() *Config {
|
|
return &Config{
|
|
Host: DEFAULT_HOST,
|
|
Port: DEFAULT_PORT,
|
|
Twitch: TwitchConfig{
|
|
ChannelName: "",
|
|
ClientID: "",
|
|
ClientSecret: "",
|
|
},
|
|
}
|
|
}
|
|
|
|
func LoadConfig() (*Config, error) {
|
|
configFile, err := os.OpenFile(configFilePath, os.O_CREATE | os.O_RDWR, 0600)
|
|
if err != nil { return nil, fmt.Errorf("open %s: %v", configFilePath, err) }
|
|
defer configFile.Close()
|
|
|
|
config := DefaultConfig()
|
|
if stat, err := configFile.Stat(); err != nil {
|
|
return nil, fmt.Errorf("stat %s: %v", configFilePath, err)
|
|
} else if stat.Size() == 0 {
|
|
if err := toml.NewEncoder(configFile).Encode(config); err != nil {
|
|
return nil, fmt.Errorf("write %s: %v", configFilePath, err)
|
|
}
|
|
return config, nil
|
|
}
|
|
|
|
if err := toml.NewDecoder(configFile).Decode(config); err != nil {
|
|
return nil, fmt.Errorf("read %s: %v", configFilePath, err)
|
|
}
|
|
return config, nil
|
|
}
|