module Rambling
  module TrieCompressor
    def compressed?
      if instance_variable_defined?(:@is_compressed)
        @is_compressed
      else
        @parent.nil? ? false : @parent.compressed?
      end
    end

    protected

    def compress_own_tree!
      if @children.size == 1 and not terminal? and not @letter.nil?
        merge_with!(@children.values.first)
        compress_own_tree!
      end

      @children.values.each { |node| node.compress_own_tree! }

      self
    end

    private

    def merge_with!(child)
      new_letter = (@letter.to_s + child.letter.to_s).to_sym

      rehash_on_parent!(@letter, new_letter)
      redefine_self!(new_letter, child)

      @children.values.each { |node| node.parent = self }
    end

    def rehash_on_parent!(old_letter, new_letter)
      return if @parent.nil?

      @parent.delete(old_letter)
      @parent[new_letter] = self
    end

    def redefine_self!(new_letter, merged_node)
      @letter = new_letter
      @children = merged_node.children
      @is_terminal = merged_node.terminal?
    end
  end
end