lib/synvert/rewriter/action.rb in synvert-0.0.9 vs lib/synvert/rewriter/action.rb in synvert-0.0.10

@@ -39,10 +39,22 @@
     def end_pos
       @node.loc.expression.end_pos
     end
 
     def rewritten_code
-      @node.rewritten_source(@code)
+      if rewritten_source.split("\n").length > 1
+        "\n\n" + rewritten_source.split("\n").map { |line|
+          indent(@node) + line
+        }.join("\n")
+      else
+        rewritten_source
+      end
+    end
+
+  private
+
+    def indent(node)
+      ' ' * node.indent
     end
   end
 
   class Rewriter::AppendAction < Rewriter::Action
     def begin_pos