Trying trings out with SATOSA

This commit is contained in:
Denis Arh
2018-09-17 16:01:05 +02:00
parent e348b3e2f4
commit 6e039e1782
2 changed files with 97 additions and 2 deletions
+85
View File
@@ -0,0 +1,85 @@
package sam
import (
"context"
"github.com/coreos/go-oidc"
"github.com/davecgh/go-spew/spew"
"golang.org/x/oauth2"
"net/http"
)
type (
openIdConnect struct {
provider *oidc.Provider
verifier *oidc.IDTokenVerifier
config oauth2.Config
}
)
func OpenIdConnect(ctx context.Context, issuer string, cfg oauth2.Config) (c *openIdConnect, err error) {
c = &openIdConnect{}
c.provider, err = oidc.NewProvider(ctx, issuer)
if err != nil {
return nil, err
}
// Configure an OpenID Connect aware OAuth2 client.
c.config = oauth2.Config{
ClientID: cfg.ClientID,
ClientSecret: cfg.ClientSecret,
RedirectURL: cfg.RedirectURL,
// Discovery returns the OAuth2 endpoints.
Endpoint: c.provider.Endpoint(),
// "openid" is a required scope for OpenID Connect flows.
Scopes: []string{oidc.ScopeOpenID, "profile", "email"},
}
c.verifier = c.provider.Verifier(&oidc.Config{ClientID: cfg.ClientID})
return
}
func (c *openIdConnect) HandleRedirect(w http.ResponseWriter, r *http.Request) {
state := "@todo"
http.Redirect(w, r, c.config.AuthCodeURL(state), http.StatusFound)
}
func (c *openIdConnect) HandleOAuth2Callback(w http.ResponseWriter, r *http.Request) {
var ctx = r.Context()
// @todo check state
// Verify state and errors.
oauth2Token, err := c.config.Exchange(ctx, r.URL.Query().Get("code"))
if err != nil {
// handle error
}
// Extract the ID Token from OAuth2 token.
rawIDToken, ok := oauth2Token.Extra("id_token").(string)
if !ok {
// handle missing token
}
// Parse and verify ID Token payload.
idToken, err := c.verifier.Verify(ctx, rawIDToken)
if err != nil {
// handle error
}
// Extract custom claims
var claims struct {
Email string `json:"email"`
Verified bool `json:"email_verified"`
}
if err := idToken.Claims(&claims); err != nil {
// handle error
}
spew.Dump()
}
+12 -2
View File
@@ -8,16 +8,17 @@ import (
"github.com/SentimensRG/ctx"
"github.com/SentimensRG/ctx/sigctx"
authService "github.com/crusttech/crust/auth/service"
samService "github.com/crusttech/crust/sam/service"
"github.com/go-chi/chi"
"github.com/go-chi/cors"
"github.com/pkg/errors"
"github.com/titpetric/factory"
"github.com/titpetric/factory/resputil"
"golang.org/x/oauth2"
authService "github.com/crusttech/crust/auth/service"
"github.com/crusttech/crust/internal/auth"
"github.com/crusttech/crust/sam/rest"
samService "github.com/crusttech/crust/sam/service"
"github.com/crusttech/crust/sam/websocket"
)
@@ -74,6 +75,15 @@ func Start() error {
r := chi.NewRouter()
r.Use(handleCORS)
if oidc, err := OpenIdConnect(ctx, "https://accounts.google.com", oauth2.Config{}); err != nil {
return errors.Wrap(err, "Could not initialize OIDC")
} else {
r.Route("/oidc/satosa", func(r chi.Router) {
r.Get("/", oidc.HandleRedirect)
r.Get("/callback", oidc.HandleOAuth2Callback)
})
}
// Only protect application routes with JWT
r.Group(func(r chi.Router) {
r.Use(jwtAuth.Verifier(), jwtAuth.Authenticator())