wsy182/Cloudreve
1
0
Fork 0
mirror of https://github.com/halejohn/Cloudreve.git synced 2026-01-26 09:34:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: OneDrive OAuth callback / edit policy

Browse Source
This commit is contained in:
HFO4
2020-03-02 16:20:33 +08:00
parent 2471dacec1
commit be8aa7b01d
5 changed files with 109 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
service/callback/oauth.go
View File

@@ -3,6 +3,7 @@ package callback
import (
"context"
model "github.com/HFO4/cloudreve/models"
"github.com/HFO4/cloudreve/pkg/cache"
"github.com/HFO4/cloudreve/pkg/filesystem/driver/onedrive"
"github.com/HFO4/cloudreve/pkg/serializer"
"github.com/HFO4/cloudreve/pkg/util"
@@ -11,13 +12,21 @@ import (
// OneDriveOauthService OneDrive 授权回调服务
type OneDriveOauthService struct {
Code string `form:"code" binding:"required"`
Code string `form:"code"`
Error string `form:"error"`
ErrorMsg string `form:"error_description"`
}
// Auth 更新认证信息
func (service *OneDriveOauthService) Auth(c *gin.Context) serializer.Response {
policyId := util.GetSession(c, "onedrive_oauth_policy").(uint)
policy, err := model.GetPolicyByID(policyId)
if service.Error != "" {
return serializer.ParamErr(service.ErrorMsg, nil)
}
policyID := util.GetSession(c, "onedrive_oauth_policy").(uint)
util.DeleteSession(c, "onedrive_oauth_policy")
policy, err := model.GetPolicyByID(policyID)
if err != nil {
return serializer.Err(serializer.CodeNotFound, "存储策略不存在", nil)
}
@@ -37,5 +46,7 @@ func (service *OneDriveOauthService) Auth(c *gin.Context) serializer.Response {
return serializer.DBErr("无法更新 RefreshToken", err)
}
cache.Deletes([]string{client.Policy.AccessKey}, "onedrive_")
return serializer.Response{}
}