refactor: migrate to new cosy

model/config_backup.go

@@ -1,7 +1,7 @@
 package model
 
 import (
-	"github.com/0xJacky/Nginx-UI/internal/logger"
+	"github.com/uozi-tech/cosy/logger"
 	"os"
 	"path/filepath"
 )

model/model.go

@@ -1,15 +1,8 @@
 package model
 
 import (
-	"fmt"
-	"github.com/0xJacky/Nginx-UI/internal/logger"
-	"github.com/0xJacky/Nginx-UI/settings"
-	"github.com/gin-gonic/gin"
-	"gorm.io/driver/sqlite"
 	"gorm.io/gen"
 	"gorm.io/gorm"
-	gormlogger "gorm.io/gorm/logger"
-	"path"
 	"time"
 )
 
@@ -38,47 +31,18 @@ func GenerateAllModel() []any {
 		BanIP{},
 		Config{},
 		Passkey{},
+		SiteCategory{},
 	}
 }
 
-func logMode() gormlogger.Interface {
-	switch settings.ServerSettings.RunMode {
-	case gin.ReleaseMode:
-		return gormlogger.Default.LogMode(gormlogger.Warn)
-	default:
-		fallthrough
-	case gin.DebugMode:
-		return gormlogger.Default.LogMode(gormlogger.Info)
-	}
+func Use(tx *gorm.DB) {
+	db = tx
 }
 
 func UseDB() *gorm.DB {
 	return db
 }
 
-func Init() *gorm.DB {
-	dbPath := path.Join(path.Dir(settings.ConfPath), fmt.Sprintf("%s.db", settings.ServerSettings.Database))
-
-	var err error
-	db, err = gorm.Open(sqlite.Open(dbPath), &gorm.Config{
-		Logger:                                   logMode(),
-		PrepareStmt:                              true,
-		DisableForeignKeyConstraintWhenMigrating: true,
-	})
-
-	if err != nil {
-		logger.Fatal(err.Error())
-	}
-
-	// Migrate the schema
-	err = db.AutoMigrate(GenerateAllModel()...)
-	if err != nil {
-		logger.Fatal(err.Error())
-	}
-
-	return db
-}
-
 type Pagination struct {
 	Total       int64 `json:"total"`
 	PerPage     int   `json:"per_page"`

model/site_category.go

@@ -0,0 +1,7 @@
+package model
+
+type SiteCategory struct {
+	Model
+	Name        string `json:"name"`
+	SyncNodeIds []int  `json:"sync_node_ids" gorm:"serializer:json"`
+}