Платформа ЦРНП "Мирокод" для разработки проектов
https://git.mirocod.ru
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.0 KiB
75 lines
2.0 KiB
package goth |
|
|
|
import ( |
|
"context" |
|
"fmt" |
|
"net/http" |
|
|
|
"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 |
|
}
|
|
|