feat: trigger code action on function only if some annotation omitted

Author: ankddev

compiler-core/src/language_server/code_action.rs

@@ -1428,6 +1428,14 @@ impl<'ast> ast::visit::Visit<'ast> for AnnotateTopLevelDefinitions<'_> {
     }
 
     fn visit_typed_function(&mut self, fun: &'ast ast::TypedFunction) {
+        // Don't annotate already annotated arguments
+        let arguments_to_annotate = fun.arguments().iter().filter(|argument| argument.annotation.is_none());
+        let return_annotation_needed = fun.return_annotation.is_none();
+
+        if arguments_to_annotate.is_empty() && !return_annotation_needed {
+            return;
+        }
+
         // Create new printer to ignore type variables from other definitions
         let mut printer = Printer::new_without_type_variables(&self.module.ast.names);
         collect_type_variables(&mut printer, fun);
@@ -1446,20 +1454,15 @@ impl<'ast> ast::visit::Visit<'ast> for AnnotateTopLevelDefinitions<'_> {
         }
 
         // Annotate each argument separately
-        for argument in fun.arguments.iter() {
-            // Don't annotate the argument if it's already annotated
-            if argument.annotation.is_some() {
-                continue;
-            }
-
+        for argument in arguments_to_annotate {
             self.edits.insert(
                 argument.location.end,
                 format!(": {}", printer.print_type(&argument.type_)),
             );
         }
 
         // Annotate the return type if it isn't already annotated
-        if fun.return_annotation.is_none() {
+        if return_annotation_needed {
             self.edits.insert(
                 fun.location.end,
                 format!(" -> {}", printer.print_type(&fun.return_type)),