Small refactor

crystal-lang · Sep 18, 2016 · 25a52af · 25a52af
1 parent 56325e0
commit 25a52af
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/src/compiler/crystal/tools/formatter.cr b/src/compiler/crystal/tools/formatter.cr
@@ -2776,20 +2776,23 @@ module Crystal
         if body.is_a?(Expressions)
           body.expressions = body.expressions[to_skip..-1]
           if body.expressions.empty?
-            case node
-            when Def, ExceptionHandler
-              node.body = Nop.new
-            end
+            set_body(node, Nop.new)
           end
         else
-          case node
-          when Def, ExceptionHandler
-            node.body = Nop.new
-          end
+          set_body(node, Nop.new)
         end
       end
     end
 
+    def set_body(node, body)
+      case node
+      when Def
+        node.body = body
+      when ExceptionHandler
+        node.body = body
+      end
+    end
+
     def visit(node : IsA)
       if node.nil_check?
         accept node.obj