atlas-cli 0.2.0

Machine Learning Lifecycle & Transparency Manager - Create and verify manifests for ML models and datasets
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use crate::error::{Error, Result};
use crate::storage::traits::{ManifestMetadata, StorageBackend};
use atlas_c2pa_lib::manifest::Manifest;
use reqwest;

#[allow(dead_code)]
pub struct RekorStorage {
    client: reqwest::blocking::Client,
    base_url: String,
}

impl RekorStorage {
    pub fn new() -> Result<Self> {
        Self::new_with_url("https://rekor.sigstore.dev".to_string())
    }

    pub fn new_with_url(url: String) -> Result<Self> {
        Ok(RekorStorage {
            client: reqwest::blocking::Client::new(),
            base_url: url,
        })
    }
}

impl StorageBackend for RekorStorage {
    fn get_base_uri(&self) -> String {
        self.base_url.clone()
    }

    fn store_manifest(&self, _manifest: &Manifest) -> Result<String> {
        // TODO: Implement actual storage
        println!("Would store manifest at: {}", self.base_url);
        Ok("dummy-manifest-id".to_string()) // Return a dummy manifest ID for now
    }

    fn retrieve_manifest(&self, _id: &str) -> Result<Manifest> {
        todo!("Implement manifest retrieval")
    }

    fn list_manifests(&self) -> Result<Vec<ManifestMetadata>> {
        todo!("Implement manifest listing")
    }
    fn delete_manifest(&self, _id: &str) -> Result<()> {
        // Implement deletion for Rekor if supported, or return error
        Err(Error::Storage(
            "Delete operation not supported for Rekor storage".to_string(),
        ))
    }
    fn as_any(&self) -> &dyn std::any::Any {
        self
    }
}