181 lines
5.9 KiB
Go
181 lines
5.9 KiB
Go
package pathprocessor
|
|
|
|
import (
|
|
"context"
|
|
"math/big"
|
|
"strings"
|
|
|
|
"github.com/ethereum/go-ethereum"
|
|
"github.com/ethereum/go-ethereum/accounts/abi"
|
|
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
|
"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/ierc1155"
|
|
"github.com/status-im/status-go/eth-node/types"
|
|
"github.com/status-im/status-go/rpc"
|
|
"github.com/status-im/status-go/services/utils"
|
|
walletCommon "github.com/status-im/status-go/services/wallet/common"
|
|
pathProcessorCommon "github.com/status-im/status-go/services/wallet/router/pathprocessor/common"
|
|
"github.com/status-im/status-go/services/wallet/wallettypes"
|
|
"github.com/status-im/status-go/transactions"
|
|
)
|
|
|
|
type ERC1155TxArgs struct {
|
|
wallettypes.SendTxArgs
|
|
TokenID *hexutil.Big `json:"tokenId"`
|
|
Recipient common.Address `json:"recipient"`
|
|
Amount *hexutil.Big `json:"amount"`
|
|
}
|
|
|
|
type ERC1155Processor struct {
|
|
rpcClient *rpc.Client
|
|
transactor transactions.TransactorIface
|
|
}
|
|
|
|
func NewERC1155Processor(rpcClient *rpc.Client, transactor transactions.TransactorIface) *ERC1155Processor {
|
|
return &ERC1155Processor{rpcClient: rpcClient, transactor: transactor}
|
|
}
|
|
|
|
func createERC1155ErrorResponse(err error) error {
|
|
return createErrorResponse(pathProcessorCommon.ProcessorERC1155Name, err)
|
|
}
|
|
|
|
func (s *ERC1155Processor) Name() string {
|
|
return pathProcessorCommon.ProcessorERC1155Name
|
|
}
|
|
|
|
func (s *ERC1155Processor) AvailableFor(params ProcessorInputParams) (bool, error) {
|
|
return params.FromChain.ChainID == params.ToChain.ChainID && params.ToToken == nil, nil
|
|
}
|
|
|
|
func (s *ERC1155Processor) CalculateFees(params ProcessorInputParams) (*big.Int, *big.Int, error) {
|
|
return walletCommon.ZeroBigIntValue(), walletCommon.ZeroBigIntValue(), nil
|
|
}
|
|
|
|
func (s *ERC1155Processor) PackTxInputData(params ProcessorInputParams) ([]byte, error) {
|
|
abi, err := abi.JSON(strings.NewReader(ierc1155.Ierc1155ABI))
|
|
if err != nil {
|
|
return []byte{}, createERC1155ErrorResponse(err)
|
|
}
|
|
|
|
id, err := walletCommon.GetTokenIdFromSymbol(params.FromToken.Symbol)
|
|
if err != nil {
|
|
return []byte{}, createERC1155ErrorResponse(err)
|
|
}
|
|
|
|
return abi.Pack("safeTransferFrom",
|
|
params.FromAddr,
|
|
params.ToAddr,
|
|
id,
|
|
params.AmountIn,
|
|
[]byte{},
|
|
)
|
|
}
|
|
|
|
func (s *ERC1155Processor) EstimateGas(params ProcessorInputParams) (uint64, error) {
|
|
if params.TestsMode {
|
|
if params.TestEstimationMap != nil {
|
|
if val, ok := params.TestEstimationMap[s.Name()]; ok {
|
|
return val.Value, val.Err
|
|
}
|
|
}
|
|
return 0, ErrNoEstimationFound
|
|
}
|
|
|
|
ethClient, err := s.rpcClient.EthClient(params.FromChain.ChainID)
|
|
if err != nil {
|
|
return 0, createERC1155ErrorResponse(err)
|
|
}
|
|
|
|
value := new(big.Int)
|
|
|
|
input, err := s.PackTxInputData(params)
|
|
if err != nil {
|
|
return 0, createERC1155ErrorResponse(err)
|
|
}
|
|
|
|
msg := ethereum.CallMsg{
|
|
From: params.FromAddr,
|
|
To: ¶ms.FromToken.Address,
|
|
Value: value,
|
|
Data: input,
|
|
}
|
|
|
|
estimation, err := ethClient.EstimateGas(context.Background(), msg)
|
|
if err != nil {
|
|
return 0, createERC1155ErrorResponse(err)
|
|
}
|
|
increasedEstimation := float64(estimation) * pathProcessorCommon.IncreaseEstimatedGasFactor
|
|
return uint64(increasedEstimation), nil
|
|
}
|
|
|
|
// TODO: remove this struct once mobile switches to the new approach
|
|
func (s *ERC1155Processor) sendOrBuild(sendArgs *MultipathProcessorTxArgs, signerFn bind.SignerFn, lastUsedNonce int64) (tx *ethTypes.Transaction, err error) {
|
|
ethClient, err := s.rpcClient.EthClient(sendArgs.ChainID)
|
|
if err != nil {
|
|
return tx, createERC1155ErrorResponse(err)
|
|
}
|
|
|
|
contract, err := ierc1155.NewIerc1155(common.Address(*sendArgs.ERC1155TransferTx.To), ethClient)
|
|
if err != nil {
|
|
return tx, createERC1155ErrorResponse(err)
|
|
}
|
|
|
|
var nonce uint64
|
|
if lastUsedNonce < 0 {
|
|
nonce, err = s.transactor.NextNonce(s.rpcClient, sendArgs.ChainID, sendArgs.ERC1155TransferTx.From)
|
|
if err != nil {
|
|
return tx, createERC1155ErrorResponse(err)
|
|
}
|
|
} else {
|
|
nonce = uint64(lastUsedNonce) + 1
|
|
}
|
|
|
|
argNonce := hexutil.Uint64(nonce)
|
|
sendArgs.ERC1155TransferTx.Nonce = &argNonce
|
|
txOpts := sendArgs.ERC1155TransferTx.ToTransactOpts(signerFn)
|
|
from := common.Address(sendArgs.ERC1155TransferTx.From)
|
|
tx, err = contract.SafeTransferFrom(
|
|
txOpts, from,
|
|
sendArgs.ERC1155TransferTx.Recipient,
|
|
sendArgs.ERC1155TransferTx.TokenID.ToInt(),
|
|
sendArgs.ERC1155TransferTx.Amount.ToInt(),
|
|
[]byte{},
|
|
)
|
|
if err != nil {
|
|
return tx, createERC1155ErrorResponse(err)
|
|
}
|
|
err = s.transactor.StoreAndTrackPendingTx(from, sendArgs.ERC1155TransferTx.Symbol, sendArgs.ChainID, sendArgs.ERC1155TransferTx.MultiTransactionID, tx)
|
|
if err != nil {
|
|
return tx, createERC1155ErrorResponse(err)
|
|
}
|
|
return tx, nil
|
|
}
|
|
|
|
func (s *ERC1155Processor) Send(sendArgs *MultipathProcessorTxArgs, lastUsedNonce int64, verifiedAccount *account.SelectedExtKey) (hash types.Hash, usedNonce uint64, err error) {
|
|
tx, err := s.sendOrBuild(sendArgs, utils.GetSigner(sendArgs.ChainID, sendArgs.ERC1155TransferTx.From, verifiedAccount.AccountKey.PrivateKey), lastUsedNonce)
|
|
if err != nil {
|
|
return hash, 0, createERC1155ErrorResponse(err)
|
|
}
|
|
return types.Hash(tx.Hash()), tx.Nonce(), nil
|
|
}
|
|
|
|
func (s *ERC1155Processor) BuildTransaction(sendArgs *MultipathProcessorTxArgs, lastUsedNonce int64) (*ethTypes.Transaction, uint64, error) {
|
|
tx, err := s.sendOrBuild(sendArgs, nil, lastUsedNonce)
|
|
return tx, tx.Nonce(), err
|
|
}
|
|
|
|
func (s *ERC1155Processor) BuildTransactionV2(sendArgs *wallettypes.SendTxArgs, lastUsedNonce int64) (*ethTypes.Transaction, uint64, error) {
|
|
return s.transactor.ValidateAndBuildTransaction(sendArgs.FromChainID, *sendArgs, lastUsedNonce)
|
|
}
|
|
|
|
func (s *ERC1155Processor) CalculateAmountOut(params ProcessorInputParams) (*big.Int, error) {
|
|
return params.AmountIn, nil
|
|
}
|
|
|
|
func (s *ERC1155Processor) GetContractAddress(params ProcessorInputParams) (common.Address, error) {
|
|
return params.FromToken.Address, nil
|
|
}
|