Added enumerable capabilities. Closes #5.

lib/rambling-trie.rb

@@ -3,6 +3,7 @@
   'children_hash_deferer',
   'compressor',
   'branches',
+  'enumerable',
   'node',
   'root',
   'version'

lib/rambling-trie/enumerable.rb

@@ -0,0 +1,20 @@
+module Rambling
+  module Trie
+    # Provides enumerable behavior to the Trie data structure.
+    module Enumerable
+      include ::Enumerable
+
+      alias_method :size, :count
+
+      # Calls block once for each of the words contained in the trie. If no block given, an Enumerator is returned.
+      def each(&block)
+        enumerator = Enumerator.new do |words|
+          words << as_word if terminal?
+          children.each { |key, child| child.each { |word| words << word } }
+        end
+
+        block.nil? ? enumerator : enumerator.each(&block)
+      end
+    end
+  end
+end

lib/rambling-trie/node.rb

@@ -5,6 +5,7 @@ class Node
       include ChildrenHashDeferer
       include Compressor
       include Branches
+      include Enumerable
 
       # Letter or letters corresponding to this node.
       # @return [Symbol, nil] the corresponding letter(s) or nil.

spec/lib/rambling-trie/enumerable_spec.rb

@@ -0,0 +1,37 @@
+require 'spec_helper'
+
+module Rambling
+  module Trie
+    describe Enumerable do
+      let(:root) { Root.new }
+      let(:words) { %w(add some words and another word) }
+
+      before :each do
+        words.each { |word| root << word.clone }
+      end
+
+      describe '#each' do
+        it 'returns an enumerator' do
+          root.each.should be_a(Enumerator)
+        end
+
+        it 'includes every word contained in the trie' do
+          root.each { |word| words.should include(word) }
+          root.count.should == words.count
+        end
+      end
+
+      describe '#size' do
+        it 'delegates to #count' do
+          root.size.should == words.size
+        end
+      end
+
+      it 'includes the core Enumerable module' do
+        root.all? { |word| words.include? word }.should be_true
+        root.any? { |word| word.start_with? 's' }.should be_true
+        root.to_a.should =~ words
+      end
+    end
+  end
+end