all repos — paprika @ db1b9b0a0c4855bc2946ef41aa87ce019a33c73e

go rewrite of taigabot

config/config.go (view raw)

 1package config
 2
 3import (
 4	"io"
 5	"log"
 6	"os"
 7
 8	"gopkg.in/yaml.v3"
 9)
10
11type Config struct {
12	Nick string
13	Pass string
14	Host string
15	Sasl string
16	Tls bool
17	Channels []string
18	DbPath string `yaml:"db-path"`
19	ApiKeys map[string]string `yaml:"api-keys"`
20}
21
22var C Config
23
24func init() {
25	configPath := ""
26	inC := false
27	initMode := false
28	for _, v := range os.Args {
29		switch v {
30		case "-c", "--config":
31			inC = true
32		case "-h", "--help":
33			usage()
34		default:
35			if inC {
36				configPath = v
37			} else if v == "init" {
38				initMode = true
39			} else if v == "help" {
40				usage()
41			}
42		}
43	}
44
45	if initMode {
46		createConfig(configPath)
47	}
48
49	var file *os.File
50	var err error
51	if configPath == "" {
52		file = findConfig()
53	} else {
54		file, err = os.Open(configPath)
55		if err != nil {
56			log.Printf("Error: Could not open config path.")
57			log.Fatalf("- %s", err)
58		}
59	}
60	defer file.Close()
61
62	data, err := io.ReadAll(file)
63	err = yaml.Unmarshal(data, &C)
64	if err != nil {
65		log.Fatal(err)
66	}
67
68	if C.DbPath == "" {
69		C.DbPath = getDbPath()
70	}
71}
72