CCIP - add price reader interfaces to ChainAccessor and move some generic types

Author: ogtownsend

pkg/loop/internal/pb/ccipocr3/chainaccessor.pb.go

@@ -30,6 +30,13 @@ service ChainAccessor {
   rpc GetExpectedNextSequenceNumber(GetExpectedNextSequenceNumberRequest) returns (GetExpectedNextSequenceNumberResponse);
   rpc GetTokenPriceUSD(GetTokenPriceUSDRequest) returns (GetTokenPriceUSDResponse);
   rpc GetFeeQuoterDestChainConfig(GetFeeQuoterDestChainConfigRequest) returns (GetFeeQuoterDestChainConfigResponse);
+  
+  // USDCMessageReader methods
+  rpc MessagesByTokenID(MessagesByTokenIDRequest) returns (MessagesByTokenIDResponse);
+  
+  // PriceReader methods
+  rpc GetFeedPricesUSD(GetFeedPricesUSDRequest) returns (GetFeedPricesUSDResponse);
+  rpc GetFeeQuoterTokenUpdates(GetFeeQuoterTokenUpdatesRequest) returns (GetFeeQuoterTokenUpdatesResponse);
 }
 
 // AllAccessors request/response messages
@@ -337,4 +344,39 @@ message FeeQuoterDestChainConfig {
   uint32 gas_price_staleness_threshold = 19;      
   bool enforce_out_of_order = 15;
   bytes chain_family_selector = 16; // [4]byte in Go
+}
+
+// USDCMessageReader request/response messages
+message MessagesByTokenIDRequest {
+  uint64 source_chain_selector = 1;
+  uint64 dest_chain_selector = 2;
+  map<string, RampTokenAmount> tokens = 3; // key is MessageTokenID string representation
+}
+
+message MessagesByTokenIDResponse {
+  map<string, bytes> messages = 1; // key is MessageTokenID string representation, value is message bytes
+}
+
+// PriceReader request/response messages
+message GetFeedPricesUSDRequest {
+  repeated string tokens = 1; // UnknownEncodedAddress
+}
+
+message GetFeedPricesUSDResponse {
+  map<string, BigInt> prices = 1; // key is UnknownEncodedAddress, value is price
+}
+
+message GetFeeQuoterTokenUpdatesRequest {
+  repeated string tokens = 1; // UnknownEncodedAddress
+  uint64 chain_selector = 2;
+}
+
+message GetFeeQuoterTokenUpdatesResponse {
+  map<string, TimestampedBig> token_updates = 1; // key is UnknownEncodedAddress
+}
+
+// Helper message types
+message MessageTokenID {
+  uint64 seq_nr = 1;
+  int32 index = 2;
 }

pkg/loop/internal/pb/ccipocr3/chainaccessor.proto

@@ -290,6 +290,51 @@ func (c *chainAccessorClient) GetFeeQuoterDestChainConfig(ctx context.Context, d
 	return pbToFeeQuoterDestChainConfigDetailed(resp.Config), nil
 }
 
