refactor: use generic manager

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

Author: gusinacio

Cargo.lock

@@ -1,15 +1,15 @@
 [workspace]
 members = [
-    "crates/allocation",
-    "crates/attestation",
-    "crates/config",
-    "crates/dips",
-    "crates/monitor",
-    "crates/query",
-    "crates/service",
-    "crates/tap-agent",
-    "crates/test-assets",
-    "crates/watcher",
+  "crates/allocation",
+  "crates/attestation",
+  "crates/config",
+  "crates/dips",
+  "crates/monitor",
+  "crates/query",
+  "crates/service",
+  "crates/tap-agent",
+  "crates/test-assets",
+  "crates/watcher",
 ]
 resolver = "2"
 
@@ -20,54 +20,54 @@ opt-level = 3
 clap = "4.4.3"
 lazy_static = "1.4.0"
 axum = { version = "0.7.9", default-features = false, features = [
-    "tokio",
-    "http1",
-    "http2",
+  "tokio",
+  "http1",
+  "http2",
 ] }
-tokio = "1.40"
+tokio = "1"
 prometheus = "0.13.3"
 anyhow = { version = "1.0.72" }
 thiserror = "1.0.49"
 async-trait = "0.1.83"
 eventuals = "0.6.7"
 base64 = "0.22.1"
 reqwest = { version = "0.12", features = [
-    "charset",
-    "h2",
+  "charset",
+  "h2",
 ], default-features = false }
 serde = { version = "1.0.206", default-features = false }
 serde_json = "1.0.124"
 sqlx = { version = "0.8.2", features = [
-    "bigdecimal",
-    "chrono",
-    "json",
-    "macros",
-    "migrate",
-    "postgres",
-    "runtime-tokio",
-    "rust_decimal",
-    "uuid",
+  "bigdecimal",
+  "chrono",
+  "json",
+  "macros",
+  "migrate",
+  "postgres",
+  "runtime-tokio",
+  "rust_decimal",
+  "uuid",
 ], default-features = false }
 uuid = { version = "1.11.0", features = ["v7"] }
 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 = "6af1add", default-features = false }
-tap_aggregator = { git = "https://github.com/semiotic-ai/timeline-aggregation-protocol", rev = "6af1add", default-features = false }
+tap_core = { git = "https://github.com/semiotic-ai/timeline-aggregation-protocol", rev = "1fc51a3", default-features = false }
+tap_aggregator = { git = "https://github.com/semiotic-ai/timeline-aggregation-protocol", rev = "1fc51a3", default-features = false }
 tracing-subscriber = { version = "0.3", features = [
-    "json",
-    "env-filter",
-    "ansi",
+  "json",
+  "env-filter",
+  "ansi",
 ], default-features = false }
