nushell/src/commands/to_url.rs

88 lines
2.8 KiB
Rust
Raw Normal View History

2019-09-19 06:25:29 +02:00
use crate::commands::WholeStreamCommand;
use crate::data::value;
2019-09-19 06:25:29 +02:00
use crate::prelude::*;
use nu_protocol::{ReturnSuccess, Signature, UntaggedValue, Value};
use nu_errors::ShellError;
2019-09-19 06:25:29 +02:00
pub struct ToURL;
impl WholeStreamCommand for ToURL {
fn name(&self) -> &str {
"to-url"
}
fn signature(&self) -> Signature {
Signature::build("to-url")
}
fn usage(&self) -> &str {
"Convert table into url-encoded text"
}
fn run(
&self,
args: CommandArgs,
registry: &CommandRegistry,
) -> Result<OutputStream, ShellError> {
to_url(args, registry)
}
}
fn to_url(args: CommandArgs, registry: &CommandRegistry) -> Result<OutputStream, ShellError> {
let args = args.evaluate_once(registry)?;
let tag = args.name_tag();
let input = args.input;
let stream = async_stream! {
let input: Vec<Value> = input.values.collect().await;
2019-09-19 06:25:29 +02:00
for value in input {
match value {
Value { value: UntaggedValue::Row(row), .. } => {
2019-09-19 06:25:29 +02:00
let mut row_vec = vec![];
for (k,v) in row.entries {
match v.as_string() {
Ok(s) => {
row_vec.push((k.clone(), s.to_string()));
2019-09-19 06:25:29 +02:00
}
_ => {
yield Err(ShellError::labeled_error_with_secondary(
"Expected table with string values",
"requires table with strings",
&tag,
2019-09-19 06:25:29 +02:00
"value originates from here",
v.tag,
))
}
}
}
match serde_urlencoded::to_string(row_vec) {
Ok(s) => {
yield ReturnSuccess::value(value::string(s).into_value(&tag));
2019-09-19 06:25:29 +02:00
}
_ => {
yield Err(ShellError::labeled_error(
"Failed to convert to url-encoded",
"cannot url-encode",
&tag,
2019-09-19 06:25:29 +02:00
))
}
}
}
Value { tag: value_tag, .. } => {
2019-09-19 06:25:29 +02:00
yield Err(ShellError::labeled_error_with_secondary(
"Expected a table from pipeline",
"requires table input",
&tag,
2019-09-19 06:25:29 +02:00
"value originates from here",
value_tag.span,
2019-09-19 06:25:29 +02:00
))
}
}
}
};
Ok(stream.to_output_stream())
}