EmbDoc fixes

Author: kddnewton

lib/syntax_tree/node.rb

@@ -4090,7 +4090,8 @@ def deconstruct_keys(_keys)
     def format(q)
       q.group do
         q.group do
-          q.text("def ")
+          q.text("def")
+          q.text(" ") if target || name.comments.empty?
 
           if target
             q.format(target)
@@ -4872,6 +4873,25 @@ class EmbDoc < Node
     def initialize(value:, location:)
       @value = value
       @location = location
+
+      @leading = false
+      @trailing = false
+    end
+
+    def leading!
+      @leading = true
+    end
+
+    def leading?
+      @leading
+    end
+
+    def trailing!
+      @trailing = true
+    end
+
+    def trailing?
+      @trailing
     end
 
     def inline?
@@ -4908,7 +4928,13 @@ def deconstruct_keys(_keys)
     end
 
     def format(q)
-      q.trim
+      if (q.parent.is_a?(DefNode) && q.parent.endless?) ||
+           q.parent.is_a?(Statements)
+        q.trim
+      else
+        q.breakable_return
+      end
+
       q.text(value)
     end
 
@@ -10465,6 +10491,7 @@ def deconstruct_keys(_keys)
 
     def format(q)
       q.text(":")
+      q.text("\\") if value.comments.any?
       q.format(value)
     end
 

test/fixtures/call.rb

@@ -60,3 +60,8 @@
 %
 a b do
 end.c d
+%
+self.
+=begin
+=end
+  to_s

test/fixtures/def.rb

@@ -23,3 +23,19 @@ def foo() # comment
 def foo( # comment
 )
 end
+%
+def
+=begin
+=end
+a
+end
+%
+def a()
+=begin
+=end
+=1
+-
+def a() =
+=begin
+=end
+  1

test/fixtures/symbols.rb

@@ -19,3 +19,8 @@
 %I[foo] # comment
 %
 %I{foo[]}
+%
+:\
+=begin
+=end
+symbol