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
124
125
126
127
128
129
130
131
|
package db
import (
"encoding/json"
"fmt"
"log"
"github.com/ponzu-cms/ponzu/system/admin/config"
"github.com/ponzu-cms/ponzu/system/item"
"github.com/boltdb/bolt"
"github.com/nilslice/jwt"
)
var store *bolt.DB
// Close exports the abillity to close our db file. Should be called with defer
// after call to Init() from the same place.
func Close() {
err := store.Close()
if err != nil {
log.Println(err)
}
}
// Init creates a db connection, initializes db with required info, sets secrets
func Init() {
fmt.Println("db.Init inside db package")
if store != nil {
fmt.Println("db.Init already intialized store")
return
}
var err error
store, err = bolt.Open("system.db", 0666, nil)
if err != nil {
log.Fatalln(err)
}
err = store.Update(func(tx *bolt.Tx) error {
// initialize db with all content type buckets & sorted bucket for type
for t := range item.Types {
_, err := tx.CreateBucketIfNotExists([]byte(t))
if err != nil {
return err
}
_, err = tx.CreateBucketIfNotExists([]byte(t + "__sorted"))
if err != nil {
return err
}
}
// init db with other buckets as needed
buckets := []string{"__config", "__users", "__contentIndex", "__addons"}
for _, name := range buckets {
_, err := tx.CreateBucketIfNotExists([]byte(name))
if err != nil {
return err
}
}
// seed db with configs structure if not present
b := tx.Bucket([]byte("__config"))
if b.Get([]byte("settings")) == nil {
j, err := json.Marshal(&config.Config{})
if err != nil {
return err
}
err = b.Put([]byte("settings"), j)
if err != nil {
return err
}
}
clientSecret := ConfigCache("client_secret")
if clientSecret != "" {
jwt.Secret([]byte(clientSecret))
}
return nil
})
if err != nil {
log.Fatalln("Coudn't initialize db with buckets.", err)
}
fmt.Println("db initialization completed")
// invalidate cache on system start
err = InvalidateCache()
if err != nil {
log.Fatalln("Failed to invalidate cache.", err)
}
fmt.Println("Cache invalidated")
go func() {
for t := range item.Types {
SortContent(t)
}
fmt.Println("content sorted (from goroutine)")
}()
}
// SystemInitComplete checks if there is at least 1 admin user in the db which
// would indicate that the system has been configured to the minimum required.
func SystemInitComplete() bool {
complete := false
err := store.View(func(tx *bolt.Tx) error {
users := tx.Bucket([]byte("__users"))
err := users.ForEach(func(k, v []byte) error {
complete = true
return nil
})
if err != nil {
return err
}
return nil
})
if err != nil {
complete = false
log.Fatalln(err)
}
return complete
}
|
