lib/csl/pretty_printer.rb in csl-1.0.0.pre1 vs lib/csl/pretty_printer.rb in csl-1.0.0.pre2

@@ -6,27 +6,29 @@
     end
     
     def to_xml
       tags.flatten.join
     end
-    
+
     def pretty_print
-      pp(tags).join("\n")
+      preamble << tags.map { |t| pp t }.join("\n")
     end
 
     private
     
 		def tabwidth
 			2
 		end
 		
-    def pp(tags, level = 0)
-      tags.map do |tag|
-        if tag.respond_to?(:map)
-          pp tag, level + 1
-        else
-          ' ' * (level * tabwidth) + tag.to_s
-        end
+		def preamble
+		  ''
+		end
+		
+    def pp(tag, level = 0)
+      if tag.is_a?(Array)
+        tag.map { |t| pp t, level + 1 }.join("\n")
+      else
+        (' ' * (level * tabwidth)) << tag.to_s
       end
     end
     
   end
 end
\ No newline at end of file