Add message logging and search server side

storage/storage.go

@@ -0,0 +1,53 @@
+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()
+}