llvy.ltd/casdoor
1
0
Fork 0
mirror of https://github.com/casdoor/casdoor.git synced 2025-07-03 12:30:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Change provider interface.

Browse Source
This commit is contained in:
Yang Luo
2021-03-23 23:23:59 +08:00
parent fa141a0edb
commit c24de7d180
5 changed files with 156 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
idp/github.go
View File

@ -15,6 +15,7 @@
package idp
import (
"context"
"encoding/json"
"io/ioutil"
"net/http"
@ -23,9 +24,35 @@ import (
"golang.org/x/oauth2"
)
type GithubIdProvider struct{}
type GithubIdProvider struct {
Client *http.Client
Config *oauth2.Config
ClientId string
ClientSecret string
RedirectUrl string
}
func (idp *GithubIdProvider) GetConfig() *oauth2.Config {
func NewGithubIdProvider(clientId string, clientSecret string, redirectUrl string) *GithubIdProvider {
idp := &GithubIdProvider{
ClientId: clientId,
ClientSecret: clientSecret,
RedirectUrl: redirectUrl,
}
config := idp.getConfig()
config.ClientID = clientId
config.ClientSecret = clientSecret
config.RedirectURL = redirectUrl
idp.Config = config
return idp
}
func (idp *GithubIdProvider) SetHttpClient(client *http.Client) {
idp.Client = client
}
func (idp *GithubIdProvider) getConfig() *oauth2.Config {
var endpoint = oauth2.Endpoint{
AuthURL: "https://github.com/login/oauth/authorize",
TokenURL: "https://github.com/login/oauth/access_token",
@ -39,7 +66,12 @@ func (idp *GithubIdProvider) GetConfig() *oauth2.Config {
return config
}
func (idp *GithubIdProvider) getEmail(httpClient *http.Client, token *oauth2.Token) string {
func (idp *GithubIdProvider) GetToken(code string) (*oauth2.Token, error) {
ctx := context.WithValue(oauth2.NoContext, oauth2.HTTPClient, idp.Client)
return idp.Config.Exchange(ctx, code)
}
func (idp *GithubIdProvider) getEmail(token *oauth2.Token) string {
res := ""
type GithubEmail struct {
@ -55,7 +87,7 @@ func (idp *GithubIdProvider) getEmail(httpClient *http.Client, token *oauth2.Tok
panic(err)
}
req.Header.Add("Authorization", "token "+token.AccessToken)
response, err := httpClient.Do(req)
response, err := idp.Client.Do(req)
if err != nil {
panic(err)
}
@ -75,7 +107,7 @@ func (idp *GithubIdProvider) getEmail(httpClient *http.Client, token *oauth2.Tok
return res
}
func (idp *GithubIdProvider) getLoginAndAvatar(httpClient *http.Client, token *oauth2.Token) (string, string) {
func (idp *GithubIdProvider) getLoginAndAvatar(token *oauth2.Token) (string, string) {
type GithubUser struct {
Login string `json:"login"`
AvatarUrl string `json:"avatar_url"`
@ -87,7 +119,7 @@ func (idp *GithubIdProvider) getLoginAndAvatar(httpClient *http.Client, token *o
panic(err)
}
req.Header.Add("Authorization", "token "+token.AccessToken)
resp, err := httpClient.Do(req)
resp, err := idp.Client.Do(req)
if err != nil {
panic(err)
}
@ -101,20 +133,20 @@ func (idp *GithubIdProvider) getLoginAndAvatar(httpClient *http.Client, token *o
return githubUser.Login, githubUser.AvatarUrl
}
func (idp *GithubIdProvider) GetUserInfo(httpClient *http.Client, token *oauth2.Token) (string, string, string, error) {
var email, username, avatarUrl string
func (idp *GithubIdProvider) GetUserInfo(token *oauth2.Token) (*UserInfo, error) {
userInfo := &UserInfo{}
var wg sync.WaitGroup
wg.Add(2)
go func() {
email = idp.getEmail(httpClient, token)
userInfo.Email = idp.getEmail(token)
wg.Done()
}()
go func() {
username, avatarUrl = idp.getLoginAndAvatar(httpClient, token)
userInfo.Username, userInfo.AvatarUrl = idp.getLoginAndAvatar(token)
wg.Done()
}()
wg.Wait()
return email, username, avatarUrl, nil
return userInfo, nil
}

52
idp/google.go
View File

@ -15,6 +15,7 @@
package idp
import (
"context"
"encoding/json"
"errors"
"io/ioutil"
@ -23,9 +24,35 @@ import (
"golang.org/x/oauth2"
)
type GoogleIdProvider struct{}
type GoogleIdProvider struct {
Client *http.Client
Config *oauth2.Config
ClientId string
ClientSecret string
RedirectUrl string
}
func (idp *GoogleIdProvider) GetConfig() *oauth2.Config {
func NewGoogleIdProvider(clientId string, clientSecret string, redirectUrl string) *GithubIdProvider {
idp := &GithubIdProvider{
ClientId: clientId,
ClientSecret: clientSecret,
RedirectUrl: redirectUrl,
}
config := idp.getConfig()
config.ClientID = clientId
config.ClientSecret = clientSecret
config.RedirectURL = redirectUrl
idp.Config = config
return idp
}
func (idp *GoogleIdProvider) SetHttpClient(client *http.Client) {
idp.Client = client
}
func (idp *GoogleIdProvider) getConfig() *oauth2.Config {
var endpoint = oauth2.Endpoint{
AuthURL: "https://accounts.google.com/o/oauth2/auth",
TokenURL: "https://accounts.google.com/o/oauth2/token",
@ -39,15 +66,20 @@ func (idp *GoogleIdProvider) GetConfig() *oauth2.Config {
return config
}
func (idp *GoogleIdProvider) GetUserInfo(httpClient *http.Client, token *oauth2.Token) (string, string, string, error) {
var email, username, avatarUrl string
func (idp *GoogleIdProvider) GetToken(code string) (*oauth2.Token, error) {
ctx := context.WithValue(oauth2.NoContext, oauth2.HTTPClient, idp.Client)
return idp.Config.Exchange(ctx, code)
}
func (idp *GoogleIdProvider) GetUserInfo(token *oauth2.Token) (*UserInfo, error) {
userInfo := &UserInfo{}
type userInfoFromGoogle struct {
Picture string `json:"picture"`
Email string `json:"email"`
}
resp, err := httpClient.Get("https://www.googleapis.com/oauth2/v2/userinfo?alt=json&access_token=" + token.AccessToken)
resp, err := idp.Client.Get("https://www.googleapis.com/oauth2/v2/userinfo?alt=json&access_token=" + token.AccessToken)
defer resp.Body.Close()
contents, err := ioutil.ReadAll(resp.Body)
var tempUser userInfoFromGoogle
@ -55,12 +87,12 @@ func (idp *GoogleIdProvider) GetUserInfo(httpClient *http.Client, token *oauth2.
if err != nil {
panic(err)
}
email = tempUser.Email
avatarUrl = tempUser.Picture
userInfo.Email = tempUser.Email
userInfo.AvatarUrl = tempUser.Picture
if email == "" {
return email, username, avatarUrl, errors.New("google email is empty, please try again")
if userInfo.Email == "" {
return userInfo, errors.New("google email is empty, please try again")
}
return email, username, avatarUrl, nil
return userInfo, nil
}

21
idp/provider.go
View File

@ -20,18 +20,25 @@ import (
"golang.org/x/oauth2"
)
type IdProvider interface {
GetConfig() *oauth2.Config
GetUserInfo(httpClient *http.Client, token *oauth2.Token) (string, string, string, error)
type UserInfo struct {
Username string
Email string
AvatarUrl string
}
func GetIdProvider(providerType string, clientId string) IdProvider {
type IdProvider interface {
SetHttpClient(client *http.Client)
GetToken(code string) (*oauth2.Token, error)
GetUserInfo(token *oauth2.Token) (*UserInfo, error)
}
func GetIdProvider(providerType string, clientId string, clientSecret string, redirectUrl string) IdProvider {
if providerType == "github" {
return &GithubIdProvider{}
return NewGithubIdProvider(clientId, clientSecret, redirectUrl)
} else if providerType == "google" {
return &GoogleIdProvider{}
return NewGoogleIdProvider(clientId, clientSecret, redirectUrl)
} else if providerType == "qq" {
return &QqIdProvider{ClientId: clientId}
return NewQqIdProvider(clientId, clientSecret, redirectUrl)
}
return nil

57
idp/qq.go
View File

@ -15,6 +15,7 @@
package idp
import (
"context"
"encoding/json"
"errors"
"fmt"
@ -26,10 +27,34 @@ import (
)
type QqIdProvider struct {
Client *http.Client
Config *oauth2.Config
ClientId string
ClientSecret string
RedirectUrl string
}
func (idp *QqIdProvider) GetConfig() *oauth2.Config {
func NewQqIdProvider(clientId string, clientSecret string, redirectUrl string) *QqIdProvider {
idp := &QqIdProvider{
ClientId: clientId,
ClientSecret: clientSecret,
RedirectUrl: redirectUrl,
}
config := idp.getConfig()
config.ClientID = clientId
config.ClientSecret = clientSecret
config.RedirectURL = redirectUrl
idp.Config = config
return idp
}
func (idp *QqIdProvider) SetHttpClient(client *http.Client) {
idp.Client = client
}
func (idp *QqIdProvider) getConfig() *oauth2.Config {
var endpoint = oauth2.Endpoint{
TokenURL: "https://graph.qq.com/oauth2.0/token",
}
@ -42,8 +67,13 @@ func (idp *QqIdProvider) GetConfig() *oauth2.Config {
return config
}
func (idp *QqIdProvider) GetUserInfo(httpClient *http.Client, token *oauth2.Token) (string, string, string, error) {
var email, username, avatarUrl string
func (idp *QqIdProvider) GetToken(code string) (*oauth2.Token, error) {
ctx := context.WithValue(oauth2.NoContext, oauth2.HTTPClient, idp.Client)
return idp.Config.Exchange(ctx, code)
}
func (idp *QqIdProvider) GetUserInfo(token *oauth2.Token) (*UserInfo, error) {
userInfo := &UserInfo{}
type userInfoFromQq struct {
Ret int `json:"ret"`
@ -53,7 +83,7 @@ func (idp *QqIdProvider) GetUserInfo(httpClient *http.Client, token *oauth2.Toke
getOpenIdUrl := fmt.Sprintf("https://graph.qq.com/oauth2.0/me?access_token=%s", token)
openIdResponse, err := httpClient.Get(getOpenIdUrl)
openIdResponse, err := idp.Client.Get(getOpenIdUrl)
if err != nil {
panic(err)
}
@ -65,25 +95,24 @@ func (idp *QqIdProvider) GetUserInfo(httpClient *http.Client, token *oauth2.Toke
openId := openIdRegRes[0][1]
if openId == "" {
return "", "", "", errors.New("openId is empty")
return userInfo, errors.New("openId is empty")
}
getUserInfoUrl := fmt.Sprintf("https://graph.qq.com/user/get_user_info?access_token=%s&oauth_consumer_key=%s&openid=%s", token, idp.ClientId, openId)
getUserInfoResponse, err := httpClient.Get(getUserInfoUrl)
getUserInfoResponse, err := idp.Client.Get(getUserInfoUrl)
if err != nil {
panic(err)
}
defer getUserInfoResponse.Body.Close()
userInfoContent, err := ioutil.ReadAll(getUserInfoResponse.Body)
var userInfo userInfoFromQq
err = json.Unmarshal(userInfoContent, &userInfo)
if err != nil || userInfo.Ret != 0 {
return "", "", "", err
var info userInfoFromQq
err = json.Unmarshal(userInfoContent, &info)
if err != nil || info.Ret != 0 {
return userInfo, err
}
email = ""
username = userInfo.Nickname
avatarUrl = userInfo.AvatarUrl
userInfo.Username = info.Nickname
userInfo.AvatarUrl = userInfo.AvatarUrl
return email, username, avatarUrl, nil
return userInfo, nil
}