don't try to unwrap functions with comments in

Author: treuherz

compiler-core/src/language_server/code_action.rs

@@ -8411,6 +8411,15 @@ impl<'a> UnwrapAnonymousFunction<'a> {
             _ => return,
         }
 
+        // We can't apply to functions with comments in (yet)
+        if self
+            .module
+            .extra
+            .has_comment_between(location.start, location.end)
+        {
+            return;
+        }
+
         // We can only apply to anonymous functions containing a single function call
         let [
             TypedStatement::Expression(TypedExpr::Call {

compiler-core/src/language_server/tests/action.rs

@@ -10059,7 +10059,43 @@ fn op(first a, second b) {
     );
 }
 
+#[test]
+fn dont_unwrap_anonymous_function_with_comment_after() {
+    assert_no_code_actions!(
+        UNWRAP_ANONYMOUS_FUNCTION,
+        "pub fn main() {
+  fn(a) {
+    op(a)
+    // look out!
+  }
+}
+
+fn op(a) {
+  todo
+}
+",
+        find_position_of("fn(a)").to_selection()
+    );
+}
 
+#[test]
+fn dont_unwrap_anonymous_function_with_comment_before() {
+    assert_no_code_actions!(
+        UNWRAP_ANONYMOUS_FUNCTION,
+        "pub fn main() {
+  fn(a) {
+    // look out!
+    op(a)
+  }
+}
+
+fn op(a) {
+  todo
+}
+",
+        find_position_of("fn(a)").to_selection()
+    );
+}
 
 #[test]
 fn unwrap_anonymous_function_unavailable_when_args_discarded() {

compiler-core/src/parse/extra.rs

@@ -40,6 +40,7 @@ impl ModuleExtra {
         self.first_comment_between(start, end).is_some()
     }
 
+    // TODO: this won't necessarily find the first comment, just any comment
     pub(crate) fn first_comment_between(&self, start: u32, end: u32) -> Option<SrcSpan> {
         self.comments
             .binary_search_by(|comment| {