Feat: OneDrive OAuth

service/admin/policy.go

@@ -2,16 +2,20 @@ package admin
 
 import (
 	"bytes"
+	"context"
 	"encoding/json"
 	"fmt"
 	model "github.com/HFO4/cloudreve/models"
 	"github.com/HFO4/cloudreve/pkg/auth"
+	"github.com/HFO4/cloudreve/pkg/cache"
 	"github.com/HFO4/cloudreve/pkg/conf"
 	"github.com/HFO4/cloudreve/pkg/filesystem/driver/cos"
+	"github.com/HFO4/cloudreve/pkg/filesystem/driver/onedrive"
 	"github.com/HFO4/cloudreve/pkg/filesystem/driver/oss"
 	"github.com/HFO4/cloudreve/pkg/request"
 	"github.com/HFO4/cloudreve/pkg/serializer"
 	"github.com/HFO4/cloudreve/pkg/util"
+	"github.com/gin-gonic/gin"
 	cossdk "github.com/tencentyun/cos-go-sdk-v5"
 	"net/http"
 	"net/url"
@@ -43,10 +47,34 @@ type AddPolicyService struct {
 
 // PolicyService 存储策略ID服务
 type PolicyService struct {
-	ID     uint   `json:"id" binding:"required"`
+	ID     uint   `uri:"id" json:"id" binding:"required"`
 	Region string `json:"region"`
 }
 
+// GetOAuth 获取 OneDrive OAuth 地址
+func (service *PolicyService) GetOAuth(c *gin.Context) serializer.Response {
+	policy, err := model.GetPolicyByID(service.ID)
+	if err != nil || policy.Type != "onedrive" {
+		return serializer.Err(serializer.CodeNotFound, "存储策略不存在", nil)
+	}
+
+	client, err := onedrive.NewClient(&policy)
+	if err != nil {
+		return serializer.Err(serializer.CodeInternalSetting, "无法初始化 OneDrive 客户端", err)
+	}
+
+	util.SetSession(c, map[string]interface{}{
+		"onedrive_oauth_policy": policy.ID,
+	})
+
+	cache.Deletes([]string{policy.BucketName}, "onedrive_")
+
+	return serializer.Response{Data: client.OAuthURL(context.Background(), []string{
+		"offline_access",
+		"files.readwrite.all",
+	})}
+}
+
 // AddSCF 创建回调云函数
 func (service *PolicyService) AddSCF() serializer.Response {
 	policy, err := model.GetPolicyByID(service.ID)

service/callback/oauth.go

@@ -0,0 +1,41 @@
+package callback
+
+import (
+	"context"
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/filesystem/driver/onedrive"
+	"github.com/HFO4/cloudreve/pkg/serializer"
+	"github.com/HFO4/cloudreve/pkg/util"
+	"github.com/gin-gonic/gin"
+)
+
+// OneDriveOauthService OneDrive 授权回调服务
+type OneDriveOauthService struct {
+	Code string `form:"code" binding:"required"`
+}
+
+// Auth 更新认证信息
+func (service *OneDriveOauthService) Auth(c *gin.Context) serializer.Response {
+	policyId := util.GetSession(c, "onedrive_oauth_policy").(uint)
+	policy, err := model.GetPolicyByID(policyId)
+	if err != nil {
+		return serializer.Err(serializer.CodeNotFound, "存储策略不存在", nil)
+	}
+
+	client, err := onedrive.NewClient(&policy)
+	if err != nil {
+		return serializer.Err(serializer.CodeInternalSetting, "无法初始化 OneDrive 客户端", err)
+	}
+
+	credential, err := client.ObtainToken(context.Background(), onedrive.WithCode(service.Code))
+	if err != nil {
+		return serializer.Err(serializer.CodeInternalSetting, "AccessToken 获取失败", err)
+	}
+
+	// 更新存储策略的 RefreshToken
+	if err := client.Policy.UpdateAccessKey(credential.RefreshToken); err != nil {
+		return serializer.DBErr("无法更新 RefreshToken", err)
+	}
+
+	return serializer.Response{}
+}