mirror of
https://github.com/nushell/nushell
synced 2024-11-10 23:24:14 +00:00
add import lists
This commit is contained in:
parent
abb0d7bd22
commit
0108a935ed
4 changed files with 60 additions and 1 deletions
|
@ -397,6 +397,24 @@ pub fn parse_use(
|
|||
|
||||
new_exports
|
||||
}
|
||||
ImportPatternMember::List { names } => {
|
||||
let mut output = vec![];
|
||||
|
||||
for (name, span) in names {
|
||||
let mut new_exports: Vec<(Vec<u8>, usize)> = exports
|
||||
.iter()
|
||||
.filter_map(|x| if &x.0 == name { Some(x.clone()) } else { None })
|
||||
.collect();
|
||||
|
||||
if new_exports.is_empty() {
|
||||
error = error.or(Some(ParseError::ExportNotFound(*span)))
|
||||
} else {
|
||||
output.append(&mut new_exports)
|
||||
}
|
||||
}
|
||||
|
||||
output
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
|
|
|
@ -1376,7 +1376,7 @@ pub fn parse_type(_working_set: &StateWorkingSet, bytes: &[u8]) -> Type {
|
|||
}
|
||||
|
||||
pub fn parse_import_pattern(
|
||||
working_set: &StateWorkingSet,
|
||||
working_set: &mut StateWorkingSet,
|
||||
span: Span,
|
||||
) -> (ImportPattern, Option<ParseError>) {
|
||||
let source = working_set.get_span_contents(span);
|
||||
|
@ -1409,6 +1409,38 @@ pub fn parse_import_pattern(
|
|||
},
|
||||
error,
|
||||
)
|
||||
} else if tail.starts_with(b"[") {
|
||||
let (result, err) = parse_list_expression(working_set, tail_span, &SyntaxShape::String);
|
||||
error = error.or(err);
|
||||
|
||||
let mut output = vec![];
|
||||
|
||||
match result {
|
||||
Expression {
|
||||
expr: Expr::List(list),
|
||||
..
|
||||
} => {
|
||||
for l in list {
|
||||
let contents = working_set.get_span_contents(l.span);
|
||||
output.push((contents.to_vec(), l.span));
|
||||
}
|
||||
|
||||
(
|
||||
ImportPattern {
|
||||
head,
|
||||
members: vec![ImportPatternMember::List { names: output }],
|
||||
},
|
||||
error,
|
||||
)
|
||||
}
|
||||
_ => (
|
||||
ImportPattern {
|
||||
head,
|
||||
members: vec![],
|
||||
},
|
||||
Some(ParseError::ExportNotFound(result.span)),
|
||||
),
|
||||
}
|
||||
} else {
|
||||
(
|
||||
ImportPattern {
|
||||
|
|
|
@ -4,6 +4,7 @@ use crate::Span;
|
|||
pub enum ImportPatternMember {
|
||||
Glob { span: Span },
|
||||
Name { name: Vec<u8>, span: Span },
|
||||
List { names: Vec<(Vec<u8>, Span)> },
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
|
|
|
@ -374,3 +374,11 @@ fn module_imports_4() -> TestResult {
|
|||
"not find import",
|
||||
)
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn module_imports_5() -> TestResult {
|
||||
run_test(
|
||||
r#"module foo { def a [] { 1 }; def b [] { 2 }; def c [] { 3 } }; use foo.[a, c]; c"#,
|
||||
"3",
|
||||
)
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue