0g-chain/x/evmutil/keeper/msg_server.go

190 lines
4.8 KiB
Go
Raw Normal View History

package keeper
import (
"context"
"fmt"
sdk "github.com/cosmos/cosmos-sdk/types"
2024-05-01 03:17:24 +00:00
"github.com/0glabs/0g-chain/x/evmutil/types"
)
type msgServer struct {
keeper Keeper
}
// NewMsgServerImpl returns an implementation of the evmutil MsgServer interface
// for the provided Keeper.
func NewMsgServerImpl(keeper Keeper) types.MsgServer {
return &msgServer{keeper: keeper}
}
var _ types.MsgServer = msgServer{}
////////////////////////////
// EVM-native assets -> Cosmos SDK
////////////////////////////
// ConvertCoinToERC20 handles a MsgConvertCoinToERC20 message to convert
2024-05-01 05:53:58 +00:00
// sdk.Coin to 0gChain EVM tokens.
func (s msgServer) ConvertCoinToERC20(
goCtx context.Context,
msg *types.MsgConvertCoinToERC20,
) (*types.MsgConvertCoinToERC20Response, error) {
ctx := sdk.UnwrapSDKContext(goCtx)
initiator, err := sdk.AccAddressFromBech32(msg.Initiator)
if err != nil {
return nil, fmt.Errorf("invalid Initiator address: %w", err)
}
receiver, err := types.NewInternalEVMAddressFromString(msg.Receiver)
if err != nil {
return nil, fmt.Errorf("invalid Receiver address: %w", err)
}
if err := s.keeper.ConvertCoinToERC20(
ctx,
initiator,
receiver,
*msg.Amount,
); err != nil {
return nil, err
}
ctx.EventManager().EmitEvent(
sdk.NewEvent(
sdk.EventTypeMessage,
sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
sdk.NewAttribute(sdk.AttributeKeySender, msg.Initiator),
),
)
return &types.MsgConvertCoinToERC20Response{}, nil
}
// ConvertERC20ToCoin handles a MsgConvertERC20ToCoin message to convert
2024-05-01 05:53:58 +00:00
// sdk.Coin to 0gChain EVM tokens.
func (s msgServer) ConvertERC20ToCoin(
goCtx context.Context,
msg *types.MsgConvertERC20ToCoin,
) (*types.MsgConvertERC20ToCoinResponse, error) {
ctx := sdk.UnwrapSDKContext(goCtx)
initiator, err := types.NewInternalEVMAddressFromString(msg.Initiator)
if err != nil {
return nil, fmt.Errorf("invalid initiator address: %w", err)
}
receiver, err := sdk.AccAddressFromBech32(msg.Receiver)
if err != nil {
return nil, fmt.Errorf("invalid receiver address: %w", err)
}
2024-05-01 03:56:00 +00:00
contractAddr, err := types.NewInternalEVMAddressFromString(msg.ZgChainERC20Address)
if err != nil {
return nil, fmt.Errorf("invalid contract address: %w", err)
}
if err := s.keeper.ConvertERC20ToCoin(
ctx,
initiator,
receiver,
contractAddr,
msg.Amount,
); err != nil {
return nil, err
}
ctx.EventManager().EmitEvent(
sdk.NewEvent(
sdk.EventTypeMessage,
sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
sdk.NewAttribute(sdk.AttributeKeySender, msg.Initiator),
),
)
return &types.MsgConvertERC20ToCoinResponse{}, nil
}
////////////////////////////
// Cosmos SDK-native assets -> EVM
////////////////////////////
// ConvertCosmosCoinToERC20 converts a native sdk.Coin to an ERC20.
// If no ERC20 contract has been deployed for the given denom, a new
// contract will be deployed and registered to the module.
func (s msgServer) ConvertCosmosCoinToERC20(
goCtx context.Context,
msg *types.MsgConvertCosmosCoinToERC20,
) (*types.MsgConvertCosmosCoinToERC20Response, error) {
ctx := sdk.UnwrapSDKContext(goCtx)
initiator, err := sdk.AccAddressFromBech32(msg.Initiator)
if err != nil {
return nil, fmt.Errorf("invalid initiator address: %w", err)
}
receiver, err := types.NewInternalEVMAddressFromString(msg.Receiver)
if err != nil {
return nil, fmt.Errorf("invalid receiver address: %w", err)
}
if err := s.keeper.ConvertCosmosCoinToERC20(
ctx,
initiator,
receiver,
*msg.Amount,
); err != nil {
return nil, err
}
ctx.EventManager().EmitEvent(
sdk.NewEvent(
sdk.EventTypeMessage,
sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
sdk.NewAttribute(sdk.AttributeKeySender, msg.Initiator),
),
)
return &types.MsgConvertCosmosCoinToERC20Response{}, nil
}
// ConvertCosmosCoinFromERC20 converts an ERC20 representation of a cosmos-native asset
// back into an sdk.Coin.
func (s msgServer) ConvertCosmosCoinFromERC20(
goCtx context.Context,
msg *types.MsgConvertCosmosCoinFromERC20,
) (*types.MsgConvertCosmosCoinFromERC20Response, error) {
ctx := sdk.UnwrapSDKContext(goCtx)
initiator, err := types.NewInternalEVMAddressFromString(msg.Initiator)
if err != nil {
return nil, fmt.Errorf("invalid initiator address: %w", err)
}
receiver, err := sdk.AccAddressFromBech32(msg.Receiver)
if err != nil {
return nil, fmt.Errorf("invalid receiver address: %w", err)
}
if err := s.keeper.ConvertCosmosCoinFromERC20(
ctx,
initiator,
receiver,
*msg.Amount,
); err != nil {
return nil, err
}
ctx.EventManager().EmitEvent(
sdk.NewEvent(
sdk.EventTypeMessage,
sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
sdk.NewAttribute(sdk.AttributeKeySender, msg.Initiator),
),
)
return &types.MsgConvertCosmosCoinFromERC20Response{}, nil
}