forked from Open-CT/openct-tasks
62 lines
1.2 KiB
Go
62 lines
1.2 KiB
Go
package controllers
|
|
|
|
import (
|
|
_ "embed"
|
|
|
|
"github.com/astaxie/beego"
|
|
"github.com/casdoor/casdoor-go-sdk/auth"
|
|
)
|
|
|
|
//go:embed token_jwt_key.pem
|
|
var JwtPublicKey string
|
|
|
|
func init() {
|
|
InitAuthConfig()
|
|
}
|
|
|
|
func InitAuthConfig() {
|
|
casdoorEndpoint := beego.AppConfig.String("casdoorEndpoint")
|
|
clientId := beego.AppConfig.String("clientId")
|
|
clientSecret := beego.AppConfig.String("clientSecret")
|
|
casdoorOrganization := beego.AppConfig.String("casdoorOrganization")
|
|
casdoorApplication := beego.AppConfig.String("casdoorApplication")
|
|
|
|
auth.InitConfig(casdoorEndpoint, clientId, clientSecret, JwtPublicKey, casdoorOrganization, casdoorApplication)
|
|
}
|
|
|
|
func (c *ApiController) Signin() {
|
|
code := c.Input().Get("code")
|
|
state := c.Input().Get("state")
|
|
|
|
token, err := auth.GetOAuthToken(code, state)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
claims, err := auth.ParseJwtToken(token.AccessToken)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
claims.AccessToken = token.AccessToken
|
|
c.SetSessionClaims(claims)
|
|
|
|
c.ResponseOk(claims)
|
|
}
|
|
|
|
func (c *ApiController) Signout() {
|
|
c.SetSessionClaims(nil)
|
|
|
|
c.ResponseOk()
|
|
}
|
|
|
|
func (c *ApiController) GetAccount() {
|
|
if c.RequireSignedIn() {
|
|
return
|
|
}
|
|
|
|
claims := c.GetSessionClaims()
|
|
|
|
c.ResponseOk(claims)
|
|
}
|