add support for cbom and use enum to define mapping

Author: dejanb

modules/ingestor/src/service/sbom/cyclonedx.rs

@@ -5,6 +5,7 @@ use crate::{
 };
 use sea_orm::TransactionTrait;
 use serde_cyclonedx::cyclonedx::v_1_6::Component;
+use std::str::FromStr;
 use tracing::instrument;
 use trustify_common::{hashing::Digests, id::Id};
 use trustify_entity::labels::Labels;
@@ -76,17 +77,43 @@ impl<'g> CyclonedxLoader<'g> {
     }
 }
 
+enum Kind {
+    AIBOM,
+    CBOM,
+}
+
+impl Kind {
+    fn as_str(&self) -> &'static str {
+        match self {
+            Kind::AIBOM => "aibom",
+            Kind::CBOM => "cbom",
+        }
+    }
+}
+
+impl FromStr for Kind {
+    type Err = ();
+
+    fn from_str(s: &str) -> Result<Self, Self::Err> {
+        match s {
+            "machine-learning-model" => Ok(Kind::AIBOM),
+            "cryptographic-asset" => Ok(Kind::CBOM),
+            _ => Err(()),
+        }
+    }
+}
+
 fn extract_labels(components: Option<&Vec<Component>>, labels_in: Labels) -> Labels {
     let mut labels = Labels::new().add("type", "cyclonedx");
 
-    // find if there are machine learning model components in the SBOM
-    if let Some(components) = components
-        && components
-            .iter()
-            .any(|c| c.type_ == "machine-learning-model")
-    {
-        labels = labels.add("ai", "machine-learning-model");
+    if let Some(components) = components {
+        for component in components {
+            if let Ok(kind) = Kind::from_str(&component.type_) {
+                labels = labels.add("kind", kind.as_str());
+            }
+        }
     }
+
     if !labels_in.is_empty() {
         return labels.extend(labels_in.0.clone());
     }