huber 0.3.7

Huber package, simplify `GitHub release` package management with a live awesome list of GitHub project
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 log::{debug, info};
use simpledi_rs::di::{DIContainer, DIContainerExtTrait, DependencyInjectTrait};

use huber_common::model::config::{Config, ConfigPath};
use huber_common::result::Result;

use crate::service::ServiceTrait;

lazy_static! {
    static ref DEFAULT_CONFIG: Config = Config::new();
}

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

unsafe impl Send for ConfigService {}

unsafe impl Sync for ConfigService {}

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

impl ConfigService {
    pub(crate) 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) -> Result<Config> {
        let path = DEFAULT_CONFIG.config_file()?;
        let config = self.container.get::<Config>().unwrap();

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

            let f = File::open(&path)?;
            Ok(serde_yaml::from_reader::<File, Config>(f)?)
        } else {
            debug!(
                "Getting the config from runtime, because {:?} does not exists",
                path
            );

            let c = Config {
                log_level: config.log_level.clone(),
                output_format: config.output_format.clone(),
                home_dir: config.home_dir.clone(),
                github_token: config.github_token.clone(),
                github_key: config.github_key.clone(),
            };
            Ok(c)
        }
    }

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

        info!("Updating the config {:?}", path);

        if path.exists() {
            let _ = remove_file(&path);
        }
        let f = File::create(&path)?;

        Ok(serde_yaml::to_writer(f, &config)?)
    }
}