mirror of
https://github.com/nushell/nushell.git
synced 2024-12-17 12:41:34 +01:00
76019f434e
* custom value trait * functions for custom value trait * custom trait behind flag * open dataframe command * command to-df for basic types * follow path for dataframe * dataframe operations * dataframe not default feature * custom as default feature * corrected examples in command
390 lines
11 KiB
Rust
390 lines
11 KiB
Rust
// use std::path::PathBuf;
|
|
|
|
use std::path::PathBuf;
|
|
use std::str::FromStr;
|
|
|
|
use chrono::{DateTime, FixedOffset};
|
|
// use nu_path::expand_path;
|
|
use nu_protocol::ast::{CellPath, PathMember};
|
|
use nu_protocol::ShellError;
|
|
use nu_protocol::{Range, Spanned, Value};
|
|
|
|
pub trait FromValue: Sized {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError>;
|
|
}
|
|
|
|
impl FromValue for Value {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
Ok(v.clone())
|
|
}
|
|
}
|
|
|
|
impl FromValue for Spanned<i64> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Int { val, span } => Ok(Spanned {
|
|
item: *val,
|
|
span: *span,
|
|
}),
|
|
Value::Filesize { val, span } => Ok(Spanned {
|
|
item: *val as i64,
|
|
span: *span,
|
|
}),
|
|
Value::Duration { val, span } => Ok(Spanned {
|
|
item: *val as i64,
|
|
span: *span,
|
|
}),
|
|
|
|
v => Err(ShellError::CantConvert(
|
|
"integer".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for i64 {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Int { val, .. } => Ok(*val),
|
|
Value::Filesize { val, .. } => Ok(*val as i64),
|
|
Value::Duration { val, .. } => Ok(*val as i64),
|
|
|
|
v => Err(ShellError::CantConvert(
|
|
"integer".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for Spanned<f64> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Int { val, span } => Ok(Spanned {
|
|
item: *val as f64,
|
|
span: *span,
|
|
}),
|
|
Value::Float { val, span } => Ok(Spanned {
|
|
item: *val,
|
|
span: *span,
|
|
}),
|
|
|
|
v => Err(ShellError::CantConvert(
|
|
"float".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for f64 {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Float { val, .. } => Ok(*val),
|
|
Value::Int { val, .. } => Ok(*val as f64),
|
|
v => Err(ShellError::CantConvert(
|
|
"float".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for Spanned<usize> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Int { val, span } => Ok(Spanned {
|
|
item: *val as usize,
|
|
span: *span,
|
|
}),
|
|
Value::Filesize { val, span } => Ok(Spanned {
|
|
item: *val as usize,
|
|
span: *span,
|
|
}),
|
|
Value::Duration { val, span } => Ok(Spanned {
|
|
item: *val as usize,
|
|
span: *span,
|
|
}),
|
|
|
|
v => Err(ShellError::CantConvert(
|
|
"integer".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for usize {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Int { val, .. } => Ok(*val as usize),
|
|
Value::Filesize { val, .. } => Ok(*val as usize),
|
|
Value::Duration { val, .. } => Ok(*val as usize),
|
|
|
|
v => Err(ShellError::CantConvert(
|
|
"integer".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for String {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
// FIXME: we may want to fail a little nicer here
|
|
match v {
|
|
Value::CellPath { val, .. } => Ok(val.into_string()),
|
|
Value::String { val, .. } => Ok(val.clone()),
|
|
v => Err(ShellError::CantConvert(
|
|
"string".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for Spanned<String> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
Ok(Spanned {
|
|
item: match v {
|
|
Value::CellPath { val, .. } => val.into_string(),
|
|
Value::String { val, .. } => val.clone(),
|
|
v => {
|
|
return Err(ShellError::CantConvert(
|
|
"string".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
))
|
|
}
|
|
},
|
|
span: v.span()?,
|
|
})
|
|
}
|
|
}
|
|
|
|
impl FromValue for Vec<String> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
// FIXME: we may want to fail a little nicer here
|
|
match v {
|
|
Value::List { vals, .. } => vals
|
|
.iter()
|
|
.map(|val| match val {
|
|
Value::String { val, .. } => Ok(val.clone()),
|
|
c => Err(ShellError::CantConvert(
|
|
"string".into(),
|
|
c.get_type().to_string(),
|
|
c.span()?,
|
|
)),
|
|
})
|
|
.collect::<Result<Vec<String>, ShellError>>(),
|
|
v => Err(ShellError::CantConvert(
|
|
"string".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for CellPath {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
let span = v.span()?;
|
|
match v {
|
|
Value::CellPath { val, .. } => Ok(val.clone()),
|
|
Value::String { val, .. } => Ok(CellPath {
|
|
members: vec![PathMember::String {
|
|
val: val.clone(),
|
|
span,
|
|
}],
|
|
}),
|
|
Value::Int { val, .. } => Ok(CellPath {
|
|
members: vec![PathMember::Int {
|
|
val: *val as usize,
|
|
span,
|
|
}],
|
|
}),
|
|
x => Err(ShellError::CantConvert(
|
|
"cell path".into(),
|
|
x.get_type().to_string(),
|
|
span,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for bool {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Bool { val, .. } => Ok(*val),
|
|
v => Err(ShellError::CantConvert(
|
|
"bool".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for Spanned<bool> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Bool { val, span } => Ok(Spanned {
|
|
item: *val,
|
|
span: *span,
|
|
}),
|
|
v => Err(ShellError::CantConvert(
|
|
"bool".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for DateTime<FixedOffset> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Date { val, .. } => Ok(*val),
|
|
v => Err(ShellError::CantConvert(
|
|
"date".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for Spanned<DateTime<FixedOffset>> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Date { val, span } => Ok(Spanned {
|
|
item: *val,
|
|
span: *span,
|
|
}),
|
|
v => Err(ShellError::CantConvert(
|
|
"date".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for Range {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Range { val, .. } => Ok((**val).clone()),
|
|
v => Err(ShellError::CantConvert(
|
|
"range".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for Spanned<Range> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Range { val, span } => Ok(Spanned {
|
|
item: (**val).clone(),
|
|
span: *span,
|
|
}),
|
|
v => Err(ShellError::CantConvert(
|
|
"range".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for Vec<u8> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::Binary { val, .. } => Ok(val.clone()),
|
|
Value::String { val, .. } => Ok(val.bytes().collect()),
|
|
v => Err(ShellError::CantConvert(
|
|
"binary data".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FromValue for Spanned<PathBuf> {
|
|
fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
match v {
|
|
Value::String { val, span } => Ok(Spanned {
|
|
item: PathBuf::from_str(val)
|
|
.map_err(|err| ShellError::FileNotFoundCustom(err.to_string(), *span))?,
|
|
span: *span,
|
|
}),
|
|
v => Err(ShellError::CantConvert(
|
|
"range".into(),
|
|
v.get_type().to_string(),
|
|
v.span()?,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
// impl FromValue for Dictionary {
|
|
// fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
// match v {
|
|
// Value {
|
|
// value: UntaggedValue::Row(r),
|
|
// ..
|
|
// } => Ok(r.clone()),
|
|
// v => Err(ShellError::type_error("row", v.spanned_type_name())),
|
|
// }
|
|
// }
|
|
// }
|
|
|
|
// impl FromValue for CapturedBlock {
|
|
// fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
// match v {
|
|
// Value {
|
|
// value: UntaggedValue::Block(b),
|
|
// ..
|
|
// } => Ok((**b).clone()),
|
|
// Value {
|
|
// value: UntaggedValue::Row(_),
|
|
// ..
|
|
// } => {
|
|
// let mut shell_error = ShellError::type_error("block", v.spanned_type_name());
|
|
// shell_error.notes.push(
|
|
// "Note: you can access columns using dot. eg) $it.column or (ls).column".into(),
|
|
// );
|
|
// Err(shell_error)
|
|
// }
|
|
// v => Err(ShellError::type_error("block", v.spanned_type_name())),
|
|
// }
|
|
// }
|
|
// }
|
|
|
|
// impl FromValue for Vec<Value> {
|
|
// fn from_value(v: &Value) -> Result<Self, ShellError> {
|
|
// match v {
|
|
// Value {
|
|
// value: UntaggedValue::Table(t),
|
|
// ..
|
|
// } => Ok(t.clone()),
|
|
// Value {
|
|
// value: UntaggedValue::Row(_),
|
|
// ..
|
|
// } => Ok(vec![v.clone()]),
|
|
// v => Err(ShellError::type_error("table", v.spanned_type_name())),
|
|
// }
|
|
// }
|
|
// }
|