54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package keeper
|
|
|
|
import (
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/authz"
|
|
)
|
|
|
|
// InitGenesis new authz genesis
|
|
func (k Keeper) InitGenesis(ctx sdk.Context, data *authz.GenesisState) {
|
|
now := ctx.BlockTime()
|
|
for _, entry := range data.Authorization {
|
|
// ignore expired authorizations
|
|
if entry.Expiration != nil && entry.Expiration.Before(now) {
|
|
continue
|
|
}
|
|
|
|
grantee, err := sdk.AccAddressFromBech32(entry.Grantee)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
granter, err := sdk.AccAddressFromBech32(entry.Granter)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
a, ok := entry.Authorization.GetCachedValue().(authz.Authorization)
|
|
if !ok {
|
|
panic("expected authorization")
|
|
}
|
|
|
|
err = k.SaveGrant(ctx, grantee, granter, a, entry.Expiration)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// ExportGenesis returns a GenesisState for a given context.
|
|
func (k Keeper) ExportGenesis(ctx sdk.Context) *authz.GenesisState {
|
|
var entries []authz.GrantAuthorization
|
|
k.IterateGrants(ctx, func(granter, grantee sdk.AccAddress, grant authz.Grant) bool {
|
|
entries = append(entries, authz.GrantAuthorization{
|
|
Granter: granter.String(),
|
|
Grantee: grantee.String(),
|
|
Expiration: grant.Expiration,
|
|
Authorization: grant.Authorization,
|
|
})
|
|
return false
|
|
})
|
|
|
|
return authz.NewGenesisState(entries)
|
|
}
|