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

Type validation for headers command (#6918) (#7047)

Browse Source 
cargo clippy lints

tests

format

Co-authored-by: Ricardo Monteiro <ricardo.monteiro@getmanta.com>
This commit is contained in:
raccmonteiro
2022-11-10 00:43:24 +00:00
committed by GitHub
parent c4cb3a77cb
commit 7b0c0692dc
2 changed files with 117 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
crates/nu-command/src/filters/headers.rs
View File

@ -122,20 +122,42 @@ fn replace_headers(value: Value, headers: &[String]) -> Result<Value, ShellError
}
}
fn is_valid_header(value: &Value) -> bool {
matches!(
value,
Value::Nothing { span: _ }
| Value::String { val: _, span: _ }
| Value::Bool { val: _, span: _ }
| Value::Float { val: _, span: _ }
| Value::Int { val: _, span: _ }
)
}
fn extract_headers(value: &Value, config: &Config) -> Result<Vec<String>, ShellError> {
match value {
Value::Record { vals, .. } => Ok(vals
.iter()
.enumerate()
.map(|(idx, value)| {
let col = value.into_string("", config);
if col.is_empty() {
format!("column{}", idx)
} else {
col
Value::Record { vals, .. } => {
for v in vals {
if !is_valid_header(v) {
return Err(ShellError::TypeMismatch(
"compatible type: Null, String, Bool, Float, Int".to_string(),
v.span()?,
));
}
})
.collect::<Vec<String>>()),
}
Ok(vals
.iter()
.enumerate()
.map(|(idx, value)| {
let col = value.into_string("", config);
if col.is_empty() {
format!("column{}", idx)
} else {
col
}
})
.collect::<Vec<String>>())
}
Value::List { vals, span } => vals
.iter()
.map(|value| extract_headers(value, config))