Upload 4 files

Dockerfile

@@ -1,35 +1,35 @@
 # 使用基础镜像
-FROM debian:latest
+FROM golang:1.18
 
 # 安装必要的工具
 RUN apt-get update && apt-get install -y \
     wget \
-    tar
-RUN apt-get update && apt-get install -y aria2
+    tar \
+    git 
+RUN apt-get update && apt-get install -y aria2 \
+    && git clone https://github.com/cloudreve/Cloudreve.git /app 
+COPY aria2.conf /app/aria2.conf
+COPY migration.go /app/models/migration.go
 # 设置工作目录
 WORKDIR /app
 
-# 下载并解压Cloudreve
-RUN wget https://github.com/cloudreve/Cloudreve/releases/download/3.8.3/cloudreve_3.8.3_linux_amd64.tar.gz \
-    && tar -zxvf cloudreve_3.8.3_linux_amd64.tar.gz \
-    && rm cloudreve_3.8.3_linux_amd64.tar.gz
+
 
 # 更改 /app 目录的权限
-RUN chmod 777 /app
-COPY conf.ini /app/conf.ini
-COPY aria2.conf /app/aria2.conf
+RUN chmod 777 /app && go mod tidy && go build -o cloudreve
+
+
 # 赋予执行权限
 RUN chmod +x ./cloudreve
-# 创建 /aria2/data 目录
+
 RUN mkdir -p /aria2/data
 
-# 更改 /aria2/data 目录的权限
+
 RUN chmod 777 /aria2/data
-# 暴露端口
+
 EXPOSE 7860
 
-# # 启动 Cloudreve
-# CMD ["./cloudreve","-c","/app/conf.ini"]
+
 COPY start.sh /app/start.sh
 RUN chmod +x /app/start.sh
 CMD ["/app/start.sh"]

migration.go

