Fix unwanted alignment after comment

CHANGES.md

@@ -30,6 +30,7 @@ profile. This started with version 0.26.0.
 - \* Force a break around comments following an infix operator (fix non-stabilizing comments) (#2478, @gpetiot)
 - \* Fix the indentation of tuples in attributes and extensions (#2488, @Julow)
 - Fix unstable comment around docked functor argument (#2506, @Julow)
+- Fix unwanted alignment after comment (#2507, @Julow)
 
 ## 0.26.1 (2023-09-15)
 

lib/Cmts.ml

@@ -615,8 +615,8 @@ let fmt_cmts_aux t (conf : Conf.t) cmts ~fmt_code pos =
          | _ -> noop ) )
 
 (** Format comments for loc. *)
-let fmt_cmts t conf ~fmt_code ?pro ?epi ?(eol = Fmt.fmt "@\n") ?(adj = eol)
-    found loc pos =
+let fmt_cmts t conf ~fmt_code ?pro ?epi ?(eol = Fmt.fmt "@;<1000 0>")
+    ?(adj = eol) found loc pos =
   let open Fmt in
   match found with
   | None | Some [] -> noop

lib/Fmt_ast.ml

@@ -3868,10 +3868,11 @@ and fmt_module c ctx ?rec_ ?epi ?(can_sparse = false) keyword ?(eqty = "=")
       ~fit:true attrs
   in
   let intro =
-    str keyword
-    $ fmt_extension_suffix c ext
-    $ fmt_attributes c ~pre:(Break (1, 0)) attrs_before
-    $ fmt_if rec_flag " rec" $ str " " $ fmt_str_loc_opt c name
+    hvbox 2
+      ( str keyword
+      $ fmt_extension_suffix c ext
+      $ fmt_attributes c ~pre:(Break (1, 0)) attrs_before
+      $ fmt_if rec_flag " rec" $ fmt "@ " $ fmt_str_loc_opt c name )
   in
   let compact =
     Poly.(c.conf.fmt_opts.let_module.v = `Compact) || not can_sparse
@@ -4188,7 +4189,7 @@ and fmt_module_expr ?(dock_struct = true) c ({ast= m; _} as xmod) =
   | Pmod_unpack (e, ty1, ty2) ->
       let package_type sep (lid, cstrs) =
         break 1 (Params.Indent.mod_unpack_annot c.conf)
-        $ hvbox 0
+        $ hovbox 0
             ( hovbox 0 (str sep $ fmt_longident_loc c lid)
             $ fmt_package_type c ctx cstrs )
       in

lib/Params.ml

@@ -598,9 +598,7 @@ let get_if_then_else (c : Conf.t) ~first ~last ~parens_bch ~parens_prev_bch
                 ( fmt_or_k first
                     (str "if" $ fmt_opt fmt_extension_suffix)
                     (str "else if")
-                $ fmt_attributes
-                $ fmt_or (Option.is_some fmt_extension_suffix) "@ " " "
-                $ fmt_cond xcnd )
+                $ fmt_attributes $ fmt "@ " $ fmt_cond xcnd )
               $ fmt "@ " )
       ; box_keyword_and_expr=
           (fun k -> hvbox 2 (fmt_or (Option.is_some xcond) "then" "else" $ k))
@@ -610,7 +608,7 @@ let get_if_then_else (c : Conf.t) ~first ~last ~parens_bch ~parens_prev_bch
             ~wrap_breaks:
               (get_parens_breaks ~opn_hint_indent:0
                  ~cls_hint:((1, 0), (1000, -2)) )
-      ; box_expr= Some false
+      ; box_expr= None
       ; expr_pro= None
       ; expr_eol= None
       ; branch_expr

test/passing/tests/comments-no-wrap.ml.err

@@ -1,4 +1,4 @@
 Warning: tests/comments.ml:186 exceeds the margin
 Warning: tests/comments.ml:190 exceeds the margin
 Warning: tests/comments.ml:250 exceeds the margin
-Warning: tests/comments.ml:430 exceeds the margin
+Warning: tests/comments.ml:434 exceeds the margin

test/passing/tests/comments-no-wrap.ml.ref

@@ -252,8 +252,11 @@ let%map
     _ =
   ()
 
-type t = < (* a *)
-         a: int [@atr] (* b *) ; b: int (* c *) >
+type t =
+  < (* a *)
+  a: int [@atr] (* b *)
+  ; b: int
+  (* c *) >
 
 type t = < a: int (* a *) ; (* b *) .. (* c *) >
 
@@ -277,8 +280,10 @@ let _ =
       | None -> do_something ()
       | Some _ -> () (* do nothing *) )
 
-let _ = try_with (fun () -> (* comment before *)
-                            a ; b (* after b *) )
+let _ =
+  try_with (fun () ->
+      (* comment before *)
+      a ; b (* after b *) )
 
 let _ =
   match x with
@@ -296,8 +301,7 @@ type t =
 
 type t =
   (* | B *)
-  | A
-  (* A *)
+  | A (* A *)
   | C
 
 type t =

test/passing/tests/comments.ml.ref

@@ -254,8 +254,11 @@ let%map
     _ =
   ()
 
-type t = < (* a *)
-         a: int [@atr] (* b *) ; b: int (* c *) >
+type t =
+  < (* a *)
+  a: int [@atr] (* b *)
+  ; b: int
+  (* c *) >
 
 type t = < a: int (* a *) ; (* b *) .. (* c *) >
 
@@ -279,8 +282,10 @@ let _ =
       | None -> do_something ()
       | Some _ -> () (* do nothing *) )
 
-let _ = try_with (fun () -> (* comment before *)
-                            a ; b (* after b *) )
+let _ =
+  try_with (fun () ->
+      (* comment before *)
+      a ; b (* after b *) )
 
 let _ =
   match x with
@@ -298,8 +303,7 @@ type t =
 
 type t =
   (* | B *)
-  | A
-  (* A *)
+  | A (* A *)
   | C
 
 type t =

test/passing/tests/comments_args.ml.ref

@@ -8,7 +8,8 @@ let emit_wrapper_function =
     false (* is_pair_generator *)
     hoisted true (* no_injection *)
     true (* inout_wrapper *)
-    is_interceptable false (* is_memoize_impl *)
+    is_interceptable false
+    (* is_memoize_impl *)
     Rx.NonRx false
 
 [@@@ocamlformat "wrap-fun-args=false"]

test/passing/tests/exp_grouping.ml.ref

@@ -279,8 +279,8 @@ let _ =
 
 let () =
   if a
-  then begin b
-    (* asd *)
+  then begin
+    b (* asd *)
   end
 
 [@@@ocamlformat "if-then-else=k-r"]

test/passing/tests/extensions-indent.ml.ref

@@ -124,7 +124,10 @@ let _ = try%lwt Lwt.return 2 with _ -> assert false
 
 let _ =
   (* foooooooooooo *)
-  try%lwt (* fooooooooooo *) Lwt.return 2 with _ -> assert false
+  try%lwt
+    (* fooooooooooo *)
+    Lwt.return 2
+  with _ -> assert false
 
 let _ =
   try%lwt

test/passing/tests/extensions.ml.ref

@@ -124,7 +124,10 @@ let _ = try%lwt Lwt.return 2 with _ -> assert false
 
 let _ =
   (* foooooooooooo *)
-  try%lwt (* fooooooooooo *) Lwt.return 2 with _ -> assert false
+  try%lwt
+    (* fooooooooooo *)
+    Lwt.return 2
+  with _ -> assert false
 
 let _ =
   try%lwt

test/passing/tests/extensions_exp_grouping.ml.ref

@@ -76,5 +76,8 @@ let a =
   (* test *)
   Lwt.return () ;%lwt Lwt.return 1
 
-let a = f ((* test *)
-           Lwt.return () ;%lwt Lwt.return 1)
+let a =
+  f
+    ( (* test *)
+      Lwt.return () ;%lwt
+      Lwt.return 1 )

test/passing/tests/ite-compact.ml.ref

@@ -119,9 +119,11 @@ let foo =
 let foo = if cmp < 0 then (* foo *) a + b else (* foo *) a - b
 
 let foo =
-  if cmp < 0 then (* ast higher precedence than context: no parens *)
+  if cmp < 0 then
+    (* ast higher precedence than context: no parens *)
     false
-  else if cmp > 0 then (* context higher prec than ast: add parens *)
+  else if cmp > 0 then
+    (* context higher prec than ast: add parens *)
     true
   else if Poly.(assoc_of_prec prec_ast = which_child && which_child <> Non)
   then foo
@@ -137,3 +139,14 @@ let _ =
     xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
   else fun xxxxxxxxxxxxxxxxx yyyyyyyyyyyyyyyyyyyy zzzzzzzzzzz ->
     xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
+
+let _ =
+  if
+    (* foo *)
+    foo
+  then 0
+  else if
+    (* bar *)
+    bar
+  then 1
+  else 2

test/passing/tests/ite-compact_closing.ml.ref

@@ -134,9 +134,11 @@ let foo =
 let foo = if cmp < 0 then (* foo *) a + b else (* foo *) a - b
 
 let foo =
-  if cmp < 0 then (* ast higher precedence than context: no parens *)
+  if cmp < 0 then
+    (* ast higher precedence than context: no parens *)
     false
-  else if cmp > 0 then (* context higher prec than ast: add parens *)
+  else if cmp > 0 then
+    (* context higher prec than ast: add parens *)
     true
   else if Poly.(assoc_of_prec prec_ast = which_child && which_child <> Non)
   then foo
@@ -152,3 +154,14 @@ let _ =
     xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
   else fun xxxxxxxxxxxxxxxxx yyyyyyyyyyyyyyyyyyyy zzzzzzzzzzz ->
     xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
+
+let _ =
+  if
+    (* foo *)
+    foo
+  then 0
+  else if
+    (* bar *)
+    bar
+  then 1
+  else 2

test/passing/tests/ite-fit_or_vertical.ml.ref

@@ -168,3 +168,17 @@ let _ =
   else
     fun xxxxxxxxxxxxxxxxx yyyyyyyyyyyyyyyyyyyy zzzzzzzzzzz ->
   xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
+
+let _ =
+  if
+    (* foo *)
+    foo
+  then
+    0
+  else if
+    (* bar *)
+    bar
+  then
+    1
+  else
+    2

test/passing/tests/ite-fit_or_vertical_closing.ml.ref

@@ -180,3 +180,17 @@ let _ =
   else
     fun xxxxxxxxxxxxxxxxx yyyyyyyyyyyyyyyyyyyy zzzzzzzzzzz ->
   xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
+
+let _ =
+  if
+    (* foo *)
+    foo
+  then
+    0
+  else if
+    (* bar *)
+    bar
+  then
+    1
+  else
+    2

test/passing/tests/ite-fit_or_vertical_no_indicate.ml.ref

@@ -168,3 +168,17 @@ let _ =
   else
     fun xxxxxxxxxxxxxxxxx yyyyyyyyyyyyyyyyyyyy zzzzzzzzzzz ->
   xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
+
+let _ =
+  if
+    (* foo *)
+    foo
+  then
+    0
+  else if
+    (* bar *)
+    bar
+  then
+    1
+  else
+    2

test/passing/tests/ite-kr.ml.ref

@@ -203,3 +203,17 @@ let _ =
   else
     fun xxxxxxxxxxxxxxxxx yyyyyyyyyyyyyyyyyyyy zzzzzzzzzzz ->
   xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
+
+let _ =
+  if
+    (* foo *)
+    foo
+  then
+    0
+  else if
+    (* bar *)
+    bar
+  then
+    1
+  else
+    2

test/passing/tests/ite-kr_closing.ml.ref

@@ -213,3 +213,17 @@ let _ =
   else
     fun xxxxxxxxxxxxxxxxx yyyyyyyyyyyyyyyyyyyy zzzzzzzzzzz ->
   xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
+
+let _ =
+  if
+    (* foo *)
+    foo
+  then
+    0
+  else if
+    (* bar *)
+    bar
+  then
+    1
+  else
+    2

test/passing/tests/ite-kw_first.ml.ref

@@ -45,8 +45,9 @@ f
 ;;
 
 f
-  ( if and_ even
-         loooooooooooooooooooooooooooooooooooooooooooooooooooooooooonger
+  ( if
+      and_ even
+        loooooooooooooooooooooooooooooooooooooooooooooooooooooooooonger
     then ()
     else () )
 
@@ -135,10 +136,12 @@ let foo = if cmp < 0 then (* foo *) a + b else (* foo *) a - b
 
 let foo =
   if cmp < 0
-  then (* ast higher precedence than context: no parens *)
+  then
+    (* ast higher precedence than context: no parens *)
     false
   else if cmp > 0
-  then (* context higher prec than ast: add parens *)
+  then
+    (* context higher prec than ast: add parens *)
     true
   else if Poly.(assoc_of_prec prec_ast = which_child && which_child <> Non)
   then foo
@@ -156,7 +159,18 @@ let _ =
   if x
   then
     fun xxxxxxxxxxxxxxxxx yyyyyyyyyyyyyyyyyyyy zzzzzzzzzzz ->
-    xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
+      xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
   else
     fun xxxxxxxxxxxxxxxxx yyyyyyyyyyyyyyyyyyyy zzzzzzzzzzz ->
-    xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
+      xxxxxxxxx yyyyyyyyyy zzzzzzzzzzzz
+
+let _ =
+  if
+    (* foo *)
+    foo
+  then 0
+  else if
+    (* bar *)
+    bar
+  then 1
+  else 2