keep more formatting for extracted function.

joodie · Nov 4, 2011 · 99f406d · 99f406d
1 parent 9ae6a97
commit 99f406d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/clojure_refactoring/extract_method.clj b/src/clojure_refactoring/extract_method.clj
@@ -51,12 +51,14 @@
 
 (defn- make-fn-node [name args body]
   "Creates an ast representing the new function"
-  (format-ast
-   (ast/list-without-whitespace
-    (ast/symbol 'defn)
+  (ast/list
+   [(ast/symbol 'defn)
+    ast/whitespace
     name
+    (ast/make-node :whitespace ["\n  "])
     (ast/vector args)
-    (ast/strip-whitespace body))))
+    (ast/make-node :whitespace ["\n  "])
+    body]))
 
 (defn call-extracted [body toplevel extracted]
   (ast/tree-replace