mirror of
https://github.com/Llewellynvdm/starship.git
synced 2024-09-29 05:29:11 +00:00
197 lines
5.5 KiB
Rust
197 lines
5.5 KiB
Rust
|
use super::{Context, Module};
|
||
|
|
||
|
#[cfg(not(target_os = "linux"))]
|
||
|
pub fn module<'a>(_context: &'a Context) -> Option<Module<'a>> {
|
||
|
None
|
||
|
}
|
||
|
|
||
|
#[cfg(target_os = "linux")]
|
||
|
pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
|
||
|
use super::RootModuleConfig;
|
||
|
use crate::configs::container::ContainerConfig;
|
||
|
use crate::formatter::StringFormatter;
|
||
|
use crate::utils::read_file;
|
||
|
|
||
|
pub fn container_name(context: &Context) -> Option<String> {
|
||
|
use crate::utils::context_path;
|
||
|
|
||
|
if context_path(context, "/proc/vz").exists() && !context_path(context, "/proc/bc").exists()
|
||
|
{
|
||
|
// OpenVZ
|
||
|
return Some("OpenVZ".into());
|
||
|
}
|
||
|
|
||
|
if context_path(context, "/run/host/container-manager").exists() {
|
||
|
// OCI
|
||
|
return Some("OCI".into());
|
||
|
}
|
||
|
|
||
|
if context_path(context, "/run/systemd/container").exists() {
|
||
|
// systemd
|
||
|
return Some("Systemd".into());
|
||
|
}
|
||
|
|
||
|
let container_env_path = context_path(context, "/run/.containerenv");
|
||
|
|
||
|
if container_env_path.exists() {
|
||
|
// podman and others
|
||
|
|
||
|
let image_res = read_file(container_env_path)
|
||
|
.map(|s| {
|
||
|
s.lines()
|
||
|
.find_map(|l| {
|
||
|
l.starts_with("image=\"").then(|| {
|
||
|
let r = l.split_at(7).1;
|
||
|
let name = r.rfind('/').map(|n| r.split_at(n + 1).1);
|
||
|
String::from(name.unwrap_or(r).trim_end_matches('"'))
|
||
|
})
|
||
|
})
|
||
|
.unwrap_or_else(|| "podman".into())
|
||
|
})
|
||
|
.unwrap_or_else(|_| "podman".into());
|
||
|
|
||
|
return Some(image_res);
|
||
|
}
|
||
|
|
||
|
if context_path(context, "/.dockerenv").exists() {
|
||
|
// docker
|
||
|
return Some("Docker".into());
|
||
|
}
|
||
|
|
||
|
None
|
||
|
}
|
||
|
|
||
|
let mut module = context.new_module("container");
|
||
|
let config: ContainerConfig = ContainerConfig::try_load(module.config);
|
||
|
|
||
|
if config.disabled {
|
||
|
return None;
|
||
|
}
|
||
|
|
||
|
let container_name = container_name(context)?;
|
||
|
|
||
|
let parsed = StringFormatter::new(config.format).and_then(|formatter| {
|
||
|
formatter
|
||
|
.map_meta(|variable, _| match variable {
|
||
|
"symbol" => Some(config.symbol),
|
||
|
_ => None,
|
||
|
})
|
||
|
.map_style(|variable| match variable {
|
||
|
"style" => Some(Ok(config.style)),
|
||
|
_ => None,
|
||
|
})
|
||
|
.map(|variable| match variable {
|
||
|
"name" => Some(Ok(&container_name)),
|
||
|
_ => None,
|
||
|
})
|
||
|
.parse(None, Some(context))
|
||
|
});
|
||
|
|
||
|
module.set_segments(match parsed {
|
||
|
Ok(segments) => segments,
|
||
|
Err(error) => {
|
||
|
log::warn!("Error in module `container`: \n{}", error);
|
||
|
return None;
|
||
|
}
|
||
|
});
|
||
|
|
||
|
Some(module)
|
||
|
}
|
||
|
|
||
|
#[cfg(test)]
|
||
|
mod tests {
|
||
|
use crate::test::ModuleRenderer;
|
||
|
use ansi_term::Color;
|
||
|
use std::path::PathBuf;
|
||
|
|
||
|
#[test]
|
||
|
fn test_none_if_disabled() {
|
||
|
let expected = None;
|
||
|
let actual = ModuleRenderer::new("container")
|
||
|
// For a custom config
|
||
|
.config(toml::toml! {
|
||
|
[container]
|
||
|
disabled = true
|
||
|
})
|
||
|
// Run the module and collect the output
|
||
|
.collect();
|
||
|
|
||
|
assert_eq!(expected, actual);
|
||
|
}
|
||
|
|
||
|
fn containerenv(name: Option<&str>) -> std::io::Result<(Option<String>, Option<String>)> {
|
||
|
use std::io::Write;
|
||
|
|
||
|
let renderer = ModuleRenderer::new("container")
|
||
|
// For a custom config
|
||
|
.config(toml::toml! {
|
||
|
[container]
|
||
|
disabled = false
|
||
|
});
|
||
|
|
||
|
let root_path = renderer.root_path();
|
||
|
|
||
|
let mut containerenv = PathBuf::from(root_path);
|
||
|
|
||
|
containerenv.push("run");
|
||
|
std::fs::DirBuilder::new()
|
||
|
.recursive(true)
|
||
|
.create(&containerenv)?;
|
||
|
|
||
|
containerenv.push(".containerenv");
|
||
|
let mut file = std::fs::File::create(&containerenv)?;
|
||
|
if let Some(name) = name {
|
||
|
file.write_all(format!("image=\"{}\"\n", name).as_bytes())?;
|
||
|
}
|
||
|
|
||
|
// The output of the module
|
||
|
let actual = renderer
|
||
|
// Run the module and collect the output
|
||
|
.collect();
|
||
|
|
||
|
// The value that should be rendered by the module.
|
||
|
let expected = Some(format!(
|
||
|
"{} ",
|
||
|
Color::Red
|
||
|
.bold()
|
||
|
.dimmed()
|
||
|
.paint(format!("⬢ [{}]", name.unwrap_or("podman")))
|
||
|
));
|
||
|
|
||
|
Ok((actual, expected))
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
#[cfg(target_os = "linux")]
|
||
|
fn test_containerenv() -> std::io::Result<()> {
|
||
|
let (actual, expected) = containerenv(None)?;
|
||
|
|
||
|
// Assert that the actual and expected values are the same
|
||
|
assert_eq!(actual, expected);
|
||
|
|
||
|
Ok(())
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
#[cfg(target_os = "linux")]
|
||
|
fn test_containerenv_fedora() -> std::io::Result<()> {
|
||
|
let (actual, expected) = containerenv(Some("fedora-toolbox:35"))?;
|
||
|
|
||
|
// Assert that the actual and expected values are the same
|
||
|
assert_eq!(actual, expected);
|
||
|
|
||
|
Ok(())
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
#[cfg(not(target_os = "linux"))]
|
||
|
fn test_containerenv() -> std::io::Result<()> {
|
||
|
let (actual, expected) = containerenv(None)?;
|
||
|
|
||
|
// Assert that the actual and expected values are not the same
|
||
|
assert_ne!(actual, expected);
|
||
|
|
||
|
Ok(())
|
||
|
}
|
||
|
}
|