split into packages

pkg/db/db.go

@@ -0,0 +1,71 @@
+// Copyright (C) 2017 Marius Schellenberger
+
+package db
+
+import (
+	"bytes"
+	"errors"
+	"github.com/boltdb/bolt"
+)
+
+const (
+	dbFile      = "gowiki.db"
+	defUser     = "admin"
+	defPassword = "gowiki"
+	userTable   = "user"
+	pageTable   = "page"
+	bcryptCost  = 13
+	indexName   = "Index"
+	welcome     = `# Welcome to GoWiki
+This is the [Index](/wiki/Index) page.  
+You can customize it how you like.`
+)
+const (
+	maxResult       = 1e6
+	maxSearchResult = 101
+)
+
+var (
+	errUserNotExists = dbErr("user does not exist")
+	errUserExists    = dbErr("user already exist")
+)
+
+func dbErr(i interface{}) error {
+	return fmt.Errorf("db: %s", i)
+}
+
+type BoltStore struct {
+	Marshaler
+	db *bolt.DB
+}
+
+func NewBoltStore() (bs *BoltStore, err error) {
+	db, err := bolt.Open(dbFile, 0666, nil)
+	if err != nil {
+		return
+	}
+	if err = db.Update(func(tx *bolt.Tx) error {
+		if _, err := tx.CreateBucketIfNotExists([]byte(userTable)); err != nil {
+			return err
+		}
+		if _, err := tx.CreateBucketIfNotExists([]byte(pagesTable)); err != nil {
+			return err
+		}
+		return nil
+	}); err != nil {
+		return
+	}
+	bs = &BoltStore{NewGOB(), db}
+	if !bs.UserExists(defUser) {
+		err = bs.CreateUser(defUser, defPassword, true)
+		if err != nil {
+			return
+		}
+	}
+	bs.UnlockUser(defUser)
+	return
+}
+
+func (bs *BoltStore) Close() error {
+	return bs.db.Close()
+}