85 lines
2.5 KiB
Go
85 lines
2.5 KiB
Go
package keeper
|
|
|
|
import (
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
"github.com/cosmos/cosmos-sdk/x/nft"
|
|
)
|
|
|
|
// BatchMint defines a method for minting a batch of nfts
|
|
func (k Keeper) BatchMint(ctx sdk.Context,
|
|
tokens []nft.NFT,
|
|
receiver sdk.AccAddress,
|
|
) error {
|
|
checked := make(map[string]bool, len(tokens))
|
|
for _, token := range tokens {
|
|
if !checked[token.ClassId] && !k.HasClass(ctx, token.ClassId) {
|
|
return sdkerrors.Wrap(nft.ErrClassNotExists, token.ClassId)
|
|
}
|
|
|
|
if k.HasNFT(ctx, token.ClassId, token.Id) {
|
|
return sdkerrors.Wrap(nft.ErrNFTExists, token.Id)
|
|
}
|
|
|
|
checked[token.ClassId] = true
|
|
k.mintWithNoCheck(ctx, token, receiver)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// BatchBurn defines a method for burning a batch of nfts from a specific classID.
|
|
// Note: When the upper module uses this method, it needs to authenticate nft
|
|
func (k Keeper) BatchBurn(ctx sdk.Context, classID string, nftIDs []string) error {
|
|
if !k.HasClass(ctx, classID) {
|
|
return sdkerrors.Wrap(nft.ErrClassNotExists, classID)
|
|
}
|
|
for _, nftID := range nftIDs {
|
|
if !k.HasNFT(ctx, classID, nftID) {
|
|
return sdkerrors.Wrap(nft.ErrNFTNotExists, nftID)
|
|
}
|
|
if err := k.burnWithNoCheck(ctx, classID, nftID); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// BatchUpdate defines a method for updating a batch of exist nfts
|
|
// Note: When the upper module uses this method, it needs to authenticate nft
|
|
func (k Keeper) BatchUpdate(ctx sdk.Context, tokens []nft.NFT) error {
|
|
checked := make(map[string]bool, len(tokens))
|
|
for _, token := range tokens {
|
|
if !checked[token.ClassId] && !k.HasClass(ctx, token.ClassId) {
|
|
return sdkerrors.Wrap(nft.ErrClassNotExists, token.ClassId)
|
|
}
|
|
|
|
if !k.HasNFT(ctx, token.ClassId, token.Id) {
|
|
return sdkerrors.Wrap(nft.ErrNFTNotExists, token.Id)
|
|
}
|
|
checked[token.ClassId] = true
|
|
k.updateWithNoCheck(ctx, token)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// BatchTransfer defines a method for sending a batch of nfts from one account to another account from a specific classID.
|
|
// Note: When the upper module uses this method, it needs to authenticate nft
|
|
func (k Keeper) BatchTransfer(ctx sdk.Context,
|
|
classID string,
|
|
nftIDs []string,
|
|
receiver sdk.AccAddress,
|
|
) error {
|
|
if !k.HasClass(ctx, classID) {
|
|
return sdkerrors.Wrap(nft.ErrClassNotExists, classID)
|
|
}
|
|
for _, nftID := range nftIDs {
|
|
if !k.HasNFT(ctx, classID, nftID) {
|
|
return sdkerrors.Wrap(nft.ErrNFTNotExists, nftID)
|
|
}
|
|
if err := k.transferWithNoCheck(ctx, classID, nftID, receiver); err != nil {
|
|
return sdkerrors.Wrap(nft.ErrNFTNotExists, nftID)
|
|
}
|
|
}
|
|
return nil
|
|
}
|