author | Ollivier Robert <roberto@keltia.net> |
Tue, 11 Apr 2017 11:51:47 +0200 | |
changeset 66 | 965586c1e3ed |
parent 59 | dd2e302b56d9 |
child 67 | 1ff7afce37fe |
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 { |
66
965586c1e3ed
Change all ID into string, I guess this was to avoir overflow.
Ollivier Robert <roberto@keltia.net>
parents:
59
diff
changeset
|
36 |
ID string |
36
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 |
|
44
f676c2646270
Actually allocate storage before Unmarshall().
Ollivier Robert <roberto@keltia.net>
parents:
42
diff
changeset
|
60 |
cnf := Config{} |
f676c2646270
Actually allocate storage before Unmarshall().
Ollivier Robert <roberto@keltia.net>
parents:
42
diff
changeset
|
61 |
err = toml.Unmarshal(buf, &cnf) |
42
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
62 |
if err != nil { |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
63 |
return c, fmt.Errorf("Error parsing toml %s: %v", file, err) |
23 | 64 |
} |
44
f676c2646270
Actually allocate storage before Unmarshall().
Ollivier Robert <roberto@keltia.net>
parents:
42
diff
changeset
|
65 |
c = &cnf |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
66 |
return |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
67 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
68 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
69 |
func loadInstance(name string) (s *Server, err error) { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
70 |
// Load instance-specific file |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
71 |
file := filepath.Join(baseDir, name+".token") |
23 | 72 |
|
42
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
73 |
log.Printf("instance is %s", file) |
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
74 |
|
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
75 |
// Check if there is any config file |
49
b48dafc1fd4f
Unshadow variable, fix test.
Ollivier Robert <roberto@keltia.net>
parents:
44
diff
changeset
|
76 |
if _, err = os.Stat(file); err == nil { |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
77 |
// Read it |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
78 |
buf, err := ioutil.ReadFile(file) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
79 |
if err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
80 |
return s, fmt.Errorf("Can not read %s", file) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
81 |
} |
23 | 82 |
|
51
4a0533d00053
Allocate stuff the same way everywhere.
Ollivier Robert <roberto@keltia.net>
parents:
50
diff
changeset
|
83 |
sc := Server{} |
50 | 84 |
err = toml.Unmarshal(buf, &sc) |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
85 |
if err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
86 |
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
|
87 |
} |
50 | 88 |
s = &sc |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
89 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
90 |
return |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
91 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
92 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
93 |
func GetInstanceList() (list []string) { |
56
e25d0d0dc8ed
Repeat after me, filepath.Join uses "," to separate args, not "+"
Ollivier Robert <roberto@keltia.net>
parents:
51
diff
changeset
|
94 |
list, err := filepath.Glob(filepath.Join(baseDir, "*.token")) |
e25d0d0dc8ed
Repeat after me, filepath.Join uses "," to separate args, not "+"
Ollivier Robert <roberto@keltia.net>
parents:
51
diff
changeset
|
95 |
log.Printf("basedir=%s", filepath.Join(baseDir, "*.token")) |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
96 |
if err != nil { |
59 | 97 |
log.Printf("warning, no *.token files in %s", baseDir) |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
98 |
list = nil |
23 | 99 |
} |
56
e25d0d0dc8ed
Repeat after me, filepath.Join uses "," to separate args, not "+"
Ollivier Robert <roberto@keltia.net>
parents:
51
diff
changeset
|
100 |
log.Printf("list=%v", list) |
23 | 101 |
return |
102 |
} |
|
103 |
||
104 |
// 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
|
105 |
func LoadConfig(name string) (s *Server, err error) { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
106 |
// Load global file |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
107 |
gFile := filepath.Join(baseDir, DefaultName) |
23 | 108 |
|
42
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
109 |
log.Printf("global is %s", gFile) |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
110 |
c, err := loadGlobal(gFile) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
111 |
if err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
112 |
return |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
113 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
114 |
if name == "" { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
115 |
s, err = loadInstance(c.Default) |
23 | 116 |
} else { |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
117 |
s, err = loadInstance(name) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
118 |
} |
23 | 119 |
|
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
120 |
return s, err |
23 | 121 |
} |
122 |
||
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
123 |
func (c *Config) Write() (err error) { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
124 |
if err = os.MkdirAll(baseDir, 0700); err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
125 |
log.Fatalf("error creating configuration directory: %s", baseDir) |
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 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
128 |
var sc []byte |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
129 |
|
42
2bebcae94ce1
Add a few checks & logging.
Ollivier Robert <roberto@keltia.net>
parents:
36
diff
changeset
|
130 |
if sc, err = toml.Marshal(*c); err != nil { |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
131 |
log.Fatalf("error saving configuration") |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
132 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
133 |
err = ioutil.WriteFile(filepath.Join(baseDir, DefaultName), sc, 0600) |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
134 |
return |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
135 |
} |
23 | 136 |
|
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
137 |
func (s *Server) WriteToken(instance string) (err error) { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
138 |
if err = os.MkdirAll(baseDir, 0700); err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
139 |
log.Fatalf("error creating configuration directory: %s", baseDir) |
23 | 140 |
} |
36
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 |
var sc []byte |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
143 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
144 |
if sc, err = toml.Marshal(s); err != nil { |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
145 |
log.Fatalf("error saving configuration") |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
146 |
} |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
147 |
|
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
148 |
full := instance + ".token" |
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
149 |
err = ioutil.WriteFile(filepath.Join(baseDir, full), sc, 0600) |
23 | 150 |
return |
36
b3c341207b30
Properly load/write configuration files.
Ollivier Robert <roberto@keltia.net>
parents:
23
diff
changeset
|
151 |
} |