rest argument shouldn't report arity mismatches in language server

mattwparas · Dec 25, 2023 · a3e9c46 · a3e9c46
1 parent 5721aea
commit a3e9c46
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/crates/steel-language-server/src/diagnostics.rs b/crates/steel-language-server/src/diagnostics.rs
@@ -169,8 +169,6 @@ impl DiagnosticGenerator for StaticArityChecker {
                         .trim_end_matches(interned.resolve())
                         .trim_end_matches("__%#__");
 
-                    log::debug!("Searching for: {} in {}", name, module_path_to_check);
-
                     let module = context
                         .engine
                         .modules()
@@ -195,7 +193,17 @@ impl DiagnosticGenerator for StaticArityChecker {
                     // Don't include rest args for now
                     if let Some(d) = top_level_define {
                         if let ExprKind::LambdaFunction(l) = &d.body {
-                            if !l.rest {
+                            let mut rest = false;
+
+                            for arg in &l.args {
+                                if let Some(ident) = arg.atom_identifier() {
+                                    if ident.resolve() == "." {
+                                        rest = true;
+                                    }
+                                }
+                            }
+
+                            if !rest && !l.rest {
                                 arity_checker.known_functions.insert(id, l.args.len());
                             }
                         }