feat(db): replace boltdb with sqlite

src/db/rdb/sqlite/sqlite.go

@@ -1,16 +1,21 @@
 package sqlite
 
 import (
+	"context"
 	"database/sql"
+	"encoding/json"
 	"fmt"
-	_ "github.com/mattn/go-sqlite3"
+	"sync"
+	"time"
 
-	"github.com/ihexxa/quickshare/src/db/rdb"
+	"github.com/ihexxa/quickshare/src/db"
+	_ "github.com/mattn/go-sqlite3"
 )
 
 type SQLite struct {
-	rdb.IDB
+	db.IDB
 	dbPath string
+	mtx    *sync.RWMutex
 }
 
 func NewSQLite(dbPath string) (*SQLite, error) {
@@ -24,3 +29,159 @@ func NewSQLite(dbPath string) (*SQLite, error) {
 		dbPath: dbPath,
 	}, nil
 }
+
+func NewSQLiteStore(db db.IDB) (*SQLiteStore, error) {
+	return &SQLiteStore{
+		db:  db,
+		mtx: &sync.RWMutex{},
+	}, nil
+}
+
+func (st *SQLiteStore) Lock() {
+	st.mtx.Lock()
+}
+
+func (st *SQLiteStore) Unlock() {
+	st.mtx.Unlock()
+}
+
+func (st *SQLiteStore) RLock() {
+	st.mtx.RLock()
+}
+
+func (st *SQLiteStore) RUnlock() {
+	st.mtx.RUnlock()
+}
+
+func (st *SQLiteStore) IsInited() bool {
+	// always try to init the db
+	return false
+}
+
+func (st *SQLiteStore) Init(ctx context.Context, rootName, rootPwd string, cfg *db.SiteConfig) error {
+	err := st.InitUserTable(ctx, rootName, rootPwd)
+	if err != nil {
+		return err
+	}
+
+	if err = st.InitFileTables(ctx); err != nil {
+		return err
+	}
+
+	return st.InitConfigTable(ctx, cfg)
+}
+
+func (st *SQLiteStore) InitUserTable(ctx context.Context, rootName, rootPwd string) error {
+	_, err := st.db.ExecContext(
+		ctx,
+		`create table if not exists t_user (
+			id bigint not null,
+			name varchar not null,
+			pwd varchar not null,
+			role integer not null,
+			used_space bigint not null,
+			quota varchar not null,
+			preference varchar not null,
+			primary key(id)
+		)`,
+	)
+	if err != nil {
+		return err
+	}
+
+	admin := &db.User{
+		ID:   0,
+		Name: rootName,
+		Pwd:  rootPwd,
+		Role: db.AdminRole,
+		Quota: &db.Quota{
+			SpaceLimit:         db.DefaultSpaceLimit,
+			UploadSpeedLimit:   db.DefaultUploadSpeedLimit,
+			DownloadSpeedLimit: db.DefaultDownloadSpeedLimit,
+		},
+		Preferences: &db.DefaultPreferences,
+	}
+	visitor := &db.User{
+		ID:   db.VisitorID,
+		Name: db.VisitorName,
+		Pwd:  rootPwd,
+		Role: db.VisitorRole,
+		Quota: &db.Quota{
+			SpaceLimit:         0,
+			UploadSpeedLimit:   db.VisitorUploadSpeedLimit,
+			DownloadSpeedLimit: db.VisitorDownloadSpeedLimit,
+		},
+		Preferences: &db.DefaultPreferences,
+	}
+	for _, user := range []*db.User{admin, visitor} {
+		err = st.AddUser(ctx, user)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (st *SQLiteStore) InitFileTables(ctx context.Context) error {
+	_, err := st.db.ExecContext(
+		ctx,
+		`create table if not exists t_file_info (
+			path varchar not null,
+			user bigint not null,
+			parent varchar not null,
+			name varchar not null,
+			is_dir boolean not null,
+			size bigint not null,
+			share_id varchar not null,
+			info varchar not null,
+			primary key(path)
+		)`,
+	)
+	if err != nil {
+		return err
+	}
+
+	_, err = st.db.ExecContext(
+		ctx,
+		`create table if not exists t_file_uploading (
+			real_path varchar not null,
+			tmp_path varchar not null,
+			user bigint not null,
+			size bigint not null,
+			uploaded bigint not null,
+			primary key(real_path)
+		)`,
+	)
+	return err
+}
+
+func (st *SQLiteStore) InitConfigTable(ctx context.Context, cfg *db.SiteConfig) error {
+	st.Lock()
+	defer st.Unlock()
+
+	_, err := st.db.ExecContext(
+		ctx,
+		`create table if not exists t_config (
+			id bigint not null,
+			config varchar not null,
+			modified datetime not null,
+			primary key(id)
+		)`,
+	)
+	if err != nil {
+		return err
+	}
+
+	cfgStr, err := json.Marshal(cfg)
+	if err != nil {
+		return err
+	}
+	_, err = st.db.ExecContext(
+		ctx,
+		`insert into t_config
+		(id, config, modified) values (?, ?, ?)`,
+		0, cfgStr, time.Now(),
+	)
+	return err
+}