0g-chain/x/issuance/keeper/issuance.go
2021-02-04 11:35:24 -07:00

243 lines
7.6 KiB
Go

package keeper
import (
"fmt"
sdk "github.com/cosmos/cosmos-sdk/types"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
supplyexported "github.com/cosmos/cosmos-sdk/x/supply/exported"
"github.com/kava-labs/kava/x/issuance/types"
)
// IssueTokens mints new tokens and sends them to the receiver address
func (k Keeper) IssueTokens(ctx sdk.Context, tokens sdk.Coin, owner, receiver sdk.AccAddress) error {
asset, found := k.GetAsset(ctx, tokens.Denom)
if !found {
return sdkerrors.Wrapf(types.ErrAssetNotFound, "denom: %s", tokens.Denom)
}
if !owner.Equals(asset.Owner) {
return sdkerrors.Wrapf(types.ErrNotAuthorized, "owner: %s, address: %s", asset.Owner, owner)
}
if asset.Paused {
return sdkerrors.Wrapf(types.ErrAssetPaused, "denom: %s", tokens.Denom)
}
if asset.Blockable {
blocked, _ := k.checkBlockedAddress(asset, receiver)
if blocked {
return sdkerrors.Wrapf(types.ErrAccountBlocked, "address: %s", receiver)
}
}
acc := k.accountKeeper.GetAccount(ctx, receiver)
_, ok := acc.(supplyexported.ModuleAccountI)
if ok {
return sdkerrors.Wrapf(types.ErrIssueToModuleAccount, "address: %s", receiver)
}
// for rate-limited assets, check that the issuance isn't over the limit
if asset.RateLimit.Active {
err := k.IncrementCurrentAssetSupply(ctx, tokens)
if err != nil {
return err
}
}
// mint new tokens
err := k.supplyKeeper.MintCoins(ctx, types.ModuleAccountName, sdk.NewCoins(tokens))
if err != nil {
return err
}
// send to receiver
err = k.supplyKeeper.SendCoinsFromModuleToAccount(ctx, types.ModuleAccountName, receiver, sdk.NewCoins(tokens))
if err != nil {
return err
}
ctx.EventManager().EmitEvent(
sdk.NewEvent(
types.EventTypeIssue,
sdk.NewAttribute(types.AttributeKeyIssueAmount, tokens.String()),
),
)
return nil
}
// RedeemTokens sends tokens from the owner address to the module account and burns them
func (k Keeper) RedeemTokens(ctx sdk.Context, tokens sdk.Coin, owner sdk.AccAddress) error {
asset, found := k.GetAsset(ctx, tokens.Denom)
if !found {
return sdkerrors.Wrapf(types.ErrAssetNotFound, "denom: %s", tokens.Denom)
}
if !owner.Equals(asset.Owner) {
return sdkerrors.Wrapf(types.ErrNotAuthorized, "owner: %s, address: %s", asset.Owner, owner)
}
if asset.Paused {
return sdkerrors.Wrapf(types.ErrAssetPaused, "denom: %s", tokens.Denom)
}
coins := sdk.NewCoins(tokens)
err := k.supplyKeeper.SendCoinsFromAccountToModule(ctx, owner, types.ModuleAccountName, coins)
if err != nil {
return err
}
err = k.supplyKeeper.BurnCoins(ctx, types.ModuleAccountName, coins)
if err != nil {
return err
}
ctx.EventManager().EmitEvent(
sdk.NewEvent(
types.EventTypeRedeem,
sdk.NewAttribute(types.AttributeKeyRedeemAmount, tokens.String()),
),
)
return nil
}
// BlockAddress adds an address to the blocked list
func (k Keeper) BlockAddress(ctx sdk.Context, denom string, owner, blockedAddress sdk.AccAddress) error {
asset, found := k.GetAsset(ctx, denom)
if !found {
return sdkerrors.Wrapf(types.ErrAssetNotFound, "denom: %s", denom)
}
if !asset.Blockable {
return sdkerrors.Wrap(types.ErrAssetUnblockable, denom)
}
if !owner.Equals(asset.Owner) {
return sdkerrors.Wrapf(types.ErrNotAuthorized, "owner: %s, address: %s", asset.Owner, owner)
}
blocked, _ := k.checkBlockedAddress(asset, blockedAddress)
if blocked {
return sdkerrors.Wrapf(types.ErrAccountAlreadyBlocked, "address: %s", blockedAddress)
}
account := k.accountKeeper.GetAccount(ctx, blockedAddress)
if account == nil {
return sdkerrors.Wrapf(types.ErrAccountNotFound, "address: %s", blockedAddress)
}
asset.BlockedAddresses = append(asset.BlockedAddresses, blockedAddress)
k.SetAsset(ctx, asset)
ctx.EventManager().EmitEvent(
sdk.NewEvent(
types.EventTypeBlock,
sdk.NewAttribute(types.AttributeKeyBlock, blockedAddress.String()),
sdk.NewAttribute(types.AttributeKeyDenom, asset.Denom),
),
)
return nil
}
// UnblockAddress removes an address from the blocked list
func (k Keeper) UnblockAddress(ctx sdk.Context, denom string, owner, addr sdk.AccAddress) error {
asset, found := k.GetAsset(ctx, denom)
if !found {
return sdkerrors.Wrapf(types.ErrAssetNotFound, "denom: %s", denom)
}
if !asset.Blockable {
return sdkerrors.Wrap(types.ErrAssetUnblockable, denom)
}
if !owner.Equals(asset.Owner) {
return sdkerrors.Wrapf(types.ErrNotAuthorized, "owner: %s, address: %s", asset.Owner, owner)
}
blocked, i := k.checkBlockedAddress(asset, addr)
if !blocked {
return sdkerrors.Wrapf(types.ErrAccountAlreadyUnblocked, "address: %s", addr)
}
blockedAddrs := k.removeBlockedAddress(asset.BlockedAddresses, i)
asset.BlockedAddresses = blockedAddrs
k.SetAsset(ctx, asset)
ctx.EventManager().EmitEvent(
sdk.NewEvent(
types.EventTypeUnblock,
sdk.NewAttribute(types.AttributeKeyUnblock, addr.String()),
sdk.NewAttribute(types.AttributeKeyDenom, asset.Denom),
),
)
return nil
}
// SetPauseStatus pauses/un-pauses an asset
func (k Keeper) SetPauseStatus(ctx sdk.Context, owner sdk.AccAddress, denom string, status bool) error {
asset, found := k.GetAsset(ctx, denom)
if !found {
return sdkerrors.Wrapf(types.ErrAssetNotFound, "denom: %s", denom)
}
if !owner.Equals(asset.Owner) {
return sdkerrors.Wrapf(types.ErrNotAuthorized, "owner: %s, address: %s", asset.Owner, owner)
}
if asset.Paused == status {
return nil
}
asset.Paused = !asset.Paused
k.SetAsset(ctx, asset)
ctx.EventManager().EmitEvent(
sdk.NewEvent(
types.EventTypePause,
sdk.NewAttribute(types.AttributeKeyPauseStatus, fmt.Sprintf("%t", status)),
sdk.NewAttribute(types.AttributeKeyDenom, asset.Denom),
),
)
return nil
}
// SeizeCoinsForBlockableAssets seizes coins from blocked addresses for assets that have blocking enabled
func (k Keeper) SeizeCoinsForBlockableAssets(ctx sdk.Context) error {
params := k.GetParams(ctx)
for _, asset := range params.Assets {
if asset.Blockable {
err := k.SeizeCoinsFromBlockedAddresses(ctx, asset.Denom)
if err != nil {
return err
}
}
}
return nil
}
// SeizeCoinsFromBlockedAddresses checks blocked addresses for coins of the input denom and transfers them to the owner account
func (k Keeper) SeizeCoinsFromBlockedAddresses(ctx sdk.Context, denom string) error {
asset, found := k.GetAsset(ctx, denom)
if !found {
return sdkerrors.Wrapf(types.ErrAssetNotFound, "denom: %s", denom)
}
for _, address := range asset.BlockedAddresses {
account := k.accountKeeper.GetAccount(ctx, address)
if account == nil {
// avoids a potential panic
// this could happen if, for example, an account was pruned from state but remained in the block list,
continue
}
coinsAmount := account.GetCoins().AmountOf(denom)
if !coinsAmount.IsPositive() {
continue
}
coins := sdk.NewCoins(sdk.NewCoin(denom, coinsAmount))
err := k.supplyKeeper.SendCoinsFromAccountToModule(ctx, address, types.ModuleAccountName, coins)
if err != nil {
return err
}
err = k.supplyKeeper.SendCoinsFromModuleToAccount(ctx, types.ModuleAccountName, asset.Owner, coins)
if err != nil {
return err
}
ctx.EventManager().EmitEvent(
sdk.NewEvent(
types.EventTypeSeize,
sdk.NewAttribute(sdk.AttributeKeyAmount, coins.String()),
sdk.NewAttribute(types.AttributeKeyAddress, address.String()),
),
)
}
return nil
}
func (k Keeper) checkBlockedAddress(asset types.Asset, checkAddress sdk.AccAddress) (bool, int) {
for i, address := range asset.BlockedAddresses {
if address.Equals(checkAddress) {
return true, i
}
}
return false, 0
}
func (k Keeper) removeBlockedAddress(blockedAddrs []sdk.AccAddress, i int) []sdk.AccAddress {
blockedAddrs[len(blockedAddrs)-1], blockedAddrs[i] = blockedAddrs[i], blockedAddrs[len(blockedAddrs)-1]
return blockedAddrs[:len(blockedAddrs)-1]
}