Flatten code action removes comments

[MrQubo]

Describe the bug

"Flatten nested attribute set" code action removes comments from code.

To Reproduce

{
  a = {
    b =
      # IMPORTANT COMMENT
      2;
  };
}

Put cursor on a and run "Flatten nested attribute set" code action.
Result:

{
  a.b = 2;
}

Expected behavior

{
  a.b =
    # IMPORTANT COMMENT
    2;
}

[MrQubo]

More complicated test case:

{
  a = {
    b.c =
      # IMPORTANT COMMENT
      2;
    # AND COMMENT HERE
    d =
      # ANOTHER COMMENT
      {
        # NESTED COMMENT TOO
        e = 1;
      } # WHAT ABOUT HERE
    ; # AND HERE
  };
}