cloudreve/service/callback/upload.go

package callback

import (
	"fmt"
	"github.com/cloudreve/Cloudreve/v4/application/dependency"
	"github.com/cloudreve/Cloudreve/v4/inventory"
	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/onedrive"
	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/fs"
	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/manager"
	"github.com/gin-gonic/gin"
)

// RemoteUploadCallbackService 远程存储上传回调请求服务
type RemoteUploadCallbackService struct {
}

// UploadCallbackService OOS/七牛云存储上传回调请求服务
type UploadCallbackService struct {
	Name       string `json:"name"`
	SourceName string `json:"source_name"`
	PicInfo    string `json:"pic_info"`
	Size       int64  `json:"size"`
}

// UpyunCallbackService 又拍云上传回调请求服务
type UpyunCallbackService struct {
	Code       int    `form:"code" binding:"required"`
	Message    string `form:"message" binding:"required"`
	SourceName string `form:"url" binding:"required"`
	Width      string `form:"image-width"`
	Height     string `form:"image-height"`
	Size       uint64 `form:"file_size"`
}

// OneDriveCallback OneDrive 客户端回调正文
type OneDriveCallback struct {
	Meta *onedrive.FileInfo
}

// S3Callback S3 客户端回调正文
type S3Callback struct {
}

// ProcessCallback 处理上传结果回调
func ProcessCallback(c *gin.Context) error {
	dep := dependency.FromContext(c)
	user := inventory.UserFromContext(c)
	m := manager.NewFileManager(dep, user)
	defer m.Recycle()

	// 获取上传会话
	uploadSession := c.MustGet(manager.UploadSessionCtx).(*fs.UploadSession)

	_, err := m.CompleteUpload(c, uploadSession)
	if err != nil {
		return fmt.Errorf("failed to complete upload: %w", err)
	}

	return nil
}

//// PreProcess 对OneDrive客户端回调进行预处理验证
//func (service *OneDriveCallback) PreProcess(c *gin.Context) serializer.Response {
//	// 创建文件系统
//	fs, err := filesystem.NewFileSystemFromCallback(c)
//	if err != nil {
//		return serializer.ErrDeprecated(serializer.CodeCreateFSError, "", err)
//	}
//	defer fs.Recycle()
//
//	// 获取回调会话
//	uploadSession := c.MustGet(filesystem.UploadSessionCtx).(*serializer.UploadSession)
//
//	// 获取文件信息
//	info, err := fs.Handler.(onedrive.Driver).Client.Meta(context.Background(), "", uploadSession.SavePath)
//	if err != nil {
//		return serializer.ErrDeprecated(serializer.CodeQueryMetaFailed, "", err)
//	}
//
//	// 验证与回调会话中是否一致
//	actualPath := strings.TrimPrefix(uploadSession.SavePath, "/")
//	isSizeCheckFailed := uploadSession.Size != info.Size
//
//	// SharePoint 会对 Office 文档增加 meta data 导致文件大小不一致，这里增加 1 MB 宽容
//	// See: https://github.com/OneDrive/onedrive-api-docs/issues/935
//	if (strings.Contains(fs.Policy.OptionsSerialized.OdDriver, "sharepoint.com") || strings.Contains(fs.Policy.OptionsSerialized.OdDriver, "sharepoint.cn")) && isSizeCheckFailed && (info.Size > uploadSession.Size) && (info.Size-uploadSession.Size <= 1048576) {
//		isSizeCheckFailed = false
//	}
//
//	if isSizeCheckFailed || !strings.EqualFold(info.GetSourcePath(), actualPath) {
//		fs.Handler.(onedrive.Driver).Client.Delete(context.Background(), []string{info.GetSourcePath()})
//		return serializer.ErrDeprecated(serializer.CodeMetaMismatch, "", err)
//	}
//	service.Meta = info
//	return ProcessCallback(c)
//}
//

//
//// PreProcess 对S3客户端回调进行预处理
//func (service *S3Callback) PreProcess(c *gin.Context) serializer.Response {
//	// 创建文件系统
//	fs, err := filesystem.NewFileSystemFromCallback(c)
//	if err != nil {
//		return serializer.ErrDeprecated(serializer.CodeCreateFSError, "", err)
//	}
//	defer fs.Recycle()
//
//	// 获取回调会话
//	uploadSession := c.MustGet(filesystem.UploadSessionCtx).(*serializer.UploadSession)
//
//	// 获取文件信息
//	info, err := fs.Handler.(*s3.Driver).Meta(context.Background(), uploadSession.SavePath)
//	if err != nil {
//		return serializer.ErrDeprecated(serializer.CodeMetaMismatch, "", err)
//	}
//
//	// 验证实际文件信息与回调会话中是否一致
//	if uploadSession.Size != info.Size {
//		return serializer.ErrDeprecated(serializer.CodeMetaMismatch, "", err)
//	}
//
//	return ProcessCallback(service, c)
//}
//
//// PreProcess 对OneDrive客户端回调进行预处理验证
//func (service *UploadCallbackService) PreProcess(c *gin.Context) serializer.Response {
//	// 创建文件系统
//	fs, err := filesystem.NewFileSystemFromCallback(c)
//	if err != nil {
//		return serializer.ErrDeprecated(serializer.CodeCreateFSError, "", err)
//	}
//	defer fs.Recycle()
//
//	// 获取回调会话
//	uploadSession := c.MustGet(filesystem.UploadSessionCtx).(*serializer.UploadSession)
//
//	// 验证文件大小
//	if uploadSession.Size != service.Size {
//		fs.Handler.Delete(context.Background(), []string{uploadSession.SavePath})
//		return serializer.ErrDeprecated(serializer.CodeMetaMismatch, "", err)
//	}
//
//	return ProcessCallback(service, c)
//}