ibc.go

package keeper

import (
	"errors"
	"fmt"
	"math/big"
	"strings"

	"github.com/ethereum/go-ethereum/common"

	ethermint "github.com/FiboChain/fbc/app/types"
	sdk "github.com/FiboChain/fbc/libs/cosmos-sdk/types"
	ibctransfertypes "github.com/FiboChain/fbc/libs/ibc-go/modules/apps/transfer/types"
	ibcclienttypes "github.com/FiboChain/fbc/libs/ibc-go/modules/core/02-client/types"
	tmtypes "github.com/FiboChain/fbc/libs/tendermint/types"
	"github.com/FiboChain/fbc/x/erc20/types"
	evmtypes "github.com/FiboChain/fbc/x/evm/types"
)

// OnMintVouchers after minting vouchers on this chain
func (k Keeper) OnMintVouchers(ctx sdk.Context, vouchers sdk.SysCoins, receiver string) error {
	if err := k.ConvertVouchers(ctx, receiver, vouchers); err != nil {
		k.Logger(ctx).Error(
			fmt.Sprintf("Failed to convert vouchers to evm tokens for receiver %s, coins %s. Receive error %s",
				receiver, vouchers.String(), err))
		return err
	}
	return nil
}

// OnUnescrowNatives after unescrow natives on this chain
func (k Keeper) OnUnescrowNatives(ctx sdk.Context, natives sdk.SysCoins, receiver string) error {
	if err := k.ConvertNatives(ctx, receiver, natives); err != nil {
		k.Logger(ctx).Error(
			fmt.Sprintf("Failed to convert natives to evm tokens for receiver %s, coins %s. Receive error %s",
				receiver, natives.String(), err))
		return err
	}
	return nil
}

// ConvertVouchers convert vouchers into evm tokens.
func (k Keeper) ConvertVouchers(ctx sdk.Context, from string, vouchers sdk.SysCoins) error {
	if len(strings.TrimSpace(from)) == 0 {
		return errors.New("empty from address string is not allowed")
	}
	fromAddr, err := sdk.AccAddressFromBech32(from)
	if err != nil {
		return err
	}

	params := k.GetParams(ctx)
	for _, c := range vouchers {
		// okc1:xxb----->okc2:ibc/xxb---->okc2:erc20/xxb
		if err := k.ConvertVoucherToERC20(ctx, fromAddr, c, params.EnableAutoDeployment); err != nil {
			return err
		}
	}

	return nil
}

// ConvertNatives convert natives into evm tokens.
func (k Keeper) ConvertNatives(ctx sdk.Context, from string, vouchers sdk.SysCoins) error {
	if len(strings.TrimSpace(from)) == 0 {
		return errors.New("empty from address string is not allowed")
	}
	fromAddr, err := sdk.AccAddressFromBech32(from)
	if err != nil {
		return err
	}

	for _, c := range vouchers {
		// if there is a contract associated with this native coin,
		// the native coin come from native erc20
		// okc1:erc20/xxb----->okc2:ibc/xxb---->okc1:ibc/yyb---->okc2:erc20/xxb
		if contract, found := k.GetContractByDenom(ctx, c.Denom); found {
			if err := k.ConvertNativeToERC20(ctx, fromAddr, c, contract); err != nil {
				return err
			}
		}
	}

	return nil
}

