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

use std::env;
static ENVIRONMENT_VAR:   &'static str = "RUST_IOT_ENVIRONMENT";
pub static PRODUCTION_STRING: &'static str =  "PRODUCTION" ;
pub static TEST_STRING:       &'static str =  "TEST";

/**
 * Get the current environment, PRODUCTION or TEST
 */
pub fn get_current_env() -> String {
    env::var(ENVIRONMENT_VAR).unwrap_or_else(|_| {println!("environment setting not found, using test environment"); return TEST_STRING.to_owned()})
}

/**
 * Database connection constants.
 */
struct DbVars {
    host_var: &'static str,
    db_var: &'static str , 
    user_var: &'static str ,
    pass_var: &'static str
}

impl DbVars {
    fn new_test() -> DbVars {
        DbVars{ 
            host_var: "SQL_HOST_TEST",
            db_var: "SQL_DB_NAME_TEST",
            user_var: "SQL_USERNAME_TEST",
            pass_var: "SQL_PASSWORD_TEST"
        }   
    }
    fn new_prod() -> DbVars {
        DbVars {
            host_var: "SQL_HOST",
            db_var: "SQL_DB_NAME",
            user_var: "SQL_USERNAME",
            pass_var: "SQL_PASSWORD"   
        }
    }
}

/**
 * Database connection environment vars
 */
pub struct DbValues {
    pub host: String,
    pub db_name: String,
    pub user: String,
    pub pass: String
}

/**
 * Returns the current database variables from the environment
 */
pub fn db_environment_values() -> DbValues {
    let cur_env_val = get_current_env();
    let is_production = String::from(PRODUCTION_STRING) == cur_env_val;
    let current_vars = match is_production {
        false => DbVars::new_test(),
        true =>  DbVars::new_prod()
    };

    let host = env::var(current_vars.host_var ).unwrap_or_else(|_| panic!(format!("Error reading environment variable {}", current_vars.host_var)));
    let db_name = env::var(current_vars.db_var).unwrap_or_else(|_| panic!(format!("Error reading environment variable {}", current_vars.db_var  )));
    let user = env::var(current_vars.user_var ).unwrap_or_else(|_| panic!(format!("Error reading environment variable {}", current_vars.user_var)));
    let pass = env::var(current_vars.pass_var ).unwrap_or_else(|_| panic!(format!("Error reading environment variable {}", current_vars.pass_var)));

    DbValues {
        host: host,
        db_name: db_name,
        user: user,
        pass: pass
    }
}