dispatch/storage/storage.go

package storage

import (
	"log"
	"os"
	"os/user"
	"path"

	"github.com/khlieng/name_pending/Godeps/_workspace/src/github.com/boltdb/bolt"
)

var (
	appDir string

	db *bolt.DB

	bucketUsers    = []byte("Users")
	bucketServers  = []byte("Servers")
	bucketChannels = []byte("Channels")
	bucketMessages = []byte("Messages")
)

func Initialize(clean bool) {
	var err error
	currentUser, _ := user.Current()
	appDir = path.Join(currentUser.HomeDir, ".name_pending")

	if clean {
		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()
}