Feat: OneDrive OAuth callback / edit policy

pull/247/head
HFO4 5 years ago
parent 2471dacec1
commit be8aa7b01d

@ -205,3 +205,25 @@ func AdminOneDriveOAuth(c *gin.Context) {
c.JSON(200, ErrorResponse(err)) c.JSON(200, ErrorResponse(err))
} }
} }
// AdminGetPolicy 获取存储策略详情
func AdminGetPolicy(c *gin.Context) {
var service admin.PolicyService
if err := c.ShouldBindUri(&service); err == nil {
res := service.Get()
c.JSON(200, res)
} else {
c.JSON(200, ErrorResponse(err))
}
}
// AdminDeletePolicy 删除存储策略
func AdminDeletePolicy(c *gin.Context) {
var service admin.PolicyService
if err := c.ShouldBindUri(&service); err == nil {
res := service.Delete()
c.JSON(200, res)
} else {
c.JSON(200, ErrorResponse(err))
}
}

@ -5,6 +5,8 @@ import (
"github.com/HFO4/cloudreve/pkg/util" "github.com/HFO4/cloudreve/pkg/util"
"github.com/HFO4/cloudreve/service/callback" "github.com/HFO4/cloudreve/service/callback"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"net/url"
"strconv"
) )
// RemoteCallback 远程上传回调 // RemoteCallback 远程上传回调
@ -82,7 +84,13 @@ func OneDriveOAuth(c *gin.Context) {
var callbackBody callback.OneDriveOauthService var callbackBody callback.OneDriveOauthService
if err := c.ShouldBindQuery(&callbackBody); err == nil { if err := c.ShouldBindQuery(&callbackBody); err == nil {
res := callbackBody.Auth(c) res := callbackBody.Auth(c)
c.JSON(200, res) redirect, _ := url.Parse("/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(301, "/#"+redirect.String())
} else { } else {
c.JSON(200, ErrorResponse(err)) c.JSON(200, ErrorResponse(err))
} }

@ -345,6 +345,10 @@ func InitMasterRouter() *gin.Engine {
policy.POST("scf", controllers.AdminAddSCF) policy.POST("scf", controllers.AdminAddSCF)
// 获取 OneDrive OAuth URL // 获取 OneDrive OAuth URL
policy.GET(":id/oauth", controllers.AdminOneDriveOAuth) policy.GET(":id/oauth", controllers.AdminOneDriveOAuth)
// 获取 存储策略
policy.GET(":id", controllers.AdminGetPolicy)
// 删除 存储策略
policy.DELETE(":id", controllers.AdminDeletePolicy)
} }
} }

@ -21,6 +21,7 @@ import (
"net/url" "net/url"
"os" "os"
"path/filepath" "path/filepath"
"strings"
"time" "time"
) )
@ -51,6 +52,52 @@ type PolicyService struct {
Region string `json:"region"` Region string `json:"region"`
} }
// Delete 删除存储策略
func (service *PolicyService) Delete() serializer.Response {
policy, err := model.GetPolicyByID(service.ID)
if err != nil {
return serializer.Err(serializer.CodeNotFound, "存储策略不存在", err)
}
// 检查是否有文件使用
total := 0
row := model.DB.Model(&model.File{}).Where("policy_id = ?", service.ID).
Select("count(id)").Row()
row.Scan(&total)
if total > 0 {
return serializer.ParamErr(fmt.Sprintf("有 %d 个文件仍在使用此存储策略,请先删除这些文件", total), nil)
}
// 检查用户组使用
var groups []model.Group
model.DB.Model(&model.Group{}).Where(
"policies like ? OR policies like ? OR policies like ? OR policies like ?",
fmt.Sprintf("[%d,%%", service.ID),
fmt.Sprintf("%%,%d]", service.ID),
fmt.Sprintf("%%,%d,%%", service.ID),
fmt.Sprintf("%%[%d]%%", service.ID),
).Find(&groups)
if len(groups) > 0 {
return serializer.ParamErr(fmt.Sprintf("有 %d 个用户组绑定了此存储策略,请先解除绑定", len(groups)), nil)
}
model.DB.Delete(&policy)
policy.ClearCache()
return serializer.Response{}
}
// Get 获取存储策略详情
func (service *PolicyService) Get() serializer.Response {
policy, err := model.GetPolicyByID(service.ID)
if err != nil {
return serializer.Err(serializer.CodeNotFound, "存储策略不存在", err)
}
return serializer.Response{Data: policy}
}
// GetOAuth 获取 OneDrive OAuth 地址 // GetOAuth 获取 OneDrive OAuth 地址
func (service *PolicyService) GetOAuth(c *gin.Context) serializer.Response { func (service *PolicyService) GetOAuth(c *gin.Context) serializer.Response {
policy, err := model.GetPolicyByID(service.ID) policy, err := model.GetPolicyByID(service.ID)
@ -195,9 +242,20 @@ func (service *SlaveTestService) Test() serializer.Response {
// Add 添加存储策略 // Add 添加存储策略
func (service *AddPolicyService) Add() serializer.Response { func (service *AddPolicyService) Add() serializer.Response {
if err := model.DB.Create(&service.Policy).Error; err != nil { service.Policy.DirNameRule = strings.TrimPrefix(service.Policy.DirNameRule, "/")
return serializer.ParamErr("存储策略添加失败", err)
if service.Policy.ID > 0 {
if err := model.DB.Save(&service.Policy).Error; err != nil {
return serializer.ParamErr("存储策略保存失败", err)
}
} else {
if err := model.DB.Create(&service.Policy).Error; err != nil {
return serializer.ParamErr("存储策略添加失败", err)
}
} }
service.Policy.ClearCache()
return serializer.Response{Data: service.Policy.ID} return serializer.Response{Data: service.Policy.ID}
} }

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

Loading…
Cancel
Save