Fix handling of stab comments

Author: doorgan

lib/elixir/lib/code/comments.ex

@@ -354,7 +354,6 @@ defmodule Code.Comments do
   defp merge_mixed_comments({:->, _, [left, right]} = quoted, state) do
     start_line = get_line(right)
     end_line = get_end_line({:__block__, state.parent_meta, [quoted]}, start_line)
-    block_start = get_line({:__block__, state.parent_meta, [quoted]})
 
     {right, comments} =
       case right do
@@ -376,20 +375,6 @@ defmodule Code.Comments do
           end
       end
 
-    {quoted, comments} =
-      case left do
-        [] ->
-          {leading_comments, comments} =
-            split_leading_comments(comments, block_start, start_line)
-
-          quoted = append_comments(quoted, :leading_comments, leading_comments)
-
-          {quoted, comments}
-
-        _ ->
-          {quoted, comments}
-      end
-
     quoted = put_args(quoted, [left, right])
 
     {quoted, %{state | comments: comments}}

lib/elixir/lib/code/formatter.ex

@@ -1886,9 +1886,13 @@ defmodule Code.Formatter do
     comments = merge_algebra_with_comments(comments, @empty)
 
     # If there are any comments before the ->, we hoist them up above the fn
-    doc = case comments do
+    doc =
+      case comments do
       [] -> doc
-      [comments] -> line(comments, doc)
+      [comment] -> line(comment, doc)
+      comments ->
+        comments_doc = comments |> Enum.reduce(&line(&2, &1)) |> force_unfit()
+        line(comments_doc, doc)
     end
 
     {doc, state}