Improve completions inside of a pipeline

2021-09-10 20:07:18 +12:00
parent a8ba00b250
commit bfd05772ef
5 changed files with 16 additions and 7 deletions
--- a/crates/nu-parser/src/flatten.rs
+++ b/crates/nu-parser/src/flatten.rs
@ -10,6 +10,7 @@ pub enum FlatShape {
    Range,
    InternalCall,
    External,
+    ExternalArg,
    Literal,
    Operator,
    Signature,
@ -55,8 +56,14 @@ pub fn flatten_expression(
            }
            output
        }
-        Expr::ExternalCall(..) => {
-            vec![(expr.span, FlatShape::External)]
+        Expr::ExternalCall(name, args) => {
+            let mut output = vec![(*name, FlatShape::External)];
+
+            for arg in args {
+                output.push((*arg, FlatShape::ExternalArg));
+            }
+
+            output
        }
        Expr::Garbage => {
            vec![(expr.span, FlatShape::Garbage)]