1 files changed, 11 insertions, 12 deletions
diff --git a/lib/reports/sales_by_shop.rb b/lib/reports/sales_by_shop.rb
index 66caf63..2c8a70b 100644
--- a/lib/reports/sales_by_shop.rb
+++ b/lib/reports/sales_by_shop.rb
@@ -1,20 +1,19 @@
 module SalesReporter
   module Reports
     class SalesByShop
-      def render(sales)
-        sales_by_shop = {}
+      def initialize
+        @shops = Hash.new { |h,k| h[k] = { :count => 0, :total => 0, :sales => [] } }
+      end
 
-        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
+      def add_sale(sale, row)
+        shop = row['Shop']
+        @shops[shop][:sales] << sale
+        @shops[shop][:count] += sale.quantity
+        @shops[shop][:total] += sale.amount
+      end
 
-        sorted = sales_by_shop.sort { |a,b| b[1][:count] <=> a[1][:count] }
+      def render
+        sorted = @shops.sort { |a,b| b[1][:count] <=> a[1][:count] }
 
         puts "Shop:               Streams:  Revenue:"
         puts "------------------------------------------"