use super::{Node, NodeContext, NodeKind, NodeRequest, NodeResponse}; use crate::{config::DynNode, error::ServiceError}; use anyhow::Result; use futures::Future; use hyper::{ header::{HeaderName, HeaderValue}, HeaderMap, }; use serde::Deserialize; use std::{collections::BTreeMap, pin::Pin, str::FromStr, sync::Arc}; pub struct HeadersKind; #[derive(Deserialize)] pub struct Headers { headers: HeaderMapWrap, inner: DynNode, } impl NodeKind for HeadersKind { fn name(&self) -> &'static str { "headers" } fn instanciate(&self, config: serde_yaml::Value) -> Result> { Ok(Arc::new(serde_yaml::from_value::(config)?)) } } impl Node for Headers { fn handle<'a>( &'a self, context: &'a mut NodeContext, request: NodeRequest, ) -> Pin> + Send + Sync + 'a>> { Box::pin(async move { let mut resp = self.inner.handle(context, request).await?; resp.headers_mut() .extend(self.headers.0.clone().into_iter()); Ok(resp) }) } } struct HeaderMapWrap(HeaderMap); impl<'de> Deserialize<'de> for HeaderMapWrap { fn deserialize(deserializer: D) -> std::result::Result where D: serde::Deserializer<'de>, { let headers = BTreeMap::::deserialize(deserializer)?; let mut hm = HeaderMap::new(); for (k, v) in headers { hm.insert( HeaderName::from_str(&k).map_err(|e| serde::de::Error::custom(format!("{e}")))?, HeaderValue::from_str(&v).map_err(|e| serde::de::Error::custom(format!("{e}")))?, ); } Ok(HeaderMapWrap(hm)) } }