huber 1.0.6

Huber, simplify GitHub package management
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

use std::fs::{remove_file, File};
use std::sync::Arc;

use anyhow::anyhow;
use huber_common::model::config::{Config, ConfigPath};
use lazy_static::lazy_static;
use log::debug;
use simpledi_rs::di::{DIContainer, DIContainerExtTrait, DependencyInjectTrait};

use crate::error::HuberError;
use crate::service::ServiceTrait;

lazy_static! {
    pub static ref DEFAULT_CONFIG: Config = Default::default();
}

#[derive(Debug)]
pub struct ConfigService {
    pub container: Option<Arc<DIContainer>>,
}

unsafe impl Send for ConfigService {}

unsafe impl Sync for ConfigService {}

pub trait ConfigTrait {
    fn get(&self) -> anyhow::Result<Config>;
    fn update(&self, config: &Config) -> anyhow::Result<()>;
}

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

impl ConfigService {
    pub fn new() -> Self {
        Self { container: None }
    }
}

impl ServiceTrait for ConfigService {}

impl DependencyInjectTrait for ConfigService {
    fn inject(&mut self, container: Arc<DIContainer>) {
        self.container = Some(container);
    }
}

impl ConfigTrait for ConfigService {
    fn get(&self) -> anyhow::Result<Config> {
        let config = self
            .container
            .get::<Config>()
            .expect("Failed to find config");
        let config_path = config.config_file()?;

        if config_path.exists() {
            debug!("Getting the config from {:?}", config_path);

            let f = File::open(&config_path)?;
            return Ok(serde_yaml::from_reader::<File, Config>(f)?);
        }

        Err(anyhow!(HuberError::ConfigNotFound(
            config_path.to_string_lossy().to_string()
        )))
    }

    fn update(&self, config: &Config) -> anyhow::Result<()> {
        let path = config.config_file()?;

        debug!("Updating the config {:?}: {:?}", path, config);
        if path.exists() {
            let _ = remove_file(&path);
        }
        let f = File::create(&path)?;
        serde_yaml::to_writer(f, &config)?;

        Ok(())
    }
}