middleware.go

// Copyright 2020 CleverGo. All rights reserved.
// Use of this source code is governed by a MIT style license that can be found
// in the LICENSE file.

package authmiddleware

import (
	"context"
	"log"

	"clevergo.tech/auth"
	"clevergo.tech/clevergo"
)

// GetIdentity retrieves a authenticated identity from context.
func GetIdentity(ctx context.Context) auth.Identity {
	return auth.GetIdentity(ctx)
}

// New returns a middleware with the given authenticator.
func New(authenticator auth.Authenticator) clevergo.MiddlewareFunc {
	return func(next clevergo.Handle) clevergo.Handle {
		return func(c *clevergo.Context) error {
			identity, err := authenticator.Authenticate(c.Request, c.Response)
			if err != nil {
				log.Println(err)
				authenticator.Challenge(c.Request, c.Response)
			} else {
				c.WithValue(auth.IdentityKey, identity)
			}
			return next(c)
		}
	}
}