initial commit

backend/database/db.go

@@ -0,0 +1,79 @@
+package database
+
+import (
+	"os"
+	"path"
+	"s-ui/config"
+	"s-ui/database/model"
+
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+)
+
+var db *gorm.DB
+
+func initUser() error {
+	var count int64
+	err := db.Model(&model.User{}).Count(&count).Error
+	if err != nil {
+		return err
+	}
+	if count == 0 {
+		user := &model.User{
+			Username: "admin",
+			Password: "admin",
+		}
+		return db.Create(user).Error
+	}
+	return nil
+}
+
+func InitDB(dbPath string) error {
+	dir := path.Dir(dbPath)
+	err := os.MkdirAll(dir, 01740)
+	if err != nil {
+		return err
+	}
+
+	var gormLogger logger.Interface
+
+	if config.IsDebug() {
+		gormLogger = logger.Default
+	} else {
+		gormLogger = logger.Discard
+	}
+
+	c := &gorm.Config{
+		Logger: gormLogger,
+	}
+	db, err = gorm.Open(sqlite.Open(dbPath), c)
+	if err != nil {
+		return err
+	}
+
+	err = db.AutoMigrate(
+		&model.Setting{},
+		&model.User{},
+		&model.Stats{},
+		&model.Client{},
+		&model.Changes{},
+	)
+	if err != nil {
+		return err
+	}
+	err = initUser()
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func GetDB() *gorm.DB {
+	return db
+}
+
+func IsNotFound(err error) bool {
+	return err == gorm.ErrRecordNotFound
+}

backend/database/model/model.go

@@ -0,0 +1,48 @@
+package model
+
+import "encoding/json"
+
+type Setting struct {
+	Id    uint   `json:"id" form:"id" gorm:"primaryKey;autoIncrement"`
+	Key   string `json:"key" form:"key"`
+	Value string `json:"value" form:"value"`
+}
+
+type User struct {
+	Id         uint   `json:"id" form:"id" gorm:"primaryKey;autoIncrement"`
+	Username   string `json:"username" form:"username"`
+	Password   string `json:"password" form:"password"`
+	LastLogins string `json:"lastLogin"`
+}
+
+type Client struct {
+	Id       uint   `json:"id" form:"id" gorm:"primaryKey;autoIncrement"`
+	Enable   bool   `json:"enable" form:"enable"`
+	Name     string `json:"name" form:"name"`
+	Config   string `json:"config" form:"config"`
+	Inbounds string `json:"inbounds" form:"inbounds"`
+	Links    string `json:"links" form:"links"`
+	Volume   int64  `json:"volume" form:"volume"`
+	Expiry   int64  `json:"expiry" form:"expiry"`
+	Down     int64  `json:"down" form:"down"`
+	Up       int64  `json:"up" form:"up"`
+}
+
+type Stats struct {
+	Id        uint64 `json:"id" gorm:"primaryKey;autoIncrement"`
+	DateTime  int64  `json:"dateTime"`
+	Resource  string `json:"resource"`
+	Tag       string `json:"tag"`
+	Direction bool   `json:"direction"`
+	Traffic   int64  `json:"traffic"`
+}
+
+type Changes struct {
+	Id       uint64          `json:"id" gorm:"primaryKey;autoIncrement"`
+	DateTime int64           `json:"dateTime"`
+	Actor    string          `json:"Actor"`
+	Key      string          `json:"key" form:"key"`
+	Action   string          `json:"action" form:"action"`
+	Index    uint            `json:"index" form:"index"`
+	Obj      json.RawMessage `json:"obj" form:"obj"`
+}