Init V4 community edition (#2265)
* Init V4 community edition * Init V4 community edition
This commit is contained in:
AaronLiu
@@ -1,140 +1,126 @@
|
||||
package controllers
|
||||
|
||||
import (
|
||||
model "github.com/cloudreve/Cloudreve/v3/models"
|
||||
"path"
|
||||
"strconv"
|
||||
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/serializer"
|
||||
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
||||
"github.com/cloudreve/Cloudreve/v3/service/callback"
|
||||
"fmt"
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/driver/upyun"
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/fs"
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/manager"
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/logging"
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/serializer"
|
||||
"github.com/cloudreve/Cloudreve/v4/pkg/util"
|
||||
"github.com/cloudreve/Cloudreve/v4/service/callback"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/qiniu/go-sdk/v7/auth/qbox"
|
||||
)
|
||||
|
||||
// RemoteCallback 远程上传回调
|
||||
func RemoteCallback(c *gin.Context) {
|
||||
var callbackBody callback.RemoteUploadCallbackService
|
||||
if err := c.ShouldBindJSON(&callbackBody); err == nil {
|
||||
res := callback.ProcessCallback(callbackBody, c)
|
||||
c.JSON(200, res)
|
||||
} else {
|
||||
c.JSON(200, ErrorResponse(err))
|
||||
}
|
||||
}
|
||||
|
||||
// QiniuCallback 七牛上传回调
|
||||
func QiniuCallback(c *gin.Context) {
|
||||
var callbackBody callback.UploadCallbackService
|
||||
if err := c.ShouldBindJSON(&callbackBody); err == nil {
|
||||
res := callback.ProcessCallback(callbackBody, c)
|
||||
if res.Code != 0 {
|
||||
c.JSON(401, serializer.GeneralUploadCallbackFailed{Error: res.Msg})
|
||||
} else {
|
||||
c.JSON(200, res)
|
||||
}
|
||||
} else {
|
||||
c.JSON(401, ErrorResponse(err))
|
||||
}
|
||||
}
|
||||
|
||||
// OSSCallback 阿里云OSS上传回调
|
||||
func OSSCallback(c *gin.Context) {
|
||||
var callbackBody callback.UploadCallbackService
|
||||
if err := c.ShouldBindJSON(&callbackBody); err == nil {
|
||||
if callbackBody.PicInfo == "," {
|
||||
callbackBody.PicInfo = ""
|
||||
}
|
||||
res := callback.ProcessCallback(callbackBody, c)
|
||||
c.JSON(200, res)
|
||||
} else {
|
||||
c.JSON(200, ErrorResponse(err))
|
||||
}
|
||||
}
|
||||
|
||||
// UpyunCallback 又拍云上传回调
|
||||
func UpyunCallback(c *gin.Context) {
|
||||
var callbackBody callback.UpyunCallbackService
|
||||
if err := c.ShouldBind(&callbackBody); err == nil {
|
||||
if callbackBody.Code != 200 {
|
||||
util.Log().Debug(
|
||||
"Upload callback returned error code:%d, message: %s",
|
||||
callbackBody.Code,
|
||||
callbackBody.Message,
|
||||
)
|
||||
// RemoteCallback process callback request to complete upload
|
||||
func ProcessCallback(failedStatusCode int, generalResp bool) gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
err := callback.ProcessCallback(c)
|
||||
if err != nil {
|
||||
if generalResp {
|
||||
c.JSON(failedStatusCode, serializer.GeneralUploadCallbackFailed{Error: err.Error()})
|
||||
} else {
|
||||
c.JSON(failedStatusCode, serializer.Err(c, err))
|
||||
}
|
||||
return
|
||||
}
|
||||
res := callback.ProcessCallback(callbackBody, c)
|
||||
c.JSON(200, res)
|
||||
} else {
|
||||
c.JSON(200, ErrorResponse(err))
|
||||
|
||||
c.JSON(200, serializer.Response{})
|
||||
}
|
||||
}
|
||||
|
||||
// OneDriveCallback OneDrive上传完成客户端回调
|
||||
func OneDriveCallback(c *gin.Context) {
|
||||
var callbackBody callback.OneDriveCallback
|
||||
if err := c.ShouldBindJSON(&callbackBody); err == nil {
|
||||
res := callbackBody.PreProcess(c)
|
||||
c.JSON(200, res)
|
||||
} else {
|
||||
c.JSON(200, ErrorResponse(err))
|
||||
// QiniuCallbackAuth 七牛回调签名验证
|
||||
func QiniuCallbackValidate(c *gin.Context) {
|
||||
session := c.MustGet(manager.UploadSessionCtx).(*fs.UploadSession)
|
||||
|
||||
// 验证回调是否来自qiniu
|
||||
mac := qbox.NewMac(session.Policy.AccessKey, session.Policy.SecretKey)
|
||||
ok, err := mac.VerifyCallback(c.Request)
|
||||
if err != nil {
|
||||
util.Log().Debug("Failed to verify callback request: %s", err)
|
||||
c.JSON(401, serializer.GeneralUploadCallbackFailed{Error: "Failed to verify callback request."})
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
|
||||
if !ok {
|
||||
c.JSON(401, serializer.GeneralUploadCallbackFailed{Error: "Invalid signature."})
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
|
||||
c.Next()
|
||||
}
|
||||
|
||||
// OSSCallbackValidate 阿里云OSS上传回调
|
||||
func OSSCallbackValidate(c *gin.Context) {
|
||||
var callbackBody callback.UploadCallbackService
|
||||
if err := c.ShouldBindJSON(&callbackBody); err == nil {
|
||||
uploadSession := c.MustGet(manager.UploadSessionCtx).(*fs.UploadSession)
|
||||
if uploadSession.Props.Size != callbackBody.Size {
|
||||
l := logging.FromContext(c)
|
||||
l.Error("Callback validate failed: size mismatch, expected: %d, actual:%d", uploadSession.Props.Size, callbackBody.Size)
|
||||
c.JSON(401,
|
||||
serializer.GeneralUploadCallbackFailed{
|
||||
Error: fmt.Sprintf("size mismatch"),
|
||||
})
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
|
||||
c.Next()
|
||||
} else {
|
||||
c.JSON(401, ErrorResponse(err))
|
||||
c.Abort()
|
||||
}
|
||||
}
|
||||
|
||||
// UpyunCallbackAuth 又拍云回调签名验证
|
||||
func UpyunCallbackAuth(c *gin.Context) {
|
||||
uploadSession := c.MustGet(manager.UploadSessionCtx).(*fs.UploadSession)
|
||||
l := logging.FromContext(c)
|
||||
if err := upyun.ValidateCallback(c, uploadSession); err != nil {
|
||||
l.Error("Failed to verify callback request: %s", err)
|
||||
|
||||
c.JSON(401, serializer.GeneralUploadCallbackFailed{Error: "Failed to verify callback request."})
|
||||
}
|
||||
|
||||
c.Next()
|
||||
}
|
||||
|
||||
// OneDriveOAuth OneDrive 授权回调
|
||||
func OneDriveOAuth(c *gin.Context) {
|
||||
var callbackBody callback.OauthService
|
||||
if err := c.ShouldBindQuery(&callbackBody); err == nil {
|
||||
res := callbackBody.OdAuth(c)
|
||||
redirect := model.GetSiteURL()
|
||||
redirect.Path = path.Join(redirect.Path, "/admin/policy")
|
||||
queries := redirect.Query()
|
||||
queries.Add("code", strconv.Itoa(res.Code))
|
||||
queries.Add("msg", res.Msg)
|
||||
queries.Add("err", res.Error)
|
||||
redirect.RawQuery = queries.Encode()
|
||||
c.Redirect(303, redirect.String())
|
||||
} else {
|
||||
c.JSON(200, ErrorResponse(err))
|
||||
}
|
||||
//var callbackBody callback.OauthService
|
||||
//if err := c.ShouldBindQuery(&callbackBody); err == nil {
|
||||
// res := callbackBody.OdAuth(c)
|
||||
// redirect := model.GetSiteURL()
|
||||
// redirect.Path = path.Join(redirect.Path, "/admin/policy")
|
||||
// queries := redirect.Query()
|
||||
// queries.Add("code", strconv.Itoa(res.Code))
|
||||
// queries.Add("msg", res.Msg)
|
||||
// queries.Add("err", res.Error)
|
||||
// redirect.RawQuery = queries.Encode()
|
||||
// c.Redirect(303, redirect.String())
|
||||
//} else {
|
||||
// c.JSON(200, ErrorResponse(err))
|
||||
//}
|
||||
}
|
||||
|
||||
// GoogleDriveOAuth Google Drive 授权回调
|
||||
func GoogleDriveOAuth(c *gin.Context) {
|
||||
var callbackBody callback.OauthService
|
||||
if err := c.ShouldBindQuery(&callbackBody); err == nil {
|
||||
res := callbackBody.GDriveAuth(c)
|
||||
redirect := model.GetSiteURL()
|
||||
redirect.Path = path.Join(redirect.Path, "/admin/policy")
|
||||
queries := redirect.Query()
|
||||
queries.Add("code", strconv.Itoa(res.Code))
|
||||
queries.Add("msg", res.Msg)
|
||||
queries.Add("err", res.Error)
|
||||
redirect.RawQuery = queries.Encode()
|
||||
c.Redirect(303, redirect.String())
|
||||
} else {
|
||||
c.JSON(200, ErrorResponse(err))
|
||||
}
|
||||
}
|
||||
|
||||
// COSCallback COS上传完成客户端回调
|
||||
func COSCallback(c *gin.Context) {
|
||||
var callbackBody callback.COSCallback
|
||||
if err := c.ShouldBindQuery(&callbackBody); err == nil {
|
||||
res := callbackBody.PreProcess(c)
|
||||
c.JSON(200, res)
|
||||
} else {
|
||||
c.JSON(200, ErrorResponse(err))
|
||||
}
|
||||
}
|
||||
|
||||
// S3Callback S3上传完成客户端回调
|
||||
func S3Callback(c *gin.Context) {
|
||||
var callbackBody callback.S3Callback
|
||||
if err := c.ShouldBindQuery(&callbackBody); err == nil {
|
||||
res := callbackBody.PreProcess(c)
|
||||
c.JSON(200, res)
|
||||
} else {
|
||||
c.JSON(200, ErrorResponse(err))
|
||||
}
|
||||
//var callbackBody callback.OauthService
|
||||
//if err := c.ShouldBindQuery(&callbackBody); err == nil {
|
||||
// res := callbackBody.GDriveAuth(c)
|
||||
// redirect := model.GetSiteURL()
|
||||
// redirect.Path = path.Join(redirect.Path, "/admin/policy")
|
||||
// queries := redirect.Query()
|
||||
// queries.Add("code", strconv.Itoa(res.Code))
|
||||
// queries.Add("msg", res.Msg)
|
||||
// queries.Add("err", res.Error)
|
||||
// redirect.RawQuery = queries.Encode()
|
||||
// c.Redirect(303, redirect.String())
|
||||
//} else {
|
||||
// c.JSON(200, ErrorResponse(err))
|
||||
//}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user