forked from gitea/gitea
76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
|
package goth
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"net/http"
|
||
|
|
||
|
"golang.org/x/net/context"
|
||
|
"golang.org/x/oauth2"
|
||
|
)
|
||
|
|
||
|
// Provider needs to be implemented for each 3rd party authentication provider
|
||
|
// e.g. Facebook, Twitter, etc...
|
||
|
type Provider interface {
|
||
|
Name() string
|
||
|
SetName(name string)
|
||
|
BeginAuth(state string) (Session, error)
|
||
|
UnmarshalSession(string) (Session, error)
|
||
|
FetchUser(Session) (User, error)
|
||
|
Debug(bool)
|
||
|
RefreshToken(refreshToken string) (*oauth2.Token, error) //Get new access token based on the refresh token
|
||
|
RefreshTokenAvailable() bool //Refresh token is provided by auth provider or not
|
||
|
}
|
||
|
|
||
|
const NoAuthUrlErrorMessage = "an AuthURL has not been set"
|
||
|
|
||
|
// Providers is list of known/available providers.
|
||
|
type Providers map[string]Provider
|
||
|
|
||
|
var providers = Providers{}
|
||
|
|
||
|
// UseProviders adds a list of available providers for use with Goth.
|
||
|
// Can be called multiple times. If you pass the same provider more
|
||
|
// than once, the last will be used.
|
||
|
func UseProviders(viders ...Provider) {
|
||
|
for _, provider := range viders {
|
||
|
providers[provider.Name()] = provider
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// GetProviders returns a list of all the providers currently in use.
|
||
|
func GetProviders() Providers {
|
||
|
return providers
|
||
|
}
|
||
|
|
||
|
// GetProvider returns a previously created provider. If Goth has not
|
||
|
// been told to use the named provider it will return an error.
|
||
|
func GetProvider(name string) (Provider, error) {
|
||
|
provider := providers[name]
|
||
|
if provider == nil {
|
||
|
return nil, fmt.Errorf("no provider for %s exists", name)
|
||
|
}
|
||
|
return provider, nil
|
||
|
}
|
||
|
|
||
|
// ClearProviders will remove all providers currently in use.
|
||
|
// This is useful, mostly, for testing purposes.
|
||
|
func ClearProviders() {
|
||
|
providers = Providers{}
|
||
|
}
|
||
|
|
||
|
// ContextForClient provides a context for use with oauth2.
|
||
|
func ContextForClient(h *http.Client) context.Context {
|
||
|
if h == nil {
|
||
|
return oauth2.NoContext
|
||
|
}
|
||
|
return context.WithValue(oauth2.NoContext, oauth2.HTTPClient, h)
|
||
|
}
|
||
|
|
||
|
// HTTPClientWithFallBack to be used in all fetch operations.
|
||
|
func HTTPClientWithFallBack(h *http.Client) *http.Client {
|
||
|
if h != nil {
|
||
|
return h
|
||
|
}
|
||
|
return http.DefaultClient
|
||
|
}
|