added cookie auth to RemoteStateService

Author: idky137

Cargo.lock

@@ -1863,9 +1863,9 @@ dependencies = [
 
 [[package]]
 name = "http"
-version = "1.2.0"
+version = "1.3.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f16ca2af56261c99fba8bac40a10251ce8188205a4c448fbb745a2e4daa76fea"
+checksum = "f4a85d31aea989eead29a3aaf9e1115a180df8282431156e533de47660892565"
 dependencies = [
  "bytes",
  "fnv",
@@ -6422,6 +6422,7 @@ dependencies = [
  "hex",
  "hex-literal",
  "howudoin",
+ "http",
  "human_bytes",
  "humantime-serde",
  "indexmap 2.7.1",

Cargo.toml

@@ -67,6 +67,7 @@ halo2 = "0.3.0"
 hex = "0.4.3"
 hex-literal = "0.4.1"
 howudoin = "0.1.2"
+http = "1.3.1"
 http-body-util = "0.1.2"
 human_bytes = { version = "0.4.3", default-features = false }
 humantime = "2.1.0"

zebra-state/Cargo.toml

@@ -51,6 +51,7 @@ elasticsearch = [
 # Remote support for the [`ReadStateService`].
 remote_read_state_service = [
     "dep:url", 
+    "dep:http",
     "dep:jsonrpsee", 
     "jsonrpsee/http-client",
     "jsonrpsee/client-core",
@@ -86,6 +87,7 @@ tracing = { workspace = true }
 
 # remote_read_state_service specific dependencies.
 url = { workspace = true, optional = true }
+http = { workspace = true, optional = true }
 jsonrpsee = { workspace = true, optional = true }
 
 # elasticsearch specific dependencies.

zebra-state/src/service.rs

@@ -2168,11 +2168,14 @@ pub fn init_test_services(
 /// Holds functionality for remote use of the [`ReadStateService`] ([`RemoteStateService`]).
 #[cfg(feature = "remote_read_state_service")]
 pub mod remote {
+    use std::fs;
+
     use super::*;
 
+    use http::header::COOKIE;
     use jsonrpsee::{
         core::client::ClientT,
-        http_client::{HttpClient, HttpClientBuilder},
+        http_client::{HeaderMap, HttpClient, HttpClientBuilder},
     };
     use url::Url;
 
@@ -2199,16 +2202,28 @@ pub mod remote {
     }
 
     impl RemoteStateService {
-        /// Creates a new remote, read-only state service client.
+        /// Creates a new remote, read-only state service client with optional cookie authentication.
         #[allow(dead_code)]
-        pub fn new(url: Url) -> Result<Self, BoxError> {
-            let client = HttpClientBuilder::default()
-                .request_timeout(Duration::from_secs(5))
-                .build(url.as_str())?;
-
-            let remote_read_state_service = Self { client };
+        pub fn new(url: Url, cookie: Option<String>) -> Result<Self, BoxError> {
+            let mut builder = HttpClientBuilder::default().request_timeout(Duration::from_secs(5));
+
+            if let Some(cookie_path) = cookie {
+                let cookie_content =
+                    fs::read_to_string(&cookie_path).map_err(|e| Box::new(e) as BoxError)?;
+                let cookie_content = cookie_content.trim();
+
+                let mut headers = HeaderMap::new();
+                headers.insert(
+                    COOKIE,
+                    cookie_content
+                        .parse()
+                        .map_err(|e| Box::new(e) as BoxError)?,
+                );
+                builder = builder.set_headers(headers);
+            }
 
-            Ok(remote_read_state_service)
+            let client = builder.build(url.as_str())?;
+            Ok(Self { client })
         }
 
         /// Sends a request to the server an waits on a response.