init
This commit is contained in:
132
service/callback/oauth.go
Normal file
132
service/callback/oauth.go
Normal file
@ -0,0 +1,132 @@
|
||||
package callback
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
model "github.com/cloudreve/Cloudreve/v3/models"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/cache"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/filesystem/driver/googledrive"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/filesystem/driver/onedrive"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/serializer"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/samber/lo"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// 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.ParamErr(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.ParamErr(fmt.Sprintf("Missing required scope: %s", missing), nil)
|
||||
}
|
||||
|
||||
policyID, ok := util.GetSession(c, "googledrive_oauth_policy").(uint)
|
||||
if !ok {
|
||||
return serializer.Err(serializer.CodeNotFound, "", nil)
|
||||
}
|
||||
|
||||
util.DeleteSession(c, "googledrive_oauth_policy")
|
||||
|
||||
policy, err := model.GetPolicyByID(policyID)
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodePolicyNotExist, "", nil)
|
||||
}
|
||||
|
||||
client, err := googledrive.NewClient(&policy)
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodeInternalSetting, "Failed to initialize Google Drive client", err)
|
||||
}
|
||||
|
||||
credential, err := client.ObtainToken(c, service.Code, "")
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodeInternalSetting, "Failed to fetch AccessToken", err)
|
||||
}
|
||||
|
||||
// 更新存储策略的 RefreshToken
|
||||
client.Policy.AccessKey = credential.RefreshToken
|
||||
if err := client.Policy.SaveAndClearCache(); err != nil {
|
||||
return serializer.DBErr("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.ParamErr(service.ErrorMsg, nil)
|
||||
}
|
||||
|
||||
policyID, ok := util.GetSession(c, "onedrive_oauth_policy").(uint)
|
||||
if !ok {
|
||||
return serializer.Err(serializer.CodeNotFound, "", nil)
|
||||
}
|
||||
|
||||
util.DeleteSession(c, "onedrive_oauth_policy")
|
||||
|
||||
policy, err := model.GetPolicyByID(policyID)
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodePolicyNotExist, "", nil)
|
||||
}
|
||||
|
||||
client, err := onedrive.NewClient(&policy)
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodeInternalSetting, "Failed to initialize OneDrive client", err)
|
||||
}
|
||||
|
||||
credential, err := client.ObtainToken(c, onedrive.WithCode(service.Code))
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodeInternalSetting, "Failed to fetch AccessToken", err)
|
||||
}
|
||||
|
||||
// 更新存储策略的 RefreshToken
|
||||
client.Policy.AccessKey = credential.RefreshToken
|
||||
if err := client.Policy.SaveAndClearCache(); err != nil {
|
||||
return serializer.DBErr("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.Err(serializer.CodeInternalSetting, "Failed to query SharePoint site 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
|
||||
}
|
||||
261
service/callback/upload.go
Normal file
261
service/callback/upload.go
Normal file
@ -0,0 +1,261 @@
|
||||
package callback
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
model "github.com/cloudreve/Cloudreve/v3/models"
|
||||
"strings"
|
||||
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/filesystem"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/filesystem/driver/cos"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/filesystem/driver/onedrive"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/filesystem/driver/s3"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/filesystem/fsctx"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/serializer"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// CallbackProcessService 上传请求回调正文接口
|
||||
type CallbackProcessService interface {
|
||||
GetBody() serializer.UploadCallback
|
||||
}
|
||||
|
||||
// RemoteUploadCallbackService 远程存储上传回调请求服务
|
||||
type RemoteUploadCallbackService struct {
|
||||
Data serializer.UploadCallback `json:"data" binding:"required"`
|
||||
}
|
||||
|
||||
// GetBody 返回回调正文
|
||||
func (service RemoteUploadCallbackService) GetBody() serializer.UploadCallback {
|
||||
return service.Data
|
||||
}
|
||||
|
||||
// UploadCallbackService OOS/七牛云存储上传回调请求服务
|
||||
type UploadCallbackService struct {
|
||||
Name string `json:"name"`
|
||||
SourceName string `json:"source_name"`
|
||||
PicInfo string `json:"pic_info"`
|
||||
Size uint64 `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
|
||||
}
|
||||
|
||||
// COSCallback COS 客户端回调正文
|
||||
type COSCallback struct {
|
||||
Bucket string `form:"bucket"`
|
||||
Etag string `form:"etag"`
|
||||
}
|
||||
|
||||
// S3Callback S3 客户端回调正文
|
||||
type S3Callback struct {
|
||||
}
|
||||
|
||||
// GetBody 返回回调正文
|
||||
func (service UpyunCallbackService) GetBody() serializer.UploadCallback {
|
||||
res := serializer.UploadCallback{}
|
||||
if service.Width != "" {
|
||||
res.PicInfo = service.Width + "," + service.Height
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
// GetBody 返回回调正文
|
||||
func (service UploadCallbackService) GetBody() serializer.UploadCallback {
|
||||
return serializer.UploadCallback{
|
||||
PicInfo: service.PicInfo,
|
||||
}
|
||||
}
|
||||
|
||||
// GetBody 返回回调正文
|
||||
func (service OneDriveCallback) GetBody() serializer.UploadCallback {
|
||||
var picInfo = "0,0"
|
||||
if service.Meta.Image.Width != 0 {
|
||||
picInfo = fmt.Sprintf("%d,%d", service.Meta.Image.Width, service.Meta.Image.Height)
|
||||
}
|
||||
return serializer.UploadCallback{
|
||||
PicInfo: picInfo,
|
||||
}
|
||||
}
|
||||
|
||||
// GetBody 返回回调正文
|
||||
func (service COSCallback) GetBody() serializer.UploadCallback {
|
||||
return serializer.UploadCallback{
|
||||
PicInfo: "",
|
||||
}
|
||||
}
|
||||
|
||||
// GetBody 返回回调正文
|
||||
func (service S3Callback) GetBody() serializer.UploadCallback {
|
||||
return serializer.UploadCallback{
|
||||
PicInfo: "",
|
||||
}
|
||||
}
|
||||
|
||||
// ProcessCallback 处理上传结果回调
|
||||
func ProcessCallback(service CallbackProcessService, c *gin.Context) serializer.Response {
|
||||
callbackBody := service.GetBody()
|
||||
|
||||
// 创建文件系统
|
||||
fs, err := filesystem.NewFileSystemFromCallback(c)
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodeCreateFSError, err.Error(), err)
|
||||
}
|
||||
defer fs.Recycle()
|
||||
|
||||
// 获取上传会话
|
||||
uploadSession := c.MustGet(filesystem.UploadSessionCtx).(*serializer.UploadSession)
|
||||
|
||||
// 查找上传会话创建的占位文件
|
||||
file, err := model.GetFilesByUploadSession(uploadSession.Key, fs.User.ID)
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodeUploadSessionExpired, "LocalUpload session file placeholder not exist", err)
|
||||
}
|
||||
|
||||
fileData := fsctx.FileStream{
|
||||
Size: uploadSession.Size,
|
||||
Name: uploadSession.Name,
|
||||
VirtualPath: uploadSession.VirtualPath,
|
||||
SavePath: uploadSession.SavePath,
|
||||
Mode: fsctx.Nop,
|
||||
Model: file,
|
||||
LastModified: uploadSession.LastModified,
|
||||
}
|
||||
|
||||
// 占位符未扣除容量需要校验和扣除
|
||||
if !fs.Policy.IsUploadPlaceholderWithSize() {
|
||||
fs.Use("AfterUpload", filesystem.HookValidateCapacity)
|
||||
fs.Use("AfterUpload", filesystem.HookChunkUploaded)
|
||||
}
|
||||
|
||||
fs.Use("AfterUpload", filesystem.HookPopPlaceholderToFile(callbackBody.PicInfo))
|
||||
fs.Use("AfterValidateFailed", filesystem.HookDeleteTempFile)
|
||||
err = fs.Upload(context.Background(), &fileData)
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodeUploadFailed, err.Error(), err)
|
||||
}
|
||||
|
||||
return serializer.Response{}
|
||||
}
|
||||
|
||||
// PreProcess 对OneDrive客户端回调进行预处理验证
|
||||
func (service *OneDriveCallback) PreProcess(c *gin.Context) serializer.Response {
|
||||
// 创建文件系统
|
||||
fs, err := filesystem.NewFileSystemFromCallback(c)
|
||||
if err != nil {
|
||||
return serializer.Err(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.Err(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.Err(serializer.CodeMetaMismatch, "", err)
|
||||
}
|
||||
service.Meta = info
|
||||
return ProcessCallback(service, c)
|
||||
}
|
||||
|
||||
// PreProcess 对COS客户端回调进行预处理
|
||||
func (service *COSCallback) PreProcess(c *gin.Context) serializer.Response {
|
||||
// 创建文件系统
|
||||
fs, err := filesystem.NewFileSystemFromCallback(c)
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodeCreateFSError, "", err)
|
||||
}
|
||||
defer fs.Recycle()
|
||||
|
||||
// 获取回调会话
|
||||
uploadSession := c.MustGet(filesystem.UploadSessionCtx).(*serializer.UploadSession)
|
||||
|
||||
// 获取文件信息
|
||||
info, err := fs.Handler.(cos.Driver).Meta(context.Background(), uploadSession.SavePath)
|
||||
if err != nil {
|
||||
return serializer.Err(serializer.CodeMetaMismatch, "", err)
|
||||
}
|
||||
|
||||
// 验证实际文件信息与回调会话中是否一致
|
||||
if uploadSession.Size != info.Size || uploadSession.Key != info.CallbackKey {
|
||||
return serializer.Err(serializer.CodeMetaMismatch, "", err)
|
||||
}
|
||||
|
||||
return ProcessCallback(service, c)
|
||||
}
|
||||
|
||||
// PreProcess 对S3客户端回调进行预处理
|
||||
func (service *S3Callback) PreProcess(c *gin.Context) serializer.Response {
|
||||
// 创建文件系统
|
||||
fs, err := filesystem.NewFileSystemFromCallback(c)
|
||||
if err != nil {
|
||||
return serializer.Err(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.Err(serializer.CodeMetaMismatch, "", err)
|
||||
}
|
||||
|
||||
// 验证实际文件信息与回调会话中是否一致
|
||||
if uploadSession.Size != info.Size {
|
||||
return serializer.Err(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.Err(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.Err(serializer.CodeMetaMismatch, "", err)
|
||||
}
|
||||
|
||||
return ProcessCallback(service, c)
|
||||
}
|
||||
Reference in New Issue
Block a user