Support jwt auth on blocksdata client

clients/blocksdata/r2_client.go

@@ -11,7 +11,9 @@ import (
 	"time"
 
 	backoff "github.com/cenkalti/backoff/v4"
+	"github.com/ethereum/go-ethereum/accounts/keystore"
 	"github.com/forta-network/forta-core-go/protocol"
+	"github.com/forta-network/forta-core-go/security"
 	"github.com/forta-network/forta-core-go/utils/httpclient"
 	"google.golang.org/protobuf/proto"
 )
@@ -24,13 +26,15 @@ const (
 
 type blocksDataClient struct {
 	dispatcherURL *url.URL
+	key           *keystore.Key
 }
 
-func NewBlocksDataClient(dispatcherURL string) *blocksDataClient {
+func NewBlocksDataClient(dispatcherURL string, key *keystore.Key) *blocksDataClient {
 	u, _ := url.Parse(dispatcherURL)
 
 	return &blocksDataClient{
 		dispatcherURL: u,
+		key:           key,
 	}
 }
 
@@ -54,6 +58,18 @@ func (c *blocksDataClient) GetBlocksData(bucket int64) (_ *protocol.BlocksData,
 	var item PresignedURLItem
 
 	err = backoff.Retry(func() error {
+		req, err := http.NewRequest(http.MethodGet, dispatcherUrl, nil)
+		if err != nil {
+			return backoff.Permanent(err)
+		}
+
+		jwt, err := security.CreateScannerJWT(c.key, nil)
+		if err != nil {
+			return backoff.Permanent(err)
+		}
+
+		req.Header.Set("Authorization", "Bearer "+jwt)
+
 		resp, err := httpclient.Default.Get(dispatcherUrl)
 		if err != nil {
 			return err

cmd/json-rpc/main.go

@@ -18,7 +18,7 @@ func initJsonRpcProxy(ctx context.Context, cfg config.Config) (*jrp.JsonRpcProxy
 }
 
 func initJsonRpcCache(ctx context.Context, cfg config.Config, botRegistry registry.BotRegistry) (*jrpcache.JsonRpcCache, error) {
-	return jrpcache.NewJsonRpcCache(ctx, cfg.JsonRpcCache, botRegistry)
+	return jrpcache.NewJsonRpcCache(ctx, cfg, botRegistry)
 }
 
 func initServices(ctx context.Context, cfg config.Config) ([]services.Service, error) {

services/json-rpc/cache/json_rpc_cache.go

@@ -7,6 +7,7 @@ import (
 	"strconv"
 	"time"
 
+	"github.com/ethereum/go-ethereum/accounts/keystore"
 	"github.com/forta-network/forta-core-go/domain"
 	"github.com/forta-network/forta-core-go/protocol"
 	"github.com/forta-network/forta-core-go/utils"
@@ -38,6 +39,7 @@ type JsonRpcCache struct {
 	botAuthenticator clients.IPAuthenticator
 	botRegistry      registry.BotRegistry
 	msgClient        clients.MessageClient
+	key              *keystore.Key
 
 	server *http.Server
 
@@ -46,15 +48,21 @@ type JsonRpcCache struct {
 	blocksDataClient clients.BlocksDataClient
 }
 
-func NewJsonRpcCache(ctx context.Context, cfg config.JsonRpcCacheConfig, botRegistry registry.BotRegistry) (*JsonRpcCache, error) {
+func NewJsonRpcCache(ctx context.Context, cfg config.Config, botRegistry registry.BotRegistry) (*JsonRpcCache, error) {
 	botAuthenticator, err := clients.NewBotAuthenticator(ctx)
 	if err != nil {
 		return nil, err
 	}
 
+	key, err := config.LoadKeyInContainer(cfg)
+	if err != nil {
+		return nil, err
+	}
+
 	return &JsonRpcCache{
 		ctx:              ctx,
-		cfg:              cfg,
+		key:              key,
+		cfg:              cfg.JsonRpcCache,
 		botAuthenticator: botAuthenticator,
 		botRegistry:      botRegistry,
 		msgClient:        messaging.NewClient("json-rpc-cache", fmt.Sprintf("%s:%s", config.DockerNatsContainerName, config.DefaultNatsPort)),
@@ -73,7 +81,7 @@ func (c *JsonRpcCache) Start() error {
 		Handler: r,
 	}
 
-	c.blocksDataClient = blocksdata.NewBlocksDataClient(c.cfg.DispatcherURL)
+	c.blocksDataClient = blocksdata.NewBlocksDataClient(c.cfg.DispatcherURL, c.key)
 
 	utils.GoListenAndServe(c.server)