feat(db): add site store

src/db/sitestore/site_store.go

@@ -0,0 +1,115 @@
+package sitestore
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"sync"
+	"time"
+
+	"github.com/ihexxa/quickshare/src/kvstore"
+)
+
+const (
+	InitNs      = "SiteInit"
+	SiteNs      = "Site"
+	InitTimeKey = "initTime"
+	SiteCfgKey  = "siteCfg"
+)
+
+var (
+	ErrNotFound = errors.New("site config not found")
+)
+
+func IsNotFound(err error) bool {
+	return err == ErrNotFound
+}
+
+type ISiteStore interface {
+	SetClientCfg(cfg *ClientConfig) error
+	GetCfg() (*SiteConfig, error)
+}
+
+type ClientConfig struct {
+	SiteName string    `json:"siteName"`
+	SiteDesc string    `json:"siteDesc"`
+	Bg       *BgConfig `json:"bg"`
+}
+
+type BgConfig struct {
+	Url      string `json:"url"`
+	Repeat   string `json:"repeat"`
+	Position string `json:"position"`
+	Align    string `json:"align"`
+}
+
+type SiteConfig struct {
+	ClientCfg *ClientConfig `json:"clientCfg"`
+}
+
+type SiteStore struct {
+	mtx   *sync.RWMutex
+	store kvstore.IKVStore
+}
+
+func NewSiteStore(store kvstore.IKVStore) (*SiteStore, error) {
+	_, ok := store.GetStringIn(InitNs, InitTimeKey)
+	if !ok {
+		var err error
+		for _, nsName := range []string{
+			InitNs,
+			SiteNs,
+		} {
+			if err = store.AddNamespace(nsName); err != nil {
+				return nil, err
+			}
+		}
+	}
+
+	err := store.SetStringIn(InitNs, InitTimeKey, fmt.Sprintf("%d", time.Now().Unix()))
+	if err != nil {
+		return nil, err
+	}
+
+	return &SiteStore{
+		store: store,
+		mtx:   &sync.RWMutex{},
+	}, nil
+}
+
+func (fi *SiteStore) SetClientCfg(cfg *ClientConfig) error {
+	fi.mtx.Lock()
+	defer fi.mtx.Unlock()
+
+	siteCfg := &SiteConfig{}
+	cfgStr, ok := fi.store.GetStringIn(SiteNs, SiteCfgKey)
+	if ok {
+		err := json.Unmarshal([]byte(cfgStr), siteCfg)
+		if err != nil {
+			return err
+		}
+	}
+	siteCfg.ClientCfg = cfg
+
+	cfgBytes, err := json.Marshal(siteCfg)
+	if err != nil {
+		return err
+	}
+	return fi.store.SetStringIn(SiteNs, SiteCfgKey, string(cfgBytes))
+}
+
+func (fi *SiteStore) GetCfg() (*SiteConfig, error) {
+	fi.mtx.RLock()
+	defer fi.mtx.RUnlock()
+
+	cfgStr, ok := fi.store.GetStringIn(SiteNs, SiteCfgKey)
+	if !ok {
+		return nil, ErrNotFound
+	}
+	siteCfg := &SiteConfig{}
+	err := json.Unmarshal([]byte(cfgStr), siteCfg)
+	if err != nil {
+		return nil, err
+	}
+	return siteCfg, nil
+}

src/db/sitestore/site_store_test.go

@@ -0,0 +1,57 @@
+package sitestore
+
+import (
+	"io/ioutil"
+	"os"
+	"reflect"
+	"testing"
+
+	"github.com/ihexxa/quickshare/src/kvstore/boltdbpvd"
+)
+
+func TestSiteStore(t *testing.T) {
+
+	testSiteMethods := func(t *testing.T, store ISiteStore) {
+		siteCfg := &SiteConfig{
+			ClientCfg: &ClientConfig{
+				SiteName: "quickshare",
+				SiteDesc: "simpel file sharing",
+				Bg: &BgConfig{
+					Url:      "/imgs/bg.jpg",
+					Repeat:   "no-repeat",
+					Position: "fixed",
+					Align:    "center",
+				},
+			},
+		}
+
+		err := store.SetClientCfg(siteCfg.ClientCfg)
+		if err != nil {
+			t.Fatal(err)
+		}
+		newSiteCfg, err := store.GetCfg()
+		if err != nil {
+			t.Fatal(err)
+		} else if !reflect.DeepEqual(newSiteCfg, siteCfg) {
+			t.Fatalf("not equal new(%v) original(%v)", newSiteCfg, siteCfg)
+		}
+	}
+
+	t.Run("Get/Set", func(t *testing.T) {
+		rootPath, err := ioutil.TempDir("./", "quickshare_sitestore_test_")
+		if err != nil {
+			t.Fatal(err)
+		}
+		defer os.RemoveAll(rootPath)
+
+		kvstore := boltdbpvd.New(rootPath, 1024)
+		defer kvstore.Close()
+
+		store, err := NewSiteStore(kvstore)
+		if err != nil {
+			t.Fatal("fail to new kvstore", err)
+		}
+
+		testSiteMethods(t, store)
+	})
+}