class SensitiveWords

# File lib/sensitive_words.rb, line 16
def get_dict_file_hash(path)
  tree = {}
  file = File.open(path, 'r')
  if file
    file.each_line do |line|
      line = line.chomp
      next if line.empty?
      node = nil
      line.chars.each do |c|
        if node
          node[c] ||= {}
          node = node[c]
        else
          tree[c] ||= {}
          node = tree[c]
        end
      end
      node[:end] = :id
    end
  end
  tree
ensure
  file.close if file
end

# File lib/sensitive_words.rb, line 8
def load_dict(dict_path)
  new_dict = get_dict_file_hash(dict_path)
  dict = @@dict.merge new_dict
  @@dict = dict
rescue Errno::ENOENT => boom
  puts "#{boom.class} - #{boom.message}"
end

# File lib/sensitive_words.rb, line 53
def initialize(input)
  @input = input
  @words = []
end

# File lib/sensitive_words.rb, line 41
def sensitive_words(input,max=nil)
  ins = SensitiveWords.new(input)
  max = max.to_i
  if max > 0
    ins.sensitive_words(max)
  else
    ins.all_sensitive_words
  end
end

# File lib/sensitive_words.rb, line 78
def all_sensitive_words
  @node, @words = @@dict, []
  @word, @queue = '', []

  @input.chars.each do |char|
    loop do 
      break if @queue.empty?
      chr = @queue.shift
      process_check(chr, true)
    end
    process_check(char)
  end

  process_check('')
  @words
end

# File lib/sensitive_words.rb, line 59
def sensitive_words(max)
  @node, @words = @@dict, []
  @word, @queue = '', []

  @input.chars.each do |char|
    break if @words.size >= max
    loop do 
      break if @queue.empty?
      chr = @queue.shift
      process_check(chr, true)
    end
    process_check(char)
  end

  process_check('')
  @words.first(max)
end

# File lib/sensitive_words.rb, line 97
def process_check(char,queuing=false)

  match, word = nil, nil

  if @node[char]
    @word << char
    @node = @node[char]
    match = :id
  else
    if @node[:end]
      word = @word
    end
    lth = @word.length
    if lth > 0
      if queuing
        @queue.unshift char
      else
        if lth > 1
          @queue += @word.chars.last(lth-1)
        end
        @queue << char
      end
    end

    @node = @@dict
    @word = ''
  end

  if !match && word
    @words << word
  end
end