0g-chain/x/issuance/keeper/issuance.go

254 lines
7.9 KiB
Go
Raw Normal View History

package keeper
import (
"fmt"
"strings"
errorsmod "cosmossdk.io/errors"
2024-05-01 03:17:24 +00:00
"github.com/0glabs/0g-chain/x/issuance/types"
sdk "github.com/cosmos/cosmos-sdk/types"
authtypes "github.com/cosmos/cosmos-sdk/x/auth/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 errorsmod.Wrapf(types.ErrAssetNotFound, "denom: %s", tokens.Denom)
}
if strings.Compare(owner.String(), asset.Owner) != 0 {
return errorsmod.Wrapf(types.ErrNotAuthorized, "owner: %s, address: %s", asset.Owner, owner)
}
if asset.Paused {
return errorsmod.Wrapf(types.ErrAssetPaused, "denom: %s", tokens.Denom)
}
if asset.Blockable {
blocked, _ := k.checkBlockedAddress(asset, receiver.String())
if blocked {
return errorsmod.Wrapf(types.ErrAccountBlocked, "address: %s", receiver)
}
}
acc := k.accountKeeper.GetAccount(ctx, receiver)
_, ok := acc.(authtypes.ModuleAccountI)
if ok {
return errorsmod.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.bankKeeper.MintCoins(ctx, types.ModuleAccountName, sdk.NewCoins(tokens))
if err != nil {
return err
}
// send to receiver
err = k.bankKeeper.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 errorsmod.Wrapf(types.ErrAssetNotFound, "denom: %s", tokens.Denom)
}
if strings.Compare(owner.String(), asset.Owner) != 0 {
return errorsmod.Wrapf(types.ErrNotAuthorized, "owner: %s, address: %s", asset.Owner, owner)
}
if asset.Paused {
return errorsmod.Wrapf(types.ErrAssetPaused, "denom: %s", tokens.Denom)
}
coins := sdk.NewCoins(tokens)
err := k.bankKeeper.SendCoinsFromAccountToModule(ctx, owner, types.ModuleAccountName, coins)
if err != nil {
return err
}
err = k.bankKeeper.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 errorsmod.Wrapf(types.ErrAssetNotFound, "denom: %s", denom)
}
if !asset.Blockable {
return errorsmod.Wrap(types.ErrAssetUnblockable, denom)
}
if strings.Compare(owner.String(), asset.Owner) != 0 {
return errorsmod.Wrapf(types.ErrNotAuthorized, "owner: %s, address: %s", asset.Owner, owner)
}
blocked, _ := k.checkBlockedAddress(asset, blockedAddress.String())
if blocked {
return errorsmod.Wrapf(types.ErrAccountAlreadyBlocked, "address: %s", blockedAddress)
}
account := k.accountKeeper.GetAccount(ctx, blockedAddress)
if account == nil {
return errorsmod.Wrapf(types.ErrAccountNotFound, "address: %s", blockedAddress)
}
asset.BlockedAddresses = append(asset.BlockedAddresses, blockedAddress.String())
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 errorsmod.Wrapf(types.ErrAssetNotFound, "denom: %s", denom)
}
if !asset.Blockable {
return errorsmod.Wrap(types.ErrAssetUnblockable, denom)
}
if strings.Compare(owner.String(), asset.Owner) != 0 {
return errorsmod.Wrapf(types.ErrNotAuthorized, "owner: %s, address: %s", asset.Owner, owner)
}
blocked, i := k.checkBlockedAddress(asset, addr.String())
if !blocked {
return errorsmod.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 errorsmod.Wrapf(types.ErrAssetNotFound, "denom: %s", denom)
}
if strings.Compare(owner.String(), asset.Owner) != 0 {
return errorsmod.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 errorsmod.Wrapf(types.ErrAssetNotFound, "denom: %s", denom)
}
for _, address := range asset.BlockedAddresses {
addrBech32, err := sdk.AccAddressFromBech32(address)
if err != nil {
return err
}
account := k.accountKeeper.GetAccount(ctx, addrBech32)
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 := k.bankKeeper.GetAllBalances(ctx, addrBech32).AmountOf(denom)
if !coinsAmount.IsPositive() {
continue
}
coins := sdk.NewCoins(sdk.NewCoin(denom, coinsAmount))
err = k.bankKeeper.SendCoinsFromAccountToModule(ctx, addrBech32, types.ModuleAccountName, coins)
if err != nil {
return err
}
ownerBech32, err := sdk.AccAddressFromBech32(asset.Owner)
if err != nil {
return err
}
err = k.bankKeeper.SendCoinsFromModuleToAccount(ctx, types.ModuleAccountName, ownerBech32, coins)
if err != nil {
return err
}
ctx.EventManager().EmitEvent(
sdk.NewEvent(
types.EventTypeSeize,
sdk.NewAttribute(sdk.AttributeKeyAmount, coins.String()),
sdk.NewAttribute(types.AttributeKeyAddress, address),
),
)
}
return nil
}
func (k Keeper) checkBlockedAddress(asset types.Asset, checkAddress string) (bool, int) {
for i, address := range asset.BlockedAddresses {
if strings.Compare(address, checkAddress) == 0 {
return true, i
}
}
return false, 0
}
func (k Keeper) removeBlockedAddress(blockedAddrs []string, i int) []string {
blockedAddrs[len(blockedAddrs)-1], blockedAddrs[i] = blockedAddrs[i], blockedAddrs[len(blockedAddrs)-1]
return blockedAddrs[:len(blockedAddrs)-1]
}