Report sales by shop.

2 files changed, 40 insertions, 2 deletions

diff --git a/lib/reports/sales_by_shop.rb b/lib/reports/sales_by_shop.rb
new file mode 100644
index 0000000..7308b1d
--- /dev/null
+++ b/lib/reports/sales_by_shop.rb
@@ -0,0 +1,37 @@
+module SalesReporter
+  module Reports
+    class SalesByShop
+      def self.render(sales)
+        sales_by_shop = {}
+
+        sales.each do |s|
+          shop = s.shop
+          if sales_by_shop[shop]
+            sales_by_shop[shop][:count] += s.quantity
+            sales_by_shop[shop][:total] += s.amount
+          else
+            sales_by_shop[shop] = {:count => s.quantity, :total => s.amount}
+          end
+        end
+
+        sorted = sales_by_shop.sort { |a,b| b[1][:count] <=> a[1][:count] }
+
+        puts "Shop:               Streams:  Revenue:"
+        puts "------------------------------------------"
+
+        total_rev = 0
+        total_streams = 0
+
+        sorted.each do |key, row|
+          puts "#{key.slice(0, 19).ljust(20)}:       #{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"
+
+      end
+    end
+  end
+end
diff --git a/lib/sale.rb b/lib/sale.rb
index 057927c..17c6f18 100644
--- a/lib/sale.rb
+++ b/lib/sale.rb
@@ -1,13 +1,14 @@
 module SalesReporter
   class Sale
-    attr_reader :date, :amount, :quantity, :country
+    attr_reader :date, :amount, :quantity, :country, :shop
 
-    def initialize(date_, amount, quantity, country)
+    def initialize(date_, amount, quantity, country, shop)
       @date = sanitize_date(date_)
       raise TypeError('sanitize failed') unless @date.is_a? Date
       @amount = amount
       @quantity = quantity
       @country = country
+      @shop = shop
     end
 
     private