author | Ollivier Robert <roberto@keltia.net> |
Mon, 10 Apr 2017 17:14:56 +0200 | |
changeset 42 | 2bebcae94ce1 |
parent 36 | b3c341207b30 |
child 44 | f676c2646270 |
permissions | -rw-r--r-- |
23 | 1 |
// config.go |
2 |
// |
|
3 |
// This file implements the configuration part for when you need the API |
|
4 |
// key to modify things in the Mastodon configuration and manage measurements. |
|
5 |
||
8 | 6 |
package gondole |
23 | 7 |
|
8 |
import ( |
|
9 |
"fmt" |
|
10 |
"io/ioutil" |
|
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
11 |
"log" |
23 | 12 |
"os" |
13 |
"path/filepath" |
|
14 |
||
15 |
"github.com/naoina/toml" |
|
16 |
) |
|
17 |
||
18 |
/* |
|
19 |
Assume the application is registered if $HOME/.config/<gondole>/config.toml already exist |
|
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
20 |
We will store the per-instance token into $HOME/.config/<gondole>/<site>.token |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
21 |
*/ |
23 | 22 |
|
23 |
const ( |
|
24 |
DefaultName = "config.toml" |
|
25 |
) |
|
26 |
||
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
27 |
var ( |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
28 |
baseDir = filepath.Join(os.Getenv("HOME"), |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
29 |
".config", |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
30 |
"gondole", |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
31 |
) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
32 |
) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
33 |
|
23 | 34 |
// Config holds our parameters |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
35 |
type Server struct { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
36 |
ID int64 |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
37 |
Name string |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
38 |
BearerToken string |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
39 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
40 |
|
23 | 41 |
type Config struct { |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
42 |
Default string |
23 | 43 |
} |
44 |
||
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
45 |
func loadGlobal(file string) (c *Config, err error) { |
42
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
46 |
log.Printf("file=%s", file) |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
47 |
// Check if there is any config file |
42
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
48 |
_, err = os.Stat(file) |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
49 |
if err != nil { |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
50 |
return |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
51 |
} |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
52 |
|
42
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
53 |
log.Printf("file=%s, found it", file) |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
54 |
// Read it |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
55 |
buf, err := ioutil.ReadFile(file) |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
56 |
if err != nil { |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
57 |
return c, fmt.Errorf("Can not read %s", file) |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
58 |
} |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
59 |
|
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
60 |
err = toml.Unmarshal(buf, c) |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
61 |
if err != nil { |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
62 |
return c, fmt.Errorf("Error parsing toml %s: %v", file, err) |
23 | 63 |
} |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
64 |
return |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
65 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
66 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
67 |
func loadInstance(name string) (s *Server, err error) { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
68 |
// Load instance-specific file |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
69 |
file := filepath.Join(baseDir, name+".token") |
23 | 70 |
|
42
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
71 |
log.Printf("instance is %s", file) |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
72 |
|
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
73 |
// Check if there is any config file |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
74 |
if _, err := os.Stat(file); err == nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
75 |
// Read it |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
76 |
buf, err := ioutil.ReadFile(file) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
77 |
if err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
78 |
return s, fmt.Errorf("Can not read %s", file) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
79 |
} |
23 | 80 |
|
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
81 |
err = toml.Unmarshal(buf, s) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
82 |
if err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
83 |
return s, fmt.Errorf("Error parsing toml %s: %v", file, err) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
84 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
85 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
86 |
return |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
87 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
88 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
89 |
func GetInstanceList() (list []string) { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
90 |
list, err := filepath.Glob(filepath.Join(baseDir + "*.token")) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
91 |
if err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
92 |
log.Printf("warning, no *.token files") |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
93 |
list = nil |
23 | 94 |
} |
95 |
return |
|
96 |
} |
|
97 |
||
98 |
// LoadConfig reads a file as a TOML document and return the structure |
|
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
99 |
func LoadConfig(name string) (s *Server, err error) { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
100 |
// Load global file |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
101 |
gFile := filepath.Join(baseDir, DefaultName) |
23 | 102 |
|
42
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
103 |
log.Printf("global is %s", gFile) |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
104 |
c, err := loadGlobal(gFile) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
105 |
if err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
106 |
return |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
107 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
108 |
if name == "" { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
109 |
s, err = loadInstance(c.Default) |
23 | 110 |
} else { |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
111 |
s, err = loadInstance(name) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
112 |
} |
23 | 113 |
|
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
114 |
return s, err |
23 | 115 |
} |
116 |
||
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
117 |
func (c *Config) Write() (err error) { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
118 |
if err = os.MkdirAll(baseDir, 0700); err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
119 |
log.Fatalf("error creating configuration directory: %s", baseDir) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
120 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
121 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
122 |
var sc []byte |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
123 |
|
42
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
124 |
if sc, err = toml.Marshal(*c); err != nil { |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
125 |
log.Fatalf("error saving configuration") |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
126 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
127 |
err = ioutil.WriteFile(filepath.Join(baseDir, DefaultName), sc, 0600) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
128 |
return |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
129 |
} |
23 | 130 |
|
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
131 |
func (s *Server) WriteToken(instance string) (err error) { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
132 |
if err = os.MkdirAll(baseDir, 0700); err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
133 |
log.Fatalf("error creating configuration directory: %s", baseDir) |
23 | 134 |
} |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
135 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
136 |
var sc []byte |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
137 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
138 |
if sc, err = toml.Marshal(s); err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
139 |
log.Fatalf("error saving configuration") |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
140 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
141 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
142 |
full := instance + ".token" |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
143 |
err = ioutil.WriteFile(filepath.Join(baseDir, full), sc, 0600) |
23 | 144 |
return |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
145 |
} |