mirror of https://github.com/cloudreve/Cloudreve
146 lines
4.9 KiB
Go
146 lines
4.9 KiB
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)
|
|
//}
|