Multi users data [1/N]

Signed-off-by: Yuan Si <do4suki@gmail.com>
2023-04-12 16:05:24 +08:00
parent fe1e012127
commit dd2da649c1
15 changed files with 1501 additions and 225 deletions
@@ -4,21 +4,32 @@
 package config

 import (
+	"honoka-chan/utils"
+	"os"
+
 	_ "github.com/mattn/go-sqlite3"
 	"xorm.io/xorm"
 )

 var (
-	ConfName = "config.yml"
-	Conf     = &AppConfigs{}
-	MainEng  *xorm.Engine
-	UserEng  *xorm.Engine
+	ConfName  = "config.yml"
+	Conf      = &AppConfigs{}
+	ExampleDb = "assets/data.example.db"
+	MainDb    = "assets/main.db"
+	UserDb    = "assets/data.db"
+	MainEng   *xorm.Engine
+	UserEng   *xorm.Engine
 )

 func init() {
 	Conf = Load(ConfName)

-	eng, err := xorm.NewEngine("sqlite3", "assets/main.db")
+	_, err := os.Stat(UserDb)
+	if err != nil {
+		utils.WriteAllText(UserDb, utils.ReadAllText(ExampleDb))
+	}
+
+	eng, err := xorm.NewEngine("sqlite3", MainDb)
 	if err != nil {
 		panic(err)
 	}
@@ -30,7 +41,7 @@ func init() {
 	MainEng.SetMaxOpenConns(50)
 	MainEng.SetMaxIdleConns(10)

-	eng, err = xorm.NewEngine("sqlite3", "assets/account.db")
+	eng, err = xorm.NewEngine("sqlite3", UserDb)
 	if err != nil {
 		panic(err)
 	}