addcfbf5cb
* Update docs/sdk/clients.md * organize ADR directory like tendermint * docs: move spec-proposals into spec/ * remove lotion, moved to website repo * move getting-started to cosmos-hub, and voyager to website * docs: move lite/ into clients/lite/ * move introduction/ content to website repo * move resources/ content to website repo * mv sdk/clients.md to clients/clients.md * mv validators to cosmos-hub/validators * move deprecated sdk/ content to _attic * sdk/modules.md is duplicate with modules/README.md * consolidate remianing sdk/ files into a single sdk.md * move examples/ to docs/examples/ * mv docs/cosmos-hub to docs/gaia * Add keys/accounts section to localnet docs
136 lines
3.0 KiB
Go
136 lines
3.0 KiB
Go
package simplestake
|
|
|
|
import (
|
|
"github.com/tendermint/tendermint/crypto"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/bank"
|
|
)
|
|
|
|
const stakingToken = "stake"
|
|
|
|
const moduleName = "simplestake"
|
|
|
|
// simple stake keeper
|
|
type Keeper struct {
|
|
ck bank.Keeper
|
|
|
|
key sdk.StoreKey
|
|
cdc *codec.Codec
|
|
codespace sdk.CodespaceType
|
|
}
|
|
|
|
func NewKeeper(key sdk.StoreKey, bankKeeper bank.Keeper, codespace sdk.CodespaceType) Keeper {
|
|
cdc := codec.New()
|
|
codec.RegisterCrypto(cdc)
|
|
return Keeper{
|
|
key: key,
|
|
cdc: cdc,
|
|
ck: bankKeeper,
|
|
codespace: codespace,
|
|
}
|
|
}
|
|
|
|
func (k Keeper) getBondInfo(ctx sdk.Context, addr sdk.AccAddress) bondInfo {
|
|
store := ctx.KVStore(k.key)
|
|
bz := store.Get(addr)
|
|
if bz == nil {
|
|
return bondInfo{}
|
|
}
|
|
var bi bondInfo
|
|
err := k.cdc.UnmarshalBinaryLengthPrefixed(bz, &bi)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return bi
|
|
}
|
|
|
|
func (k Keeper) setBondInfo(ctx sdk.Context, addr sdk.AccAddress, bi bondInfo) {
|
|
store := ctx.KVStore(k.key)
|
|
bz, err := k.cdc.MarshalBinaryLengthPrefixed(bi)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
store.Set(addr, bz)
|
|
}
|
|
|
|
func (k Keeper) deleteBondInfo(ctx sdk.Context, addr sdk.AccAddress) {
|
|
store := ctx.KVStore(k.key)
|
|
store.Delete(addr)
|
|
}
|
|
|
|
// register a bond with the keeper
|
|
func (k Keeper) Bond(ctx sdk.Context, addr sdk.AccAddress, pubKey crypto.PubKey, stake sdk.Coin) (int64, sdk.Error) {
|
|
if stake.Denom != stakingToken {
|
|
return 0, ErrIncorrectStakingToken(k.codespace)
|
|
}
|
|
|
|
_, _, err := k.ck.SubtractCoins(ctx, addr, []sdk.Coin{stake})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
bi := k.getBondInfo(ctx, addr)
|
|
if bi.isEmpty() {
|
|
bi = bondInfo{
|
|
PubKey: pubKey,
|
|
Power: 0,
|
|
}
|
|
}
|
|
|
|
bi.Power = bi.Power + stake.Amount.Int64()
|
|
|
|
k.setBondInfo(ctx, addr, bi)
|
|
return bi.Power, nil
|
|
}
|
|
|
|
// register an unbond with the keeper
|
|
func (k Keeper) Unbond(ctx sdk.Context, addr sdk.AccAddress) (crypto.PubKey, int64, sdk.Error) {
|
|
bi := k.getBondInfo(ctx, addr)
|
|
if bi.isEmpty() {
|
|
return nil, 0, ErrInvalidUnbond(k.codespace)
|
|
}
|
|
k.deleteBondInfo(ctx, addr)
|
|
|
|
returnedBond := sdk.NewInt64Coin(stakingToken, bi.Power)
|
|
|
|
_, _, err := k.ck.AddCoins(ctx, addr, []sdk.Coin{returnedBond})
|
|
if err != nil {
|
|
return bi.PubKey, bi.Power, err
|
|
}
|
|
|
|
return bi.PubKey, bi.Power, nil
|
|
}
|
|
|
|
// FOR TESTING PURPOSES -------------------------------------------------
|
|
|
|
func (k Keeper) bondWithoutCoins(ctx sdk.Context, addr sdk.AccAddress, pubKey crypto.PubKey, stake sdk.Coin) (int64, sdk.Error) {
|
|
if stake.Denom != stakingToken {
|
|
return 0, ErrIncorrectStakingToken(k.codespace)
|
|
}
|
|
|
|
bi := k.getBondInfo(ctx, addr)
|
|
if bi.isEmpty() {
|
|
bi = bondInfo{
|
|
PubKey: pubKey,
|
|
Power: 0,
|
|
}
|
|
}
|
|
|
|
bi.Power = bi.Power + stake.Amount.Int64()
|
|
|
|
k.setBondInfo(ctx, addr, bi)
|
|
return bi.Power, nil
|
|
}
|
|
|
|
func (k Keeper) unbondWithoutCoins(ctx sdk.Context, addr sdk.AccAddress) (crypto.PubKey, int64, sdk.Error) {
|
|
bi := k.getBondInfo(ctx, addr)
|
|
if bi.isEmpty() {
|
|
return nil, 0, ErrInvalidUnbond(k.codespace)
|
|
}
|
|
k.deleteBondInfo(ctx, addr)
|
|
|
|
return bi.PubKey, bi.Power, nil
|
|
}
|