mirror of
https://github.com/Start9Labs/start-os.git
synced 2026-04-04 06:19:44 +00:00
018b7e972c
* account for no dependent config when auto configuring * fix config path * add check for config file to dependency check command
96 lines
3.1 KiB
Rust
96 lines
3.1 KiB
Rust
use std::borrow::Cow;
|
|
use std::collections::BTreeMap;
|
|
use std::path::Path;
|
|
|
|
use beau_collector::BeauCollector;
|
|
use embassy::config::action::SetResult;
|
|
use embassy::config::{spec, Config};
|
|
use linear_map::LinearMap;
|
|
|
|
pub mod rules;
|
|
|
|
use anyhow::anyhow;
|
|
pub use rules::{ConfigRuleEntry, ConfigRuleEntryWithSuggestions};
|
|
|
|
pub fn validate_configuration(
|
|
name: &str,
|
|
config: Config,
|
|
rules_path: &Path,
|
|
config_path: &Path,
|
|
) -> Result<SetResult, anyhow::Error> {
|
|
let rules: Vec<ConfigRuleEntry> = serde_yaml::from_reader(std::fs::File::open(rules_path)?)?;
|
|
let mut cfgs = LinearMap::new();
|
|
cfgs.insert(name, Cow::Borrowed(&config));
|
|
let rule_check = rules
|
|
.into_iter()
|
|
.map(|r| r.check(&config, &cfgs))
|
|
.bcollect::<Vec<_>>();
|
|
match rule_check {
|
|
Ok(_) => {
|
|
// create temp config file
|
|
serde_yaml::to_writer(
|
|
std::fs::File::create(config_path.with_extension("tmp"))?,
|
|
&config,
|
|
)?;
|
|
std::fs::rename(config_path.with_extension("tmp"), config_path)?;
|
|
// return set result
|
|
Ok(SetResult {
|
|
depends_on: BTreeMap::new(),
|
|
// sending sigterm so service is restarted - in 0.3.x services, this is whatever signal is needed to send to the process to pick up the configuration
|
|
signal: Some(nix::sys::signal::SIGTERM),
|
|
})
|
|
}
|
|
Err(e) => Err(anyhow!("{}", e)),
|
|
}
|
|
}
|
|
|
|
pub fn validate_dependency_configuration(
|
|
name: &str,
|
|
config: &Option<Config>,
|
|
parent_name: &str,
|
|
parent_config: Config,
|
|
rules_path: &Path,
|
|
) -> Result<(), anyhow::Error> {
|
|
let rules: Vec<ConfigRuleEntry> = serde_yaml::from_reader(std::fs::File::open(rules_path)?)?;
|
|
let mut cfgs = LinearMap::new();
|
|
cfgs.insert(parent_name, Cow::Borrowed(&parent_config));
|
|
if let Some(config) = config {
|
|
cfgs.insert(name, Cow::Borrowed(&config))
|
|
} else {
|
|
cfgs.insert(name, Cow::Owned(serde_json::Map::new()))
|
|
};
|
|
let rule_check = rules
|
|
.into_iter()
|
|
.map(|r| r.check(&parent_config, &cfgs))
|
|
.bcollect::<Vec<_>>();
|
|
match rule_check {
|
|
Ok(_) => Ok(()),
|
|
Err(e) => Err(anyhow!("{}", e)),
|
|
}
|
|
}
|
|
|
|
pub fn apply_dependency_configuration(
|
|
package_id: &str,
|
|
config: Option<Config>,
|
|
dependency_id: &str,
|
|
mut dep_config: Config,
|
|
rules_path: &Path,
|
|
) -> Result<Config, anyhow::Error> {
|
|
let rules: Vec<ConfigRuleEntryWithSuggestions> =
|
|
serde_yaml::from_reader(std::fs::File::open(rules_path)?)?;
|
|
let mut cfgs = LinearMap::new();
|
|
cfgs.insert(dependency_id, Cow::Owned(dep_config.clone()));
|
|
match config {
|
|
Some(config) => cfgs.insert(package_id, Cow::Owned(config.clone())),
|
|
None => cfgs.insert(package_id, Cow::Owned(serde_json::Map::new())),
|
|
};
|
|
let rule_check = rules
|
|
.into_iter()
|
|
.map(|r| r.apply(dependency_id, &mut dep_config, &mut cfgs))
|
|
.bcollect::<Vec<_>>();
|
|
match rule_check {
|
|
Ok(_) => Ok(dep_config),
|
|
Err(e) => Err(anyhow!("{}", e)),
|
|
}
|
|
}
|