status-go/services/wallet/bridge/transfer.go

package bridge

import (
	"context"
	"math/big"
	"strings"

	"github.com/ethereum/go-ethereum"
	"github.com/ethereum/go-ethereum/accounts/abi"
	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/common/hexutil"
	ethTypes "github.com/ethereum/go-ethereum/core/types"
	"github.com/status-im/status-go/account"
	"github.com/status-im/status-go/contracts/ierc20"
	"github.com/status-im/status-go/eth-node/types"
	"github.com/status-im/status-go/params"
	"github.com/status-im/status-go/rpc"
	"github.com/status-im/status-go/services/wallet/token"
	"github.com/status-im/status-go/transactions"
)

type TransferBridge struct {
	rpcClient  *rpc.Client
	transactor transactions.TransactorIface
}

func NewTransferBridge(rpcClient *rpc.Client, transactor transactions.TransactorIface) *TransferBridge {
	return &TransferBridge{rpcClient: rpcClient, transactor: transactor}
}

func (s *TransferBridge) Name() string {
	return "Transfer"
}

func (s *TransferBridge) AvailableFor(from, to *params.Network, token *token.Token, toToken *token.Token) (bool, error) {
	return from.ChainID == to.ChainID && token != nil && toToken == nil, nil
}

func (s *TransferBridge) CalculateFees(from, to *params.Network, token *token.Token, amountIn *big.Int) (*big.Int, *big.Int, error) {
	return big.NewInt(0), big.NewInt(0), nil
}

func (s *TransferBridge) PackTxInputData(contractType string, fromNetwork *params.Network, toNetwork *params.Network, from common.Address, to common.Address, token *token.Token, amountIn *big.Int) ([]byte, error) {
	if token.IsNative() {
		return []byte("eth_sendRawTransaction"), nil
	} else {
		abi, err := abi.JSON(strings.NewReader(ierc20.IERC20ABI))
		if err != nil {
			return []byte{}, err
		}
		return abi.Pack("transfer",
			to,
			amountIn,
		)
	}
}

func (s *TransferBridge) EstimateGas(fromNetwork *params.Network, toNetwork *params.Network, from common.Address, to common.Address, token *token.Token, toToken *token.Token, amountIn *big.Int) (uint64, error) {
	estimation := uint64(0)
	var err error

	input, err := s.PackTxInputData("", fromNetwork, toNetwork, from, to, token, amountIn)
	if err != nil {
		return 0, err
	}

	if token.IsNative() {
		estimation, err = s.transactor.EstimateGas(fromNetwork, from, to, amountIn, input)
		if err != nil {
			return 0, err
		}
	} else {
		ethClient, err := s.rpcClient.EthClient(fromNetwork.ChainID)
		if err != nil {
			return 0, err
		}

		ctx := context.Background()

		msg := ethereum.CallMsg{
			From: from,
			To:   &token.Address,
			Data: input,
		}

		estimation, err = ethClient.EstimateGas(ctx, msg)
		if err != nil {
			return 0, err
		}

	}

	increasedEstimation := float64(estimation) * IncreaseEstimatedGasFactor
	return uint64(increasedEstimation), nil
}

func (s *TransferBridge) BuildTx(network, _ *params.Network, fromAddress common.Address, toAddress common.Address, token *token.Token, amountIn *big.Int, bonderFee *big.Int) (*ethTypes.Transaction, error) {
	toAddr := types.Address(toAddress)
	if token.IsNative() {
		sendArgs := &TransactionBridge{
			TransferTx: &transactions.SendTxArgs{
				From:  types.Address(fromAddress),
				To:    &toAddr,
				Value: (*hexutil.Big)(amountIn),
				Data:  types.HexBytes("0x0"),
			},
			ChainID: network.ChainID,
		}

		return s.BuildTransaction(sendArgs)
	}
	abi, err := abi.JSON(strings.NewReader(ierc20.IERC20ABI))
	if err != nil {
		return nil, err
	}
	input, err := abi.Pack("transfer",
		toAddress,
		amountIn,
	)
	if err != nil {
		return nil, err
	}
	sendArgs := &TransactionBridge{
		TransferTx: &transactions.SendTxArgs{
			From:  types.Address(fromAddress),
			To:    &toAddr,
			Value: (*hexutil.Big)(big.NewInt(0)),
			Data:  input,
		},
		ChainID: network.ChainID,
	}

	return s.BuildTransaction(sendArgs)
}

func (s *TransferBridge) Send(sendArgs *TransactionBridge, verifiedAccount *account.SelectedExtKey) (types.Hash, error) {
	return s.transactor.SendTransactionWithChainID(sendArgs.ChainID, *sendArgs.TransferTx, verifiedAccount)
}

func (s *TransferBridge) BuildTransaction(sendArgs *TransactionBridge) (*ethTypes.Transaction, error) {
	return s.transactor.ValidateAndBuildTransaction(sendArgs.ChainID, *sendArgs.TransferTx)
}

func (s *TransferBridge) CalculateAmountOut(from, to *params.Network, amountIn *big.Int, symbol string) (*big.Int, error) {
	return amountIn, nil
}

func (s *TransferBridge) GetContractAddress(network *params.Network, token *token.Token) (common.Address, error) {
	return common.Address{}, nil
}