Merge pull request #5 from graphops/chris/x-gateway-id

feat: add support for x-gateway-id

Author: sahra-karakoc

Cargo.lock

@@ -24,3 +24,4 @@ axum = "0.8.4"
 tokio-util = { version = "0.7.15" }
 http = "1.3.1"
 tracing = "0.1"
+once_cell = "1.20"

Cargo.toml

@@ -1,8 +1,21 @@
 // SPDX-License-Identifier: Apache-2.0
-pub const GATEWAY_URL: &str = "https://gateway.thegraph.com/api";
+use once_cell::sync::Lazy;
+use std::collections::HashMap;
+
 pub const GRAPH_NETWORK_SUBGRAPH_ARBITRUM: &str = "QmdKXcBUHR3UyURqVRQHu1oV6VUkBrhi2vNvMx3bNDnUCc";
 pub const GATEWAY_QOS_ORACLE: &str = "QmZmb6z87QmqBLmkMhaqWy7h2GLF1ey8Qj7YSRuqSGMjeH";
 
+// Gateway ID to URL mapping
+pub static GATEWAY_REGISTRY: Lazy<HashMap<&'static str, &'static str>> = Lazy::new(|| {
+    let mut m = HashMap::new();
+    m.insert("edgeandnode", "https://gateway.thegraph.com/api");
+    m.insert("graphops", "https://graph-gateway.graphops.xyz/api");
+    m
+});
+
+// Default gateway ID
+pub const DEFAULT_GATEWAY_ID: &str = "edgeandnode";
+
 pub const SUBGRAPH_SERVER_INSTRUCTIONS: &str = "**Interacting with The Graph Subgraphs**
 **IMPORTANT: ALWAYS verify query volumes using `get_deployment_30day_query_counts` for any potential subgraph candidate *before* selecting or querying it. This step is NON-OPTIONAL. Failure to do so may result in using outdated or irrelevant data.**
 **Follow this sequence strictly:**

src/constants.rs

@@ -13,4 +13,6 @@ pub enum SubgraphError {
     JsonError(#[from] serde_json::Error),
     #[error("Internal processing error: {0}")]
     InternalProcessingError(String),
+    #[error("{0}")]
+    InvalidGatewayId(String),
 }