From e41a830bc7ddfce463a1e3f0289fbbae804272a9 Mon Sep 17 00:00:00 2001
From: Lucca Martins <lucca_martins30@yahoo.com.br>
Date: Tue, 27 May 2025 02:21:41 -0300
Subject: [PATCH] eth_getTdByNumber method (#1552)

* get td by number method

* fix ci

* fixing lint
---
 eth/api_backend.go                       |  7 +++++++
 internal/ethapi/api.go                   | 26 ++++++++++++++++++++++++
 internal/ethapi/api_test.go              |  4 ++++
 internal/ethapi/backend.go               |  2 ++
 internal/ethapi/transaction_args_test.go |  4 ++++
 5 files changed, 43 insertions(+)

diff --git a/eth/api_backend.go b/eth/api_backend.go
index ef872ca9a2..ac435c1ba9 100644
--- a/eth/api_backend.go
+++ b/eth/api_backend.go
@@ -285,6 +285,13 @@ func (b *EthAPIBackend) GetTd(ctx context.Context, hash common.Hash) *big.Int {
 	return nil
 }
 
+func (b *EthAPIBackend) GetTdByNumber(ctx context.Context, blockNr rpc.BlockNumber) *big.Int {
+	if header, err := b.HeaderByNumber(ctx, blockNr); header != nil && err == nil {
+		return b.eth.blockchain.GetTd(header.Hash(), uint64(blockNr.Int64()))
+	}
+	return nil
+}
+
 func (b *EthAPIBackend) GetEVM(ctx context.Context, msg *core.Message, state *state.StateDB, header *types.Header, vmConfig *vm.Config, blockCtx *vm.BlockContext) *vm.EVM {
 	if vmConfig == nil {
 		vmConfig = b.eth.blockchain.GetVMConfig()
diff --git a/internal/ethapi/api.go b/internal/ethapi/api.go
index eec4b6b0ca..847a8997b9 100644
--- a/internal/ethapi/api.go
+++ b/internal/ethapi/api.go
@@ -769,6 +769,32 @@ func (api *BlockChainAPI) GetBlockReceipts(ctx context.Context, blockNrOrHash rp
 	return result, nil
 }
 
+// GetTdByHash returns a map containing the total difficulty (hex-encoded) for the given block hash.
+func (api *BlockChainAPI) GetTdByHash(ctx context.Context, hash common.Hash) map[string]interface{} {
+	td := api.b.GetTd(ctx, hash)
+	if td == nil {
+		return nil
+	}
+
+	resp := make(map[string]interface{}, 2)
+	resp["blockHash"] = hash.Hex()
+	resp["totalDifficulty"] = hexutil.EncodeBig(td)
+	return resp
+}
+
+// GetTdByNumber returns a map containing the total difficulty (hex-encoded) for the given block number.
+func (api *BlockChainAPI) GetTdByNumber(ctx context.Context, blockNr rpc.BlockNumber) map[string]interface{} {
+	td := api.b.GetTdByNumber(ctx, blockNr)
+	if td == nil {
+		return nil
+	}
+
+	resp := make(map[string]interface{}, 2)
+	resp["blockNumber"] = hexutil.EncodeUint64(uint64(blockNr.Int64()))
+	resp["totalDifficulty"] = hexutil.EncodeBig(td)
+	return resp
+}
+
 // OverrideAccount indicates the overriding fields of account during the execution
 // of a message call.
 // Note, state and stateDiff can't be specified at the same time. If state is
diff --git a/internal/ethapi/api_test.go b/internal/ethapi/api_test.go
index d453b03070..2db3175dac 100644
--- a/internal/ethapi/api_test.go
+++ b/internal/ethapi/api_test.go
@@ -568,6 +568,10 @@ func (b testBackend) GetTd(ctx context.Context, hash common.Hash) *big.Int {
 	}
 	return big.NewInt(1)
 }
+func (b testBackend) GetTdByNumber(ctx context.Context, blockNr rpc.BlockNumber) *big.Int {
+	panic("not implemented")
+}
+
 func (b testBackend) GetEVM(ctx context.Context, msg *core.Message, state *state.StateDB, header *types.Header, vmConfig *vm.Config, blockContext *vm.BlockContext) *vm.EVM {
 	if vmConfig == nil {
 		vmConfig = b.chain.GetVMConfig()
diff --git a/internal/ethapi/backend.go b/internal/ethapi/backend.go
index 97f8d3c6cb..920f14cc4b 100644
--- a/internal/ethapi/backend.go
+++ b/internal/ethapi/backend.go
@@ -70,6 +70,8 @@ type Backend interface {
 	Pending() (*types.Block, types.Receipts, *state.StateDB)
 	GetReceipts(ctx context.Context, hash common.Hash) (types.Receipts, error)
 	GetEVM(ctx context.Context, msg *core.Message, state *state.StateDB, header *types.Header, vmConfig *vm.Config, blockCtx *vm.BlockContext) *vm.EVM
+	GetTd(ctx context.Context, hash common.Hash) *big.Int
+	GetTdByNumber(ctx context.Context, blockNr rpc.BlockNumber) *big.Int
 	SubscribeChainEvent(ch chan<- core.ChainEvent) event.Subscription
 	SubscribeChainHeadEvent(ch chan<- core.ChainHeadEvent) event.Subscription
 
diff --git a/internal/ethapi/transaction_args_test.go b/internal/ethapi/transaction_args_test.go
index 39b9f93350..7ea5c841d3 100644
--- a/internal/ethapi/transaction_args_test.go
+++ b/internal/ethapi/transaction_args_test.go
@@ -470,3 +470,7 @@ func (b *backendMock) PurgeWhitelistedMilestone() {}
 func (b backendMock) PeerStats() interface{} {
 	return nil
 }
+
+func (b backendMock) GetTdByNumber(ctx context.Context, blockNr rpc.BlockNumber) *big.Int {
+	panic("not implemented")
+}