feat: add getfrom

KyberNetwork · Apr 8, 2024 · c24d773 · c24d773
1 parent a6432b0
commit c24d773
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/pkg/mev/blxr_bundle_sender.go b/pkg/mev/blxr_bundle_sender.go
@@ -9,8 +9,10 @@ import (
 	"net/http"
 	"strconv"
 
+	"github.com/ethereum/go-ethereum"
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/core/types"
+	"github.com/ethereum/go-ethereum/ethclient/gethclient"
 	"github.com/flashbots/mev-share-node/mevshare"
 )
 
@@ -32,6 +34,10 @@ type BloxrouteClient struct {
 	enabledBuilders []BlxrBuilder
 }
 
+func (s *BloxrouteClient) EstimateBundleGas(ctx context.Context, messages []ethereum.CallMsg, overrides *map[common.Address]gethclient.OverrideAccount) ([]uint64, error) {
+	return nil, nil
+}
+
 func (s *BloxrouteClient) MevSimulateBundle(
 	_ uint64,
 	_ common.Hash,

diff --git a/pkg/mev/bundle_sender.go b/pkg/mev/bundle_sender.go
@@ -135,6 +135,7 @@ func (s *Client) flashbotBackrunSendBundle(
 }
 
 func (s *Client) EstimateBundleGas(
+	_ context.Context,
 	messages []ethereum.CallMsg,
 	overrides *map[common.Address]gethclient.OverrideAccount,
 ) ([]uint64, error) {
@@ -146,7 +147,7 @@ func (s *Client) EstimateBundleGas(
 	var gasEstimateCost []hexutil.Uint64
 
 	err := s.ethClient.Client().Call(
-		&gasEstimateCost, "eth_estimateGasBundle",
+		&gasEstimateCost, ETHEstimateGasBundleMethod,
 		map[string]interface{}{
 			"transactions": bundles,
 		}, "latest", overrides,

diff --git a/pkg/mev/pkg.go b/pkg/mev/pkg.go
@@ -13,6 +13,7 @@ import (
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/common/hexutil"
 	"github.com/ethereum/go-ethereum/core/types"
+	"github.com/ethereum/go-ethereum/ethclient/gethclient"
 	"github.com/flashbots/mev-share-node/mevshare"
 )
 
@@ -35,6 +36,7 @@ const (
 	ETHSendBundleMethod             = "eth_sendBundle"
 	EthCallBundleMethod             = "eth_callBundle"
 	ETHCancelBundleMethod           = "eth_cancelBundle"
+	ETHEstimateGasBundleMethod      = "eth_estimateGasBundle"
 	MevSendBundleMethod             = "mev_sendBundle"
 	MaxBlockFromTarget              = 3
 )
@@ -57,6 +59,11 @@ type IBundleSender interface {
 		ctx context.Context, bundleUUID string,
 	) error
 	SimulateBundle(ctx context.Context, blockNumber uint64, txs ...*types.Transaction) (SendBundleResponse, error)
+	EstimateBundleGas(
+		ctx context.Context,
+		messages []ethereum.CallMsg,
+		overrides *map[common.Address]gethclient.OverrideAccount,
+	) ([]uint64, error)
 	MevSimulateBundle(
 		blockNumber uint64,
 		pendingTxHash common.Hash,
@@ -196,3 +203,8 @@ func ToCallArg(msg ethereum.CallMsg) interface{} {
 	}
 	return arg
 }
+
+func GetFrom(tx *types.Transaction) (common.Address, error) {
+	from, err := types.Sender(types.LatestSignerForChainID(tx.ChainId()), tx)
+	return from, err
+}