@@ -0,0 +1,218 @@
+package model
+
+import (
+	"context"
+	"github.com/cloudreve/Cloudreve/v3/models/scripts/invoker"
+	"github.com/cloudreve/Cloudreve/v3/pkg/cache"
+	"github.com/cloudreve/Cloudreve/v3/pkg/conf"
+	"github.com/cloudreve/Cloudreve/v3/pkg/util"
+	"github.com/fatih/color"
+	"github.com/hashicorp/go-version"
+	"github.com/jinzhu/gorm"
+	"sort"
+	"strings"
+)
+
+// 是否需要迁移
+func needMigration() bool {
+	var setting Setting
+	return DB.Where("name = ?", "db_version_"+conf.RequiredDBVersion).First(&setting).Error != nil
+}
+
+// 执行数据迁移
+func migration() {
+	// 确认是否需要执行迁移
+	if !needMigration() {
+		util.Log().Info("Database version fulfilled, skip schema migration.")
+		return
+
+	}
+
+	util.Log().Info("Start initializing database schema...")
+
+	// 清除所有缓存
+	if instance, ok := cache.Store.(*cache.RedisStore); ok {
+		instance.DeleteAll()
+	}
+
+	// 自动迁移模式
+	if conf.DatabaseConfig.Type == "mysql" {
+		DB = DB.Set("gorm:table_options", "ENGINE=InnoDB")
+	}
+
+	DB.AutoMigrate(&User{}, &Setting{}, &Group{}, &Policy{}, &Folder{}, &File{}, &Share{},
+		&Task{}, &Download{}, &Tag{}, &Webdav{}, &Node{}, &SourceLink{})
+
+	// 创建初始存储策略
+	addDefaultPolicy()
+
+	// 创建初始用户组
+	addDefaultGroups()
+
+	// 创建初始管理员账户
+	addDefaultUser()
+
+	// 创建初始节点
+	addDefaultNode()
+
+	// 向设置数据表添加初始设置
+	addDefaultSettings()
+
+	// 执行数据库升级脚本
+	execUpgradeScripts()
+
+	util.Log().Info("Finish initializing database schema.")
+
+}
+
+func addDefaultPolicy() {
+	_, err := GetPolicyByID(uint(1))
+	// 未找到初始存储策略时，则创建
+	if gorm.IsRecordNotFoundError(err) {
+		defaultPolicy := Policy{
+			Name:               "Default storage policy",
+			Type:               "local",
+			MaxSize:            0,
+			AutoRename:         true,
+			DirNameRule:        "uploads/{uid}/{path}",
+			FileNameRule:       "{uid}_{randomkey8}_{originname}",
+			IsOriginLinkEnable: false,
+			OptionsSerialized: PolicyOption{
+				ChunkSize: 25 << 20, // 25MB
+			},
+		}
+		if err := DB.Create(&defaultPolicy).Error; err != nil {
+			util.Log().Panic("Failed to create default storage policy: %s", err)
+		}
+	}
+}
+
+func addDefaultSettings() {
+	for _, value := range defaultSettings {
+		DB.Where(Setting{Name: value.Name}).Create(&value)
+	}
+}
+
+func addDefaultGroups() {
+	_, err := GetGroupByID(1)
+	// 未找到初始管理组时，则创建
+	if gorm.IsRecordNotFoundError(err) {
+		defaultAdminGroup := Group{
+			Name:          "Admin",
+			PolicyList:    []uint{1},
+			MaxStorage:    1 * 1024 * 1024 * 1024,
+			ShareEnabled:  true,
+			WebDAVEnabled: true,
+			OptionsSerialized: GroupOption{
+				ArchiveDownload:  true,
+				ArchiveTask:      true,
+				ShareDownload:    true,
+				Aria2:            true,
+				SourceBatchSize:  1000,
+				Aria2BatchSize:   50,
+				RedirectedSource: true,
+				AdvanceDelete:    true,
+			},
+		}
+		if err := DB.Create(&defaultAdminGroup).Error; err != nil {
+			util.Log().Panic("Failed to create admin user group: %s", err)
+		}
+	}
+
+	err = nil
+	_, err = GetGroupByID(2)
+	// 未找到初始注册会员时，则创建
+	if gorm.IsRecordNotFoundError(err) {
+		defaultAdminGroup := Group{
+			Name:          "User",
+			PolicyList:    []uint{1},
+			MaxStorage:    1 * 1024 * 1024 * 1024,
+			ShareEnabled:  true,
+			WebDAVEnabled: true,
+			OptionsSerialized: GroupOption{
+				ShareDownload:    true,
+				SourceBatchSize:  10,
+				Aria2BatchSize:   1,
+				RedirectedSource: true,
+			},
+		}
+		if err := DB.Create(&defaultAdminGroup).Error; err != nil {
+			util.Log().Panic("Failed to create initial user group: %s", err)
+		}
+	}
+
+	err = nil
+	_, err = GetGroupByID(3)
+	// 未找到初始游客用户组时，则创建
+	if gorm.IsRecordNotFoundError(err) {
+		defaultAdminGroup := Group{
+			Name:       "Anonymous",
+			PolicyList: []uint{},
+			Policies:   "[]",
+			OptionsSerialized: GroupOption{
+				ShareDownload: true,
+			},
+		}
+		if err := DB.Create(&defaultAdminGroup).Error; err != nil {
+			util.Log().Panic("Failed to create anonymous user group: %s", err)
+		}
+	}
+}
+
+func addDefaultUser() {
+	_, err := GetUserByID(1)
+	password := util.RandStringRunes(8)
+
+	// 未找到初始用户时，则创建
+	if gorm.IsRecordNotFoundError(err) {
+		defaultUser := NewUser()
+		defaultUser.Email = "[email protected]"
+		defaultUser.Nick = "admin"
+		defaultUser.Status = Active
+		defaultUser.GroupID = 1
+		err := defaultUser.SetPassword(password)
+		if err != nil {
+			util.Log().Panic("Failed to create password: %s", err)
+		}
+		if err := DB.Create(&defaultUser).Error; err != nil {
+			util.Log().Panic("Failed to create initial root user: %s", err)
+		}
+
+		c := color.New(color.FgWhite).Add(color.BgBlack).Add(color.Bold)
+		util.Log().Info("Admin user name: " + c.Sprint("[email protected]"))
+		util.Log().Info("Admin password: " + c.Sprint(password))
+	}
+}
+
+func addDefaultNode() {
+	_, err := GetNodeByID(1)
+
+	if gorm.IsRecordNotFoundError(err) {
+		defaultAdminGroup := Node{
+			Name:   "Master (Local machine)",
+			Status: NodeActive,
+			Type:   MasterNodeType,
+			Aria2OptionsSerialized: Aria2Option{
+				Interval: 10,
+				Timeout:  10,
+			},
+		}
+		if err := DB.Create(&defaultAdminGroup).Error; err != nil {
+			util.Log().Panic("Failed to create initial node: %s", err)
+		}
+	}
+}
+
+func execUpgradeScripts() {
+	s := invoker.ListPrefix("UpgradeTo")
+	versions := make([]*version.Version, len(s))
+	for i, raw := range s {
+		v, _ := version.NewVersion(strings.TrimPrefix(raw, "UpgradeTo"))
+		versions[i] = v
+	}
+	sort.Sort(version.Collection(versions))
+
+	for i := 0; i < len(versions); i++ {
+		invoker.RunDBScript("UpgradeTo"+versions[i].String(), context.Background())
+	}
+}

start.sh

@@ -2,4 +2,4 @@
 # 启动 aria2
 aria2c --conf-path=/app/aria2.conf & 
 # 启动 Cloudreve
-./cloudreve -c /app/conf.ini
+./cloudreve