exclude functions with capture holes

The existing Expand Function Capture refactor does this.

Author: treuherz

language-server/src/code_action.rs

@@ -11087,9 +11087,10 @@ impl<'ast> ast::visit::Visit<'ast> for WrapInAnonymousFunction<'ast> {
             return;
         }
 
-        let is_excluded = match expression {
-            TypedExpr::Fn { kind, .. } if kind.is_anonymous() => true,
-            _ => false,
+        let is_excluded = if let TypedExpr::Fn { kind, .. } = expression {
+            kind.is_anonymous() || kind.is_capture()
+        } else {
+            false
         };
 
         if let Type::Fn { arguments, .. } = &*expression.type_()

language-server/src/tests/action.rs

@@ -7272,6 +7272,27 @@ pub fn add(n, m) { n + m }
     );
 }
 
+#[test]
+fn expand_function_capture_in_pipeline() {
+    assert_code_action!(
+        EXPAND_FUNCTION_CAPTURE,
+        "pub fn main() {
+  1 |> wibble(2, _) |> wobble
+}
+
+fn wibble(a, b) {
+  todo
+}
+
+fn wobble(i) {
+  todo
+}
+
+",
+        find_position_of("wibble").to_selection()
+    );
+}
+
 #[test]
 fn generate_dynamic_decoder() {
     assert_code_action!(
@@ -12546,48 +12567,6 @@ fn wobble(i) {
     );
 }
 
-#[test]
-fn wrap_multiargument_pipeline_step_in_anonymous_function() {
-    assert_code_action!(
-        WRAP_IN_ANONYMOUS_FUNCTION,
-        "pub fn main() {
-  1 |> wibble(2) |> wobble
-}
-
-fn wibble(a, b) {
-  todo
-}
-
-fn wobble(i) {
-  todo
-}
-
-",
-        find_position_of("wibble").to_selection()
-    );
-}
-
-#[test]
-fn wrap_capturing_pipeline_step_in_anonymous_function() {
-    assert_code_action!(
-        WRAP_IN_ANONYMOUS_FUNCTION,
-        "pub fn main() {
-  1 |> wibble(2, _) |> wobble
-}
-
-fn wibble(a, b) {
-  todo
-}
-
-fn wobble(i) {
-  todo
-}
-
-",
-        find_position_of("wibble").to_selection()
-    );
-}
-
 #[test]
 fn wrap_final_pipeline_step_in_anonymous_function() {
     assert_code_action!(

…pipeline_step_in_anonymous_function.snap …expand_function_capture_in_pipeline.snaplanguage-server/src/tests/snapshots/gleam_language_server__tests__action__wrap_capturing_pipeline_step_in_anonymous_function.snap renamed to language-server/src/tests/snapshots/gleam_language_server__tests__action__expand_function_capture_in_pipeline.snap language-server/src/tests/snapshots/gleam_language_server__tests__action__wrap_capturing_pipeline_step_in_anonymous_function.snap renamed to language-server/src/tests/snapshots/gleam_language_server__tests__action__expand_function_capture_in_pipeline.snap

@@ -20,7 +20,7 @@ fn wobble(i) {
 
 ----- AFTER ACTION
 pub fn main() {
-  1 |> fn(int) { wibble(2, _)(int) } |> wobble
+  1 |> fn(int) { wibble(2, int) } |> wobble
 }
 
 fn wibble(a, b) {