/
config.go
105 lines (90 loc) · 2.01 KB
/
config.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package main
import (
"bufio"
"bytes"
"fmt"
"io/ioutil"
"os"
"strconv"
"github.com/BurntSushi/toml"
)
type Config struct {
Yahoo YahooConfig `toml:"Yahoo"`
Slack SlackConfig `toml:"Slack"`
General GeneralConfig `toml:"General"`
}
type YahooConfig struct {
Token string `toml:"token"`
}
type SlackConfig struct {
Token string `toml:"token"`
Channel string `toml:"channel"`
NDaysAgo int `toml:"n_days_ago"`
}
type GeneralConfig struct {
Filename string `toml:"filename"`
}
const configFile = "config.toml"
// 設定ファイルの存在チェック
func (c *Config) FileExists() bool {
_, err := os.Stat(configFile)
return err == nil
}
func (c *Config) scan() string {
sc := bufio.NewScanner(os.Stdin)
sc.Scan()
input := sc.Text()
return input
}
func (c *Config) LoadConfig() error {
_, err := toml.DecodeFile(configFile, &c)
if err != nil {
return err
}
return nil
}
func (c *Config) CreateConfig() error {
var yahooConfig YahooConfig
fmt.Print("Input YahooToken: ")
yahooConfig.Token = c.scan()
var slackConfig SlackConfig
fmt.Print("Input SlackToken: ")
slackConfig.Token = c.scan()
fmt.Print("Input SlackChannel: ")
slackConfig.Channel = c.scan()
fmt.Print("Input SlackNDaysAgo: ")
nDaysAgo, err := strconv.Atoi(c.scan())
if err != nil {
return err
}
slackConfig.NDaysAgo = nDaysAgo
var generalConfig GeneralConfig
fmt.Print("Input Upload Filename (.gif): ")
generalConfig.Filename = c.scan()
c.Yahoo = yahooConfig
c.Slack = slackConfig
c.General = generalConfig
var buffer bytes.Buffer
encoder := toml.NewEncoder(&buffer)
if err := encoder.Encode(c); err != nil {
return err
}
ioutil.WriteFile(configFile, []byte(buffer.String()), 0644)
return nil
}
func NewConfig() (*Config, error) {
c := &Config{}
if isFile := c.FileExists(); isFile {
if err := c.LoadConfig(); err != nil {
return nil, err
}
} else {
if err := c.CreateConfig(); err != nil {
return nil, err
}
if err := c.LoadConfig(); err != nil {
return nil, err
}
}
return c, nil
}