73 lines
2 KiB
Rust
73 lines
2 KiB
Rust
use crate::policy::{Action, Policy};
|
|
|
|
use std::{io::Read, net::SocketAddr, str::FromStr};
|
|
|
|
pub struct Config {
|
|
pub listen_addr: SocketAddr,
|
|
pub pass_addr: SocketAddr,
|
|
pub default_action: Action,
|
|
pub challenge_timeout: u64,
|
|
pub policy_groups: Vec<Vec<Policy>>,
|
|
}
|
|
|
|
impl Config {
|
|
pub fn from_file(path: &str) -> Self {
|
|
let mut config_file = std::fs::File::open(path).expect("Cannot open config file");
|
|
let mut config_str = String::new();
|
|
config_file
|
|
.read_to_string(&mut config_str)
|
|
.expect("Cannot read config file");
|
|
let config_yaml = saphyr::Yaml::load_from_str(&config_str).expect("Error parsing config");
|
|
let config_doc = &config_yaml[0];
|
|
|
|
Config {
|
|
listen_addr: config_doc["listen"]
|
|
.as_str()
|
|
.expect("Missing listen address in config")
|
|
.parse()
|
|
.expect("Invalid listen address"),
|
|
pass_addr: config_doc["pass"]
|
|
.as_str()
|
|
.expect("Missing pass address in config")
|
|
.parse()
|
|
.expect("Invalid pass address"),
|
|
challenge_timeout: config_doc["challenge-timeout"]
|
|
.as_i64()
|
|
.expect("Missing challenge timeout in config") as u64,
|
|
default_action: Action::from_str(
|
|
config_doc["default-action"]
|
|
.as_str()
|
|
.expect("Missing default action in config"),
|
|
)
|
|
.expect("Invalid default action"),
|
|
policy_groups: config_doc["policy-groups"]
|
|
.as_vec()
|
|
.expect("Missing policies in config")
|
|
.iter()
|
|
.map(|policy_group| {
|
|
policy_group
|
|
.as_vec()
|
|
.expect("Missing policies in config")
|
|
.iter()
|
|
.map(|policy| Policy {
|
|
name: policy["name"]
|
|
.as_str()
|
|
.expect("Expected policy name string")
|
|
.to_string(),
|
|
first_line_regex: policy["first-line"]
|
|
.as_str()
|
|
.expect("Expected policy first line regex string")
|
|
.to_string(),
|
|
action: Action::from_str(
|
|
policy["action"]
|
|
.as_str()
|
|
.expect("Expected policy action string"),
|
|
)
|
|
.expect("Invalid policy action"),
|
|
})
|
|
.collect()
|
|
})
|
|
.collect(),
|
|
}
|
|
}
|
|
}
|