mirror of
https://github.com/oauth2-proxy/oauth2-proxy.git
synced 2024-11-28 09:08:44 +02:00
98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
package providers
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
"github.com/oauth2-proxy/oauth2-proxy/pkg/apis/sessions"
|
|
"github.com/oauth2-proxy/oauth2-proxy/pkg/requests"
|
|
)
|
|
|
|
// DigitalOceanProvider represents a DigitalOcean based Identity Provider
|
|
type DigitalOceanProvider struct {
|
|
*ProviderData
|
|
}
|
|
|
|
var _ Provider = (*DigitalOceanProvider)(nil)
|
|
|
|
const (
|
|
digitalOceanProviderName = "DigitalOcean"
|
|
digitalOceanDefaultScope = "read"
|
|
)
|
|
|
|
var (
|
|
// Default Login URL for DigitalOcean.
|
|
// Pre-parsed URL of https://cloud.digitalocean.com/v1/oauth/authorize.
|
|
digitalOceanDefaultLoginURL = &url.URL{
|
|
Scheme: "https",
|
|
Host: "cloud.digitalocean.com",
|
|
Path: "/v1/oauth/authorize",
|
|
}
|
|
|
|
// Default Redeem URL for DigitalOcean.
|
|
// Pre-parsed URL of https://cloud.digitalocean.com/v1/oauth/token.
|
|
digitalOceanDefaultRedeemURL = &url.URL{
|
|
Scheme: "https",
|
|
Host: "cloud.digitalocean.com",
|
|
Path: "/v1/oauth/token",
|
|
}
|
|
|
|
// Default Profile URL for DigitalOcean.
|
|
// Pre-parsed URL of https://cloud.digitalocean.com/v2/account.
|
|
digitalOceanDefaultProfileURL = &url.URL{
|
|
Scheme: "https",
|
|
Host: "api.digitalocean.com",
|
|
Path: "/v2/account",
|
|
}
|
|
)
|
|
|
|
// NewDigitalOceanProvider initiates a new DigitalOceanProvider
|
|
func NewDigitalOceanProvider(p *ProviderData) *DigitalOceanProvider {
|
|
p.setProviderDefaults(providerDefaults{
|
|
name: digitalOceanProviderName,
|
|
loginURL: digitalOceanDefaultLoginURL,
|
|
redeemURL: digitalOceanDefaultRedeemURL,
|
|
profileURL: digitalOceanDefaultProfileURL,
|
|
validateURL: digitalOceanDefaultProfileURL,
|
|
scope: digitalOceanDefaultScope,
|
|
})
|
|
return &DigitalOceanProvider{ProviderData: p}
|
|
}
|
|
|
|
func getDigitalOceanHeader(accessToken string) http.Header {
|
|
header := make(http.Header)
|
|
header.Set("Content-Type", "application/json")
|
|
header.Set("Authorization", fmt.Sprintf("Bearer %s", accessToken))
|
|
return header
|
|
}
|
|
|
|
// GetEmailAddress returns the Account email address
|
|
func (p *DigitalOceanProvider) GetEmailAddress(ctx context.Context, s *sessions.SessionState) (string, error) {
|
|
if s.AccessToken == "" {
|
|
return "", errors.New("missing access token")
|
|
}
|
|
|
|
json, err := requests.New(p.ProfileURL.String()).
|
|
WithContext(ctx).
|
|
WithHeaders(getDigitalOceanHeader(s.AccessToken)).
|
|
Do().
|
|
UnmarshalJSON()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
email, err := json.GetPath("account", "email").String()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return email, nil
|
|
}
|
|
|
|
// ValidateSessionState validates the AccessToken
|
|
func (p *DigitalOceanProvider) ValidateSessionState(ctx context.Context, s *sessions.SessionState) bool {
|
|
return validateToken(ctx, p, s.AccessToken, getDigitalOceanHeader(s.AccessToken))
|
|
}
|