# Copyright (C) 2014 Harald Eilertsen
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

module SalesReporter
  module Reports
    class SalesByCountry
      def initialize
        @countries = Hash.new { |h,k| h[k] = { :count => 0, :total => 0, :sales => [] } }
      end

      def add_sale(sale, row)
        c = row['Country']
        @countries[c][:sales] << sale
        @countries[c][:count] += sale.quantity
        @countries[c][:total] += sale.amount
      end

      def render
        sorted = @countries.sort { |a,b| b[1][:count] <=> a[1][:count] }

        puts "Country: Streams:  Revenue:"
        puts "------------------------------------------"

        total_rev = 0
        total_streams = 0

        sorted.each do |key, row|
          puts "#{key}:       #{row[:count].to_s.rjust(7)} #{row[:total].round(5).to_s.rjust(10)} EUR"
          total_rev += row[:total]
          total_streams += row[:count]
        end

        puts "------------------------------------------"
        puts "Total:    #{total_streams.to_s.rjust(7)} #{total_rev.round(5).to_s.rjust(10)} EUR"
        puts "=========================================="
      end
    end
  end
end