keel/net/http/middleware/cookietokenprovider.go

package middleware

import (
	"net/http"

	"github.com/pkg/errors"
)

type (
	CookieTokenProviderOptions struct{}
	CookieTokenProviderOption  func(*CookieTokenProviderOptions)
)

// GetDefaultCookieTokenOptions returns the default options
func GetDefaultCookieTokenOptions() CookieTokenProviderOptions {
	return CookieTokenProviderOptions{}
}

func CookieTokenProvider(cookieName string, opts ...CookieTokenProviderOption) TokenProvider {
	options := GetDefaultCookieTokenOptions()
	for _, opt := range opts {
		if opt != nil {
			opt(&options)
		}
	}
	return func(r *http.Request) (string, error) {
		if cookie, err := r.Cookie(cookieName); errors.Is(err, http.ErrNoCookie) {
			return "", nil
		} else if err != nil {
			return "", errors.New("failed to retrieve cookie")
		} else {
			return cookie.Value, nil
		}
	}
}