// ConvertVoucherToERC20 convert voucher into evm token.
func (k Keeper) ConvertVoucherToERC20(ctx sdk.Context, from sdk.AccAddress, voucher sdk.SysCoin, autoDeploy bool) error {
	k.Logger(ctx).Info("convert vouchers into evm tokens",
		"fromBech32", from.String(),
		"fromEth", common.BytesToAddress(from.Bytes()).String(),
		"voucher", voucher.String())

	if !types.IsValidIBCDenom(voucher.Denom) {
		return fmt.Errorf("coin %s is not supported for wrapping", voucher.Denom)
	}

	var err error
	contract, found := k.GetContractByDenom(ctx, voucher.Denom)
	if !found {
		// automated deployment contracts
		if !autoDeploy {
			k.Logger(ctx).Error("no contract found and not auto deploy for the denom", "denom", voucher.Denom)
			return types.ErrNoContractNotAuto
		}
		contract, err = k.DeployModuleERC20(ctx, voucher.Denom)
		if err != nil {
			return err
		}
		k.SetContractForDenom(ctx, voucher.Denom, contract)
		k.Logger(ctx).Info("contract created for coin", "contract", contract.String(), "denom", voucher.Denom)

		ctx.EventManager().EmitEvents(sdk.Events{
			sdk.NewEvent(
				types.EventTypDeployModuleERC20,
				sdk.NewAttribute(types.AttributeKeyContractAddr, contract.String()),
				sdk.NewAttribute(ibctransfertypes.AttributeKeyDenom, voucher.Denom),
			),
		})
	}

	// 1. transfer voucher from user address to contact address in bank
	if err := k.bankKeeper.SendCoins(ctx, from, sdk.AccAddress(contract.Bytes()), sdk.NewCoins(voucher)); err != nil {
		return err
	}
	// 2. call contract, mint token to user address in contract
	if _, err := k.CallModuleERC20(
		ctx,
		contract,
		types.ContractMintMethod,
		common.BytesToAddress(from.Bytes()),
		voucher.Amount.BigInt()); err != nil {
		return err
	}

	ctx.EventManager().EmitEvents(sdk.Events{
		sdk.NewEvent(
			types.EventTypLock,
			sdk.NewAttribute(types.AttributeKeyFrom, from.String()),
			sdk.NewAttribute(types.AttributeKeyTo, contract.String()),
			sdk.NewAttribute(sdk.AttributeKeyAmount, voucher.String()),
		),
		sdk.NewEvent(
			types.EventTypCallModuleERC20,
			sdk.NewAttribute(types.AttributeKeyContractAddr, contract.String()),
			sdk.NewAttribute(types.AttributeKeyContractMethod, types.ContractMintMethod),
			sdk.NewAttribute(ibctransfertypes.AttributeKeyReceiver, from.String()),
			sdk.NewAttribute(ibctransfertypes.AttributeKeyAmount, voucher.Amount.BigInt().String()),
		),
	})
	return nil
}

// ConvertNativeToERC20 convert native into evm token.
func (k Keeper) ConvertNativeToERC20(ctx sdk.Context, from sdk.AccAddress, native sdk.SysCoin, contract common.Address) error {
	k.Logger(ctx).Info("convert native into evm tokens",
		"fromBech32", from.String(),
		"fromEth", common.BytesToAddress(from.Bytes()).String(),
		"native", native.String(),
		"contract", contract.String())

	// 1. transfer native from user address to module address and burn them
	if err := k.supplyKeeper.SendCoinsFromAccountToModule(ctx, from, types.ModuleName, sdk.NewCoins(native)); err != nil {
		return err
	}
	if err := k.supplyKeeper.BurnCoins(ctx, types.ModuleName, sdk.NewCoins(native)); err != nil {
		return err
	}

	// 2. call contract, mint token to user address in contract
	if _, err := k.CallModuleERC20(
		ctx,
		contract,
		types.ContractMintMethod,
		common.BytesToAddress(from.Bytes()),
		native.Amount.BigInt()); err != nil {
		return err
	}

	ctx.EventManager().EmitEvents(sdk.Events{
		sdk.NewEvent(
			types.EventTypBurn,
			sdk.NewAttribute(types.AttributeKeyFrom, from.String()),
			sdk.NewAttribute(sdk.AttributeKeyAmount, native.String()),
		),
		sdk.NewEvent(
			types.EventTypCallModuleERC20,
			sdk.NewAttribute(types.AttributeKeyContractAddr, contract.String()),
			sdk.NewAttribute(types.AttributeKeyContractMethod, types.ContractMintMethod),
			sdk.NewAttribute(ibctransfertypes.AttributeKeyReceiver, from.String()),
			sdk.NewAttribute(ibctransfertypes.AttributeKeyAmount, native.Amount.BigInt().String()),
		),
	})
	return nil
}

