Init V4 community edition (#2265)

* Init V4 community edition

* Init V4 community edition

routers/controllers/slave.go

@@ -1,138 +1,118 @@
 package controllers
 
 import (
-	"context"
+	"fmt"
 
-	"github.com/cloudreve/Cloudreve/v3/pkg/request"
-	"github.com/cloudreve/Cloudreve/v3/pkg/serializer"
-	"github.com/cloudreve/Cloudreve/v3/service/admin"
-	"github.com/cloudreve/Cloudreve/v3/service/aria2"
-	"github.com/cloudreve/Cloudreve/v3/service/explorer"
-	"github.com/cloudreve/Cloudreve/v3/service/node"
+	"github.com/cloudreve/Cloudreve/v4/pkg/cluster"
+	"github.com/cloudreve/Cloudreve/v4/pkg/downloader"
+	"github.com/cloudreve/Cloudreve/v4/pkg/downloader/slave"
+	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/fs"
+	"github.com/cloudreve/Cloudreve/v4/pkg/request"
+	"github.com/cloudreve/Cloudreve/v4/pkg/serializer"
+	"github.com/cloudreve/Cloudreve/v4/service/admin"
+	"github.com/cloudreve/Cloudreve/v4/service/explorer"
+	"github.com/cloudreve/Cloudreve/v4/service/node"
 	"github.com/gin-gonic/gin"
 )
 
 // SlaveUpload 从机文件上传
 func SlaveUpload(c *gin.Context) {
-	// 创建上下文
-	ctx, cancel := context.WithCancel(context.Background())
-	defer cancel()
-
-	var service explorer.UploadService
-	if err := c.ShouldBindUri(&service); err == nil {
-		res := service.SlaveUpload(ctx, c)
-		c.JSON(200, res)
+	service := ParametersFromContext[*explorer.UploadService](c, explorer.UploadParameterCtx{})
+	err := service.SlaveUpload(c)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
 		request.BlackHole(c.Request.Body)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+		c.Abort()
+		return
 	}
+
+	c.JSON(200, serializer.Response{})
 }
 
 // SlaveGetUploadSession 从机创建上传会话
 func SlaveGetUploadSession(c *gin.Context) {
-	// 创建上下文
-	ctx, cancel := context.WithCancel(context.Background())
-	defer cancel()
-
-	var service explorer.SlaveCreateUploadSessionService
-	if err := c.ShouldBindJSON(&service); err == nil {
-		res := service.Create(ctx, c)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+	service := ParametersFromContext[*explorer.SlaveCreateUploadSessionService](c, explorer.SlaveCreateUploadSessionParamCtx{})
+	if err := service.Create(c); err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
+
+	c.JSON(200, serializer.Response{})
 }
 
 // SlaveDeleteUploadSession 从机删除上传会话
 func SlaveDeleteUploadSession(c *gin.Context) {
-	// 创建上下文
-	ctx, cancel := context.WithCancel(context.Background())
-	defer cancel()
+	service := ParametersFromContext[*explorer.SlaveDeleteUploadSessionService](c, explorer.SlaveDeleteUploadSessionParamCtx{})
+	if err := service.Delete(c); err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
+	}
 
-	var service explorer.UploadSessionService
-	if err := c.ShouldBindUri(&service); err == nil {
-		res := service.SlaveDelete(ctx, c)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+	c.JSON(200, serializer.Response{})
+}
+
+// SlaveServeEntity download entity content
+func SlaveServeEntity(c *gin.Context) {
+	service := ParametersFromContext[*explorer.EntityDownloadService](c, explorer.EntityDownloadParameterCtx{})
+	err := service.SlaveServe(c)
+	if err != nil {
+		c.JSON(400, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
 }
 
-// SlaveDownload 从机文件下载,此请求返回的HTTP状态码不全为200
-func SlaveDownload(c *gin.Context) {
-	// 创建上下文
-	ctx, cancel := context.WithCancel(context.Background())
-	defer cancel()
-
-	var service explorer.SlaveDownloadService
-	if err := c.ShouldBindUri(&service); err == nil {
-		res := service.ServeFile(ctx, c, true)
-		if res.Code != 0 {
-			c.JSON(400, res)
-		}
-	} else {
-		c.JSON(400, ErrorResponse(err))
+// SlaveMeta retrieve media metadata
+func SlaveMeta(c *gin.Context) {
+	service := ParametersFromContext[*explorer.SlaveMetaService](c, explorer.SlaveMetaParamCtx{})
+	res, err := service.MediaMeta(c)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
-}
 
-// SlavePreview 从机文件预览
-func SlavePreview(c *gin.Context) {
-	// 创建上下文
-	ctx, cancel := context.WithCancel(context.Background())
-	defer cancel()
-
-	var service explorer.SlaveDownloadService
-	if err := c.ShouldBindUri(&service); err == nil {
-		res := service.ServeFile(ctx, c, false)
-		if res.Code != 0 {
-			c.JSON(200, res)
-		}
-	} else {
-		c.JSON(200, ErrorResponse(err))
-	}
+	c.JSON(200, serializer.NewResponseWithGobData(c, res))
 }
 
 // SlaveThumb 从机文件缩略图
 func SlaveThumb(c *gin.Context) {
-	// 创建上下文
-	ctx, cancel := context.WithCancel(context.Background())
-	defer cancel()
-
-	var service explorer.SlaveFileService
-	if err := c.ShouldBindUri(&service); err == nil {
-		res := service.Thumb(ctx, c)
-		if res.Code != 0 {
-			c.JSON(200, res)
-		}
-	} else {
-		c.JSON(200, ErrorResponse(err))
+	service := ParametersFromContext[*explorer.SlaveThumbService](c, explorer.SlaveThumbParamCtx{})
+	err := service.Thumb(c)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
 }
 
 // SlaveDelete 从机删除
 func SlaveDelete(c *gin.Context) {
-	// 创建上下文
-	ctx, cancel := context.WithCancel(context.Background())
-	defer cancel()
-
-	var service explorer.SlaveFilesService
-	if err := c.ShouldBindJSON(&service); err == nil {
-		res := service.Delete(ctx, c)
-		c.JSON(200, res)
+	service := ParametersFromContext[*explorer.SlaveDeleteFileService](c, explorer.SlaveDeleteFileParamCtx{})
+	if failed, err := service.Delete(c); err != nil {
+		c.JSON(200, serializer.NewResponseWithGobData(c, serializer.Response{
+			Code:  serializer.CodeNotFullySuccess,
+			Data:  failed,
+			Msg:   fmt.Sprintf("Failed to delete %d files(s)", len(failed)),
+			Error: err.Error(),
+		}))
 	} else {
-		c.JSON(200, ErrorResponse(err))
+		c.JSON(200, serializer.Response{Data: ""})
 	}
 }
 
 // SlavePing 从机测试
 func SlavePing(c *gin.Context) {
-	var service admin.SlavePingService
-	if err := c.ShouldBindJSON(&service); err == nil {
-		res := service.Test()
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+	service := ParametersFromContext[*admin.SlavePingService](c, admin.SlavePingParameterCtx{})
+	if err := service.Test(c); err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
+
+	c.JSON(200, serializer.Response{})
 }
 
 // SlaveList 从机列出文件
@@ -146,101 +126,167 @@ func SlaveList(c *gin.Context) {
 	}
 }
 
-// SlaveHeartbeat 接受主机心跳包
-func SlaveHeartbeat(c *gin.Context) {
-	var service serializer.NodePingReq
-	if err := c.ShouldBindJSON(&service); err == nil {
-		res := node.HandleMasterHeartbeat(&service)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+// SlaveDownloadTaskCreate creates a download task on slave
+func SlaveDownloadTaskCreate(c *gin.Context) {
+	service := ParametersFromContext[*slave.CreateSlaveDownload](c, node.CreateSlaveDownloadTaskParamCtx{})
+	d := c.MustGet(downloader.DownloaderCtxKey).(downloader.Downloader)
+	handle, err := d.CreateTask(c, service.Url, service.Options)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
+
+	c.JSON(200, serializer.NewResponseWithGobData(c, handle))
 }
 
-// SlaveAria2Create 创建 Aria2 任务
-func SlaveAria2Create(c *gin.Context) {
-	var service serializer.SlaveAria2Call
-	if err := c.ShouldBindJSON(&service); err == nil {
-		res := aria2.Add(c, &service)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+// SlaveDownloadTaskStatus 查询从机 Aria2 任务状态
+func SlaveDownloadTaskStatus(c *gin.Context) {
+	service := ParametersFromContext[*slave.GetSlaveDownload](c, node.GetSlaveDownloadTaskParamCtx{})
+	d := c.MustGet(downloader.DownloaderCtxKey).(downloader.Downloader)
+	info, err := d.Info(c, service.Handle)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
+
+	c.JSON(200, serializer.NewResponseWithGobData(c, info))
 }
 
-// SlaveAria2Status 查询从机 Aria2 任务状态
-func SlaveAria2Status(c *gin.Context) {
-	var service serializer.SlaveAria2Call
-	if err := c.ShouldBindJSON(&service); err == nil {
-		res := aria2.SlaveStatus(c, &service)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+// SlaveCancelDownloadTask 取消从机离线下载任务
+func SlaveCancelDownloadTask(c *gin.Context) {
+	service := ParametersFromContext[*slave.CancelSlaveDownload](c, node.CancelSlaveDownloadTaskParamCtx{})
+	d := c.MustGet(downloader.DownloaderCtxKey).(downloader.Downloader)
+	err := d.Cancel(c, service.Handle)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
+
+	c.JSON(200, serializer.Response{})
 }
 
-// SlaveCancelAria2Task 取消从机离线下载任务
-func SlaveCancelAria2Task(c *gin.Context) {
-	var service serializer.SlaveAria2Call
-	if err := c.ShouldBindJSON(&service); err == nil {
-		res := aria2.SlaveCancel(c, &service)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+// SlaveSelectFilesToDownload 从机选取离线下载文件
+func SlaveSelectFilesToDownload(c *gin.Context) {
+	service := ParametersFromContext[*slave.SetSlaveFilesToDownload](c, node.SelectSlaveDownloadFilesParamCtx{})
+	d := c.MustGet(downloader.DownloaderCtxKey).(downloader.Downloader)
+	err := d.SetFilesToDownload(c, service.Handle, service.Args...)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
+
+	c.JSON(200, serializer.Response{})
 }
 
-// SlaveSelectTask 从机选取离线下载文件
-func SlaveSelectTask(c *gin.Context) {
-	var service serializer.SlaveAria2Call
-	if err := c.ShouldBindJSON(&service); err == nil {
-		res := aria2.SlaveSelect(c, &service)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+// SlaveTestDownloader 从机测试下载器连接
+func SlaveTestDownloader(c *gin.Context) {
+	d := c.MustGet(downloader.DownloaderCtxKey).(downloader.Downloader)
+	res, err := d.Test(c)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
-}
 
-// SlaveCreateTransferTask 从机创建中转任务
-func SlaveCreateTransferTask(c *gin.Context) {
-	var service serializer.SlaveTransferReq
-	if err := c.ShouldBindJSON(&service); err == nil {
-		res := explorer.CreateTransferTask(c, &service)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
-	}
-}
-
-// SlaveNotificationPush 处理从机发送的消息推送
-func SlaveNotificationPush(c *gin.Context) {
-	var service node.SlaveNotificationService
-	if err := c.ShouldBindUri(&service); err == nil {
-		res := service.HandleSlaveNotificationPush(c)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
-	}
+	c.JSON(200, serializer.Response{Data: res})
 }
 
 // SlaveGetOauthCredential 从机获取主机的OneDrive存储策略凭证
-func SlaveGetOauthCredential(c *gin.Context) {
-	var service node.OauthCredentialService
-	if err := c.ShouldBindUri(&service); err == nil {
-		res := service.Get(c)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+func SlaveGetCredential(c *gin.Context) {
+	service := ParametersFromContext[*node.OauthCredentialService](c, node.OauthCredentialParamCtx{})
+	cred, err := service.Get(c)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
+
+	c.JSON(200, serializer.NewResponseWithGobData(c, cred))
 }
 
-// SlaveSelectTask 从机删除离线下载临时文件
-func SlaveDeleteTempFile(c *gin.Context) {
-	var service serializer.SlaveAria2Call
-	if err := c.ShouldBindJSON(&service); err == nil {
-		res := aria2.SlaveDeleteTemp(c, &service)
-		c.JSON(200, res)
-	} else {
-		c.JSON(200, ErrorResponse(err))
+// SlaveCreateTask creates tasks and register it in registry
+func SlaveCreateTask(c *gin.Context) {
+	service := ParametersFromContext[*cluster.CreateSlaveTask](c, node.CreateSlaveTaskParamCtx{})
+	taskId, err := node.CreateTaskInSlave(service, c)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
 	}
+
+	c.JSON(200, serializer.NewResponseWithGobData(c, taskId))
+}
+
+// SlaveCreateTask creates tasks and register it in registry
+func SlaveGetTask(c *gin.Context) {
+	service := ParametersFromContext[*node.GetSlaveTaskService](c, node.GetSlaveTaskParamCtx{})
+	task, err := service.Get(c)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
+	}
+
+	c.JSON(200, serializer.NewResponseWithGobData(c, task))
+}
+
+func SlaveCleanupFolder(c *gin.Context) {
+	service := ParametersFromContext[*cluster.FolderCleanup](c, node.FolderCleanupParamCtx{})
+	if err := node.Cleanup(service, c); err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
+	}
+
+	c.JSON(200, serializer.Response{})
+}
+
+func StatelessPrepareUpload(c *gin.Context) {
+	service := ParametersFromContext[*fs.StatelessPrepareUploadService](c, node.StatelessPrepareUploadParamCtx{})
+	uploadSession, err := node.StatelessPrepareUpload(service, c)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
+	}
+
+	c.JSON(200, serializer.NewResponseWithGobData(c, uploadSession))
+}
+
+func StatelessCompleteUpload(c *gin.Context) {
+	service := ParametersFromContext[*fs.StatelessCompleteUploadService](c, node.StatelessCompleteUploadParamCtx{})
+	_, err := node.StatelessCompleteUpload(service, c)
+	if err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
+	}
+
+	c.JSON(200, serializer.Response{})
+}
+
+func StatelessOnUploadFailed(c *gin.Context) {
+	service := ParametersFromContext[*fs.StatelessOnUploadFailedService](c, node.StatelessOnUploadFailedParamCtx{})
+	if err := node.StatelessOnUploadFailed(service, c); err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
+	}
+
+	c.JSON(200, serializer.Response{})
+}
+
+func StatelessCreateFile(c *gin.Context) {
+	service := ParametersFromContext[*fs.StatelessCreateFileService](c, node.StatelessCreateFileParamCtx{})
+	if err := node.StatelessCreateFile(service, c); err != nil {
+		c.JSON(200, serializer.Err(c, err))
+		c.Abort()
+		return
+	}
+
+	c.JSON(200, serializer.Response{})
 }