feat: use thegraph-core alloy crate

Signed-off-by: Lorenzo Delgado <[email protected]>

Author: LNSD

Cargo.lock

@@ -17,16 +17,6 @@ resolver = "2"
 opt-level = 3
 
 [workspace.dependencies]
-alloy = { version = "=0.5.4", features = [
-    "kzg",
-    "signer-mnemonic",
-    "dyn-abi",
-    "sol-types",
-    "signer-local",
-    "eip712",
-    "rlp",
-    "signers",
-], default-features = false }
 clap = "4.4.3"
 lazy_static = "1.4.0"
 axum = { version = "0.7.9", default-features = false, features = [
@@ -62,14 +52,24 @@ 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 = "61b47b3", default-features = false }
-tap_aggregator = { git = "https://github.com/semiotic-ai/timeline-aggregation-protocol", rev = "61b47b3", default-features = false }
+tap_core = { git = "https://github.com/semiotic-ai/timeline-aggregation-protocol", rev = "1c6e29f", default-features = false }
+tap_aggregator = { git = "https://github.com/semiotic-ai/timeline-aggregation-protocol", rev = "1c6e29f", default-features = false }
 tracing-subscriber = { version = "0.3", features = [
     "json",
     "env-filter",
     "ansi",
 ], default-features = false }
-thegraph-core = { git = "https://github.com/edgeandnode/toolshed", rev = "1663534fc1738e2db1b11cb54b5bb478ee970d40" }
+thegraph-core = { version = "0.9.6", features = [
+    "attestation",
+    "alloy-eip712",
+    "alloy-sol-types",
+    "alloy-rlp",
+    "alloy-signers",
+    "alloy-signer-local",
+    "alloy-signer-mnemonic",
+    "alloy-kzg",
+    "serde"
+] }
 thegraph-graphql-http = { version = "0.3.2", features = ["reqwest"] }
 graphql_client = { version = "0.14.0", features = ["reqwest-rustls"] }
 bip39 = "2.0.0"

Cargo.toml

@@ -6,6 +6,5 @@ edition = "2021"
 [dependencies]
 indexer-query = { path = "../query" }
 serde = { workspace = true, features = ["derive"] }
-alloy.workspace = true
 thegraph-core.workspace = true
 anyhow.workspace = true

crates/allocation/Cargo.toml

@@ -3,10 +3,12 @@
 
 use std::str::FromStr;
 
-use alloy::primitives::U256;
 use indexer_query::allocations_query;
 use serde::{Deserialize, Deserializer};
-use thegraph_core::{Address, DeploymentId};
+use thegraph_core::{
+    alloy::primitives::{Address, U256},
+    DeploymentId,
+};
 
 #[derive(Clone, Debug, PartialEq, Eq)]
 pub struct Allocation {

crates/allocation/src/lib.rs

@@ -5,7 +5,6 @@ edition = "2021"
 
 [dependencies]
 indexer-allocation = { path = "../allocation" }
-alloy.workspace = true
 thegraph-core.workspace = true
 anyhow.workspace = true
 

crates/attestation/Cargo.toml

@@ -1,16 +1,20 @@
 // Copyright 2023-, Edge & Node, GraphOps, and Semiotic Labs.
 // SPDX-License-Identifier: Apache-2.0
 
-use alloy::{
-    dyn_abi::Eip712Domain,
-    signers::{
-        k256,
-        local::{coins_bip39::English, MnemonicBuilder, PrivateKeySigner},
+use indexer_allocation::Allocation;
+use thegraph_core::{
+    alloy::{
+        primitives::{Address, ChainId},
+        signers::{
+            k256,
+            local::{coins_bip39::English, MnemonicBuilder, PrivateKeySigner},
+        },
+        sol_types::Eip712Domain,
     },
+    attestation,
+    attestation::Attestation,
+    DeploymentId,
 };
-use thegraph_core::{attestation, Address, Attestation, ChainId, DeploymentId};
-
-use indexer_allocation::Allocation;
 
 pub fn derive_key_pair(
     indexer_mnemonic: &str,
@@ -92,7 +96,7 @@ fn wallet_for_allocation(
     // range [0, 100] and checking for a match
     for i in 0..100 {
         // The allocation was either created at the epoch it intended to or one
-        // epoch later. So try both both.
+        // epoch later. So try both.
         for created_at_epoch in [allocation.created_at_epoch, allocation.created_at_epoch - 1] {
             // The allocation ID is the address of a unique key pair, we just
             // need to find the right one by enumerating them all
@@ -117,12 +121,12 @@ fn wallet_for_allocation(
 
 #[cfg(test)]
 mod tests {
-    use alloy::primitives::U256;
     use std::str::FromStr;
-    use test_log::test;
 
     use indexer_allocation::{Allocation, AllocationStatus, SubgraphDeployment};
     use test_assets::DISPUTE_MANAGER_ADDRESS;
+    use test_log::test;
+    use thegraph_core::alloy::primitives::{address, U256};
 
     use super::*;
 
@@ -142,7 +146,7 @@ mod tests {
             )
             .unwrap()
             .address(),
-            Address::from_str("0xfa44c72b753a66591f241c7dc04e8178c30e13af").unwrap()
+            address!("fa44c72b753a66591f241c7dc04e8178c30e13af")
         );
 
         assert_eq!(
@@ -157,7 +161,7 @@ mod tests {
             )
             .unwrap()
             .address(),
-            Address::from_str("0xa171cd12c3dde7eb8fe7717a0bcd06f3ffa65658").unwrap()
+            address!("a171cd12c3dde7eb8fe7717a0bcd06f3ffa65658")
         );
     }
 
@@ -166,7 +170,7 @@ mod tests {
         // Note that we use `derive_key_pair` to derive the private key
 
         let allocation = Allocation {
-            id: Address::from_str("0xa171cd12c3dde7eb8fe7717a0bcd06f3ffa65658").unwrap(),
+            id: address!("a171cd12c3dde7eb8fe7717a0bcd06f3ffa65658"),
             status: AllocationStatus::Null,
             subgraph_deployment: SubgraphDeployment {
                 id: DeploymentId::from_str(
@@ -192,7 +196,7 @@ mod tests {
                     INDEXER_OPERATOR_MNEMONIC,
                     &allocation,
                     1,
-                    *DISPUTE_MANAGER_ADDRESS
+                    DISPUTE_MANAGER_ADDRESS
                 )
                 .unwrap()
                 .signer
@@ -213,7 +217,7 @@ mod tests {
 
         let allocation = Allocation {
             // Purposefully wrong address
-            id: Address::from_str("0xdeadbeefcafebabedeadbeefcafebabedeadbeef").unwrap(),
+            id: address!("deadbeefcafebabedeadbeefcafebabedeadbeef"),
             status: AllocationStatus::Null,
             subgraph_deployment: SubgraphDeployment {
                 id: DeploymentId::from_str(
@@ -237,7 +241,7 @@ mod tests {
             INDEXER_OPERATOR_MNEMONIC,
             &allocation,
             1,
-            *DISPUTE_MANAGER_ADDRESS
+            DISPUTE_MANAGER_ADDRESS
         )
         .is_err());
     }

crates/attestation/src/lib.rs

@@ -4,7 +4,6 @@ version = "1.2.2"
 edition = "2021"
 
 [dependencies]
-alloy.workspace = true
 serde.workspace = true
 thegraph-core.workspace = true
 tracing.workspace = true

crates/config/Cargo.toml

@@ -1,29 +1,27 @@
 // Copyright 2023-, Edge & Node, GraphOps, and Semiotic Labs.
 // SPDX-License-Identifier: Apache-2.0
 
-use bigdecimal::{BigDecimal, FromPrimitive, ToPrimitive};
-use figment::{
-    providers::{Env, Format, Toml},
-    Figment,
-};
-use serde_repr::Deserialize_repr;
-use serde_with::DurationSecondsWithFrac;
 use std::{
     collections::HashMap,
+    env,
     net::{Ipv4Addr, SocketAddr, SocketAddrV4},
     path::PathBuf,
     str::FromStr,
     time::Duration,
 };
-use tracing::warn;
 
-use alloy::primitives::Address;
+use bigdecimal::{BigDecimal, FromPrimitive, ToPrimitive};
 use bip39::Mnemonic;
+use figment::{
+    providers::{Env, Format, Toml},
+    Figment,
+};
 use regex::Regex;
 use serde::Deserialize;
-use serde_with::serde_as;
-use std::env;
-use thegraph_core::DeploymentId;
+use serde_repr::Deserialize_repr;
+use serde_with::{serde_as, DurationSecondsWithFrac};
+use thegraph_core::{alloy::primitives::Address, DeploymentId};
+use tracing::warn;
 use url::Url;
 
 use crate::NonZeroGRT;
@@ -423,15 +421,15 @@ pub struct RavRequestConfig {
 
 #[cfg(test)]
 mod tests {
-    use alloy::primitives::FixedBytes;
+    use std::{env, fs, path::PathBuf};
+
     use figment::value::Uncased;
     use sealed_test::prelude::*;
-    use std::{env, fs, path::PathBuf, str::FromStr};
+    use thegraph_core::alloy::primitives::address;
     use tracing_test::traced_test;
 
-    use crate::{Config, ConfigPrefix};
-
     use super::{DatabaseConfig, SHARED_PREFIX};
+    use crate::{Config, ConfigPrefix};
 
     #[test]
     fn test_minimal_config() {
@@ -451,9 +449,7 @@ mod tests {
         )
         .unwrap();
         max_config.dips = Some(crate::DipsConfig {
-            allowed_payers: vec![thegraph_core::Address(
-                FixedBytes::<20>::from_str("0x3333333333333333333333333333333333333333").unwrap(),
-            )],
+            allowed_payers: vec![address!("3333333333333333333333333333333333333333")],
             cancellation_time_tolerance: None,
         });
 

crates/config/src/config.rs

@@ -4,9 +4,8 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
-alloy.workspace = true 
 thiserror.workspace = true
 anyhow.workspace = true
 alloy-sol-types = "=0.8.15"
-alloy-rlp = "0.3.9" 
+alloy-rlp = "0.3.9"
 thegraph-core.workspace = true

crates/dips/Cargo.toml

@@ -1,17 +1,17 @@
 // Copyright 2023-, Edge & Node, GraphOps, and Semiotic Labs.
 // SPDX-License-Identifier: Apache-2.0
 
-use std::str::FromStr;
-use std::time::{Duration, SystemTime, UNIX_EPOCH};
-
-pub use alloy;
-pub use alloy_rlp;
-
-use alloy::core::primitives::Address;
-use alloy::rlp::{RlpDecodable, RlpEncodable};
-use alloy::{primitives::PrimitiveSignature as Signature, signers::SignerSync};
-use alloy_rlp::Encodable;
-use thegraph_core::alloy_sol_types::{sol, Eip712Domain, SolStruct};
+use std::{
+    str::FromStr,
+    time::{Duration, SystemTime, UNIX_EPOCH},
+};
+
+use alloy_rlp::{Encodable, RlpDecodable, RlpEncodable};
+use thegraph_core::alloy::{
+    primitives::{Address, PrimitiveSignature as Signature},
+    signers::SignerSync,
+    sol_types::{sol, Eip712Domain, SolStruct},
+};
 use thiserror::Error;
 
 sol! {
@@ -212,10 +212,14 @@ impl SignedCancellationRequest {
 mod test {
     use std::time::{Duration, SystemTime, UNIX_EPOCH};
 
-    use alloy::primitives::{Address, FixedBytes, U256};
-    use alloy::signers::local::PrivateKeySigner;
-    use alloy::sol_types::SolStruct;
-    use thegraph_core::attestation::eip712_domain;
+    use thegraph_core::{
+        alloy::{
+            primitives::{Address, FixedBytes, U256},
+            signers::local::PrivateKeySigner,
+            sol_types::SolStruct,
+        },
+        attestation::eip712_domain,
+    };
 
     use crate::{
         AgreementVoucherValidationError, CancellationRequest, CancellationRequestValidationError,