Fix: catch internal error in OneDrive OAuth callback

whaliendev · Apr 7, 2020 · 1a1543f · 1a1543f
1 parent 4aef12b
commit 1a1543f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/assets b/assets
diff --git a/go.mod b/go.mod
@@ -9,7 +9,7 @@ require (
 	github.com/duo-labs/webauthn v0.0.0-20191119193225-4bf9a0f776d4
 	github.com/fatih/color v1.7.0
 	github.com/gin-contrib/cors v1.3.0
-	github.com/gin-contrib/gzip v0.0.2-0.20200226035851-25bef2ef21e8 // indirect
+	github.com/gin-contrib/gzip v0.0.2-0.20200226035851-25bef2ef21e8
 	github.com/gin-contrib/sessions v0.0.1
 	github.com/gin-contrib/static v0.0.0-20191128031702-f81c604d8ac2
 	github.com/gin-gonic/gin v1.5.0

diff --git a/service/callback/oauth.go b/service/callback/oauth.go
@@ -23,7 +23,11 @@ func (service *OneDriveOauthService) Auth(c *gin.Context) serializer.Response {
 		return serializer.ParamErr(service.ErrorMsg, nil)
 	}
 
-	policyID := util.GetSession(c, "onedrive_oauth_policy").(uint)
+	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)