nushell/src/commands/enter.rs

212 lines
7.4 KiB
Rust
Raw Normal View History

2019-06-13 23:47:25 +02:00
use crate::commands::command::CommandAction;
use crate::errors::ShellError;
use crate::object::{Primitive, Value};
2019-06-22 05:43:37 +02:00
use crate::parser::Spanned;
2019-06-13 23:47:25 +02:00
use crate::prelude::*;
use std::path::{Path, PathBuf};
pub fn enter(args: CommandArgs) -> Result<OutputStream, ShellError> {
2019-06-22 05:43:37 +02:00
let path = match args.nth(0) {
None => return Err(ShellError::maybe_labeled_error(
2019-06-15 20:36:17 +02:00
"open requires a path or url",
"missing path",
args.name_span,
2019-06-22 05:43:37 +02:00
)),
Some(p) => p,
};
2019-06-13 23:47:25 +02:00
2019-06-16 01:03:49 +02:00
let span = args.name_span;
2019-06-13 23:47:25 +02:00
let cwd = args
2019-06-22 05:43:37 +02:00
.env()
2019-06-13 23:47:25 +02:00
.lock()
.unwrap()
2019-06-15 19:52:55 +02:00
.front()
2019-06-13 23:47:25 +02:00
.unwrap()
.path()
.to_path_buf();
2019-06-22 05:43:37 +02:00
2019-06-13 23:47:25 +02:00
let mut full_path = PathBuf::from(cwd);
2019-06-22 05:43:37 +02:00
let (file_extension, contents) = match path.item() {
2019-06-13 23:47:25 +02:00
Value::Primitive(Primitive::String(s)) => {
if s.starts_with("http:") || s.starts_with("https:") {
let response = reqwest::get(s);
match response {
Ok(mut r) => match r.text() {
Ok(s) => {
let fname = r
.url()
.path_segments()
.and_then(|segments| segments.last())
.and_then(|name| if name.is_empty() { None } else { Some(name) })
.and_then(|name| {
PathBuf::from(name)
.extension()
.map(|name| name.to_string_lossy().to_string())
});
(fname, s)
}
Err(_) => {
return Err(ShellError::labeled_error(
"Web page contents corrupt",
"received garbled data",
2019-06-22 05:43:37 +02:00
args.expect_nth(0)?.span,
2019-06-13 23:47:25 +02:00
));
}
},
Err(_) => {
return Err(ShellError::labeled_error(
"URL could not be opened",
"url not found",
2019-06-22 05:43:37 +02:00
args.expect_nth(0)?.span,
2019-06-13 23:47:25 +02:00
));
}
}
} else {
full_path.push(Path::new(&s));
match std::fs::read_to_string(&full_path) {
Ok(s) => (
full_path
.extension()
.map(|name| name.to_string_lossy().to_string()),
s,
),
Err(_) => {
return Err(ShellError::labeled_error(
"File cound not be opened",
"file not found",
2019-06-22 05:43:37 +02:00
args.expect_nth(0)?.span,
2019-06-13 23:47:25 +02:00
));
}
}
}
}
_ => {
return Err(ShellError::labeled_error(
"Expected string value for filename",
"expected filename",
2019-06-22 05:43:37 +02:00
args.expect_nth(0)?.span,
2019-06-13 23:47:25 +02:00
));
}
};
let mut stream = VecDeque::new();
2019-06-22 05:43:37 +02:00
let file_extension = match args.nth(1) {
2019-06-13 23:47:25 +02:00
Some(Spanned {
item: Value::Primitive(Primitive::String(s)),
2019-06-15 04:24:13 +02:00
span,
}) => {
if s == "--raw" {
None
} else if s == "--json" {
Some("json".to_string())
} else if s == "--xml" {
Some("xml".to_string())
2019-06-16 08:43:40 +02:00
} else if s == "--ini" {
Some("ini".to_string())
2019-06-15 04:24:13 +02:00
} else if s == "--yaml" {
Some("yaml".to_string())
} else if s == "--toml" {
Some("toml".to_string())
} else {
return Err(ShellError::labeled_error(
"Unknown flag for open",
"unknown flag",
span.clone(),
));
}
2019-06-13 23:47:25 +02:00
}
2019-06-15 04:24:13 +02:00
_ => file_extension,
2019-06-13 23:47:25 +02:00
};
match file_extension {
2019-06-15 04:24:13 +02:00
Some(x) if x == "toml" => {
2019-06-13 23:47:25 +02:00
stream.push_back(ReturnValue::Action(CommandAction::Enter(
2019-06-16 01:03:49 +02:00
crate::commands::from_toml::from_toml_string_to_value(contents).map_err(
move |_| {
ShellError::maybe_labeled_error(
"Could not load as TOML",
"could not load as TOML",
span,
)
},
)?,
2019-06-13 23:47:25 +02:00
)));
}
2019-06-15 04:24:13 +02:00
Some(x) if x == "json" => {
2019-06-13 23:47:25 +02:00
stream.push_back(ReturnValue::Action(CommandAction::Enter(
2019-06-16 01:03:49 +02:00
crate::commands::from_json::from_json_string_to_value(contents).map_err(
move |_| {
ShellError::maybe_labeled_error(
"Could not load as JSON",
"could not load as JSON",
span,
)
},
)?,
2019-06-13 23:47:25 +02:00
)));
}
2019-06-15 04:24:13 +02:00
Some(x) if x == "xml" => {
2019-06-13 23:47:25 +02:00
stream.push_back(ReturnValue::Action(CommandAction::Enter(
2019-06-16 01:03:49 +02:00
crate::commands::from_xml::from_xml_string_to_value(contents).map_err(
move |_| {
ShellError::maybe_labeled_error(
"Could not load as XML",
"could not load as XML",
span,
)
},
)?,
2019-06-13 23:47:25 +02:00
)));
}
2019-06-16 08:43:40 +02:00
Some(x) if x == "ini" => {
stream.push_back(ReturnValue::Action(CommandAction::Enter(
crate::commands::from_ini::from_ini_string_to_value(contents).map_err(
move |_| {
ShellError::maybe_labeled_error(
"Could not load as INI",
"could not load as INI",
span,
)
},
)?,
)));
}
2019-06-15 04:24:13 +02:00
Some(x) if x == "yml" => {
2019-06-13 23:47:25 +02:00
stream.push_back(ReturnValue::Action(CommandAction::Enter(
2019-06-16 01:03:49 +02:00
crate::commands::from_yaml::from_yaml_string_to_value(contents).map_err(
move |_| {
ShellError::maybe_labeled_error(
"Could not load as YAML",
"could not load as YAML",
span,
)
},
)?,
2019-06-13 23:47:25 +02:00
)));
}
2019-06-15 04:24:13 +02:00
Some(x) if x == "yaml" => {
2019-06-13 23:47:25 +02:00
stream.push_back(ReturnValue::Action(CommandAction::Enter(
2019-06-16 01:03:49 +02:00
crate::commands::from_yaml::from_yaml_string_to_value(contents).map_err(
move |_| {
ShellError::maybe_labeled_error(
"Could not load as YAML",
"could not load as YAML",
span,
)
},
)?,
2019-06-13 23:47:25 +02:00
)));
}
_ => {
stream.push_back(ReturnValue::Action(CommandAction::Enter(Value::Primitive(
Primitive::String(contents),
))));
}
}
Ok(stream.boxed())
}