forked from extern/nushell
d2213d18fa
* Playground infraestructure (tests, etc) additions. A few things to note: * Nu can be started with a custom configuration file (`nu --config-file /path/to/sample_config.toml`). Useful for mocking the configuration on test runs. * When given a custom configuration file Nu will save any changes to the file supplied appropiately. * The `$nu.config-path` variable either shows the default configuration file (or the custom one, if given) * We can now run end to end tests with finer grained control (currently, since this is baseline work, standard out) This will allow to check things like exit status, assert the contents with a format, etc) * Remove (for another PR)
105 lines
2.9 KiB
Rust
105 lines
2.9 KiB
Rust
use crate::prelude::*;
|
|
use nu_engine::WholeStreamCommand;
|
|
use nu_errors::ShellError;
|
|
use nu_protocol::{
|
|
ColumnPath, Primitive, ReturnSuccess, Signature, SyntaxShape, UntaggedValue, Value,
|
|
};
|
|
|
|
pub struct SubCommand;
|
|
|
|
#[derive(Deserialize)]
|
|
pub struct Arguments {
|
|
column_path: ColumnPath,
|
|
value: Value,
|
|
}
|
|
|
|
#[async_trait]
|
|
impl WholeStreamCommand for SubCommand {
|
|
fn name(&self) -> &str {
|
|
"config set"
|
|
}
|
|
|
|
fn signature(&self) -> Signature {
|
|
Signature::build("config set")
|
|
.required("key", SyntaxShape::ColumnPath, "variable name to set")
|
|
.required("value", SyntaxShape::Any, "value to use")
|
|
}
|
|
|
|
fn usage(&self) -> &str {
|
|
"Sets a value in the config"
|
|
}
|
|
|
|
async fn run(&self, args: CommandArgs) -> Result<OutputStream, ShellError> {
|
|
set(args).await
|
|
}
|
|
|
|
fn examples(&self) -> Vec<Example> {
|
|
vec![
|
|
Example {
|
|
description: "Set auto pivoting",
|
|
example: "config set pivot_mode always",
|
|
result: None,
|
|
},
|
|
Example {
|
|
description: "Set line editor options",
|
|
example: "config set line_editor [[edit_mode, completion_type]; [emacs circular]]",
|
|
result: None,
|
|
},
|
|
Example {
|
|
description: "Set coloring options",
|
|
example: "config set color_config [[header_align header_bold]; [left $true]]",
|
|
result: None,
|
|
},
|
|
Example {
|
|
description: "Set nested options",
|
|
example: "config set color_config.header_color white",
|
|
result: None,
|
|
},
|
|
]
|
|
}
|
|
}
|
|
|
|
pub async fn set(args: CommandArgs) -> Result<OutputStream, ShellError> {
|
|
let name = args.call_info.name_tag.clone();
|
|
let scope = args.scope.clone();
|
|
let (
|
|
Arguments {
|
|
column_path,
|
|
mut value,
|
|
},
|
|
_,
|
|
) = args.process().await?;
|
|
|
|
let path = match scope.get_var("config-path") {
|
|
Some(Value {
|
|
value: UntaggedValue::Primitive(Primitive::FilePath(path)),
|
|
..
|
|
}) => Some(path),
|
|
_ => nu_data::config::default_path().ok(),
|
|
};
|
|
|
|
let raw_entries = nu_data::config::read(&name, &path)?;
|
|
let configuration = UntaggedValue::row(raw_entries).into_value(&name);
|
|
|
|
if let UntaggedValue::Table(rows) = &value.value {
|
|
if rows.len() == 1 && rows[0].is_row() {
|
|
value = rows[0].clone();
|
|
}
|
|
}
|
|
|
|
match configuration.forgiving_insert_data_at_column_path(&column_path, value) {
|
|
Ok(Value {
|
|
value: UntaggedValue::Row(changes),
|
|
..
|
|
}) => {
|
|
config::write(&changes.entries, &path)?;
|
|
|
|
Ok(OutputStream::one(ReturnSuccess::value(
|
|
UntaggedValue::Row(changes).into_value(name),
|
|
)))
|
|
}
|
|
Ok(_) => Ok(OutputStream::empty()),
|
|
Err(reason) => Err(reason),
|
|
}
|
|
}
|