+// USDCMessageReader methods
+func (c *chainAccessorClient) MessagesByTokenID(ctx context.Context, source, dest ccipocr3.ChainSelector, tokens map[ccipocr3.MessageTokenID]ccipocr3.RampTokenAmount) (map[ccipocr3.MessageTokenID]ccipocr3.Bytes, error) {
+	resp, err := c.grpc.MessagesByTokenID(ctx, &ccipocr3pb.MessagesByTokenIDRequest{
+		SourceChainSelector: uint64(source),
+		DestChainSelector:   uint64(dest),
+		Tokens:              messageTokenIDMapToPb(tokens),
+	})
+	if err != nil {
+		return nil, err
+	}
+	return pbToMessagesByTokenID(resp.Messages)
+}
+
+// PriceReader methods
+func (c *chainAccessorClient) GetFeedPricesUSD(ctx context.Context, tokens []ccipocr3.UnknownEncodedAddress) (ccipocr3.TokenPriceMap, error) {
+	var tokenStrs []string
+	for _, token := range tokens {
+		tokenStrs = append(tokenStrs, string(token))
+	}
+
+	resp, err := c.grpc.GetFeedPricesUSD(ctx, &ccipocr3pb.GetFeedPricesUSDRequest{
+		Tokens: tokenStrs,
+	})
+	if err != nil {
+		return nil, err
+	}
+	return pbToTokenPriceMap(resp.Prices), nil
+}
+
+func (c *chainAccessorClient) GetFeeQuoterTokenUpdates(ctx context.Context, tokens []ccipocr3.UnknownEncodedAddress, chain ccipocr3.ChainSelector) (map[ccipocr3.UnknownEncodedAddress]ccipocr3.TimestampedBig, error) {
+	var tokenStrs []string
+	for _, token := range tokens {
+		tokenStrs = append(tokenStrs, string(token))
+	}
+
+	resp, err := c.grpc.GetFeeQuoterTokenUpdates(ctx, &ccipocr3pb.GetFeeQuoterTokenUpdatesRequest{
+		Tokens:        tokenStrs,
+		ChainSelector: uint64(chain),
+	})
+	if err != nil {
+		return nil, err
+	}
+	return pbToTokenUpdates(resp.TokenUpdates), nil
+}
+
 // Server implementation
 var _ ccipocr3pb.ChainAccessorServer = (*chainAccessorServer)(nil)
 
@@ -498,3 +543,62 @@ func (s *chainAccessorServer) GetFeeQuoterDestChainConfig(ctx context.Context, r
 		Config: feeQuoterDestChainConfigToPb(config),
 	}, nil
 }
+
+// USDCMessageReader server methods
+func (s *chainAccessorServer) MessagesByTokenID(ctx context.Context, req *ccipocr3pb.MessagesByTokenIDRequest) (*ccipocr3pb.MessagesByTokenIDResponse, error) {
+	tokens, err := pbToMessageTokenIDMap(req.Tokens)
+	if err != nil {
+		return nil, err
+	}
+
+	messages, err := s.impl.MessagesByTokenID(
+		ctx,
+		ccipocr3.ChainSelector(req.SourceChainSelector),
+		ccipocr3.ChainSelector(req.DestChainSelector),
+		tokens,
+	)
+	if err != nil {
+		return nil, err
+	}
+
+	return &ccipocr3pb.MessagesByTokenIDResponse{
+		Messages: messagesByTokenIDToPb(messages),
+	}, nil
+}
+
+// PriceReader server methods
+func (s *chainAccessorServer) GetFeedPricesUSD(ctx context.Context, req *ccipocr3pb.GetFeedPricesUSDRequest) (*ccipocr3pb.GetFeedPricesUSDResponse, error) {
+	var tokens []ccipocr3.UnknownEncodedAddress
+	for _, tokenStr := range req.Tokens {
+		tokens = append(tokens, ccipocr3.UnknownEncodedAddress(tokenStr))
+	}
+
+	prices, err := s.impl.GetFeedPricesUSD(ctx, tokens)
+	if err != nil {
+		return nil, err
+	}
+
+	return &ccipocr3pb.GetFeedPricesUSDResponse{
+		Prices: tokenPriceMapToPb(prices),
+	}, nil
+}
+
+func (s *chainAccessorServer) GetFeeQuoterTokenUpdates(ctx context.Context, req *ccipocr3pb.GetFeeQuoterTokenUpdatesRequest) (*ccipocr3pb.GetFeeQuoterTokenUpdatesResponse, error) {
+	var tokens []ccipocr3.UnknownEncodedAddress
+	for _, tokenStr := range req.Tokens {
+		tokens = append(tokens, ccipocr3.UnknownEncodedAddress(tokenStr))
+	}
+
+	updates, err := s.impl.GetFeeQuoterTokenUpdates(
+		ctx,
+		tokens,
+		ccipocr3.ChainSelector(req.ChainSelector),
+	)
+	if err != nil {
+		return nil, err
+	}
+
+	return &ccipocr3pb.GetFeeQuoterTokenUpdatesResponse{
+		TokenUpdates: tokenUpdatesToPb(updates),
+	}, nil
+}