nushell/src/commands/config.rs

132 lines
3.2 KiB
Rust
Raw Normal View History

use crate::prelude::*;
2019-08-15 07:02:02 +02:00
use crate::commands::WholeStreamCommand;
use crate::errors::ShellError;
2019-07-24 00:22:11 +02:00
use crate::object::{config, Value};
2019-07-15 23:16:27 +02:00
use crate::parser::hir::SyntaxType;
2019-08-02 21:15:07 +02:00
use crate::parser::registry::{self};
use std::iter::FromIterator;
pub struct Config;
2019-08-02 21:15:07 +02:00
#[derive(Deserialize)]
pub struct ConfigArgs {
2019-08-09 06:51:21 +02:00
set: Option<(Tagged<String>, Tagged<Value>)>,
get: Option<Tagged<String>>,
clear: Tagged<bool>,
remove: Option<Tagged<String>>,
path: Tagged<bool>,
2019-08-02 21:15:07 +02:00
}
2019-07-24 00:22:11 +02:00
2019-08-15 07:02:02 +02:00
impl WholeStreamCommand for Config {
fn name(&self) -> &str {
"config"
}
2019-08-02 21:15:07 +02:00
fn signature(&self) -> Signature {
Signature::build("config")
.named("set", SyntaxType::Any)
.named("get", SyntaxType::Any)
.named("remove", SyntaxType::Any)
.switch("clear")
.switch("path")
}
fn usage(&self) -> &str {
"Configuration management."
}
2019-08-02 21:15:07 +02:00
fn run(
&self,
args: CommandArgs,
registry: &registry::CommandRegistry,
) -> Result<OutputStream, ShellError> {
args.process(registry, config)?.run()
}
}
2019-08-02 21:15:07 +02:00
pub fn config(
ConfigArgs {
set,
get,
clear,
remove,
path,
}: ConfigArgs,
RunnableContext { name, .. }: RunnableContext,
) -> Result<OutputStream, ShellError> {
let mut result = crate::object::config::config(name)?;
if let Some(v) = get {
let key = v.to_string();
let value = result
.get(&key)
.ok_or_else(|| ShellError::string(&format!("Missing key {} in config", key)))?;
return Ok(
2019-07-09 06:31:26 +02:00
stream![value.clone()].into(), // futures::stream::once(futures::future::ready(ReturnSuccess::Value(value.clone()))).into(),
);
}
2019-08-02 21:15:07 +02:00
if let Some((key, value)) = set {
result.insert(key.to_string(), value.clone());
2019-08-02 21:15:07 +02:00
config::write_config(&result)?;
2019-08-09 06:51:21 +02:00
return Ok(stream![Tagged::from_simple_spanned_item(
2019-08-02 21:15:07 +02:00
Value::Object(result.into()),
value.span()
)]
.from_input_stream());
}
2019-08-09 06:51:21 +02:00
if let Tagged {
item: true,
tag: Tag { span, .. },
} = clear
{
result.clear();
config::write_config(&result)?;
return Ok(stream![Tagged::from_simple_spanned_item(
Value::Object(result.into()),
2019-08-09 06:51:21 +02:00
span
)]
.from_input_stream());
}
2019-08-09 06:51:21 +02:00
if let Tagged {
item: true,
tag: Tag { span, .. },
} = path
{
2019-08-02 21:15:07 +02:00
let path = config::config_path()?;
2019-08-09 06:51:21 +02:00
return Ok(stream![Tagged::from_simple_spanned_item(
Value::Primitive(Primitive::Path(path)),
span
)]
.from_input_stream());
2019-08-02 21:15:07 +02:00
}
if let Some(v) = remove {
let key = v.to_string();
if result.contains_key(&key) {
result.remove(&key);
2019-08-25 06:51:12 +02:00
config::write_config(&result)?;
} else {
return Err(ShellError::string(&format!(
"{} does not exist in config",
key
)));
}
2019-08-09 06:51:21 +02:00
let obj = VecDeque::from_iter(vec![Value::Object(result.into()).simple_spanned(v.span())]);
return Ok(obj.from_input_stream());
}
2019-08-09 06:51:21 +02:00
return Ok(vec![Value::Object(result.into()).simple_spanned(name)].into());
}