holm/nushell
1
1
Fork 0
forked from extern/nushell
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add experimental support for flags in custom commands (#2808)

Browse Source 
* Add experimental support for flags in custom commands

* clippy
This commit is contained in:
Jonathan Turner
2020-12-21 20:36:59 +13:00
committed by GitHub
parent 67acaae53c
commit c012d648fb
4 changed files with 148 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
crates/nu-cli/src/commands/command.rs
View File

@ -275,6 +275,39 @@ impl WholeStreamCommand for Block {
}
}
}
if let Some(args) = evaluated.args.named {
for named in &block.params.named {
let name = named.0;
if let Some(value) = args.get(name) {
if name.starts_with('$') {
ctx.scope.add_var(name, value.clone());
} else {
ctx.scope.add_var(format!("${}", name), value.clone());
}
} else if name.starts_with('$') {
ctx.scope
.add_var(name, UntaggedValue::nothing().into_untagged_value());
} else {
ctx.scope.add_var(
format!("${}", name),
UntaggedValue::nothing().into_untagged_value(),
);
}
}
} else {
for named in &block.params.named {
let name = named.0;
if name.starts_with('$') {
ctx.scope
.add_var(name, UntaggedValue::nothing().into_untagged_value());
} else {
ctx.scope.add_var(
format!("${}", name),
UntaggedValue::nothing().into_untagged_value(),
);
}
}
}
let result = run_block(&block, &ctx, input).await;
ctx.scope.exit_scope();
result.map(|x| x.to_output_stream())