// DeployModuleERC20 deploy an embed erc20 contract
func (k Keeper) DeployModuleERC20(ctx sdk.Context, denom string) (common.Address, error) {
	implContract, found := k.GetImplementTemplateContract(ctx)
	if !found {
		return common.Address{}, errors.New("not found implement contract")
	}
	proxyContract, found := k.GetProxyTemplateContract(ctx)
	if !found {
		return common.Address{}, errors.New("not found proxy contract")
	}

	// 1. deploy implement contract
	byteCode := common.Hex2Bytes(implContract.Bin)
	_, implRes, err := k.callEvmByModule(ctx, nil, big.NewInt(0), byteCode)
	if err != nil {
		return common.Address{}, err
	}

	// 2. deploy proxy contract
	byteCode = common.Hex2Bytes(proxyContract.Bin)
	implInput, err := implContract.ABI.Pack("initialize", denom, uint8(0))
	if err != nil {
		return common.Address{}, err
	}
	input, err := proxyContract.ABI.Pack("", implRes.ContractAddress, implInput)
	if err != nil {
		return common.Address{}, err
	}
	data := append(byteCode, input...)
	_, res, err := k.callEvmByModule(ctx, nil, big.NewInt(0), data)
	if err != nil {
		return common.Address{}, err
	}
	return res.ContractAddress, nil
}

// CallModuleERC20 call a method of ModuleERC20 contract
func (k Keeper) CallModuleERC20(ctx sdk.Context, contract common.Address, method string, args ...interface{}) ([]byte, error) {
	k.Logger(ctx).Info("call erc20 module contract", "contract", contract.String(), "method", method, "args", args)

	implContract, found := k.GetImplementTemplateContract(ctx)
	if !found {
		return nil, errors.New("not found implement contract")
	}
	data, err := implContract.ABI.Pack(method, args...)
	if err != nil {
		return nil, err
	}

	_, res, err := k.callEvmByModule(ctx, &contract, big.NewInt(0), data)
	if err != nil {
		return nil, fmt.Errorf("call contract failed: %s, %s, %s", contract.Hex(), method, err)
	}
	return res.Ret, nil
}

func (k Keeper) CallModuleERC20Proxy(ctx sdk.Context, contract common.Address, method string, args ...interface{}) ([]byte, error) {
	k.Logger(ctx).Info("call proxy erc20 module contract", "contract", contract.String(), "method", method, "args", args)

	proxyContract, found := k.GetProxyTemplateContract(ctx)
	if !found {
		return nil, errors.New("not found proxy contract")
	}
	data, err := proxyContract.ABI.Pack(method, args...)
	if err != nil {
		return nil, err
	}

	_, res, err := k.callEvmByModule(ctx, &contract, big.NewInt(0), data)
	if err != nil {
		return nil, fmt.Errorf("call contract failed: %s, %s, %s", contract.Hex(), method, err)
	}
	return res.Ret, nil
}

// callEvmByModule execute an evm message from native module
func (k Keeper) callEvmByModule(ctx sdk.Context, to *common.Address, value *big.Int, data []byte) (*evmtypes.ExecutionResult, *evmtypes.ResultData, error) {
	config, found := k.evmKeeper.GetChainConfig(ctx)
	if !found {
		return nil, nil, types.ErrChainConfigNotFound
	}

	chainIDEpoch, err := ethermint.ParseChainID(ctx.ChainID())
	if err != nil {
		return nil, nil, err
	}

	acc := k.accountKeeper.GetAccount(ctx, types.IbcEvmModuleBechAddr)
	if acc == nil {
		acc = k.accountKeeper.NewAccountWithAddress(ctx, types.IbcEvmModuleBechAddr)
	}
	nonce := acc.GetSequence()
	txHash := tmtypes.Tx(ctx.TxBytes()).Hash(ctx.BlockHeight())
	ethTxHash := common.BytesToHash(txHash)

	gasLimit := ctx.GasMeter().Limit()
	if gasLimit == sdk.NewInfiniteGasMeter().Limit() {
		gasLimit = k.evmKeeper.GetParams(ctx).MaxGasLimitPerTx
	}

	st := evmtypes.StateTransition{
		AccountNonce: nonce,
		Price:        big.NewInt(0),
		GasLimit:     gasLimit,
		Recipient:    to,
		Amount:       value,
		Payload:      data,
		Csdb:         evmtypes.CreateEmptyCommitStateDB(k.evmKeeper.GenerateCSDBParams(), ctx),
		ChainID:      chainIDEpoch,
		TxHash:       &ethTxHash,
		Sender:       types.IbcEvmModuleETHAddr,
		Simulate:     ctx.IsCheckTx(),
		TraceTx:      false,
		TraceTxLog:   false,
	}

	executionResult, resultData, err, innertxs, contracts := st.TransitionDb(ctx, config)
	if !ctx.IsCheckTx() && !ctx.IsTraceTx() {
		k.addEVMInnerTx(ethTxHash.Hex(), innertxs, contracts)
	}
	if err != nil {
		return nil, nil, err
	}

	st.Csdb.Commit(false) // write code to db
	acc.SetSequence(nonce + 1)
	k.accountKeeper.SetAccount(ctx, acc)

	return executionResult, resultData, err
}

