update rake task shift_sales:audit

lib/tasks/shift_sales.rake

@@ -1,9 +1,21 @@
 namespace :shift_sales do
   desc "TODO"
-  task :audit, [:shift_sale_id, :repayment] => [:environment] do |tasks, args|
+  task :audit, [:shift_sale_ids, :repayment] => [:environment] do |tasks, args|
     Rails.logger = Logger.new(STDOUT)
+
+    if args[:shift_sale_ids].empty?
+      return
+    end
+
+    if args[:shift_sale_ids].include? '..'
+      shift_sales_ids = args[:shift_sale_ids].split('..').inject { |s,e| s.to_i..e.to_i }
+    else
+      shift_sales_ids = args[:shift_sale_ids].split(' ')
+    end
+
+    shift_sales_ids.each do |shift_sale_id|
       ActiveRecord::Base.transaction do
-      shift_sale = ShiftSale.find(args[:shift_sale_id])
+        shift_sale = ShiftSale.find(shift_sale_id)
 
         total_revenue     = 0
         total_discounts   = 0
@@ -107,7 +119,7 @@ namespace :shift_sales do
             end
           else
             total_receipt     += 1
-          total_void        += 1
+            total_void        += sale.grand_total
           end
         end
 
@@ -139,5 +151,6 @@ namespace :shift_sales do
         )
       end
     end
+  end
 
 end