mirror of
https://github.com/Start9Labs/start-os.git
synced 2026-03-30 12:11:56 +00:00
0.2.5 initial commit
Makefile incomplete
This commit is contained in:
276
appmgr/src/dependencies.rs
Normal file
276
appmgr/src/dependencies.rs
Normal file
@@ -0,0 +1,276 @@
|
||||
use std::borrow::Cow;
|
||||
use std::path::Path;
|
||||
|
||||
use emver::{Version, VersionRange};
|
||||
use linear_map::LinearMap;
|
||||
use rand::SeedableRng;
|
||||
|
||||
use crate::config::{Config, ConfigRuleEntryWithSuggestions, ConfigSpec};
|
||||
use crate::manifest::ManifestLatest;
|
||||
use crate::Error;
|
||||
use crate::ResultExt as _;
|
||||
|
||||
#[derive(Clone, Debug, Fail, serde::Serialize)]
|
||||
#[serde(rename_all = "kebab-case")]
|
||||
pub enum DependencyError {
|
||||
NotInstalled, // "not-installed"
|
||||
NotRunning, // "not-running"
|
||||
IncorrectVersion {
|
||||
expected: VersionRange,
|
||||
received: Version,
|
||||
}, // { "incorrect-version": { "expected": "0.1.0", "received": "^0.2.0" } }
|
||||
ConfigUnsatisfied(Vec<String>), // { "config-unsatisfied": ["Bitcoin Core must have pruning set to manual."] }
|
||||
PointerUpdateError(String), // { "pointer-update-error": "Bitcoin Core RPC Port must not be 18332" }
|
||||
Other(String), // { "other": "Well fuck." }
|
||||
}
|
||||
impl std::fmt::Display for DependencyError {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
use DependencyError::*;
|
||||
match self {
|
||||
NotInstalled => write!(f, "Not Installed"),
|
||||
NotRunning => write!(f, "Not Running"),
|
||||
IncorrectVersion { expected, received } => write!(
|
||||
f,
|
||||
"Incorrect Version: Expected {}, Received {}",
|
||||
expected, received
|
||||
),
|
||||
ConfigUnsatisfied(rules) => {
|
||||
write!(f, "Configuration Rule(s) Violated: {}", rules.join(", "))
|
||||
}
|
||||
PointerUpdateError(e) => write!(f, "Pointer Update Caused {}", e),
|
||||
Other(e) => write!(f, "System Error: {}", e),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, serde::Serialize)]
|
||||
#[serde(rename_all = "kebab-case")]
|
||||
pub struct TaggedDependencyError {
|
||||
pub dependency: String,
|
||||
pub error: DependencyError,
|
||||
}
|
||||
impl std::fmt::Display for TaggedDependencyError {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
write!(f, "{}: {}", self.dependency, self.error)
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, Default, serde::Deserialize, serde::Serialize)]
|
||||
pub struct Dependencies(pub LinearMap<String, DepInfo>);
|
||||
|
||||
#[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
|
||||
#[serde(rename_all = "kebab-case")]
|
||||
pub struct DepInfo {
|
||||
pub version: VersionRange,
|
||||
pub optional: Option<String>,
|
||||
pub description: Option<String>,
|
||||
#[serde(default)]
|
||||
pub mount_public: bool,
|
||||
#[serde(default)]
|
||||
pub mount_shared: bool,
|
||||
#[serde(default)]
|
||||
pub config: Vec<ConfigRuleEntryWithSuggestions>,
|
||||
}
|
||||
impl DepInfo {
|
||||
pub async fn satisfied(
|
||||
&self,
|
||||
dependency_id: &str,
|
||||
dependency_config: Option<Config>, // fetch if none
|
||||
dependent_id: &str,
|
||||
dependent_config: &Config,
|
||||
) -> Result<Result<(), DependencyError>, Error> {
|
||||
let info = if let Some(info) = crate::apps::list_info().await?.remove(dependency_id) {
|
||||
info
|
||||
} else {
|
||||
return Ok(Err(DependencyError::NotInstalled));
|
||||
};
|
||||
if !&info.version.satisfies(&self.version) {
|
||||
return Ok(Err(DependencyError::IncorrectVersion {
|
||||
expected: self.version.clone(),
|
||||
received: info.version.clone(),
|
||||
}));
|
||||
}
|
||||
let dependency_config = if let Some(cfg) = dependency_config {
|
||||
cfg
|
||||
} else {
|
||||
let app_config = crate::apps::config(dependency_id).await?;
|
||||
if let Some(cfg) = app_config.config {
|
||||
cfg
|
||||
} else {
|
||||
app_config
|
||||
.spec
|
||||
.gen(&mut rand::rngs::StdRng::from_entropy(), &None)
|
||||
.unwrap_or_default()
|
||||
}
|
||||
};
|
||||
let mut errors = Vec::new();
|
||||
let mut cfgs = LinearMap::with_capacity(2);
|
||||
cfgs.insert(dependency_id, Cow::Borrowed(&dependency_config));
|
||||
cfgs.insert(dependent_id, Cow::Borrowed(dependent_config));
|
||||
for rule in self.config.iter() {
|
||||
if !(rule.entry.rule.compiled)(&dependency_config, &cfgs) {
|
||||
errors.push(rule.entry.description.clone());
|
||||
}
|
||||
}
|
||||
if !errors.is_empty() {
|
||||
return Ok(Err(DependencyError::ConfigUnsatisfied(errors)));
|
||||
}
|
||||
if crate::apps::status(dependency_id).await?.status != crate::apps::DockerStatus::Running {
|
||||
return Ok(Err(DependencyError::NotRunning));
|
||||
}
|
||||
Ok(Ok(()))
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, serde::Serialize)]
|
||||
#[serde(rename_all = "kebab-case")]
|
||||
pub struct AppDepInfo {
|
||||
#[serde(flatten)]
|
||||
pub info: DepInfo,
|
||||
pub required: bool,
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pub error: Option<DependencyError>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Default, serde::Serialize)]
|
||||
#[serde(rename_all = "kebab-case")]
|
||||
pub struct AppDependencies(pub LinearMap<String, AppDepInfo>);
|
||||
|
||||
pub async fn check_dependencies(
|
||||
manifest: ManifestLatest,
|
||||
dependent_config: &Config,
|
||||
dependent_config_spec: &ConfigSpec,
|
||||
) -> Result<AppDependencies, Error> {
|
||||
let mut deps = AppDependencies::default();
|
||||
for (dependency_id, dependency_info) in manifest.dependencies.0.into_iter() {
|
||||
let required = dependency_info.optional.is_none()
|
||||
|| dependent_config_spec.requires(&dependency_id, dependent_config);
|
||||
let error = dependency_info
|
||||
.satisfied(&dependency_id, None, &manifest.id, dependent_config)
|
||||
.await?
|
||||
.err();
|
||||
let app_dep_info = AppDepInfo {
|
||||
error,
|
||||
required,
|
||||
info: dependency_info,
|
||||
};
|
||||
deps.0.insert(dependency_id, app_dep_info);
|
||||
}
|
||||
Ok(deps)
|
||||
}
|
||||
|
||||
pub async fn auto_configure(
|
||||
dependent: &str,
|
||||
dependency: &str,
|
||||
dry_run: bool,
|
||||
) -> Result<crate::config::ConfigurationRes, Error> {
|
||||
let (dependent_config, mut dependency_config, manifest) = futures::try_join!(
|
||||
crate::apps::config_or_default(dependent),
|
||||
crate::apps::config_or_default(dependency),
|
||||
crate::apps::manifest(dependent)
|
||||
)?;
|
||||
let mut cfgs = LinearMap::new();
|
||||
cfgs.insert(dependent, Cow::Borrowed(&dependent_config));
|
||||
cfgs.insert(dependency, Cow::Owned(dependency_config.clone()));
|
||||
let dep_info = manifest
|
||||
.dependencies
|
||||
.0
|
||||
.get(dependency)
|
||||
.ok_or_else(|| failure::format_err!("{} Does Not Depend On {}", dependent, dependency))
|
||||
.no_code()?;
|
||||
for rule in &dep_info.config {
|
||||
if let Err(e) = rule.apply(dependency, &mut dependency_config, &mut cfgs) {
|
||||
log::warn!("Rule Unsatisfied After Applying Suggestions: {}", e);
|
||||
}
|
||||
}
|
||||
crate::config::configure(dependency, Some(dependency_config), None, dry_run).await
|
||||
}
|
||||
|
||||
pub async fn update_shared(dependency_id: &str) -> Result<(), Error> {
|
||||
let dependency_manifest = crate::apps::manifest(dependency_id).await?;
|
||||
if let Some(shared) = dependency_manifest.shared {
|
||||
for dependent_id in &crate::apps::dependents(dependency_id, false).await? {
|
||||
let dependent_manifest = crate::apps::manifest(&dependent_id).await?;
|
||||
if dependent_manifest
|
||||
.dependencies
|
||||
.0
|
||||
.get(dependency_id)
|
||||
.ok_or_else(|| failure::format_err!("failed to index dependent: {}", dependent_id))?
|
||||
.mount_shared
|
||||
{
|
||||
tokio::fs::create_dir_all(
|
||||
Path::new(crate::VOLUMES)
|
||||
.join(dependency_id)
|
||||
.join(&shared)
|
||||
.join(&dependent_id),
|
||||
)
|
||||
.await?;
|
||||
}
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn update_binds(dependent_id: &str) -> Result<(), Error> {
|
||||
let dependent_manifest = crate::apps::manifest(dependent_id).await?;
|
||||
let dependency_manifests = futures::future::try_join_all(
|
||||
dependent_manifest
|
||||
.dependencies
|
||||
.0
|
||||
.into_iter()
|
||||
.filter(|(_, info)| info.mount_public || info.mount_shared)
|
||||
.map(|(id, info)| async {
|
||||
crate::apps::manifest(&id).await.map(|man| (id, info, man))
|
||||
}),
|
||||
)
|
||||
.await?;
|
||||
// i just have a gut feeling this shouldn't be concurrent
|
||||
for (dependency_id, info, dependency_manifest) in dependency_manifests {
|
||||
match (dependency_manifest.public, info.mount_public) {
|
||||
(Some(public), true) => {
|
||||
let public_path = Path::new(crate::VOLUMES).join(&dependency_id).join(public);
|
||||
if let Ok(metadata) = tokio::fs::metadata(&public_path).await {
|
||||
if metadata.is_dir() {
|
||||
crate::disks::bind(
|
||||
public_path,
|
||||
Path::new(crate::VOLUMES)
|
||||
.join(&dependent_id)
|
||||
.join("start9")
|
||||
.join("public")
|
||||
.join(&dependency_id),
|
||||
true,
|
||||
)
|
||||
.await?
|
||||
}
|
||||
}
|
||||
}
|
||||
_ => (),
|
||||
}
|
||||
match (dependency_manifest.shared, info.mount_shared) {
|
||||
(Some(shared), true) => {
|
||||
let shared_path = Path::new(crate::VOLUMES)
|
||||
.join(&dependency_id)
|
||||
.join(shared)
|
||||
.join(dependent_id); // namespaced by dependent
|
||||
tokio::fs::create_dir_all(&shared_path).await?;
|
||||
if let Ok(metadata) = tokio::fs::metadata(&shared_path).await {
|
||||
if metadata.is_dir() {
|
||||
crate::disks::bind(
|
||||
shared_path,
|
||||
Path::new(crate::VOLUMES)
|
||||
.join(&dependent_id)
|
||||
.join("start9")
|
||||
.join("shared")
|
||||
.join(&dependency_id),
|
||||
false,
|
||||
)
|
||||
.await?
|
||||
}
|
||||
}
|
||||
}
|
||||
_ => (),
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
Reference in New Issue
Block a user