forked from extern/nushell
fmt
This commit is contained in:
merelymyself
parent
878b1e1603
commit
6d951751cf
@ -16,7 +16,7 @@ use std::collections::HashMap;
|
|||||||
use std::io::BufReader;
|
use std::io::BufReader;
|
||||||
|
|
||||||
use reqwest::StatusCode;
|
use reqwest::StatusCode;
|
||||||
use std::path::{PathBuf, Path};
|
use std::path::{Path, PathBuf};
|
||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
@ -108,182 +108,188 @@ impl Command for SubCommand {
|
|||||||
let output = call.has_flag("output");
|
let output = call.has_flag("output");
|
||||||
if !output {
|
if !output {
|
||||||
run_fetch(engine_state, stack, call, input)
|
run_fetch(engine_state, stack, call, input)
|
||||||
}
|
} else {
|
||||||
else {
|
|
||||||
match run_fetch(engine_state, stack, call, input) {
|
match run_fetch(engine_state, stack, call, input) {
|
||||||
Err(err) => Err(err),
|
Err(err) => Err(err),
|
||||||
Ok(value) => {
|
Ok(value) => {
|
||||||
let path:Value = call.get_flag(engine_state, stack, "output").expect("there should be a value").expect("value should be unwrappable");
|
let path: Value = call
|
||||||
|
.get_flag(engine_state, stack, "output")
|
||||||
|
.expect("there should be a value")
|
||||||
|
.expect("value should be unwrappable");
|
||||||
let bin = call.has_flag("bin");
|
let bin = call.has_flag("bin");
|
||||||
let append = call.has_flag("append");
|
let append = call.has_flag("append");
|
||||||
let span = call.head;
|
let span = call.head;
|
||||||
let path = &path.as_string().expect("path should be a string");
|
let path = &path.as_string().expect("path should be a string");
|
||||||
let path = Path::new(path);
|
let path = Path::new(path);
|
||||||
|
|
||||||
let file = match (append, path.exists()) {
|
let file = match (append, path.exists()) {
|
||||||
(true, true) => std::fs::OpenOptions::new()
|
(true, true) => std::fs::OpenOptions::new()
|
||||||
.write(true)
|
.write(true)
|
||||||
.append(true)
|
.append(true)
|
||||||
.open(path),
|
.open(path),
|
||||||
_ => std::fs::File::create(path),
|
_ => std::fs::File::create(path),
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut file = match file {
|
let mut file = match file {
|
||||||
Ok(file) => file,
|
Ok(file) => file,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
let arg_span = call.get_named_arg("output").expect("arg should exist").span;
|
let arg_span =
|
||||||
return Ok(PipelineData::Value(
|
call.get_named_arg("output").expect("arg should exist").span;
|
||||||
Value::Error {
|
return Ok(PipelineData::Value(
|
||||||
error: ShellError::GenericError(
|
Value::Error {
|
||||||
"Permission denied".into(),
|
error: ShellError::GenericError(
|
||||||
err.to_string(),
|
"Permission denied".into(),
|
||||||
Some(arg_span),
|
err.to_string(),
|
||||||
None,
|
Some(arg_span),
|
||||||
Vec::new(),
|
None,
|
||||||
),
|
Vec::new(),
|
||||||
},
|
),
|
||||||
None,
|
|
||||||
));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let ext = if bin {
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
path.extension()
|
|
||||||
.map(|name| name.to_string_lossy().to_string())
|
|
||||||
};
|
|
||||||
|
|
||||||
if let Some(ext) = ext {
|
|
||||||
let output = match engine_state.find_decl(format!("to {}", ext).as_bytes(), &[]) {
|
|
||||||
Some(converter_id) => {
|
|
||||||
let output = engine_state.get_decl(converter_id).run(
|
|
||||||
engine_state,
|
|
||||||
stack,
|
|
||||||
&Call::new(span),
|
|
||||||
value,
|
|
||||||
)?;
|
|
||||||
|
|
||||||
output.into_value(span)
|
|
||||||
}
|
|
||||||
None => value.into_value(span),
|
|
||||||
};
|
|
||||||
|
|
||||||
match output {
|
|
||||||
Value::String { val, .. } => {
|
|
||||||
if let Err(err) = file.write_all(val.as_bytes()) {
|
|
||||||
return Err(ShellError::IOError(err.to_string()));
|
|
||||||
} else {
|
|
||||||
file.flush()?
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(PipelineData::new(span))
|
|
||||||
}
|
|
||||||
Value::Binary { val, .. } => {
|
|
||||||
if let Err(err) = file.write_all(&val) {
|
|
||||||
return Err(ShellError::IOError(err.to_string()));
|
|
||||||
} else {
|
|
||||||
file.flush()?
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(PipelineData::new(span))
|
|
||||||
}
|
|
||||||
Value::List { vals, .. } => {
|
|
||||||
let val = vals
|
|
||||||
.into_iter()
|
|
||||||
.map(|it| it.as_string())
|
|
||||||
.collect::<Result<Vec<String>, ShellError>>()?
|
|
||||||
.join("\n")
|
|
||||||
+ "\n";
|
|
||||||
|
|
||||||
if let Err(err) = file.write_all(val.as_bytes()) {
|
|
||||||
return Err(ShellError::IOError(err.to_string()));
|
|
||||||
} else {
|
|
||||||
file.flush()?
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(PipelineData::new(span))
|
|
||||||
}
|
|
||||||
v => Err(ShellError::UnsupportedInput(
|
|
||||||
format!("{:?} not supported", v.get_type()),
|
|
||||||
span,
|
|
||||||
)),
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
match value {
|
|
||||||
PipelineData::ExternalStream { stdout: None, .. } => Ok(PipelineData::new(span)),
|
|
||||||
PipelineData::ExternalStream {
|
|
||||||
stdout: Some(mut stream),
|
|
||||||
..
|
|
||||||
} => {
|
|
||||||
let mut writer = BufWriter::new(file);
|
|
||||||
|
|
||||||
stream
|
|
||||||
.try_for_each(move |result| {
|
|
||||||
let buf = match result {
|
|
||||||
Ok(v) => match v {
|
|
||||||
Value::String { val, .. } => val.into_bytes(),
|
|
||||||
Value::Binary { val, .. } => val,
|
|
||||||
_ => {
|
|
||||||
return Err(ShellError::UnsupportedInput(
|
|
||||||
format!("{:?} not supported", v.get_type()),
|
|
||||||
v.span()?,
|
|
||||||
));
|
|
||||||
}
|
|
||||||
},
|
},
|
||||||
Err(err) => return Err(err),
|
None,
|
||||||
|
));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let ext = if bin {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
path.extension()
|
||||||
|
.map(|name| name.to_string_lossy().to_string())
|
||||||
|
};
|
||||||
|
|
||||||
|
if let Some(ext) = ext {
|
||||||
|
let output =
|
||||||
|
match engine_state.find_decl(format!("to {}", ext).as_bytes(), &[]) {
|
||||||
|
Some(converter_id) => {
|
||||||
|
let output = engine_state.get_decl(converter_id).run(
|
||||||
|
engine_state,
|
||||||
|
stack,
|
||||||
|
&Call::new(span),
|
||||||
|
value,
|
||||||
|
)?;
|
||||||
|
|
||||||
|
output.into_value(span)
|
||||||
|
}
|
||||||
|
None => value.into_value(span),
|
||||||
};
|
};
|
||||||
|
|
||||||
if let Err(err) = writer.write(&buf) {
|
match output {
|
||||||
return Err(ShellError::IOError(err.to_string()));
|
Value::String { val, .. } => {
|
||||||
|
if let Err(err) = file.write_all(val.as_bytes()) {
|
||||||
|
return Err(ShellError::IOError(err.to_string()));
|
||||||
|
} else {
|
||||||
|
file.flush()?
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(PipelineData::new(span))
|
||||||
}
|
}
|
||||||
Ok(())
|
Value::Binary { val, .. } => {
|
||||||
})
|
if let Err(err) = file.write_all(&val) {
|
||||||
.map(|_| PipelineData::new(span))
|
return Err(ShellError::IOError(err.to_string()));
|
||||||
|
} else {
|
||||||
|
file.flush()?
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(PipelineData::new(span))
|
||||||
|
}
|
||||||
|
Value::List { vals, .. } => {
|
||||||
|
let val = vals
|
||||||
|
.into_iter()
|
||||||
|
.map(|it| it.as_string())
|
||||||
|
.collect::<Result<Vec<String>, ShellError>>()?
|
||||||
|
.join("\n")
|
||||||
|
+ "\n";
|
||||||
|
|
||||||
|
if let Err(err) = file.write_all(val.as_bytes()) {
|
||||||
|
return Err(ShellError::IOError(err.to_string()));
|
||||||
|
} else {
|
||||||
|
file.flush()?
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(PipelineData::new(span))
|
||||||
|
}
|
||||||
|
v => Err(ShellError::UnsupportedInput(
|
||||||
|
format!("{:?} not supported", v.get_type()),
|
||||||
|
span,
|
||||||
|
)),
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
match value {
|
||||||
|
PipelineData::ExternalStream { stdout: None, .. } => {
|
||||||
|
Ok(PipelineData::new(span))
|
||||||
|
}
|
||||||
|
PipelineData::ExternalStream {
|
||||||
|
stdout: Some(mut stream),
|
||||||
|
..
|
||||||
|
} => {
|
||||||
|
let mut writer = BufWriter::new(file);
|
||||||
|
|
||||||
|
stream
|
||||||
|
.try_for_each(move |result| {
|
||||||
|
let buf = match result {
|
||||||
|
Ok(v) => match v {
|
||||||
|
Value::String { val, .. } => val.into_bytes(),
|
||||||
|
Value::Binary { val, .. } => val,
|
||||||
|
_ => {
|
||||||
|
return Err(ShellError::UnsupportedInput(
|
||||||
|
format!("{:?} not supported", v.get_type()),
|
||||||
|
v.span()?,
|
||||||
|
));
|
||||||
|
}
|
||||||
|
},
|
||||||
|
Err(err) => return Err(err),
|
||||||
|
};
|
||||||
|
|
||||||
|
if let Err(err) = writer.write(&buf) {
|
||||||
|
return Err(ShellError::IOError(err.to_string()));
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
})
|
||||||
|
.map(|_| PipelineData::new(span))
|
||||||
|
}
|
||||||
|
value => match value.into_value(span) {
|
||||||
|
Value::String { val, .. } => {
|
||||||
|
if let Err(err) = file.write_all(val.as_bytes()) {
|
||||||
|
return Err(ShellError::IOError(err.to_string()));
|
||||||
|
} else {
|
||||||
|
file.flush()?
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(PipelineData::new(span))
|
||||||
|
}
|
||||||
|
Value::Binary { val, .. } => {
|
||||||
|
if let Err(err) = file.write_all(&val) {
|
||||||
|
return Err(ShellError::IOError(err.to_string()));
|
||||||
|
} else {
|
||||||
|
file.flush()?
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(PipelineData::new(span))
|
||||||
|
}
|
||||||
|
Value::List { vals, .. } => {
|
||||||
|
let val = vals
|
||||||
|
.into_iter()
|
||||||
|
.map(|it| it.as_string())
|
||||||
|
.collect::<Result<Vec<String>, ShellError>>()?
|
||||||
|
.join("\n")
|
||||||
|
+ "\n";
|
||||||
|
|
||||||
|
if let Err(err) = file.write_all(val.as_bytes()) {
|
||||||
|
return Err(ShellError::IOError(err.to_string()));
|
||||||
|
} else {
|
||||||
|
file.flush()?
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(PipelineData::new(span))
|
||||||
|
}
|
||||||
|
v => Err(ShellError::UnsupportedInput(
|
||||||
|
format!("{:?} not supported", v.get_type()),
|
||||||
|
span,
|
||||||
|
)),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
value => match value.into_value(span) {
|
|
||||||
Value::String { val, .. } => {
|
|
||||||
if let Err(err) = file.write_all(val.as_bytes()) {
|
|
||||||
return Err(ShellError::IOError(err.to_string()));
|
|
||||||
} else {
|
|
||||||
file.flush()?
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(PipelineData::new(span))
|
|
||||||
}
|
|
||||||
Value::Binary { val, .. } => {
|
|
||||||
if let Err(err) = file.write_all(&val) {
|
|
||||||
return Err(ShellError::IOError(err.to_string()));
|
|
||||||
} else {
|
|
||||||
file.flush()?
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(PipelineData::new(span))
|
|
||||||
}
|
|
||||||
Value::List { vals, .. } => {
|
|
||||||
let val = vals
|
|
||||||
.into_iter()
|
|
||||||
.map(|it| it.as_string())
|
|
||||||
.collect::<Result<Vec<String>, ShellError>>()?
|
|
||||||
.join("\n")
|
|
||||||
+ "\n";
|
|
||||||
|
|
||||||
if let Err(err) = file.write_all(val.as_bytes()) {
|
|
||||||
return Err(ShellError::IOError(err.to_string()));
|
|
||||||
} else {
|
|
||||||
file.flush()?
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(PipelineData::new(span))
|
|
||||||
}
|
|
||||||
v => Err(ShellError::UnsupportedInput(
|
|
||||||
format!("{:?} not supported", v.get_type()),
|
|
||||||
span,
|
|
||||||
)),
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user