nushell/src/commands/shells.rs

60 lines
1.5 KiB
Rust
Raw Normal View History

use crate::commands::WholeStreamCommand;
2019-08-07 19:49:11 +02:00
use crate::prelude::*;
use nu_errors::ShellError;
use nu_protocol::{Signature, TaggedDictBuilder};
use std::sync::atomic::Ordering;
2019-08-07 19:49:11 +02:00
pub struct Shells;
impl WholeStreamCommand for Shells {
fn name(&self) -> &str {
"shells"
}
fn signature(&self) -> Signature {
Signature::build("shells")
}
fn usage(&self) -> &str {
"Display the list of current shells."
}
fn run(
&self,
args: CommandArgs,
registry: &CommandRegistry,
) -> Result<OutputStream, ShellError> {
shells(args, registry)
}
}
fn shells(args: CommandArgs, _registry: &CommandRegistry) -> Result<OutputStream, ShellError> {
2019-08-07 19:49:11 +02:00
let mut shells_out = VecDeque::new();
let tag = args.call_info.name_tag;
2019-08-07 19:49:11 +02:00
for (index, shell) in args
.shell_manager
.shells
.lock()
.map_err(|_| {
ShellError::labeled_error("Could not list shells", "could not list shells", &tag)
})?
.iter()
.enumerate()
{
let mut dict = TaggedDictBuilder::new(&tag);
2019-08-10 22:18:14 +02:00
if index == (*args.shell_manager.current_shell).load(Ordering::SeqCst) {
dict.insert_untagged(" ", "X".to_string());
2019-08-10 22:18:14 +02:00
} else {
dict.insert_untagged(" ", " ".to_string());
2019-08-10 22:18:14 +02:00
}
dict.insert_untagged("name", shell.name());
dict.insert_untagged("path", shell.path());
2019-08-07 19:49:11 +02:00
shells_out.push_back(dict.into_value());
2019-08-07 19:49:11 +02:00
}
Ok(shells_out.to_output_stream())
}