-thegraph-core = { version = "0.10.0", features = [
-    "attestation",
-    "alloy-eip712",
-    "alloy-sol-types",
-    "alloy-rlp",
-    "alloy-signers",
-    "alloy-signer-local",
-    "alloy-signer-mnemonic",
-    "serde"
+thegraph-core = { git = "https://github.com/edgeandnode/toolshed", rev = "a1d0509", features = [
+  "attestation",
+  "alloy-eip712",
+  "alloy-sol-types",
+  "alloy-rlp",
+  "alloy-signers",
+  "alloy-signer-local",
+  "alloy-signer-mnemonic",
+  "serde",
 ] }
 thegraph-graphql-http = { version = "0.3.2", features = ["reqwest"] }
 graphql_client = { version = "0.14.0", features = ["reqwest-rustls"] }

Cargo.toml

@@ -9,7 +9,7 @@ use crate::{
 };
 use anyhow::anyhow;
 use async_trait::async_trait;
-use thegraph_core::alloy::{dyn_abi::Eip712Domain, primitives::Address};
+use thegraph_core::alloy::{primitives::Address, sol_types::Eip712Domain};
 use uuid::Uuid;
 
 #[derive(Debug)]

crates/dips/src/server.rs

@@ -18,6 +18,7 @@ use axum::{
 };
 use tap_core::{
     manager::{adapters::ReceiptStore, Manager},
+    rav::ReceiptAggregateVoucher,
     receipt::{Context, SignedReceipt},
 };
 use tower_http::auth::AsyncAuthorizeRequest;
@@ -30,7 +31,7 @@ use crate::{error::IndexerServiceError, middleware::prometheus_metrics::MetricLa
 ///
 /// Requires SignedReceipt, MetricLabels and Arc<Context> extensions
 pub fn tap_receipt_authorize<T, B>(
-    tap_manager: Arc<Manager<T>>,
+    tap_manager: Arc<Manager<T, SignedReceipt, ReceiptAggregateVoucher>>,
     failed_receipt_metric: &'static prometheus::CounterVec,
 ) -> impl AsyncAuthorizeRequest<
     B,
@@ -40,7 +41,7 @@ pub fn tap_receipt_authorize<T, B>(
 > + Clone
        + Send
 where
-    T: ReceiptStore + Sync + Send + 'static,
+    T: ReceiptStore<SignedReceipt> + Sync + Send + 'static,
     B: Send,
 {
     move |request: Request<B>| {
@@ -91,7 +92,7 @@ mod tests {
         receipt::{
             checks::{Check, CheckError, CheckList, CheckResult},
             state::Checking,
-            ReceiptWithState,
+            ReceiptWithState, SignedReceipt,
         },
     };
     use test_assets::{
@@ -133,11 +134,11 @@ mod tests {
 
         struct MyCheck;
         #[async_trait::async_trait]
-        impl Check for MyCheck {
+        impl Check<SignedReceipt> for MyCheck {
             async fn check(
                 &self,
                 _: &tap_core::receipt::Context,
-                receipt: &ReceiptWithState<Checking>,
+                receipt: &ReceiptWithState<Checking, SignedReceipt>,
             ) -> CheckResult {
                 if receipt.signed_receipt().message.nonce == FAILED_NONCE {
                     Err(CheckError::Failed(anyhow::anyhow!("Failed")))

crates/service/src/middleware/auth/tap.rs

@@ -7,7 +7,7 @@ use indexer_allocation::Allocation;
 use indexer_monitor::EscrowAccounts;
 use receipt_store::{DatabaseReceipt, InnerContext};
 use sqlx::PgPool;
-use tap_core::receipt::checks::ReceiptCheck;
+use tap_core::receipt::{checks::ReceiptCheck, SignedReceipt};
 use thegraph_core::alloy::{primitives::Address, sol_types::Eip712Domain};
 use tokio::sync::{
     mpsc::{self, Sender},
@@ -48,7 +48,7 @@ impl IndexerTapContext {
         escrow_accounts: Receiver<EscrowAccounts>,
         timestamp_error_tolerance: Duration,
         receipt_max_value: u128,
-    ) -> Vec<ReceiptCheck> {
+    ) -> Vec<ReceiptCheck<SignedReceipt>> {
         vec![
             Arc::new(AllocationEligible::new(indexer_allocations)),
             Arc::new(SenderBalanceCheck::new(escrow_accounts)),

crates/service/src/tap.rs

@@ -8,7 +8,7 @@ use indexer_allocation::Allocation;
 use tap_core::receipt::{
     checks::{Check, CheckError, CheckResult},
     state::Checking,
-    ReceiptWithState,
+    ReceiptWithState, SignedReceipt,
 };
 use thegraph_core::alloy::primitives::Address;
 use tokio::sync::watch::Receiver;
@@ -25,11 +25,11 @@ impl AllocationEligible {
     }
 }
 #[async_trait::async_trait]
-impl Check for AllocationEligible {
+impl Check<SignedReceipt> for AllocationEligible {
     async fn check(
         &self,
         _: &tap_core::receipt::Context,
-        receipt: &ReceiptWithState<Checking>,
+        receipt: &ReceiptWithState<Checking, SignedReceipt>,
     ) -> CheckResult {
         let allocation_id = receipt.signed_receipt().message.allocation_id;
         if !self

crates/service/src/tap/checks/allocation_eligible.rs

@@ -11,7 +11,7 @@ use sqlx::{postgres::PgListener, PgPool};
 use tap_core::receipt::{
     checks::{Check, CheckError, CheckResult},
     state::Checking,
-    ReceiptWithState,
+    ReceiptWithState, SignedReceipt,
 };
 use thegraph_core::alloy::primitives::Address;
 
@@ -153,11 +153,11 @@ impl DenyListCheck {
 }
 
 #[async_trait::async_trait]
-impl Check for DenyListCheck {
+impl Check<SignedReceipt> for DenyListCheck {
     async fn check(
         &self,
         ctx: &tap_core::receipt::Context,
-        _: &ReceiptWithState<Checking>,
+        _: &ReceiptWithState<Checking, SignedReceipt>,
     ) -> CheckResult {
         let Sender(receipt_sender) = ctx
             .get::<Sender>()

crates/service/src/tap/checks/deny_list_check.rs

@@ -9,7 +9,7 @@ pub struct ReceiptMaxValueCheck {
 use tap_core::receipt::{
     checks::{Check, CheckError, CheckResult},
     state::Checking,
-    ReceiptWithState,
+    ReceiptWithState, SignedReceipt,
 };
 
 impl ReceiptMaxValueCheck {
@@ -19,11 +19,11 @@ impl ReceiptMaxValueCheck {
 }
 
 #[async_trait::async_trait]
-impl Check for ReceiptMaxValueCheck {
+impl Check<SignedReceipt> for ReceiptMaxValueCheck {
     async fn check(
         &self,
         _: &tap_core::receipt::Context,
-        receipt: &ReceiptWithState<Checking>,
+        receipt: &ReceiptWithState<Checking, SignedReceipt>,
     ) -> CheckResult {
         let receipt_value = receipt.signed_receipt().message.value;
 
@@ -54,7 +54,9 @@ mod tests {
     use super::*;
     use crate::tap::Eip712Domain;
 
-    fn create_signed_receipt_with_custom_value(value: u128) -> ReceiptWithState<Checking> {
+    fn create_signed_receipt_with_custom_value(
+        value: u128,
+    ) -> ReceiptWithState<Checking, SignedReceipt> {
         let index: u32 = 0;
         let wallet: PrivateKeySigner = MnemonicBuilder::<English>::default()
             .phrase("abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon about")
@@ -86,7 +88,7 @@ mod tests {
             &wallet,
         )
         .unwrap();
-        ReceiptWithState::<Checking>::new(receipt)
+        ReceiptWithState::new(receipt)
     }
 
     const RECEIPT_LIMIT: u128 = 10;

crates/service/src/tap/checks/receipt_max_val_check.rs

@@ -6,7 +6,7 @@ use indexer_monitor::EscrowAccounts;
 use tap_core::receipt::{
     checks::{Check, CheckError, CheckResult},
     state::Checking,
-    ReceiptWithState,
+    ReceiptWithState, SignedReceipt,
 };
 use thegraph_core::alloy::primitives::U256;
 use tokio::sync::watch::Receiver;
@@ -24,11 +24,11 @@ impl SenderBalanceCheck {
 }
 
 #[async_trait::async_trait]
-impl Check for SenderBalanceCheck {
+impl Check<SignedReceipt> for SenderBalanceCheck {
     async fn check(
         &self,
         ctx: &tap_core::receipt::Context,
-        _: &ReceiptWithState<Checking>,
+        _: &ReceiptWithState<Checking, SignedReceipt>,
     ) -> CheckResult {
         let escrow_accounts_snapshot = self.escrow_accounts.borrow();
 

crates/service/src/tap/checks/sender_balance_check.rs

@@ -11,7 +11,7 @@ pub struct TimestampCheck {
 use tap_core::receipt::{
     checks::{Check, CheckError, CheckResult},
     state::Checking,
-    ReceiptWithState,
+    ReceiptWithState, SignedReceipt,
 };
 
 impl TimestampCheck {
@@ -23,11 +23,11 @@ impl TimestampCheck {
 }
 
 #[async_trait::async_trait]
-impl Check for TimestampCheck {
+impl Check<SignedReceipt> for TimestampCheck {
     async fn check(
         &self,
         _: &tap_core::receipt::Context,
-        receipt: &ReceiptWithState<Checking>,
+        receipt: &ReceiptWithState<Checking, SignedReceipt>,
     ) -> CheckResult {
         let timestamp_now = SystemTime::now()
             .duration_since(SystemTime::UNIX_EPOCH)
@@ -52,7 +52,7 @@ mod tests {
     use std::time::{Duration, SystemTime};
 
     use tap_core::{
-        receipt::{checks::Check, state::Checking, Context, Receipt, ReceiptWithState},
+        receipt::{checks::Check, state::Checking, Context, Receipt, ReceiptWithState, SignedReceipt},
         signed_message::EIP712SignedMessage,
         tap_eip712_domain,
     };
@@ -66,7 +66,7 @@ mod tests {
 
     fn create_signed_receipt_with_custom_timestamp(
         timestamp_ns: u64,
-    ) -> ReceiptWithState<Checking> {
+    ) -> ReceiptWithState<Checking, SignedReceipt> {
         let index: u32 = 0;
         let wallet: PrivateKeySigner = MnemonicBuilder::<English>::default()
             .phrase("abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon about")
@@ -89,7 +89,7 @@ mod tests {
             &wallet,
         )
         .unwrap();
-        ReceiptWithState::<Checking>::new(receipt)
+        ReceiptWithState::new(receipt)
     }
 
     #[tokio::test]