carbem 0.5.0

A Rust library for retrieving carbon emission values from cloud providers
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135

//! Provider Registry Pattern for dynamic provider management

use std::collections::HashMap;

use crate::error::{CarbemError, Result};
use crate::providers::CarbonProvider;
use crate::providers::azure::{AzureConfig, AzureProvider};
use crate::providers::ibm::{IbmConfig, IbmProvider};

/// Type alias for provider factory functions
type ProviderFactory =
    Box<dyn Fn(serde_json::Value) -> Result<Box<dyn CarbonProvider + Send + Sync>> + Send + Sync>;

/// Registry for carbon emission providers
pub struct ProviderRegistry {
    factories: HashMap<String, ProviderFactory>,
}

impl ProviderRegistry {
    /// Create a new provider registry
    pub fn new() -> Self {
        let mut registry = Self {
            factories: HashMap::new(),
        };

        // Register built-in providers
        registry.register_azure();
        registry.register_ibm();

        registry
    }

    /// Register Azure provider factory
    fn register_azure(&mut self) {
        let factory: ProviderFactory = Box::new(|config_json| {
            let config: AzureConfig = serde_json::from_value(config_json)
                .map_err(|e| CarbemError::Config(format!("Invalid Azure config: {}", e)))?;

            let provider = AzureProvider::new(config)?;
            Ok(Box::new(provider) as Box<dyn CarbonProvider + Send + Sync>)
        });

        self.factories.insert("azure".to_string(), factory);
    }

    /// Register IBM provider factory
    fn register_ibm(&mut self) {
        let factory: ProviderFactory = Box::new(|config_json| {
            let config: IbmConfig = serde_json::from_value(config_json)
                .map_err(|e| CarbemError::Config(format!("Invalid IBM config: {}", e)))?;

            let provider = IbmProvider::new(config)?;
            Ok(Box::new(provider) as Box<dyn CarbonProvider + Send + Sync>)
        });

        self.factories.insert("ibm".to_string(), factory);
    }

    /// Register a custom provider factory
    pub fn register_provider<F>(&mut self, name: &str, factory: F)
    where
        F: Fn(serde_json::Value) -> Result<Box<dyn CarbonProvider + Send + Sync>>
            + Send
            + Sync
            + 'static,
    {
        self.factories.insert(name.to_string(), Box::new(factory));
    }

    /// Create a provider instance
    pub fn create_provider(
        &self,
        name: &str,
        config: serde_json::Value,
    ) -> Result<Box<dyn CarbonProvider + Send + Sync>> {
        let factory = self
            .factories
            .get(name)
            .ok_or_else(|| CarbemError::UnsupportedProvider(name.to_string()))?;

        factory(config)
    }

    /// Get list of available providers
    pub fn available_providers(&self) -> Vec<String> {
        self.factories.keys().cloned().collect()
    }
}

impl Default for ProviderRegistry {
    fn default() -> Self {
        Self::new()
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use serde_json::json;

    #[test]
    fn test_registry_creation() {
        let registry = ProviderRegistry::new();
        let providers = registry.available_providers();
        assert!(providers.contains(&"azure".to_string()));
        assert!(providers.contains(&"ibm".to_string()));
    }

    #[test]
    fn test_azure_provider_creation() {
        let registry = ProviderRegistry::new();
        let config = json!({"access_token": "test-token"});

        let result = registry.create_provider("azure", config);
        assert!(result.is_ok());
    }

    #[test]
    fn test_ibm_provider_creation() {
        let registry = ProviderRegistry::new();
        let config = json!({"api_key": "test-api-key"});

        let result = registry.create_provider("ibm", config);
        assert!(result.is_ok());
    }

    #[test]
    fn test_unknown_provider() {
        let registry = ProviderRegistry::new();
        let config = json!({});

        let result = registry.create_provider("unknown", config);
        assert!(result.is_err());
    }
}