51b8b4380d
* feat: integrate alist with casdoor * fix: casdoor as an option for login Co-authored-by: wenxuan70 <t736660416@gmail.com>
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package controllers
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/Xhofe/alist/conf"
|
|
"github.com/Xhofe/alist/server/common"
|
|
"github.com/Xhofe/alist/utils"
|
|
"github.com/casdoor/casdoor-go-sdk/casdoorsdk"
|
|
"github.com/gin-gonic/gin"
|
|
"net/http"
|
|
)
|
|
|
|
type OAuthReq struct {
|
|
Code string `json:"code"`
|
|
State string `json:"state"`
|
|
}
|
|
|
|
func Verify(c *gin.Context) {
|
|
token := c.GetHeader("Authorization")
|
|
if token != conf.Token {
|
|
common.ErrorStrResp(c, "Invalid token", 401)
|
|
return
|
|
}
|
|
common.SuccessResp(c)
|
|
}
|
|
func GetRedirectUrl(c *gin.Context) {
|
|
if !conf.GetBool("Enable Casdoor") {
|
|
common.ErrorStrResp(c, "Casdoor is not enabled", 1001)
|
|
return
|
|
}
|
|
redirectUri := generateRedirectUri(c.Request)
|
|
common.SuccessResp(c, utils.GetSignInUrl(redirectUri))
|
|
}
|
|
|
|
func generateRedirectUri(r *http.Request) string {
|
|
protocol := "http"
|
|
host := r.Host
|
|
if r.TLS != nil {
|
|
protocol = "https"
|
|
}
|
|
return fmt.Sprintf("%s://%s/@manage", protocol, host)
|
|
}
|
|
|
|
func OAuth(c *gin.Context) {
|
|
var req OAuthReq
|
|
if err := c.ShouldBind(&req); err != nil {
|
|
common.ErrorResp(c, err, 401)
|
|
return
|
|
}
|
|
|
|
if req.Code == "" || req.State == "" {
|
|
common.ErrorStrResp(c, "Invalid code or state", 400)
|
|
return
|
|
}
|
|
|
|
token, err := casdoorsdk.GetOAuthToken(req.Code, req.State)
|
|
if err != nil {
|
|
common.ErrorResp(c, err, 401)
|
|
return
|
|
}
|
|
|
|
_, err = casdoorsdk.ParseJwtToken(token.AccessToken)
|
|
if err != nil {
|
|
common.ErrorResp(c, err, 401)
|
|
return
|
|
}
|
|
|
|
common.SuccessResp(c, conf.Token)
|
|
}
|