Remove the custom constrtouctor and replace with an optional argument to override the discovery endpoit

This commit is contained in:
ydris 2022-02-16 08:50:05 +01:00
parent 4d2d193d1c
commit 713a082da1
2 changed files with 20 additions and 43 deletions

View file

@ -69,11 +69,12 @@ var (
)
type relyingParty struct {
issuer string
endpoints Endpoints
oauthConfig *oauth2.Config
oauth2Only bool
pkce bool
issuer string
DiscoveryEndpoint string
endpoints Endpoints
oauthConfig *oauth2.Config
oauth2Only bool
pkce bool
httpClient *http.Client
cookieHandler *httphelper.CookieHandler
@ -170,7 +171,7 @@ func NewRelyingPartyOIDC(issuer, clientID, clientSecret, redirectURI string, sco
return nil, err
}
}
discoveryConfiguration, err := client.Discover(rp.issuer, rp.httpClient)
discoveryConfiguration, err := client.Discover(rp.issuer, rp.httpClient, rp.DiscoveryEndpoint)
if err != nil {
return nil, err
}
@ -181,45 +182,16 @@ func NewRelyingPartyOIDC(issuer, clientID, clientSecret, redirectURI string, sco
return rp, nil
}
//NewRelyingPartyOIDCWithCustomEndpoints creates an (OIDC) RelyingParty with the given
//discoveryConfiguration, clientID, clientSecret, redirectURI, scopes and other possible configOptions
//it will use the provided end points
//This is usefull when the server does not use standard endpoint paths
func NewRelyingPartyOIDCWithCustomEndpoints(
discoveryConfiguration *oidc.DiscoveryConfiguration,
clientID,
clientSecret,
redirectURI string,
scopes []string,
options ...Option) (RelyingParty, error) {
rp := &relyingParty{
issuer: discoveryConfiguration.Issuer,
oauthConfig: &oauth2.Config{
ClientID: clientID,
ClientSecret: clientSecret,
RedirectURL: redirectURI,
Scopes: scopes,
},
httpClient: httphelper.DefaultHTTPClient,
oauth2Only: false,
}
for _, optFunc := range options {
if err := optFunc(rp); err != nil {
return nil, err
}
}
endpoints := GetEndpoints(discoveryConfiguration)
rp.oauthConfig.Endpoint = endpoints.Endpoint
rp.endpoints = endpoints
return rp, nil
}
//Option is the type for providing dynamic options to the relyingParty
type Option func(*relyingParty) error
func WithCustomDiscoveryUrl(url string) Option {
return func(rp *relyingParty) error {
rp.DiscoveryEndpoint = url
return nil
}
}
//WithCookieHandler set a `CookieHandler` for securing the various redirects
func WithCookieHandler(cookieHandler *httphelper.CookieHandler) Option {
return func(rp *relyingParty) error {