feat: use new context to pass query

Signed-off-by: Gustavo Inacio <[email protected]>

Author: gusinacio

Cargo.lock

@@ -38,7 +38,7 @@ sqlx = { version = "0.8.2", features = [
 tracing = { version = "0.1.40", default-features = false }
 bigdecimal = "0.4.3"
 build-info = "0.0.39"
-tap_core = { git = "https://github.com/semiotic-ai/timeline-aggregation-protocol", rev = "ff856d9", default-features = false }
+tap_core = { git = "https://github.com/semiotic-ai/timeline-aggregation-protocol", rev = "3fe6bc2", default-features = false }
 tracing-subscriber = { version = "0.3", features = [
     "json",
     "env-filter",

Cargo.toml

@@ -13,6 +13,7 @@ use axum_extra::TypedHeader;
 use lazy_static::lazy_static;
 use prometheus::{register_counter_vec, register_histogram_vec, CounterVec, HistogramVec};
 use reqwest::StatusCode;
+use tap_core::receipt::Context;
 use thegraph_core::DeploymentId;
 use tracing::trace;
 
@@ -126,16 +127,13 @@ where
         .as_ref()
         .map(ToString::to_string)
         .unwrap_or_default();
-    let _ = state
-        .value_check_sender
-        .tx_query
-        .send(AgoraQuery {
-            signature,
-            deployment_id: manifest_id,
-            query: query_body.query.clone(),
-            variables,
-        })
-        .await;
+    let mut ctx = Context::new();
+    ctx.insert(AgoraQuery {
+        signature,
+        deployment_id: manifest_id,
+        query: query_body.query.clone(),
+        variables,
+    });
 
     // recover the signer address
     // get escrow accounts from eventual
@@ -168,7 +166,7 @@ where
     // Verify the receipt and store it in the database
     state
         .tap_manager
-        .verify_and_store_receipt(receipt)
+        .verify_and_store_receipt(&ctx, receipt)
         .await
         .inspect_err(|_| {
             FAILED_RECEIPT

common/src/indexer_service/http/request_handler.rs

@@ -27,7 +27,11 @@ impl AllocationEligible {
 }
 #[async_trait::async_trait]
 impl Check for AllocationEligible {
-    async fn check(&self, receipt: &ReceiptWithState<Checking>) -> CheckResult {
+    async fn check(
+        &self,
+        _: &tap_core::receipt::Context,
+        receipt: &ReceiptWithState<Checking>,
+    ) -> CheckResult {
         let allocation_id = receipt.signed_receipt().message.allocation_id;
         if !self
             .indexer_allocations

common/src/tap/checks/allocation_eligible.rs

@@ -150,7 +150,11 @@ impl DenyListCheck {
 
 #[async_trait::async_trait]
 impl Check for DenyListCheck {
-    async fn check(&self, receipt: &ReceiptWithState<Checking>) -> CheckResult {
+    async fn check(
+        &self,
+        _: &tap_core::receipt::Context,
+        receipt: &ReceiptWithState<Checking>,
+    ) -> CheckResult {
         let receipt_signer = receipt
             .signed_receipt()
             .recover_signer(&self.domain_separator)
@@ -195,7 +199,7 @@ mod tests {
     use std::str::FromStr;
 
     use alloy::hex::ToHexExt;
-    use tap_core::receipt::ReceiptWithState;
+    use tap_core::receipt::{Context, ReceiptWithState};
 
     use crate::test_vectors::{self, create_signed_receipt, TAP_SENDER};
 
@@ -241,7 +245,10 @@ mod tests {
         let checking_receipt = ReceiptWithState::new(signed_receipt);
 
         // Check that the receipt is rejected
-        assert!(deny_list_check.check(&checking_receipt).await.is_err());
+        assert!(deny_list_check
+            .check(&Context::new(), &checking_receipt)
+            .await
+            .is_err());
     }
 
     #[sqlx::test(migrations = "../migrations")]
@@ -255,7 +262,10 @@ mod tests {
         // Check that the receipt is valid
         let checking_receipt = ReceiptWithState::new(signed_receipt);
 
-        deny_list_check.check(&checking_receipt).await.unwrap();
+        deny_list_check
+            .check(&Context::new(), &checking_receipt)
+            .await
+            .unwrap();
 
         // Add the sender to the denylist
         sqlx::query!(
@@ -271,7 +281,10 @@ mod tests {
 
         // Check that the receipt is rejected
         tokio::time::sleep(std::time::Duration::from_millis(100)).await;
-        assert!(deny_list_check.check(&checking_receipt).await.is_err());
+        assert!(deny_list_check
+            .check(&Context::new(), &checking_receipt)
+            .await
+            .is_err());
 
         // Remove the sender from the denylist
         sqlx::query!(
@@ -287,6 +300,9 @@ mod tests {
 
         // Check that the receipt is valid again
         tokio::time::sleep(std::time::Duration::from_millis(100)).await;
-        deny_list_check.check(&checking_receipt).await.unwrap();
+        deny_list_check
+            .check(&Context::new(), &checking_receipt)
+            .await
+            .unwrap();
     }
 }

common/src/tap/checks/deny_list_check.rs

@@ -20,7 +20,11 @@ impl ReceiptMaxValueCheck {
 
 #[async_trait::async_trait]
 impl Check for ReceiptMaxValueCheck {
-    async fn check(&self, receipt: &ReceiptWithState<Checking>) -> CheckResult {
+    async fn check(
+        &self,
+        _: &tap_core::receipt::Context,
+        receipt: &ReceiptWithState<Checking>,
+    ) -> CheckResult {
         let receipt_value = receipt.signed_receipt().message.value;
 
         if receipt_value < self.receipt_max_value {
@@ -45,6 +49,7 @@ mod tests {
     use alloy::signers::local::MnemonicBuilder;
     use alloy::signers::local::PrivateKeySigner;
     use alloy::sol_types::eip712_domain;
+    use tap_core::receipt::Context;
 
     use super::*;
     use tap_core::{
@@ -96,20 +101,29 @@ mod tests {
     async fn test_receipt_lower_than_limit() {
         let signed_receipt = create_signed_receipt_with_custom_value(RECEIPT_LIMIT - 1);
         let timestamp_check = ReceiptMaxValueCheck::new(RECEIPT_LIMIT);
-        assert!(timestamp_check.check(&signed_receipt).await.is_ok());
+        assert!(timestamp_check
+            .check(&Context::new(), &signed_receipt)
+            .await
+            .is_ok());
     }
 
     #[tokio::test]
     async fn test_receipt_higher_than_limit() {
         let signed_receipt = create_signed_receipt_with_custom_value(RECEIPT_LIMIT + 1);
         let timestamp_check = ReceiptMaxValueCheck::new(RECEIPT_LIMIT);
-        assert!(timestamp_check.check(&signed_receipt).await.is_err());
+        assert!(timestamp_check
+            .check(&Context::new(), &signed_receipt)
+            .await
+            .is_err());
     }
 
     #[tokio::test]
     async fn test_receipt_same_as_limit() {
         let signed_receipt = create_signed_receipt_with_custom_value(RECEIPT_LIMIT);
         let timestamp_check = ReceiptMaxValueCheck::new(RECEIPT_LIMIT);
-        assert!(timestamp_check.check(&signed_receipt).await.is_err());
+        assert!(timestamp_check
+            .check(&Context::new(), &signed_receipt)
+            .await
+            .is_err());
     }
 }

common/src/tap/checks/receipt_max_val_check.rs

@@ -30,7 +30,11 @@ impl SenderBalanceCheck {
 
 #[async_trait::async_trait]
 impl Check for SenderBalanceCheck {
-    async fn check(&self, receipt: &ReceiptWithState<Checking>) -> CheckResult {
+    async fn check(
+        &self,
+        _: &tap_core::receipt::Context,
+        receipt: &ReceiptWithState<Checking>,
+    ) -> CheckResult {
         let escrow_accounts_snapshot = self.escrow_accounts.value_immediate().unwrap_or_default();
 
         let receipt_signer = receipt

common/src/tap/checks/sender_balance_check.rs

@@ -23,7 +23,11 @@ impl TimestampCheck {
 
 #[async_trait::async_trait]
 impl Check for TimestampCheck {
-    async fn check(&self, receipt: &ReceiptWithState<Checking>) -> CheckResult {
+    async fn check(
+        &self,
+        _: &tap_core::receipt::Context,
+        receipt: &ReceiptWithState<Checking>,
+    ) -> CheckResult {
         let timestamp_now = SystemTime::now()
             .duration_since(SystemTime::UNIX_EPOCH)
             .map_err(|e| CheckError::Failed(e.into()))?;
@@ -55,7 +59,7 @@ mod tests {
 
     use super::*;
     use tap_core::{
-        receipt::{checks::Check, state::Checking, Receipt, ReceiptWithState},
+        receipt::{checks::Check, state::Checking, Context, Receipt, ReceiptWithState},
         signed_message::EIP712SignedMessage,
     };
 
@@ -102,7 +106,10 @@ mod tests {
         let timestamp_ns = timestamp as u64;
         let signed_receipt = create_signed_receipt_with_custom_timestamp(timestamp_ns);
         let timestamp_check = TimestampCheck::new(Duration::from_secs(30));
-        assert!(timestamp_check.check(&signed_receipt).await.is_ok());
+        assert!(timestamp_check
+            .check(&Context::new(), &signed_receipt)
+            .await
+            .is_ok());
     }
 
     #[tokio::test]
@@ -115,7 +122,10 @@ mod tests {
         let timestamp_ns = timestamp as u64;
         let signed_receipt = create_signed_receipt_with_custom_timestamp(timestamp_ns);
         let timestamp_check = TimestampCheck::new(Duration::from_secs(30));
-        assert!(timestamp_check.check(&signed_receipt).await.is_err());
+        assert!(timestamp_check
+            .check(&Context::new(), &signed_receipt)
+            .await
+            .is_err());
     }
 
     #[tokio::test]
@@ -128,6 +138,9 @@ mod tests {
         let timestamp_ns = timestamp as u64;
         let signed_receipt = create_signed_receipt_with_custom_timestamp(timestamp_ns);
         let timestamp_check = TimestampCheck::new(Duration::from_secs(30));
-        assert!(timestamp_check.check(&signed_receipt).await.is_err());
+        assert!(timestamp_check
+            .check(&Context::new(), &signed_receipt)
+            .await
+            .is_err());
     }
 }