LD: Adding additional functionality for order cycle management reports - adding hub code model, adding temp_controlled bool, adding delivery report

lib/open_food_network/order_cycle_management_report.rb

@@ -1,3 +1,5 @@
+require 'open_food_network/user_balance_calculator'
+
 module OpenFoodNetwork
   class OrderCycleManagementReport
     attr_reader :params
@@ -7,23 +9,19 @@ module OpenFoodNetwork
     end
 
     def header
-      ["First Name", "Last Name", "Email", "Phone", "Hub", "Shipping Method", "Payment Method", "Amount"]
+      if is_payment_methods?
+        ["First Name", "Last Name", "Hub", "Hub Code", "Email", "Phone", "Shipping Method", "Payment Method", "Amount", "Balance"]
+      else
+        ["First Name", "Last Name", "Hub", "Hub Code", "Delivery Address", "Delivery Postcode", "Phone", "Shipping Method", "Payment Method", "Amount", "Balance", "Temp Controlled Items?", "Special Instructions"]
+      end
     end
 
     def table
-      orders.map do |order|
-        ba = order.billing_address
-        da = order.distributor.andand.address
-        [ba.firstname,
-         ba.lastname,
-         order.email,
-         ba.phone,
-         order.distributor.andand.name,
-         order.shipping_method.andand.name,
-         order.payments.first.andand.payment_method.andand.name,
-         order.payments.first.amount
-        ]
-      end
+      if is_payment_methods?
+        orders.map { |o| payment_method_row o }
+      else
+        orders.map { |o| delivery_row o }
+      end     
     end
 
     def orders
@@ -34,6 +32,44 @@ module OpenFoodNetwork
       filter_to_order_cycle filter_to_payment_method filter_to_shipping_method orders
     end
 
+
+    private 
+
+    def payment_method_row (order)
+      ba = order.billing_address
+      da = order.distributor.andand.address
+      [ba.firstname,
+       ba.lastname,
+       order.distributor.andand.name,
+       customer_code(order.email),
+       order.email,
+       ba.phone,
+       order.shipping_method.andand.name,
+       order.payments.first.andand.payment_method.andand.name,
+       order.payments.first.amount,
+       OpenFoodNetwork::UserBalanceCalculator.new(order.user, order.distributor).balance
+      ]
+    end
+
+    def delivery_row (order)
+      ba = order.billing_address
+      da = order.distributor.andand.address
+      [ba.firstname,
+       ba.lastname,
+       order.distributor.andand.name,
+       customer_code(order.email),
+       "#{ba.address1} #{ba.address2} #{ba.city}",
+       ba.zipcode,
+       ba.phone,
+       order.shipping_method.andand.name,
+       order.payments.first.andand.payment_method.andand.name,
+       order.payments.first.amount,
+       OpenFoodNetwork::UserBalanceCalculator.new(order.user, order.distributor).balance,
+       has_temperature_controlled_items?(order),
+       order.special_instructions
+      ]
+    end
+
     def filter_to_payment_method(orders)
       if params[:payment_method_name].present?
         orders.with_payment_method_name(params[:payment_method_name])
@@ -57,5 +93,19 @@ module OpenFoodNetwork
         orders
       end
     end
+
+    def has_temperature_controlled_items?(order)
+      order.line_items.any? { |line_item| line_item.product.shipping_category.nil? ? 
+        false : line_item.product.shipping_category.temperature_controlled? }
+    end
+
+    def is_payment_methods?
+      params[:report_type] == "payment_methods"
+    end
+
+    def customer_code (email)
+      customer = Customer.where(email: email).first
+      customer.nil? ? "" : customer.code
+    end
   end
 end