完成xray启动

database/db.go

@@ -1,13 +1,15 @@
 package database
 
 import (
+	"gorm.io/driver/sqlite"
 	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
 	"io/fs"
 	"os"
 	"path"
+	"x-ui/config"
 	"x-ui/database/model"
 )
-import "gorm.io/driver/sqlite"
 
 var db *gorm.DB
 
@@ -35,6 +37,10 @@ func initInbound() error {
 	return db.AutoMigrate(&model.Inbound{})
 }
 
+func initSetting() error {
+	return db.AutoMigrate(&model.Setting{})
+}
+
 func InitDB(dbPath string) error {
 	dir := path.Dir(dbPath)
 	err := os.MkdirAll(dir, fs.ModeDir)
@@ -42,7 +48,17 @@ func InitDB(dbPath string) error {
 		return err
 	}
 
-	c := &gorm.Config{}
+	var gormLogger logger.Interface
+
+	if config.IsDebug() {
+		gormLogger = logger.Discard
+	} else {
+		gormLogger = logger.Default
+	}
+
+	c := &gorm.Config{
+		Logger: gormLogger,
+	}
 	db, err = gorm.Open(sqlite.Open(dbPath), c)
 	if err != nil {
 		return err
@@ -56,6 +72,10 @@ func InitDB(dbPath string) error {
 	if err != nil {
 		return err
 	}
+	err = initSetting()
+	if err != nil {
+		return err
+	}
 
 	return nil
 }
@@ -63,3 +83,7 @@ func InitDB(dbPath string) error {
 func GetDB() *gorm.DB {
 	return db
 }
+
+func IsNotFound(err error) bool {
+	return err == gorm.ErrRecordNotFound
+}

database/model/model.go

@@ -1,6 +1,9 @@
 package model
 
-import "time"
+import (
+	"encoding/json"
+	"x-ui/xray"
+)
 
 type Protocol string
 
@@ -20,20 +23,38 @@ type User struct {
 }
 
 type Inbound struct {
-	Id         int       `json:"id" gorm:"primaryKey;autoIncrement"`
-	UserId     int       `json:"user_id"`
-	Up         int64     `json:"up"`
-	Down       int64     `json:"down"`
-	Remark     string    `json:"remark"`
-	Enable     bool      `json:"enable"`
-	ExpiryTime time.Time `json:"expiry_time"`
+	Id         int    `json:"id" form:"id" gorm:"primaryKey;autoIncrement"`
+	UserId     int    `json:"user_id" form:"user_id"`
+	Up         int64  `json:"up" form:"up"`
+	Down       int64  `json:"down" form:"down"`
+	Remark     string `json:"remark" form:"remark"`
+	Enable     bool   `json:"enable" form:"enable"`
+	ExpiryTime int64  `json:"expiry_time" form:"expiry_time"`
 
 	// config part
-	Listen         string   `json:"listen"`
-	Port           int      `json:"port"`
-	Protocol       Protocol `json:"protocol"`
-	Settings       string   `json:"settings"`
-	StreamSettings string   `json:"stream_settings"`
-	Tag            string   `json:"tag"`
-	Sniffing       string   `json:"sniffing"`
+	Listen         string   `json:"listen" form:"listen"`
+	Port           int      `json:"port" form:"port"`
+	Protocol       Protocol `json:"protocol" form:"protocol"`
+	Settings       string   `json:"settings" form:"settings"`
+	StreamSettings string   `json:"stream_settings" form:"stream_settings"`
+	Tag            string   `json:"tag" form:"tag"`
+	Sniffing       string   `json:"sniffing" form:"sniffing"`
+}
+
+func (i *Inbound) GenXrayInboundConfig() *xray.InboundConfig {
+	return &xray.InboundConfig{
+		Listen:         i.Listen,
+		Port:           i.Port,
+		Protocol:       string(i.Protocol),
+		Settings:       json.RawMessage(i.Settings),
+		StreamSettings: json.RawMessage(i.StreamSettings),
+		Tag:            i.Tag,
+		Sniffing:       json.RawMessage(i.Sniffing),
+	}
+}
+
+type Setting struct {
+	Id    int    `json:"id" form:"id" gorm:"primaryKey;autoIncrement"`
+	Key   string `json:"key" form:"key"`
+	Value string `json:"value" form:"value"`
 }