package context

import (
	"context"

	"github.com/markbates/goth"
	"github.com/pkg/errors"
)

const keyUser = "user"

func User(ctx context.Context) *goth.User {
	user, ok := ctx.Value(keyUser).(*goth.User)
	if !ok {
		panic(errors.New("no user in context"))
	}

	return user
}

func SetUser(ctx context.Context, user *goth.User) context.Context {
	return context.WithValue(ctx, keyUser, user)
}