context.go

package passwordless

import (
	"net/http"

	"context"
)

type ctxKey int

const (
	reqKey ctxKey = 1
	rwKey  ctxKey = 2
)

// SetContext returns a Context containing the specified `ResponseWriter` and
// `Request`. If a nil Context is provided, a new one is returned.
func SetContext(ctx context.Context, rw http.ResponseWriter, r *http.Request) context.Context {
	if ctx == nil {
		ctx = context.Background()
	}
	ctx = context.WithValue(ctx, reqKey, r)
	ctx = context.WithValue(ctx, rwKey, rw)
	return ctx
}

// fromContext extracts a `ResponseWriter` and `Request` from the Context,
// assuming that `SetContext` was called previously to populate it.
func fromContext(ctx context.Context) (http.ResponseWriter, *http.Request) {
	var rw http.ResponseWriter = nil
	var req *http.Request = nil
	if ctx != nil {
		if v, ok := ctx.Value(rwKey).(http.ResponseWriter); ok {
			rw = v
		}
		if v, ok := ctx.Value(reqKey).(*http.Request); ok {
			req = v
		}
	}
	return rw, req
}