ethrpc: state overrides

Author: attente

ethrpc/ethrpc.go

@@ -487,6 +487,12 @@ func (p *Provider) CallContract(ctx context.Context, msg ethereum.CallMsg, block
 	return result, err
 }
 
+func (p *Provider) CallContractWithOverrides(ctx context.Context, msg ethereum.CallMsg, blockNum *big.Int, overrides map[common.Address]OverrideAccount) ([]byte, error) {
+	var result []byte
+	_, err := p.Do(ctx, CallContractWithOverrides(msg, blockNum, overrides).Strict(p.strictness).Into(&result))
+	return result, err
+}
+
 func (p *Provider) CallContractAtHash(ctx context.Context, msg ethereum.CallMsg, blockHash common.Hash) ([]byte, error) {
 	var result []byte
 	_, err := p.Do(ctx, CallContractAtHash(msg, blockHash).Strict(p.strictness).Into(&result))
@@ -523,6 +529,12 @@ func (p *Provider) EstimateGas(ctx context.Context, msg ethereum.CallMsg) (uint6
 	return result, err
 }
 
+func (p *Provider) EstimateGasWithOverrides(ctx context.Context, msg ethereum.CallMsg, overrides map[common.Address]OverrideAccount) (uint64, error) {
+	var result uint64
+	_, err := p.Do(ctx, EstimateGasWithOverrides(msg, overrides).Strict(p.strictness).Into(&result))
+	return result, err
+}
+
 func (p *Provider) DebugTraceBlockByNumber(ctx context.Context, blockNum *big.Int) ([]*TransactionDebugTrace, error) {
 	var result []*TransactionDebugTrace
 	_, err := p.Do(ctx, DebugTraceBlockByNumber(blockNum).Into(&result))

ethrpc/methods.go

@@ -342,6 +342,18 @@ func CallContract(msg ethereum.CallMsg, blockNum *big.Int) CallBuilder[[]byte] {
 	}
 }
 
+func CallContractWithOverrides(msg ethereum.CallMsg, blockNum *big.Int, overrides map[common.Address]OverrideAccount) CallBuilder[[]byte] {
+	params := []any{toCallArg(msg), toBlockNumArg(blockNum)}
+	if len(overrides) > 0 {
+		params = append(params, overrides)
+	}
+	return CallBuilder[[]byte]{
+		method: "eth_call",
+		params: params,
+		intoFn: hexIntoBytes,
+	}
+}
+
 func CallContractAtHash(msg ethereum.CallMsg, blockHash common.Hash) CallBuilder[[]byte] {
 	return CallBuilder[[]byte]{
 		method: "eth_call",
@@ -419,6 +431,18 @@ func EstimateGas(msg ethereum.CallMsg) CallBuilder[uint64] {
 	}
 }
 
+func EstimateGasWithOverrides(msg ethereum.CallMsg, overrides map[common.Address]OverrideAccount) CallBuilder[uint64] {
+	params := []any{toCallArg(msg)}
+	if len(overrides) > 0 {
+		params = append(params, toBlockNumArg(nil), overrides)
+	}
+	return CallBuilder[uint64]{
+		method: "eth_estimateGas",
+		params: params,
+		intoFn: hexIntoUint64,
+	}
+}
+
 type DebugTracer string
 
 const (

ethrpc/overrides.go

@@ -0,0 +1,7 @@
+package ethrpc
+
+import "github.com/0xsequence/ethkit/go-ethereum/ethclient/gethclient"
+
+// OverrideAccount specifies the state of an account to be overridden.
+// This is an alias of gethclient.OverrideAccount so callers don't need to import gethclient.
+type OverrideAccount = gethclient.OverrideAccount