up

backend/models/configs.go

@@ -0,0 +1,62 @@
+package models
+
+import "github.com/mitchellh/mapstructure"
+
+var Configs map[string]interface{}
+
+//mime信息转换位拓展名
+
+type ConfigsWeb_ struct {
+	Host            string `mapstructure:"host"`
+	Port            string `mapstructure:"port"`
+	Tls             bool   `mapstructure:"tls"`
+	CertPrivatePath string `mapstructure:"certPrivatePath"`
+	CertPublicPath  string `mapstructure:"certPublicPath"`
+}
+
+type ConfigsUser_ struct {
+	CookieTimeout       int    `mapstructure:"cookieTimeout"`
+	PassHashType        string `mapstructure:"passHashType"`
+	AvatarSavePath      string `mapstructure:"avatarSavePath"`
+	AvatarGinrouterPath string `mapstructure:"avatarGinrouterPath"`
+	AvatarPath          string `mapstructure:"avatarPath"`
+}
+
+type ConfigsFile_ struct {
+	MaxSize        uint64            `mapstructure:"maxSize"`
+	Pahts          map[string]string `mapstructure:"pahts"`
+	AllowImageMime map[string]string `mapstructure:"allowImageMime"`
+	AllowVideoMime map[string]string `mapstructure:"allowVideoMime"`
+	AllowMusicMime map[string]string `mapstructure:"allowMusicMime"`
+	AllowPdfMime   map[string]string `mapstructure:"allowPdfMime"`
+}
+
+var ConfigsWed ConfigsWeb_
+var ConfigsUser ConfigsUser_
+var ConfigsFile ConfigsFile_
+
+func ConfigAllInit() error {
+
+	//初始化数据库
+	DatabaseInit()
+
+	//读取web配置
+	err := mapstructure.Decode(Configs["web"].(map[string]interface{}), &ConfigsWed)
+	if err != nil {
+		panic(err)
+	}
+
+	//初始化user config
+	err = mapstructure.Decode(Configs["user"].(map[string]interface{}), &ConfigsUser)
+	if err != nil {
+		panic(err)
+	}
+
+	//初始化file config
+	err = mapstructure.Decode(Configs["file"].(map[string]interface{}), &ConfigsFile)
+	if err != nil {
+		panic(err)
+	}
+
+	return nil
+}

backend/models/file.go

@@ -0,0 +1,12 @@
+package models
+
+import "os"
+
+// 判断文件是否存在
+func FileExists(path string) bool {
+	_, err := os.Stat(path)
+	if err != nil {
+		return !os.IsNotExist(err)
+	}
+	return true
+}

backend/models/sql.go

@@ -0,0 +1,99 @@
+package models
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/glebarez/sqlite"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+)
+
+var DB *gorm.DB
+
+type TabFileInfo_ struct {
+	ID     uint      `gorm:"primaryKey;autoIncrement"`
+	Name   string    `gorm:"not null;size:256;index"` // 前端报告的文件名
+	Path   string    `gorm:"not null;size:300"`       //
+	Sha256 string    `gorm:"not null;size:256;index"` //
+	Mime   string    `gorm:"size:64;index"`
+	Type   string    `gorm:"size:64;index"`
+	Const  uint      `gorm:"default:1;index"`
+	Per    uint      `gorm:"default:1"`
+	UserID uint      `gorm:"not null;index"`
+	Date   time.Time `gorm:"type:datetime;default:CURRENT_TIMESTAMP"` // 默认当前时间
+}
+
+type TabUser_ struct {
+	ID    uint      `gorm:"primaryKey;autoIncrement"`                // 自增主键
+	Name  string    `gorm:"size:100;uniqueIndex"`                    // 唯一约束索引
+	Email string    `gorm:"size:255;index"`                          // 字符串长度限制100 索引
+	Pass  string    `gorm:"size:128"`                                // 建议存储哈希后的密码
+	Date  time.Time `gorm:"type:datetime;default:CURRENT_TIMESTAMP"` // 默认当前时间
+}
+
+type TabUserInfo_ struct {
+	ID         uint      `gorm:"primaryKey;autoIncrement"`
+	UserID     uint      `gorm:"not null;uniqueIndex"`
+	FirstName  string    `gorm:"size:50;null"`
+	Username   string    `gorm:"size:30;null"`
+	Birthdate  time.Time `gorm:"type:datetime;null"`
+	Gender     string    `gorm:"type:char(1);check:gender IN ('M', 'F', 'U');default:'U'"`
+	AvatarPath string    `gorm:"size:255"`
+	Region     string    `gorm:"size:50"`
+	Language   string    `gorm:"size:10;default:'zh-CN'"`
+	CreatedAt  time.Time `gorm:"type:datetime;default:CURRENT_TIMESTAMP;column:created_at"`
+}
+
+// var def_user_info = User_info{
+// 	ID:0,
+// 	UserID:0,
+// }
+
+type TabCookie_ struct {
+	ID           uint      `gorm:"primaryKey;autoIncrement"`
+	UserID       uint      `gorm:"size:16;not null"`
+	Name         string    `gorm:"size:255;not null;index"`
+	Value        string    `gorm:"size:255;not null;index"`
+	Domain       string    `gorm:"size:255;not null"`
+	Path         string    `gorm:"size:255;not null;default:/"`
+	ExpiresAt    time.Time `gorm:"type:datetime;index"`
+	CreatedAt    time.Time `gorm:"type:datetime;not null;default:CURRENT_TIMESTAMP"`
+	UpdatedAt    time.Time `gorm:"type:datetime;index;not null;default:CURRENT_TIMESTAMP"`
+	SecureFlag   bool      `gorm:"not null;default:false"`
+	HttpOnly     bool      `gorm:"not null;default:false"`
+	SameSite     string    `gorm:"size:10;not null;default:'Lax'"`
+	PartitionKey string    `gorm:"size:50"`
+}
+
+func DatabaseInit() error {
+	var err error
+	fmt.Println("database_init")
+	DatabaseConfigs := Configs["database"].(map[string]interface{})
+
+	if DatabaseConfigs["type"].(string) == "sqlite" {
+		//sqlite init
+		fmt.Println("sqlite")
+		DB, err = gorm.Open(sqlite.Open(DatabaseConfigs["path"].(string)), &gorm.Config{})
+	} else if DatabaseConfigs["type"].(string) == "mysql" {
+		//mysql init
+		fmt.Println("mysql")
+		dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", DatabaseConfigs["user"].(string), DatabaseConfigs["pass"].(string), DatabaseConfigs["host"].(string), DatabaseConfigs["port"].(string), DatabaseConfigs["name"].(string))
+		DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
+	}
+	if err != nil {
+		fmt.Println(err)
+		panic("数据库连接失败")
+	}
+
+	// 自动创建表结构
+	DB.AutoMigrate(&TabUser_{})
+
+	DB.AutoMigrate(&TabUserInfo_{})
+
+	DB.AutoMigrate(&TabCookie_{})
+
+	DB.AutoMigrate(&TabFileInfo_{})
+
+	return nil
+}