nushell/src/commands/config.rs

155 lines
3.8 KiB
Rust
Raw Normal View History

2019-08-15 07:02:02 +02:00
use crate::commands::WholeStreamCommand;
use crate::data::{config, Value};
2019-09-09 19:10:52 +02:00
use crate::errors::ShellError;
use crate::parser::hir::SyntaxType;
2019-08-02 21:15:07 +02:00
use crate::parser::registry::{self};
2019-09-10 12:08:01 +02:00
use crate::prelude::*;
use std::iter::FromIterator;
2019-09-10 12:08:01 +02:00
use std::path::PathBuf;
pub struct Config;
2019-08-02 21:15:07 +02:00
#[derive(Deserialize)]
pub struct ConfigArgs {
2019-09-10 12:08:01 +02:00
load: Option<Tagged<PathBuf>>,
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("load", SyntaxType::Path)
.named("set", SyntaxType::Any)
.named("get", SyntaxType::Any)
.named("remove", SyntaxType::Any)
2019-08-02 21:15:07 +02:00
.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 {
2019-09-10 12:08:01 +02:00
load,
2019-08-02 21:15:07 +02:00
set,
get,
clear,
remove,
path,
}: ConfigArgs,
RunnableContext { name, .. }: RunnableContext,
) -> Result<OutputStream, ShellError> {
2019-09-10 12:08:01 +02:00
let name_span = name;
let configuration = if let Some(supplied) = load {
Some(supplied.item().clone())
} else {
None
};
let mut result = crate::data::config::read(name_span, &configuration)?;
2019-08-02 21:15:07 +02:00
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)))?;
2019-09-10 12:08:01 +02:00
let mut results = VecDeque::new();
match value {
Tagged {
item: Value::Table(list),
..
} => {
for l in list {
results.push_back(ReturnSuccess::value(l.clone()));
}
}
x => results.push_back(ReturnSuccess::value(x.clone())),
}
return Ok(results.to_output_stream());
}
2019-08-02 21:15:07 +02:00
if let Some((key, value)) = set {
result.insert(key.to_string(), value.clone());
2019-09-10 12:08:01 +02:00
config::write(&result, &configuration)?;
return Ok(stream![Tagged::from_simple_spanned_item(
Value::Row(result.into()),
value.span()
)]
.from_input_stream());
}
if let Tagged {
item: true,
tag: Tag { span, .. },
} = clear
{
result.clear();
2019-09-10 12:08:01 +02:00
config::write(&result, &configuration)?;
return Ok(stream![Tagged::from_simple_spanned_item(
Value::Row(result.into()),
span
)]
.from_input_stream());
}
if let Tagged {
item: true,
tag: Tag { span, .. },
} = path
{
2019-09-10 12:08:01 +02:00
let path = config::default_path_for(&configuration)?;
2019-08-02 21:15:07 +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) {
2019-09-09 12:03:01 +02:00
result.swap_remove(&key);
2019-09-10 12:08:01 +02:00
config::write(&result, &configuration)?;
} else {
return Err(ShellError::string(&format!(
"{} does not exist in config",
key
)));
}
let obj = VecDeque::from_iter(vec![Value::Row(result.into()).simple_spanned(v.span())]);
return Ok(obj.from_input_stream());
}
return Ok(vec![Value::Row(result.into()).simple_spanned(name)].into());
}