2017-11-26 20:29:17 -08:00
|
|
|
package auth
|
|
|
|
|
|
|
|
import (
|
2017-12-20 21:23:19 -08:00
|
|
|
"github.com/cosmos/cosmos-sdk/types"
|
2018-01-06 14:22:21 -08:00
|
|
|
"github.com/cosmos/cosmos-sdk/x/store"
|
2017-11-26 20:29:17 -08:00
|
|
|
)
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
|
|
|
Usage:
|
|
|
|
|
|
|
|
import "accounts"
|
|
|
|
|
|
|
|
var acc accounts.Account
|
|
|
|
|
|
|
|
accounts.SetAccount(ctx, acc)
|
2017-12-20 21:23:19 -08:00
|
|
|
acc2 := accounts.GetAccount(ctx)
|
2017-11-26 20:29:17 -08:00
|
|
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
type contextKey int // local to the auth module
|
|
|
|
|
|
|
|
const (
|
|
|
|
// A context key of the Account variety
|
|
|
|
contextKeyAccount contextKey = iota
|
|
|
|
)
|
|
|
|
|
2018-01-06 14:22:21 -08:00
|
|
|
func SetAccount(ctx types.Context, account store.Account) types.Context {
|
2018-01-03 17:20:21 -08:00
|
|
|
return ctx.WithValueUnsafe(contextKeyAccount, account)
|
2017-11-26 20:29:17 -08:00
|
|
|
}
|
|
|
|
|
2018-01-06 14:22:21 -08:00
|
|
|
func GetAccount(ctx types.Context) store.Account {
|
|
|
|
return ctx.Value(contextKeyAccount).(store.Account)
|
2017-11-26 20:29:17 -08:00
|
|
|
}
|