-
Notifications
You must be signed in to change notification settings - Fork 0
/
load.go
123 lines (114 loc) · 2.97 KB
/
load.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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package main
import (
"io/ioutil"
"log"
"os"
"path/filepath"
"github.com/ghodss/yaml"
)
// LoadMembers loads all members with MemberStatus "Regular" and "Periodic".
// Each call to LoadMembers loads and parses them directly from the YAML files
// with no caching. Since the amount of data in these files is minimal having
// the freshest data is more important than the unnecessary optimization that
// could come from caching.
func LoadMembers() []*Member {
members := []*Member{}
regulars := LoadMembersByStatus("regular")
periodic := LoadMembersByStatus("periodic")
if regulars != nil {
members = append(members, regulars...)
}
if periodic != nil {
members = append(members, periodic...)
}
return members
}
// LoadAllMembers loads all members with any status. See LoadMembers for more.
func LoadAllMembers() []*Member {
members := []*Member{}
regulars := LoadMembersByStatus("regular")
periodic := LoadMembersByStatus("periodic")
former := LoadMembersByStatus("former")
banned := LoadMembersByStatus("banned")
if regulars != nil {
members = append(members, regulars...)
}
if periodic != nil {
members = append(members, periodic...)
}
if former != nil {
members = append(members, former...)
}
if banned != nil {
members = append(members, banned...)
}
return members
}
func LoadMembersByStatus(status string) []*Member {
if !(status == "regular" || status == "periodic" || status == "former" || status == "banned") {
log.Printf("no such status: \"%v\"", status)
return nil
}
members := []*Member{}
mdir := filepath.Join(Dir(), "members", status)
dir, err := os.Open(mdir)
if err != nil {
log.Printf("failed to load members from %v: %v", mdir, err)
return nil
}
names, _ := dir.Readdirnames(0)
for _, name := range names {
path := filepath.Join(mdir, name, "member.yml")
m := LoadMember(path)
m.Status = status
if m != nil {
members = append(members, m)
}
}
return members
}
func LoadMember(path string) *Member {
m := new(Member)
byt, err := ioutil.ReadFile(path)
if err != nil {
log.Printf("failed to read member at %v: %v", path, err)
return nil
}
err = yaml.Unmarshal(byt, m)
if err != nil {
log.Printf("failed to parse YAML for member at %v: %v", path, err)
return nil
}
return m
}
// TODO consolidate Load*
func LoadHours() Hours {
h := Hours{}
path := filepath.Join(Dir(), "schedule", "hours.yml")
byt, err := ioutil.ReadFile(path)
if err != nil {
log.Printf("failed to read Hours at %v: %v", path, err)
return nil
}
err = yaml.Unmarshal(byt, &h)
if err != nil {
log.Printf("failed to parse YAML for Hours at %v: %v", path, err)
return nil
}
return h
}
func LoadConfig() Config {
conf := Config{}
path := filepath.Join(Dir(), "config.yml")
byt, err := ioutil.ReadFile(path)
if err != nil {
log.Printf("failed to read Config at %v: %v", path, err)
return conf
}
err = yaml.Unmarshal(byt, &conf)
if err != nil {
log.Printf("failed to parse YAML for Config at %v: %v", path, err)
return conf
}
return conf
}