chore: 初始化 Cloudreve 源码仓库
This commit is contained in:
@@ -0,0 +1,123 @@
|
||||
package callback
|
||||
|
||||
import (
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/serializer"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// OauthService OAuth 存储策略授权回调服务
|
||||
type OauthService struct {
|
||||
Code string `form:"code"`
|
||||
Error string `form:"error"`
|
||||
ErrorMsg string `form:"error_description"`
|
||||
Scope string `form:"scope"`
|
||||
}
|
||||
|
||||
//
|
||||
//// GDriveAuth Google Drive 更新认证信息
|
||||
//func (service *OauthService) GDriveAuth(c *gin.Context) serializer.Response {
|
||||
// if service.Error != "" {
|
||||
// return serializer.ParamErrDeprecated(service.Error, nil)
|
||||
// }
|
||||
//
|
||||
// // validate required scope
|
||||
// if missing, found := lo.Find[string](googledrive.RequiredScope, func(item string) bool {
|
||||
// return !strings.Contains(service.Scope, item)
|
||||
// }); found {
|
||||
// return serializer.ParamErrDeprecated(fmt.Sprintf("Missing required scope: %s", missing), nil)
|
||||
// }
|
||||
//
|
||||
// policyID, ok := util.GetSession(c, "googledrive_oauth_policy").(uint)
|
||||
// if !ok {
|
||||
// return serializer.ErrDeprecated(serializer.CodeNotFound, "", nil)
|
||||
// }
|
||||
//
|
||||
// util.DeleteSession(c, "googledrive_oauth_policy")
|
||||
//
|
||||
// policy, err := model.GetPolicyByID(policyID)
|
||||
// if err != nil {
|
||||
// return serializer.ErrDeprecated(serializer.CodePolicyNotExist, "", nil)
|
||||
// }
|
||||
//
|
||||
// client, err := googledrive.NewClient(&policy)
|
||||
// if err != nil {
|
||||
// return serializer.ErrDeprecated(serializer.CodeInternalSetting, "Failed to initialize Google Drive client", err)
|
||||
// }
|
||||
//
|
||||
// credential, err := client.ObtainToken(c, service.Code, "")
|
||||
// if err != nil {
|
||||
// return serializer.ErrDeprecated(serializer.CodeInternalSetting, "Failed to fetch AccessToken", err)
|
||||
// }
|
||||
//
|
||||
// // 更新存储策略的 RefreshToken
|
||||
// client.Policy.AccessKey = credential.RefreshToken
|
||||
// if err := client.Policy.SaveAndClearCache(); err != nil {
|
||||
// return serializer.DBErrDeprecated("Failed to update RefreshToken", err)
|
||||
// }
|
||||
//
|
||||
// cache.Deletes([]string{client.Policy.AccessKey}, googledrive.TokenCachePrefix)
|
||||
// return serializer.Response{}
|
||||
//}
|
||||
|
||||
// OdAuth OneDrive 更新认证信息
|
||||
func (service *OauthService) OdAuth(c *gin.Context) serializer.Response {
|
||||
//if service.Error != "" {
|
||||
// return serializer.ParamErrDeprecated(service.ErrorMsg, nil)
|
||||
//}
|
||||
//
|
||||
//policyID, ok := util.GetSession(c, "onedrive_oauth_policy").(uint)
|
||||
//if !ok {
|
||||
// return serializer.ErrDeprecated(serializer.CodeNotFound, "", nil)
|
||||
//}
|
||||
//
|
||||
//util.DeleteSession(c, "onedrive_oauth_policy")
|
||||
//
|
||||
//policy, err := model.GetPolicyByID(policyID)
|
||||
//if err != nil {
|
||||
// return serializer.ErrDeprecated(serializer.CodePolicyNotExist, "", nil)
|
||||
//}
|
||||
//
|
||||
//client, err := onedrive.NewClient(&policy)
|
||||
//if err != nil {
|
||||
// return serializer.ErrDeprecated(serializer.CodeInternalSetting, "Failed to initialize OneDrive client", err)
|
||||
//}
|
||||
//
|
||||
//credential, err := client.ObtainToken(c, onedrive.WithCode(service.Code))
|
||||
//if err != nil {
|
||||
// return serializer.ErrDeprecated(serializer.CodeInternalSetting, "Failed to fetch AccessToken", err)
|
||||
//}
|
||||
//
|
||||
//// 更新存储策略的 RefreshToken
|
||||
//client.Policy.AccessKey = credential.RefreshToken
|
||||
//if err := client.Policy.SaveAndClearCache(); err != nil {
|
||||
// return serializer.DBErrDeprecated("Failed to update RefreshToken", err)
|
||||
//}
|
||||
//
|
||||
//cache.Deletes([]string{client.Policy.AccessKey}, "onedrive_")
|
||||
//if client.Policy.OptionsSerialized.OdDriver != "" && strings.Contains(client.Policy.OptionsSerialized.OdDriver, "http") {
|
||||
// if err := querySharePointSiteID(c, client.Policy); err != nil {
|
||||
// return serializer.ErrDeprecated(serializer.CodeInternalSetting, "Failed to query SharePoint basic ID", err)
|
||||
// }
|
||||
//}
|
||||
|
||||
return serializer.Response{}
|
||||
}
|
||||
|
||||
//func querySharePointSiteID(ctx context.Context, policy *model.Policy) error {
|
||||
//client, err := onedrive.NewClient(policy)
|
||||
//if err != nil {
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
//id, err := client.GetSiteIDByURL(ctx, client.Policy.OptionsSerialized.OdDriver)
|
||||
//if err != nil {
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
//client.Policy.OptionsSerialized.OdDriver = fmt.Sprintf("sites/%s/drive", id)
|
||||
//if err := client.Policy.SaveAndClearCache(); err != nil {
|
||||
// return err
|
||||
//}
|
||||
|
||||
//return nil
|
||||
//}
|
||||
@@ -0,0 +1,145 @@
|
||||
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)
|
||||
//}
|
||||
Reference in New Issue
Block a user