feat: add token provider to v11y client

trustification · Sep 7, 2023 · 8f788bd · 8f788bd
1 parent 1516ff6
commit 8f788bd
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/collector/osv/src/lib.rs b/collector/osv/src/lib.rs
@@ -85,7 +85,7 @@ impl AppState {
             addr: RwLock::new(None),
             connected: AtomicBool::new(false),
             collectorist_client: collectorist_client::CollectoristClient::new(collector_id, collectorist_url),
-            v11y_client: v11y_client::V11yClient::new(v11y_url),
+            v11y_client: v11y_client::V11yClient::new(v11y_url, ()),
             guac_url: RwLock::new(None),
         }
     }

diff --git a/v11y/client/Cargo.toml b/v11y/client/Cargo.toml
@@ -14,9 +14,10 @@ humantime-serde = "1.1.1"
 #clap = { version = "4", features = ["derive"] }
 reqwest = "0.11.18"
 anyhow = "1"
-chrono = { version = "0.4.26", features = [ "serde"] }
+chrono = { version = "0.4.26", features = ["serde"] }
 utoipa = { version = "3", features = ["actix_extras"] }
 
+trustification-auth = { path = "../../auth" }
 
 [dev-dependencies]
 tokio = "*"

diff --git a/v11y/client/src/lib.rs b/v11y/client/src/lib.rs
@@ -5,6 +5,7 @@ use std::hash::{Hash, Hasher};
 use chrono::{DateTime, Utc};
 use reqwest::Url;
 use serde::{Deserialize, Serialize};
+use trustification_auth::client::{TokenInjector, TokenProvider};
 use utoipa::ToSchema;
 
 #[derive(Serialize, Deserialize, Debug, ToSchema)]
@@ -170,18 +171,25 @@ impl V11yUrl {
 #[allow(unused)]
 pub struct V11yClient {
     v11y_url: V11yUrl,
+    provider: Box<dyn TokenProvider>,
 }
 
 impl V11yClient {
-    pub fn new(url: Url) -> Self {
+    pub fn new<P: TokenProvider>(url: Url, provider: P) -> Self
+    where
+        P: TokenProvider + 'static,
+    {
         Self {
             v11y_url: V11yUrl::new(url),
+            provider: Box::new(provider),
         }
     }
 
     pub async fn ingest_vulnerability(&self, vuln: &Vulnerability) -> Result<(), anyhow::Error> {
         Ok(reqwest::Client::new()
             .post(self.v11y_url.vulnerability_url())
+            .inject_token(self.provider.as_ref())
+            .await?
             .json(&vuln)
             .send()
             .await