// IbcTransferVouchers transfer vouchers to other chain by ibc
func (k Keeper) IbcTransferVouchers(ctx sdk.Context, from, to string, vouchers sdk.SysCoins) error {
	if len(strings.TrimSpace(from)) == 0 {
		return errors.New("empty from address string is not allowed")
	}
	fromAddr, err := sdk.AccAddressFromBech32(from)
	if err != nil {
		return err
	}

	if len(strings.TrimSpace(to)) == 0 {
		return errors.New("to address cannot be empty")
	}
	k.Logger(ctx).Info("transfer vouchers to other chain by ibc", "from", from, "to", to, "vouchers", vouchers)

	for _, c := range vouchers {
		if _, found := k.GetContractByDenom(ctx, c.Denom); !found {
			return fmt.Errorf("coin %s is not supported", c.Denom)
		}
		// okc2:erc20/xxb----->okc2:ibc/xxb---ibc--->okc1:xxb
		if err := k.ibcSendTransfer(ctx, fromAddr, to, c); err != nil {
			return err
		}
	}

	return nil
}

// IbcTransferNative20 transfer native20 to other chain by ibc
func (k Keeper) IbcTransferNative20(ctx sdk.Context, from, to string, native20s sdk.SysCoins, portID, channelID string) error {
	if len(strings.TrimSpace(from)) == 0 {
		return errors.New("empty from address string is not allowed")
	}
	fromAddr, err := sdk.AccAddressFromBech32(from)
	if err != nil {
		return err
	}

	if len(strings.TrimSpace(to)) == 0 {
		return errors.New("to address cannot be empty")
	}
	k.Logger(ctx).Info("transfer native20 to other chain by ibc", "from", from, "to", to, "vouchers", native20s)

	for _, c := range native20s {
		if _, found := k.GetContractByDenom(ctx, c.Denom); !found {
			return fmt.Errorf("coin %s is not supported", c.Denom)
		}
		// okc2:erc20/xxb----->okc2:ibc/xxb---ibc--->okc1:xxb
		if err := k.ibcSendTransferWithChannel(ctx, fromAddr, to, c, portID, channelID); err != nil {
			return err
		}
	}

	return nil
}

func (k Keeper) ibcSendTransfer(ctx sdk.Context, sender sdk.AccAddress, to string, coin sdk.Coin) error {
	// Coin needs to be a voucher so that we can extract the channel id from the denom
	channelID, err := k.GetSourceChannelID(ctx, coin.Denom)
	if err != nil {
		return err
	}

	// Transfer coins to receiver through IBC
	params := k.GetParams(ctx)
	timeoutTimestamp := uint64(ctx.BlockTime().UnixNano()) + params.IbcTimeout
	timeoutHeight := ibcclienttypes.ZeroHeight()

	return k.transferKeeper.SendTransfer(
		ctx,
		ibctransfertypes.PortID,
		channelID,
		sdk.NewCoinAdapter(coin.Denom, sdk.NewIntFromBigInt(coin.Amount.BigInt())),
		sender,
		to,
		timeoutHeight,
		timeoutTimestamp,
	)
}

func (k Keeper) ibcSendTransferWithChannel(ctx sdk.Context, sender sdk.AccAddress, to string, coin sdk.Coin, portID, channelID string) error {
	// Transfer coins to receiver through IBC
	params := k.GetParams(ctx)
	timeoutTimestamp := uint64(ctx.BlockTime().UnixNano()) + params.IbcTimeout
	timeoutHeight := ibcclienttypes.ZeroHeight()

	return k.transferKeeper.SendTransfer(
		ctx,
		portID,
		channelID,
		sdk.NewCoinAdapter(coin.Denom, sdk.NewIntFromBigInt(coin.Amount.BigInt())),
		sender,
		to,
		timeoutHeight,
		timeoutTimestamp,
	)
}