rambling-trie 0.0.1 → 0.0.2

data/lib/invalid_trie_operation.rb

@@ -0,0 +1,4 @@
+module Rambling
+  class InvalidTrieOperation < Exception
+  end
+end

data/lib/rambling-trie.rb

@@ -1,3 +1,4 @@
+require File.join(File.dirname(__FILE__), 'invalid_trie_operation.rb')
 require File.join(File.dirname(__FILE__), 'trie_node.rb')
 require File.join(File.dirname(__FILE__), 'trie.rb')
 

data/lib/trie_node.rb

@@ -2,14 +2,17 @@ module Rambling
   class TrieNode
     attr_reader :letter, :children
 
-    def initialize(word)
+    def initialize(word, parent = nil)
       @letter = nil
+      @word = ''
+      @parent = parent
       @is_terminal = false
       @children = {}
 
       unless word.nil?
         @letter = word.slice!(0)
         @is_terminal = word.empty?
+        @word = get_parent_letter_string if terminal?
         add_branch_from(word)
       end
 
@@ -31,6 +34,11 @@ module Rambling
       @children.has_key?(key)
     end
 
+    def as_word
+      raise InvalidTrieOperation.new() unless @letter.nil? or terminal?
+      @word
+    end
+
     def add_branch_from(word)
       unless word.empty?
         first_letter = word.slice(0)
@@ -39,7 +47,7 @@ module Rambling
           word.slice!(0)
           @children[first_letter].add_branch_from(word)
         else
-          @children[first_letter] = TrieNode.new(word)
+          @children[first_letter] = TrieNode.new(word, self)
         end
       end
     end
@@ -54,6 +62,15 @@ module Rambling
       passes_condition(word) { |node, sliced_word| node.is_word?(sliced_word) }
     end
 
+    protected
+    def get_parent_letter_string
+      if @parent.nil?
+        @letter or ''
+      else
+        @parent.get_parent_letter_string + @letter
+      end
+    end
+
     private
     def passes_condition(word, &block)
       first_letter = word.slice!(0)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rambling-trie
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -17,9 +17,10 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- lib/rambling-trie.rb
-- lib/trie.rb
-- lib/trie_node.rb
+- ./lib/invalid_trie_operation.rb
+- ./lib/rambling-trie.rb
+- ./lib/trie.rb
+- ./lib/trie_node.rb
 homepage: http://rubygems.org/gems/rambling-trie
 licenses: []
 post_install_message: