mirror of
https://github.com/teamhanko/hanko.git
synced 2025-10-29 07:40:07 +08:00
062aee4d45
Rename identities table columns for more clarity. Rename parameters,
arguments etc. to accommodate these changes.
Change that the SAML provider domain is persisted in the identities
table as the provider ID. Use the SAML Entity ID/Issuer ID of the
IdP instead.
Introduce saml identity entity (including migrations and a persister)
as a specialization of an identity to allow for determining the
correct provider name to return to the client/frontend and for assisting
in determining whether an identity is a SAML identity (i.e. SAML
identities should have a corresponding SAML Identity instance while
OAuth/OIDC entities do not).
110 lines
2.6 KiB
Go
110 lines
2.6 KiB
Go
package thirdparty
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/teamhanko/hanko/backend/config"
|
|
"golang.org/x/oauth2"
|
|
)
|
|
|
|
const (
|
|
DiscordAPIBase = "https://discord.com/api"
|
|
DiscordOauthAuthEndpoint = "https://discord.com/oauth2/authorize"
|
|
DiscordOauthTokenEndpoint = DiscordAPIBase + "/oauth2/token"
|
|
DiscordUserInfoEndpoint = DiscordAPIBase + "/users/@me"
|
|
)
|
|
|
|
var DefaultDiscordScopes = []string{
|
|
"identify",
|
|
"email",
|
|
}
|
|
|
|
type discordProvider struct {
|
|
config config.ThirdPartyProvider
|
|
oauthConfig *oauth2.Config
|
|
}
|
|
|
|
type DiscordUser struct {
|
|
ID string `json:"id"`
|
|
Username string `json:"username"`
|
|
GlobalName string `json:"global_name"`
|
|
Avatar string `json:"avatar"`
|
|
Email string `json:"email"`
|
|
Verified bool `json:"verified"`
|
|
}
|
|
|
|
// NewDiscordProvider creates a Discord third party provider.
|
|
func NewDiscordProvider(config config.ThirdPartyProvider, redirectURL string) (OAuthProvider, error) {
|
|
if !config.Enabled {
|
|
return nil, errors.New("discord provider is disabled")
|
|
}
|
|
|
|
return &discordProvider{
|
|
config: config,
|
|
oauthConfig: &oauth2.Config{
|
|
ClientID: config.ClientID,
|
|
ClientSecret: config.Secret,
|
|
Endpoint: oauth2.Endpoint{
|
|
AuthURL: DiscordOauthAuthEndpoint,
|
|
TokenURL: DiscordOauthTokenEndpoint,
|
|
},
|
|
Scopes: DefaultDiscordScopes,
|
|
RedirectURL: redirectURL,
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func (p discordProvider) AuthCodeURL(state string, opts ...oauth2.AuthCodeOption) string {
|
|
return p.oauthConfig.AuthCodeURL(state, opts...)
|
|
}
|
|
|
|
func (g discordProvider) GetOAuthToken(code string) (*oauth2.Token, error) {
|
|
return g.oauthConfig.Exchange(context.Background(), code)
|
|
}
|
|
|
|
func (g discordProvider) GetUserData(token *oauth2.Token) (*UserData, error) {
|
|
var user DiscordUser
|
|
if err := makeRequest(token, g.oauthConfig, DiscordUserInfoEndpoint, &user); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
data := &UserData{}
|
|
|
|
if user.Email != "" {
|
|
data.Emails = append(data.Emails, Email{
|
|
Email: user.Email,
|
|
Verified: user.Verified,
|
|
Primary: true,
|
|
})
|
|
}
|
|
|
|
if len(data.Emails) <= 0 {
|
|
return nil, errors.New("unable to find email with Discord provider")
|
|
}
|
|
|
|
data.Metadata = &Claims{
|
|
Issuer: DiscordAPIBase,
|
|
Subject: user.ID,
|
|
Name: user.GlobalName,
|
|
Picture: g.buildAvatarURL(user.ID, user.Avatar),
|
|
Email: user.Email,
|
|
EmailVerified: user.Verified,
|
|
}
|
|
|
|
return data, nil
|
|
}
|
|
|
|
func (g discordProvider) buildAvatarURL(userID string, avatarHash string) string {
|
|
if avatarHash == "" {
|
|
return "" // No image
|
|
}
|
|
|
|
return fmt.Sprintf("https://cdn.discordapp.com/avatars/%s/%s.png", userID, avatarHash)
|
|
}
|
|
|
|
func (g discordProvider) ID() string {
|
|
return g.config.ID
|
|
}
|