fix: fixup for --devmode

trustification · Oct 10, 2023 · dc3d655 · dc3d655
1 parent 876b916
commit dc3d655
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 1 deletion.
diff --git a/collectorist/api/src/config.rs b/collectorist/api/src/config.rs
@@ -2,13 +2,45 @@ use reqwest::Url;
 use serde::{Deserialize, Serialize};
 use std::collections::HashMap;
 use std::time::Duration;
+use trustification_infrastructure::endpoint::{self, Endpoint};
 
 #[derive(Serialize, Deserialize, Clone, Debug)]
 #[serde(rename_all = "kebab-case")]
 pub struct CollectorsConfig {
     pub collectors: HashMap<String, CollectorConfig>,
 }
 
+impl CollectorsConfig {
+    pub fn devmode() -> Self {
+        let mut collectors = HashMap::new();
+        collectors.insert(
+            "osv".into(),
+            CollectorConfig {
+                url: endpoint::CollectorOsv::url(),
+                interests: vec![Interest::Package, Interest::Vulnerability],
+                cadence: default_cadence(),
+            },
+        );
+        collectors.insert(
+            "snyk".into(),
+            CollectorConfig {
+                url: endpoint::CollectorSnyk::url(),
+                interests: vec![Interest::Package],
+                cadence: default_cadence(),
+            },
+        );
+        collectors.insert(
+            "nvd".into(),
+            CollectorConfig {
+                url: endpoint::CollectorNvd::url(),
+                interests: vec![Interest::Vulnerability],
+                cadence: default_cadence(),
+            },
+        );
+        Self { collectors }
+    }
+}
+
 #[derive(Serialize, Deserialize, Copy, Clone, Debug, PartialEq)]
 #[serde(rename_all = "kebab-case")]
 pub enum Interest {

diff --git a/collectorist/api/src/lib.rs b/collectorist/api/src/lib.rs
@@ -96,6 +96,7 @@ impl Run {
                         self.collector_config,
                         self.csub_url,
                         provider,
+                        self.devmode,
                     )
                     .await?;
 
@@ -142,6 +143,7 @@ impl Run {
         collector_config: Option<PathBuf>,
         csub_url: Url,
         provider: P,
+        devmode: bool,
     ) -> anyhow::Result<Arc<AppState>>
     where
         P: TokenProvider + Clone + 'static,
@@ -150,11 +152,14 @@ impl Run {
 
         let collectorist_config = if collectorist_config.exists() {
             serde_yaml::from_reader(File::open(collectorist_config)?)?
+        } else if devmode {
+            CollectorsConfig::devmode()
         } else {
-            log::info!(
+            log::error!(
                 "configuration file {} missing, no collectors configured",
                 collectorist_config.to_str().unwrap_or("<unknown>")
             );
+
             CollectorsConfig {
                 collectors: Default::default(),
             }