refactor: create inject_deployment middleware

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

Author: gusinacio

Cargo.lock

@@ -40,7 +40,7 @@ tap_core.workspace = true
 uuid.workspace = true
 alloy.workspace = true
 tower_governor = "0.4.0"
-tower-http = { version = "0.5.2", features = [
+tower-http = { version = "0.6.2", features = [
   "auth",
   "cors",
   "normalize-path",
@@ -57,6 +57,9 @@ bip39.workspace = true
 [dev-dependencies]
 hex-literal = "0.4.1"
 test-assets = { path = "../test-assets" }
+tower-test = "0.4.0"
+tower = "0.5.1"
+tokio-test = "0.4.4"
 
 [build-dependencies]
 build-info-build = { version = "0.0.39", default-features = false }

crates/service/Cargo.toml

@@ -4,6 +4,7 @@
 mod cli;
 mod database;
 mod error;
+mod middleware;
 mod routes;
 pub mod service;
 mod tap;

crates/service/src/lib.rs

@@ -0,0 +1,4 @@
+// Copyright 2023-, Edge & Node, GraphOps, and Semiotic Labs.
+// SPDX-License-Identifier: Apache-2.0
+
+mod inject_deployment;

crates/service/src/middleware.rs

@@ -0,0 +1,61 @@
+// Copyright 2023-, Edge & Node, GraphOps, and Semiotic Labs.
+// SPDX-License-Identifier: Apache-2.0
+
+use axum::{
+    extract::{Path, Request},
+    middleware::Next,
+    response::Response,
+    RequestExt,
+};
+use thegraph_core::DeploymentId;
+
+pub async fn deployment_middleware(mut request: Request, next: Next) -> Response {
+    let deployment_id = request.extract_parts::<Path<DeploymentId>>().await.ok();
+    if let Some(Path(deployment_id)) = deployment_id {
+        request.extensions_mut().insert(deployment_id);
+    }
+    next.run(request).await
+}
+
+#[cfg(test)]
+mod tests {
+    use super::deployment_middleware;
+    use axum::{
+        body::Body,
+        http::{Extensions, Request},
+        middleware::from_fn,
+        routing::get,
+        Router,
+    };
+    use reqwest::StatusCode;
+    use test_assets::ESCROW_SUBGRAPH_DEPLOYMENT;
+    use thegraph_core::DeploymentId;
+    use tower::ServiceExt;
+
+    #[tokio::test]
+    async fn test_deployment_middleware() {
+        let middleware = from_fn(deployment_middleware);
+
+        async fn handle(extensions: Extensions) -> Body {
+            extensions
+                .get::<DeploymentId>()
+                .expect("Should contain a deployment_id");
+            Body::empty()
+        }
+
+        let app = Router::new()
+            .route("/:deployment_id", get(handle))
+            .layer(middleware);
+
+        let res = app
+            .oneshot(
+                Request::builder()
+                    .uri(format!("/{}", *ESCROW_SUBGRAPH_DEPLOYMENT))
+                    .body(Body::empty())
+                    .unwrap(),
+            )
+            .await
+            .unwrap();
+        assert_eq!(res.status(), StatusCode::OK);
+    }
+}