feat: adds namespace header (#41)

Author: Sbcdn

src/api.rs

@@ -136,6 +136,7 @@ pub struct EndpointMiddleware {
     pub token: String,
     pub version: String,
     pub wrap: Option<String>,
+    pub namespace: Option<String>,
 }
 impl MiddleWare for EndpointMiddleware {
     fn request<E: Endpoint>(
@@ -174,6 +175,15 @@ impl MiddleWare for EndpointMiddleware {
             );
         }
 
+        // Optionally wrap response
+        if let Some(namespace) = &self.namespace {
+            info!("Middleware: adding namespace header {}", namespace);
+            req.headers_mut().append(
+                "X-Vault-Namespace",
+                http::HeaderValue::from_str(namespace.as_str()).unwrap(),
+            );
+        }
+
         Ok(())
     }
 

src/client.rs

@@ -120,6 +120,7 @@ impl VaultClient {
             token: settings.token.clone(),
             version: version_str,
             wrap: None,
+            namespace: settings.namespace.clone(),
         };
 
         let http_client = http_client
@@ -163,6 +164,8 @@ pub struct VaultClientSettings {
     pub version: u8,
     #[builder(default = "false")]
     pub wrapping: bool,
+    #[builder(default)]
+    pub namespace: Option<String>,
 }
 
 impl VaultClientSettingsBuilder {
@@ -184,6 +187,11 @@ impl VaultClientSettingsBuilder {
         self
     }
 
+    pub fn set_namespace(&mut self, str: String) -> &mut Self {
+        self.namespace = Some(Some(str));
+        self
+    }
+
     fn default_address(&self) -> Result<Url, String> {
         let address = if let Ok(address) = env::var("VAULT_ADDR") {
             info!("Using vault address from $VAULT_ADDR: {address}");