Stats module for thor

stats.thor

module Count
  def count_by
    self.inject({}) do |memo, elem|
      key = yield elem
      memo[key] ||= 0
      memo[key] += 1
      memo
    end
  end
end

class Stats < Thor
  desc :stats, "Calculates the basic stats for stdin"
  def stats
    STDIN.inject({:count => 0, :min => nil, :max => nil, :sum => 0}) do |stats, line|
       num = line.to_f
       stats[:min] ||= num
       stats[:max] ||= num
       stats[:min] = num if num < stats[:min]
       stats[:max] = num if num > stats[:max]
       stats[:sum] += num
       stats[:count] += 1
       puts stats.to_yaml
     end
  end

  desc :count_by, "Count number of occurence of the same line content"
  def count_by
    STDIN.extend(Count)
    counts = STDIN.count_by do |line|
      line.chomp
    end
    puts counts.to_yaml
    puts "Total: #{counts.inject{|sum,(key, value)| sum += value;sum}}"
  end
end