make map ~a:(fun and map (fun consistent

Author: EmileTrotignon

lib/Params.ml

@@ -176,17 +176,10 @@ module Exp = struct
       | Some ((Labelled _ | Optional _), _, _) -> true
       | _ -> false
     in
-    let is_ctx_beginend =
-      match ctx0 with
-      | Exp {pexp_desc= Pexp_beginend _; _} -> true
-      | _ -> false
-    in
     if Conf.(c.fmt_opts.ocp_indent_compat.v) then
       if last_arg || is_labelled_arg then break 1 2 else str " "
     else if is_labelled_arg then break 1 2
-    else if last_arg then break 1 0
-    else if is_ctx_beginend then break 1 0
-    else str " "
+    else break 1 0
 
   let box_fun_decl_args ~ctx ~ctx0 ?(last_arg = false) ?epi c ~parens ~kw
       ~args ~annot =
@@ -217,11 +210,8 @@ module Exp = struct
             match ctx_is_apply_and_exp_is_arg ~ctx ~ctx0 with
             | Some (_, _, true) ->
                 (* Is last arg. *) hvbox ~name (if parens then 0 else 2)
-            | Some (Nolabel, _, false) ->
-                (* TODO: Inconsistent formatting of fun args. *)
-                hovbox ~name 0
-            | Some ((Labelled _ | Optional _), _, false) -> hvbox ~name 0
-            | None -> Fn.id
+            | Some (_, _, false) -> hvbox ~name 0
+            | None -> hvbox ~name 0
         in
         (box, not c.fmt_opts.wrap_fun_args.v)
     in

test/passing/refs.ahrefs/break_fun_decl-fit_or_vertical.ml.ref

@@ -136,8 +136,12 @@ let f
 
 let new_specialised_args =
   Variable.Map.mapi
-    (fun new_inner_var______ (definition : Definition.t) :
-         Flambda.specialised_to -> ())
+    (fun
+      new_inner_var______
+      (definition : Definition.t)
+      :
+      Flambda.specialised_to
+    -> ())
     foo
 
 let new_specialised_args =

test/passing/refs.ahrefs/break_fun_decl-smart.ml.ref

@@ -129,8 +129,12 @@ let f
 
 let new_specialised_args =
   Variable.Map.mapi
-    (fun new_inner_var______ (definition : Definition.t) :
-         Flambda.specialised_to -> ())
+    (fun
+      new_inner_var______
+      (definition : Definition.t)
+      :
+      Flambda.specialised_to
+    -> ())
     foo
 
 let new_specialised_args =

test/passing/refs.ahrefs/break_fun_decl-wrap.ml.ref

@@ -110,8 +110,12 @@ let f (module Store : Irmin.Generic_key.S with type repo = repo)
 
 let new_specialised_args =
   Variable.Map.mapi
-    (fun new_inner_var______ (definition : Definition.t) :
-         Flambda.specialised_to -> ())
+    (fun
+      new_inner_var______
+      (definition : Definition.t)
+      :
+      Flambda.specialised_to
+    -> ())
     foo
 
 let new_specialised_args =

test/passing/refs.ahrefs/break_fun_decl.ml.ref

@@ -110,8 +110,12 @@ let f (module Store : Irmin.Generic_key.S with type repo = repo)
 
 let new_specialised_args =
   Variable.Map.mapi
-    (fun new_inner_var______ (definition : Definition.t) :
-         Flambda.specialised_to -> ())
+    (fun
+      new_inner_var______
+      (definition : Definition.t)
+      :
+      Flambda.specialised_to
+    -> ())
     foo
 
 let new_specialised_args =

test/passing/refs.ahrefs/class_expr.ml.err

@@ -9,7 +9,8 @@ class c (* xx *) i (* yy *) = x
 class c =
   object
     method class_infos : 'a. ('a -> 'res) -> 'a class_infos -> 'res =
-      fun _a
+      fun
+        _a
         { pci_virt; pci_params; pci_name; pci_expr; pci_loc; pci_attributes } ->
         let pci_virt = self#virtual_flag pci_virt in
         let pci_params = self#list in

test/passing/refs.ahrefs/class_expr.ml.ref

@@ -394,9 +394,7 @@ let v =
 let v =
   map x
     begin fun
-      x
-      arggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggg
-    ->
+      x arggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggggg ->
     y
   end
 
@@ -457,10 +455,7 @@ let _ =
 
 let _ =
   lazy begin fun
-         xxxxxxxxxxxxxxxxxxxxxxx
-         yyyyyyyyyyyyyyyyyyy
-         zzzzzzzzzzzzzzzzzzzzzzzz
-       ->
+         xxxxxxxxxxxxxxxxxxxxxxx yyyyyyyyyyyyyyyyyyy zzzzzzzzzzzzzzzzzzzzzzzz ->
     print_endline xxxxxxxxx;
     f xxxxxxxxxx yyyyyyyyyyyyy zzzzzzzzzzzzzzzzzzzzz
   end

test/passing/refs.ahrefs/exp_grouping.ml.ref

@@ -1 +1 @@
-Warning: fun_decl-no-wrap-fun-args.ml:39 exceeds the margin
+Warning: fun_decl-no-wrap-fun-args.ml:41 exceeds the margin

test/passing/refs.ahrefs/fun_decl-no-wrap-fun-args.ml.err

@@ -5,25 +5,27 @@ let _ = fun (x : int) : int -> (some_large_computation : int)
 let fooo = List.foooo ~f:(fun foooo foooo : bool -> foooooooooooooooooooooo)
 
 let _ =
- fun (x : int)
-     (x : int)
-     (x : int)
-     (x : int)
-     (x : int)
-     :
-     fooooooooooooooooooooooooooo foooooooooooooo foooooooooo ->
+ fun
+   (x : int)
+   (x : int)
+   (x : int)
+   (x : int)
+   (x : int)
+   :
+   fooooooooooooooooooooooooooo foooooooooooooo foooooooooo ->
   some_large_computation
 
 let _ =
- fun (x : int)
-     (x : int)
-     (x : int)
-     (x : int)
-     (x : int)
-     (x : int)
-     (x : int)
-     :
-     fooooooooooooooooooooooooooo foooooooooooooo foooooooooo ->
+ fun
+   (x : int)
+   (x : int)
+   (x : int)
+   (x : int)
+   (x : int)
+   (x : int)
+   (x : int)
+   :
+   fooooooooooooooooooooooooooo foooooooooooooo foooooooooo ->
   some_large_computation
 
 let () =
@@ -94,17 +96,19 @@ let _ =
 
 let _ =
   let _ = () in
-  fun (context : Context.t)
-      ~(local_bins : origin Appendable_list.t Filename.Map.t Memo.Lazy.t)
-    ->
+  fun
+    (context : Context.t)
+    ~(local_bins : origin Appendable_list.t Filename.Map.t Memo.Lazy.t)
+  ->
     let _ = () in
     ()
 
 let _ =
   print_endline "foo";
-  fun (context : Context.t)
-      ~(local_bins : origin Appendable_list.t Filename.Map.t Memo.Lazy.t)
-    ->
+  fun
+    (context : Context.t)
+    ~(local_bins : origin Appendable_list.t Filename.Map.t Memo.Lazy.t)
+  ->
     let _ = () in
     ()