Fix formatting of floating docs between recursive modules (#2455)

This fixes a bug that was introduced in commit bc91e42 (pr #2451)

Author: EmileTrotignon

CHANGES.md

@@ -135,7 +135,7 @@ profile. This started with version 0.26.0.
 - JaneStreet profile: doesn't align infix ops with open paren (#2204, @gpetiot)
 - Re-use the type let_binding from the parser instead of value_binding, improve the spacing of let-bindings regarding of having extension or comments (#2219, @gpetiot)
 - The `ocamlformat` package now only contains the binary, the library is available through the `ocamlformat-lib` package (#2230, @gpetiot)
-- The position of module and module type attributes is now preserved. (#2451, @emiletrotignon)
+- The position of module and module type attributes is now preserved. (#2451, #2455, @emiletrotignon)
 
 ### Added
 

test/passing/tests/comment_in_modules.ml

@@ -25,3 +25,14 @@ module A (A:sig end) (* comment *) (B: sig end) : sig end = struct end
 module A (A:sig end) (* comment *) : sig end = struct end
 
 module (* comment *) A (A : sig end) : sig end = struct end
+
+module rec A : A =
+  struct
+
+  end
+
+(** floatting *)
+
+and B : B = struct end
+(** about b *)
+

test/passing/tests/comment_in_modules.ml.ref

@@ -27,3 +27,10 @@ module A (A : sig end) (* comment *) (B : sig end) : sig end = struct end
 module A (A : sig end) : sig end = (* comment *) struct end
 
 module (* comment *) A (A : sig end) : sig end = struct end
+
+module rec A : A = struct end
+
+(** floatting *)
+
+(** about b *)
+and B : B = struct end

vendor/parser-extended/docstrings.ml

@@ -171,7 +171,7 @@ let add_text_attrs dsl attrs =
 (** {!add_text_attrs} but operate on [ext_attrs]. *)
 let add_text_attrs' text attrs' =
   let open Parsetree in
-  { attrs' with attrs_before = add_text_attrs text attrs'.attrs_before }
+  { attrs' with attrs_after = add_text_attrs text attrs'.attrs_after }
 
 (* Find the first non-info docstring in a list, attach it and return it *)
 let get_docstring ~info dsl =