nodifier 0.1.3 → 0.1.4

data/lib/node.rb

@@ -1,4 +1,5 @@
 require File.dirname(__FILE__) + '/indentation'
+require File.dirname(__FILE__) + '/stopper'
 require File.dirname(__FILE__) + '/xml_node_formatter'
 
 class Node
@@ -19,11 +20,19 @@ class Node
       return to_s(indent_level) { |node| node.label }
     end
 
-    s = indent(formatter_block.call(self), indent_level)
+    stopper = Stopper.new
+    case formatter_block.arity
+    when 1:
+      s = indent(formatter_block.call(self), indent_level)
+    when 2:
+      s = indent(formatter_block.call([self, stopper]), indent_level)
+    end
 
-    @children.each do |child|
-      s << "\n"
-      s << child.to_s(indent_level + 1, &formatter_block)
+    unless stopper.stopped?
+      @children.each do |child|
+        s << "\n"
+        s << child.to_s(indent_level + 1, &formatter_block)
+      end
     end
 
     s

data/lib/stopper.rb

@@ -0,0 +1,13 @@
+class Stopper
+  def initialize
+    @stopped = false
+  end
+
+  def stop!
+    @stopped = true
+  end
+
+  def stopped?
+    @stopped
+  end
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: nodifier
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 19
   prerelease: 
   segments: 
   - 0
   - 1
-  - 3
-  version: 0.1.3
+  - 4
+  version: 0.1.4
 platform: ruby
 authors: 
 - Daniel Tao
@@ -32,6 +32,7 @@ files:
 - lib/node.rb
 - lib/node_parser.rb
 - lib/nodifier.rb
+- lib/stopper.rb
 - lib/xml_node_formatter.rb
 has_rdoc: true
 homepage: http://github.com/dtao/nodifier