mirror of
https://gitlab.com/comunic/comunicapiv3
synced 2025-06-20 16:35:17 +00:00
Parse configuration
This commit is contained in:
97
src/data/config.rs
Normal file
97
src/data/config.rs
Normal file
@ -0,0 +1,97 @@
|
||||
use std::error::Error;
|
||||
|
||||
use yaml_rust::{Yaml, YamlLoader};
|
||||
|
||||
/// Server configuration
|
||||
///
|
||||
/// @author Pierre Hubert
|
||||
#[derive(Debug)]
|
||||
pub struct DatabaseConfig {
|
||||
pub host: String,
|
||||
pub name: String,
|
||||
pub username: String,
|
||||
pub password: String,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct Config {
|
||||
pub port: i32,
|
||||
pub listen_address: String,
|
||||
pub storage_url: String,
|
||||
pub storage_path: String,
|
||||
pub database: DatabaseConfig,
|
||||
pub proxy: Option<String>,
|
||||
pub force_https: bool,
|
||||
}
|
||||
|
||||
/// Globally available configuration
|
||||
static mut CONF: Option<Config> = None;
|
||||
|
||||
impl Config {
|
||||
fn yaml_i32(parsed: &Yaml, name: &str) -> i32 {
|
||||
parsed[name].as_i64().expect(format!("{} is missing (int)", name).as_str()) as i32
|
||||
}
|
||||
|
||||
fn yaml_str(parsed: &Yaml, name: &str) -> String {
|
||||
parsed[name].as_str().expect(format!("{} is missing (str)", name).as_str()).to_string()
|
||||
}
|
||||
|
||||
fn yaml_bool(parsed: &Yaml, name: &str) -> bool {
|
||||
parsed[name].as_bool().expect(format!("{} is missing (bool)", name).as_str())
|
||||
}
|
||||
|
||||
/// Load the configuration from a given path
|
||||
pub fn load(path: &str) -> Result<(), Box<dyn Error>> {
|
||||
|
||||
// Read & parse configuration
|
||||
let conf_str = std::fs::read_to_string(path)?;
|
||||
let parsed = YamlLoader::load_from_str(&conf_str)?;
|
||||
|
||||
if parsed.len() != 1 {
|
||||
panic!("parsed.len() != 1");
|
||||
}
|
||||
let parsed = &parsed[0];
|
||||
|
||||
// Read configuration
|
||||
let parsed_db = &parsed["database"];
|
||||
let database_conf = DatabaseConfig {
|
||||
host: Config::yaml_str(parsed_db, "host"),
|
||||
name: Config::yaml_str(parsed_db, "name"),
|
||||
username: Config::yaml_str(parsed_db, "username"),
|
||||
password: Config::yaml_str(parsed_db, "password"),
|
||||
};
|
||||
|
||||
let proxy = Config::yaml_str(parsed, "proxy");
|
||||
|
||||
let conf = Config {
|
||||
port: Config::yaml_i32(parsed, "server-port") as i32,
|
||||
listen_address: Config::yaml_str(parsed, "server-address"),
|
||||
|
||||
storage_url: Config::yaml_str(parsed, "storage-url"),
|
||||
storage_path: Config::yaml_str(parsed, "storage-path"),
|
||||
|
||||
database: database_conf,
|
||||
|
||||
proxy: match proxy.as_ref() {
|
||||
"none" => None,
|
||||
s => Some(s.to_string())
|
||||
},
|
||||
|
||||
force_https: Config::yaml_bool(parsed, "force-https"),
|
||||
};
|
||||
|
||||
// Save new configuration in memory
|
||||
unsafe {
|
||||
CONF = Some(conf);
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Get an instance of the configuration
|
||||
pub fn conf() -> &'static Config {
|
||||
unsafe {
|
||||
return &CONF.as_ref().unwrap();
|
||||
}
|
||||
}
|
1
src/data/mod.rs
Normal file
1
src/data/mod.rs
Normal file
@ -0,0 +1 @@
|
||||
pub mod config;
|
1
src/lib.rs
Normal file
1
src/lib.rs
Normal file
@ -0,0 +1 @@
|
||||
pub mod data;
|
@ -1,3 +1,11 @@
|
||||
use comunic_server::data::config::{conf, Config};
|
||||
|
||||
fn main() {
|
||||
|
||||
// Load configuration
|
||||
Config::load("config.yaml").expect("Could not load configuration!");
|
||||
|
||||
println!("{:#?}", conf());
|
||||
|
||||
println!("Hello, world!");
|
||||
}
|
||||
|
Reference in New Issue
Block a user