55 lines
2.0 KiB
Rust
55 lines
2.0 KiB
Rust
|
use ansi_term::Color;
|
||
|
|
use std::env;
|
||
|
|
|
||
|
|
use super::{Context, Module};
|
||
|
|
|
||
|
|
// IN_NIX_SHELL should be "pure" or "impure" but lorri uses "1" for "impure"
|
||
|
|
// https://github.com/target/lorri/issues/140
|
||
|
|
|
||
|
|
/// Creates a module showing if inside a nix-shell
|
||
|
|
///
|
||
|
|
/// The module will use the `$IN_NIX_SHELL` and `$name` environment variable to
|
||
|
|
/// determine if it's inside a nix-shell and the name of it.
|
||
|
|
///
|
||
|
|
/// The following options are availables:
|
||
|
|
/// - use_name (bool) // print the name of the nix-shell
|
||
|
|
/// - impure_msg (string) // change the impure msg
|
||
|
|
/// - pure_msg (string) // change the pure msg
|
||
|
|
///
|
||
|
|
/// Will display the following:
|
||
|
|
/// - name (pure) // use_name == true in a pure nix-shell
|
||
|
|
/// - name (impure) // use_name == true in an impure nix-shell
|
||
|
|
/// - pure // use_name == false in a pure nix-shell
|
||
|
|
/// - impure // use_name == false in an impure nix-shell
|
||
|
|
pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
|
||
|
|
let mut module = context.new_module("nix_shell")?;
|
||
|
|
|
||
|
|
env::var("IN_NIX_SHELL")
|
||
|
|
.ok()
|
||
|
|
.and_then(|shell_type| {
|
||
|
|
if shell_type == "1" || shell_type == "impure" {
|
||
|
|
Some(module.config_value_str("impure_msg").unwrap_or("impure"))
|
||
|
|
} else if shell_type == "pure" {
|
||
|
|
Some(module.config_value_str("pure_msg").unwrap_or("pure"))
|
||
|
|
} else {
|
||
|
|
None
|
||
|
|
}
|
||
|
|
})
|
||
|
|
.map(|shell_type| {
|
||
|
|
if module.config_value_bool("use_name").unwrap_or(false) {
|
||
|
|
match env::var("name").ok() {
|
||
|
|
Some(name) => format!("{} ({})", name, shell_type),
|
||
|
|
None => shell_type.to_string(),
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
shell_type.to_string()
|
||
|
|
}
|
||
|
|
})
|
||
|
|
.map(|segment| {
|
||
|
|
let module_color = Color::Red.bold();
|
||
|
|
module.set_style(module_color);
|
||
|
|
module.new_segment("nix_shell", &segment);
|
||
|
|
module
|
||
|
|
})
|
||
|
|
}
|