MaysWind
79 lines
2.4 KiB
Go
79 lines
2.4 KiB
Go
package oauth2
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/mayswind/ezbookkeeping/pkg/core"
|
|
"github.com/mayswind/ezbookkeeping/pkg/errs"
|
|
"github.com/mayswind/ezbookkeeping/pkg/log"
|
|
)
|
|
|
|
// CommonOAuth2Provider represents common OAuth 2.0 provider
|
|
type CommonOAuth2Provider struct {
|
|
OAuth2Provider
|
|
dataSource CommonOAuth2DataSource
|
|
}
|
|
|
|
// CommonOAuth2DataSource defines the structure of OAuth 2.0 data source
|
|
type CommonOAuth2DataSource interface {
|
|
// GetAuthUrl returns the authentication url of the data source
|
|
GetAuthUrl() string
|
|
|
|
// GetTokenUrl returns the token url of the data source
|
|
GetTokenUrl() string
|
|
|
|
// GetUserInfoRequest returns the user info request of the data source
|
|
GetUserInfoRequest() (*http.Request, error)
|
|
|
|
// GetScopes returns the scopes required by the data source
|
|
GetScopes() []string
|
|
|
|
// ParseUserInfo returns the user info by parsing the response body
|
|
ParseUserInfo(c core.Context, body []byte) (*OAuth2UserInfo, error)
|
|
}
|
|
|
|
// GetAuthUrl returns the authentication url of the common OAuth 2.0 provider
|
|
func (p *CommonOAuth2Provider) GetAuthUrl() string {
|
|
return p.dataSource.GetAuthUrl()
|
|
}
|
|
|
|
// GetTokenUrl returns the token url of the common OAuth 2.0 provider
|
|
func (p *CommonOAuth2Provider) GetTokenUrl() string {
|
|
return p.dataSource.GetTokenUrl()
|
|
}
|
|
|
|
// GetScopes returns the scopes required by the common OAuth 2.0 provider
|
|
func (p *CommonOAuth2Provider) GetScopes() []string {
|
|
return p.dataSource.GetScopes()
|
|
}
|
|
|
|
// GetUserInfo returns the user info by the common OAuth 2.0 provider
|
|
func (p *CommonOAuth2Provider) GetUserInfo(c core.Context, oauth2Client *http.Client) (*OAuth2UserInfo, error) {
|
|
req, err := p.dataSource.GetUserInfoRequest()
|
|
|
|
if err != nil {
|
|
log.Errorf(c, "[common_oauth2_provider.GetUserInfo] failed to get user info request, because %s", err.Error())
|
|
return nil, errs.ErrFailedToRequestRemoteApi
|
|
}
|
|
|
|
resp, err := oauth2Client.Do(req)
|
|
|
|
if err != nil {
|
|
log.Errorf(c, "[common_oauth2_provider.GetUserInfo] failed to get user info response, because %s", err.Error())
|
|
return nil, errs.ErrFailedToRequestRemoteApi
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
body, err := io.ReadAll(resp.Body)
|
|
|
|
log.Debugf(c, "[common_oauth2_provider.GetUserInfo] response is %s", body)
|
|
|
|
if resp.StatusCode != 200 {
|
|
log.Errorf(c, "[common_oauth2_provider.GetUserInfo] failed to get user info response, because response code is %d", resp.StatusCode)
|
|
return nil, errs.ErrFailedToRequestRemoteApi
|
|
}
|
|
|
|
return p.dataSource.ParseUserInfo(c, body)
|
|
}
|