mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-10-29 10:57:44 +09:00 
			
		
		
		
	Additional OAuth2 providers (#1010)
* add google+ * sort signin oauth2 providers based on the name so order is always the same * update auth tip for google+ * add gitlab provider * add bitbucket provider (and some go fmt) * add twitter provider * add facebook provider * add dropbox provider * add openid connect provider incl. new format of tips section in "Add New Source" * lower the amount of disk storage for each session to prevent issues while building cross platform (and disk overflow) * imports according to goimport and code style * make it possible to set custom urls to gitlab and github provider (only these could have a different host) * split up oauth2 into multiple files * small typo in comment * fix indention * fix indentation * fix new line before external import * fix layout of signin part * update "broken" dependency
This commit is contained in:
		
				
					committed by
					
						 Lunny Xiao
						Lunny Xiao
					
				
			
			
				
	
			
			
			
						parent
						
							2368bbb672
						
					
				
				
					commit
					950f2e2074
				
			
							
								
								
									
										187
									
								
								vendor/github.com/markbates/goth/providers/gitlab/gitlab.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										187
									
								
								vendor/github.com/markbates/goth/providers/gitlab/gitlab.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,187 @@ | ||||
| // Package gitlab implements the OAuth2 protocol for authenticating users through gitlab. | ||||
| // This package can be used as a reference implementation of an OAuth2 provider for Goth. | ||||
| package gitlab | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"encoding/json" | ||||
| 	"io" | ||||
| 	"io/ioutil" | ||||
| 	"net/http" | ||||
| 	"net/url" | ||||
| 	"strconv" | ||||
|  | ||||
| 	"github.com/markbates/goth" | ||||
| 	"golang.org/x/oauth2" | ||||
| 	"fmt" | ||||
| ) | ||||
|  | ||||
| // These vars define the Authentication, Token, and Profile URLS for Gitlab. If | ||||
| // using Gitlab CE or EE, you should change these values before calling New. | ||||
| // | ||||
| // Examples: | ||||
| //	gitlab.AuthURL = "https://gitlab.acme.com/oauth/authorize | ||||
| //	gitlab.TokenURL = "https://gitlab.acme.com/oauth/token | ||||
| //	gitlab.ProfileURL = "https://gitlab.acme.com/api/v3/user | ||||
| var ( | ||||
| 	AuthURL    = "https://gitlab.com/oauth/authorize" | ||||
| 	TokenURL   = "https://gitlab.com/oauth/token" | ||||
| 	ProfileURL = "https://gitlab.com/api/v3/user" | ||||
| ) | ||||
|  | ||||
| // Provider is the implementation of `goth.Provider` for accessing Gitlab. | ||||
| type Provider struct { | ||||
| 	ClientKey    string | ||||
| 	Secret       string | ||||
| 	CallbackURL  string | ||||
| 	HTTPClient   *http.Client | ||||
| 	config       *oauth2.Config | ||||
| 	providerName string | ||||
| 	authURL      string | ||||
| 	tokenURL     string | ||||
| 	profileURL   string | ||||
| } | ||||
|  | ||||
| // New creates a new Gitlab provider and sets up important connection details. | ||||
| // You should always call `gitlab.New` to get a new provider.  Never try to | ||||
| // create one manually. | ||||
| func New(clientKey, secret, callbackURL string, scopes ...string) *Provider { | ||||
| 	return NewCustomisedURL(clientKey, secret, callbackURL, AuthURL, TokenURL, ProfileURL, scopes...) | ||||
| } | ||||
|  | ||||
| // NewCustomisedURL is similar to New(...) but can be used to set custom URLs to connect to | ||||
| func NewCustomisedURL(clientKey, secret, callbackURL, authURL, tokenURL, profileURL string, scopes ...string) *Provider { | ||||
| 	p := &Provider{ | ||||
| 		ClientKey:    clientKey, | ||||
| 		Secret:       secret, | ||||
| 		CallbackURL:  callbackURL, | ||||
| 		providerName: "gitlab", | ||||
| 		profileURL:   profileURL, | ||||
| 	} | ||||
| 	p.config = newConfig(p, authURL, tokenURL, scopes) | ||||
| 	return p | ||||
| } | ||||
|  | ||||
| // Name is the name used to retrieve this provider later. | ||||
| func (p *Provider) Name() string { | ||||
| 	return p.providerName | ||||
| } | ||||
|  | ||||
| // SetName is to update the name of the provider (needed in case of multiple providers of 1 type) | ||||
| func (p *Provider) SetName(name string) { | ||||
| 	p.providerName = name | ||||
| } | ||||
|  | ||||
| func (p *Provider) Client() *http.Client { | ||||
| 	return goth.HTTPClientWithFallBack(p.HTTPClient) | ||||
| } | ||||
|  | ||||
| // Debug is a no-op for the gitlab package. | ||||
| func (p *Provider) Debug(debug bool) {} | ||||
|  | ||||
| // BeginAuth asks Gitlab for an authentication end-point. | ||||
| func (p *Provider) BeginAuth(state string) (goth.Session, error) { | ||||
| 	return &Session{ | ||||
| 		AuthURL: p.config.AuthCodeURL(state), | ||||
| 	}, nil | ||||
| } | ||||
|  | ||||
| // FetchUser will go to Gitlab and access basic information about the user. | ||||
| func (p *Provider) FetchUser(session goth.Session) (goth.User, error) { | ||||
| 	sess := session.(*Session) | ||||
| 	user := goth.User{ | ||||
| 		AccessToken:  sess.AccessToken, | ||||
| 		Provider:     p.Name(), | ||||
| 		RefreshToken: sess.RefreshToken, | ||||
| 		ExpiresAt:    sess.ExpiresAt, | ||||
| 	} | ||||
|  | ||||
| 	if user.AccessToken == "" { | ||||
| 		// data is not yet retrieved since accessToken is still empty | ||||
| 		return user, fmt.Errorf("%s cannot get user information without accessToken", p.providerName) | ||||
| 	} | ||||
|  | ||||
| 	response, err := p.Client().Get(p.profileURL + "?access_token=" + url.QueryEscape(sess.AccessToken)) | ||||
| 	if err != nil { | ||||
| 		if response != nil { | ||||
| 			response.Body.Close() | ||||
| 		} | ||||
| 		return user, err | ||||
| 	} | ||||
|  | ||||
| 	defer response.Body.Close() | ||||
|  | ||||
| 	if response.StatusCode != http.StatusOK { | ||||
| 		return user, fmt.Errorf("%s responded with a %d trying to fetch user information", p.providerName, response.StatusCode) | ||||
| 	} | ||||
|  | ||||
| 	bits, err := ioutil.ReadAll(response.Body) | ||||
| 	if err != nil { | ||||
| 		return user, err | ||||
| 	} | ||||
|  | ||||
| 	err = json.NewDecoder(bytes.NewReader(bits)).Decode(&user.RawData) | ||||
| 	if err != nil { | ||||
| 		return user, err | ||||
| 	} | ||||
|  | ||||
| 	err = userFromReader(bytes.NewReader(bits), &user) | ||||
|  | ||||
| 	return user, err | ||||
| } | ||||
|  | ||||
| func newConfig(provider *Provider, authURL, tokenURL string, scopes []string) *oauth2.Config { | ||||
| 	c := &oauth2.Config{ | ||||
| 		ClientID:     provider.ClientKey, | ||||
| 		ClientSecret: provider.Secret, | ||||
| 		RedirectURL:  provider.CallbackURL, | ||||
| 		Endpoint: oauth2.Endpoint{ | ||||
| 			AuthURL:  authURL, | ||||
| 			TokenURL: tokenURL, | ||||
| 		}, | ||||
| 		Scopes: []string{}, | ||||
| 	} | ||||
|  | ||||
| 	if len(scopes) > 0 { | ||||
| 		for _, scope := range scopes { | ||||
| 			c.Scopes = append(c.Scopes, scope) | ||||
| 		} | ||||
| 	} | ||||
| 	return c | ||||
| } | ||||
|  | ||||
| func userFromReader(r io.Reader, user *goth.User) error { | ||||
| 	u := struct { | ||||
| 		Name      string `json:"name"` | ||||
| 		Email     string `json:"email"` | ||||
| 		NickName  string `json:"username"` | ||||
| 		ID        int    `json:"id"` | ||||
| 		AvatarURL string `json:"avatar_url"` | ||||
| 	}{} | ||||
| 	err := json.NewDecoder(r).Decode(&u) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	user.Email = u.Email | ||||
| 	user.Name = u.Name | ||||
| 	user.NickName = u.NickName | ||||
| 	user.UserID = strconv.Itoa(u.ID) | ||||
| 	user.AvatarURL = u.AvatarURL | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| //RefreshTokenAvailable refresh token is provided by auth provider or not | ||||
| func (p *Provider) RefreshTokenAvailable() bool { | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| //RefreshToken get new access token based on the refresh token | ||||
| func (p *Provider) RefreshToken(refreshToken string) (*oauth2.Token, error) { | ||||
| 	token := &oauth2.Token{RefreshToken: refreshToken} | ||||
| 	ts := p.config.TokenSource(goth.ContextForClient(p.Client()), token) | ||||
| 	newToken, err := ts.Token() | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return newToken, err | ||||
| } | ||||
							
								
								
									
										63
									
								
								vendor/github.com/markbates/goth/providers/gitlab/session.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										63
									
								
								vendor/github.com/markbates/goth/providers/gitlab/session.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,63 @@ | ||||
| package gitlab | ||||
|  | ||||
| import ( | ||||
| 	"encoding/json" | ||||
| 	"errors" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/markbates/goth" | ||||
| ) | ||||
|  | ||||
| // Session stores data during the auth process with Gitlab. | ||||
| type Session struct { | ||||
| 	AuthURL      string | ||||
| 	AccessToken  string | ||||
| 	RefreshToken string | ||||
| 	ExpiresAt    time.Time | ||||
| } | ||||
|  | ||||
| var _ goth.Session = &Session{} | ||||
|  | ||||
| // GetAuthURL will return the URL set by calling the `BeginAuth` function on the Gitlab provider. | ||||
| func (s Session) GetAuthURL() (string, error) { | ||||
| 	if s.AuthURL == "" { | ||||
| 		return "", errors.New(goth.NoAuthUrlErrorMessage) | ||||
| 	} | ||||
| 	return s.AuthURL, nil | ||||
| } | ||||
|  | ||||
| // Authorize the session with Gitlab and return the access token to be stored for future use. | ||||
| func (s *Session) Authorize(provider goth.Provider, params goth.Params) (string, error) { | ||||
| 	p := provider.(*Provider) | ||||
| 	token, err := p.config.Exchange(goth.ContextForClient(p.Client()), params.Get("code")) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	if !token.Valid() { | ||||
| 		return "", errors.New("Invalid token received from provider") | ||||
| 	} | ||||
|  | ||||
| 	s.AccessToken = token.AccessToken | ||||
| 	s.RefreshToken = token.RefreshToken | ||||
| 	s.ExpiresAt = token.Expiry | ||||
| 	return token.AccessToken, err | ||||
| } | ||||
|  | ||||
| // Marshal the session into a string | ||||
| func (s Session) Marshal() string { | ||||
| 	b, _ := json.Marshal(s) | ||||
| 	return string(b) | ||||
| } | ||||
|  | ||||
| func (s Session) String() string { | ||||
| 	return s.Marshal() | ||||
| } | ||||
|  | ||||
| // UnmarshalSession wil unmarshal a JSON string into a session. | ||||
| func (p *Provider) UnmarshalSession(data string) (goth.Session, error) { | ||||
| 	s := &Session{} | ||||
| 	err := json.NewDecoder(strings.NewReader(data)).Decode(s) | ||||
| 	return s, err | ||||
| } | ||||
		Reference in New Issue
	
	Block a user