use super::{Context, Module, ModuleConfig}; use crate::configs::singularity::SingularityConfig; use crate::formatter::StringFormatter; /// Creates a module with the current Singularity image /// /// Will display the Singularity image if `$SINGULARITY_NAME` is set. pub fn module<'a>(context: &'a Context) -> Option> { let singularity_env = context.get_env("SINGULARITY_NAME")?; let mut module = context.new_module("singularity"); let config: SingularityConfig = SingularityConfig::try_load(module.config); 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 { "env" => Some(Ok(&singularity_env)), _ => None, }) .parse(None, Some(context)) }); module.set_segments(match parsed { Ok(segments) => segments, Err(error) => { log::warn!("Error in module `singularity`: \n{}", error); return None; } }); Some(module) } #[cfg(test)] mod tests { use crate::test::ModuleRenderer; use nu_ansi_term::Color; #[test] fn no_env_set() { let actual = ModuleRenderer::new("singularity").collect(); let expected = None; assert_eq!(expected, actual); } #[test] fn env_set() { let actual = ModuleRenderer::new("singularity") .env("SINGULARITY_NAME", "centos.img") .collect(); let expected = Some(format!( "{} ", Color::Blue.bold().dimmed().paint("[centos.img]") )); assert_eq!(expected, actual); } }