54 lines
953 B
Go
54 lines
953 B
Go
|
package storage
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"os/user"
|
||
|
|
"path"
|
||
|
|
|
||
|
|
"github.com/khlieng/name_pending/Godeps/_workspace/src/github.com/boltdb/bolt"
|
||
|
|
|
||
|
|
"github.com/khlieng/name_pending/args"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
appDir string
|
||
|
|
|
||
|
|
db *bolt.DB
|
||
|
|
|
||
|
|
bucketUsers = []byte("Users")
|
||
|
|
bucketServers = []byte("Servers")
|
||
|
|
bucketChannels = []byte("Channels")
|
||
|
|
bucketMessages = []byte("Messages")
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
var err error
|
||
|
|
currentUser, _ := user.Current()
|
||
|
|
appDir = path.Join(currentUser.HomeDir, ".name_pending")
|
||
|
|
|
||
|
|
if args.Development {
|
||
|
|
os.RemoveAll(appDir)
|
||
|
|
}
|
||
|
|
|
||
|
|
os.Mkdir(appDir, 0777)
|
||
|
|
os.Mkdir(path.Join(appDir, "logs"), 0777)
|
||
|
|
|
||
|
|
db, err = bolt.Open(path.Join(appDir, "data.db"), 0600, nil)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal("Could not open database file")
|
||
|
|
}
|
||
|
|
|
||
|
|
db.Update(func(tx *bolt.Tx) error {
|
||
|
|
tx.CreateBucketIfNotExists(bucketUsers)
|
||
|
|
tx.CreateBucketIfNotExists(bucketServers)
|
||
|
|
tx.CreateBucketIfNotExists(bucketChannels)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func Cleanup() {
|
||
|
|
db.Close()
|
||
|
|
}
|