openfoodfoundation/openfoodnetwork

View on GitHub
lib/reporting/reports/enterprise_fee_summary/enterprise_fees_with_tax_report_by_producer.rb

Summary

Maintainability
A
0 mins
Test Coverage
# frozen_string_literal: true

module Reporting
  module Reports
    module EnterpriseFeeSummary
      class EnterpriseFeesWithTaxReportByProducer < ReportTemplate
        attr_accessor :permissions

        def initialize(user, params = {}, render: false)
          super(user, params, render:)
          @permissions = Permissions.new(user)
        end

        def search
          report_line_items.orders
        end

        def order_permissions
          @order_permissions ||= ::Permissions::Order.new(user, ransack_params)
        end

        def report_line_items
          @report_line_items ||= Reporting::LineItems.new(order_permissions, params)
        end

        def supplier_ids_filter(line_item)
          return true if params[:supplier_id_in].blank?

          params[:supplier_id_in].include?(line_item.supplier.id.to_s)
        end

        def enterprise_fee_filtered_ids
          return @enterprise_fee_filtered_ids unless @enterprise_fee_filtered_ids.nil?

          @enterprise_fee_filtered_ids = EnterpriseFee.where(nil)
          unless enterprise_fee_ids_filter.empty?
            @enterprise_fee_filtered_ids = @enterprise_fee_filtered_ids.where(
              id: enterprise_fee_ids_filter
            )
          end
          unless enterprise_fee_owner_ids_filter.empty?
            @enterprise_fee_filtered_ids = @enterprise_fee_filtered_ids.where(
              enterprise_id: enterprise_fee_owner_ids_filter
            )
          end
          @enterprise_fee_filtered_ids = @enterprise_fee_filtered_ids.pluck(:id)
        end

        def enterprise_fee_filters?
          enterprise_fee_ids_filter + enterprise_fee_owner_ids_filter != []
        end

        def enterprise_fee_ids_filter
          ransack_params["enterprise_fee_id_in"]&.filter(&:present?) || []
        end

        def enterprise_fee_owner_ids_filter
          ransack_params["enterprise_fee_owner_id_in"]&.filter(&:present?) || []
        end

        def query_result
          # The objective is to group the orders by
          # [tax_rate, entreprise_fee, supplier_id, distributor_id and order_cycle_id]

          # The order.all_adjustment describes
          #   - the enterprise fees applied on the order
          #   - the enterprise fees applied on the line items
          #   - all the taxes including, the taxes applied on the enterprise fees
          # The access to enterprise fee tax rates is done in two steps
          #   1.we'll need to store two attributes for each order.all_adjustemnt.enterprise_fee
          #     a. originator_id will refer to the enterprise fee instance
          #     b. id of the adjustment
          #   2. order.all_adjustemnt.tax.where(adjustment_id: id,"Spree::Adjustment")
          #     - this will return the tax applied on the enterprise fees
          orders = report_line_items.list.map(&:order).uniq
          orders.flat_map(&join_enterprise_fee)
            .flat_map(&join_tax_rate)
            .flat_map(&join_supplier)
            .group_by(&group_keys)
            .each(&change_root_to_order)
        end

        def join_enterprise_fee
          proc do |order|
            query = order
              .all_adjustments
              .enterprise_fee
              .where(originator_id: enterprise_fees_related_to_incoming_exchanges_ids(order))

            if enterprise_fee_filters?
              query = query.where(originator_id: enterprise_fee_filtered_ids)
            end
            query.group('originator_id')
              .pluck("originator_id", 'array_agg(id)')
              .map do |enterprise_fee_id, enterprise_fee_adjustment_ids|
                {
                  enterprise_fee_id:,
                  enterprise_fee_adjustment_ids:,
                  order:
                }
              end
          end
        end

        def enterprise_fees_related_to_incoming_exchanges_ids(order)
          order
            .order_cycle
            .exchanges
            .incoming
            .flat_map(&:enterprise_fee_ids)
            .uniq
        end

        def join_tax_rate
          proc do |item|
            tax_rate_ids = item[:order].all_adjustments.tax.where(
              adjustable_id: item[:enterprise_fee_adjustment_ids],
              adjustable_type: "Spree::Adjustment"
            ).pluck(:originator_id)

            tax_rate_ids << nil if tax_rate_ids.empty?
            tax_rate_ids.map do |tax_rate_id|
              {
                tax_rate_id:,
                enterprise_fee_id: item[:enterprise_fee_id],
                order: item[:order],
              }
            end
          end
        end

        # For every supplier, we want to show the enteprise fees
        # applied to at least one of his products
        def join_supplier
          proc do |item|
            order = item[:order]
            enterprise_fees_per_variant = enterprise_fees_per_variant(order)
            filtered_line_items(order)
              .filter do |line_item|
                item[:enterprise_fee_id].in?(
                  enterprise_fees_per_variant.fetch(line_item.variant, [])
                )
              end
              .map do |line_item|
                {
                  tax_rate_id: item[:tax_rate_id],
                  enterprise_fee_id: item[:enterprise_fee_id],
                  supplier_id: line_item.supplier.id,
                  order:
                }
              end
          end
        end

        def filtered_line_items(order)
          order
            .line_items
            .filter(&method(:supplier_ids_filter))
        end

        # { variant: [enterprise_fee_ids] }
        def enterprise_fees_per_variant(order)
          hash = {}
          order.order_cycle.exchanges.each do |exchange|
            exchange.variants.each do |variant|
              hash[variant] ||= order.order_cycle.coordinator_fee_ids
              hash[variant] += exchange.enterprise_fee_ids
            end
          end
          hash
        end

        def group_keys
          proc do |hash|
            [
              hash[:tax_rate_id],
              hash[:enterprise_fee_id],
              hash[:supplier_id],
              hash[:order].distributor_id,
              hash[:order].order_cycle_id
            ]
          end
        end

        def change_root_to_order
          proc do |_, v|
            v.map!{ |item| item[:order] }
          end
        end

        def columns
          {
            distributor: :distributor,
            producer: :producer,
            producer_tax_status: :producer_tax_status,
            order_cycle: :order_cycle,
            enterprise_fee_name: :enterprise_fee_name,
            enterprise_fee_type: :enterprise_fee_type,
            enterprise_fee_owner: :enterprise_fee_owner,
            tax_category: :tax_category,
            tax_rate_name: :tax_rate_name,
            tax_rate: :tax_rate_amount,
            total_excl_tax: :total_excl_tax,
            tax: :tax,
            total_incl_tax: :total_incl_tax
          }
        end

        def rules
          [
            { group_by: :distributor },
            { group_by: :producer },
            {
              # TOTAL (appear second)
              group_by: :order_cycle,
              summary_row: order_cycle_totals_row,
            },
            {
              # Cost of produce (appear first)
              group_by: :order_cycle,
              summary_row_class: nil,
              summary_row_label: nil,
              summary_row: order_cycle_line_items_row,
            },
          ]
        end

        def order_cycle_totals_row
          proc do |_key, items, rows|
            supplier_id = items.first.first[2] # supplier id used in the grouped line items
            line_items = items.flat_map(&:second).uniq.map(&:line_items).flatten
              .filter do |line_item|
                line_item.supplier_id == supplier_id
              end

            tax_for_enterprise_fees = rows.map(&:tax).sum
            total_excl_tax = total_fees_excl_tax(items) + line_items_excl_tax(line_items)
            tax = tax_for_enterprise_fees + tax_for_line_items(line_items)
            {
              total_excl_tax:,
              tax:,
              total_incl_tax: total_excl_tax + tax
            }
          end
        end

        def order_cycle_line_items_row
          proc do |key, items, _rows|
            supplier_id = items.first.first[2] # supplier id used in the grouped line items
            line_items = items.flat_map(&:last).uniq.map(&:line_items).flatten
              .filter do |line_item|
                line_item.supplier_id == supplier_id
              end
            producer = producer(items.first)

            total_excl_tax = line_items_excl_tax(line_items)
            tax = tax_for_line_items(line_items)
            {
              distributor: distributor(items.first),
              producer:,
              producer_tax_status: producer_tax_status(items.first),
              order_cycle: key,
              enterprise_fee_name: I18n.t('report_line_cost_of_produce'),
              enterprise_fee_type: I18n.t('report_line_line_items'),
              enterprise_fee_owner: producer,
              total_excl_tax:,
              tax:,
              total_incl_tax: total_excl_tax + tax,
            }
          end
        end

        def total_fees_excl_tax(items)
          order_ids = items.flat_map(&:second).map(&:id).uniq
          enterprise_fee_ids = items.map(&:first).map(&:second)
          enterprise_fees_amount_for_orders(
            order_ids, enterprise_fee_ids
          ) - included_tax_for_order_ids(
            order_ids, enterprise_fee_ids
          )
        end

        def line_items_excl_tax(line_items)
          cost_of_line_items(line_items) - line_items.sum(&:included_tax)
        end

        def cost_of_line_items(line_items)
          line_items.sum(&:amount)
        end

        # This query gets called twice for each set of line_items, ideally it would be cached.
        def tax_for_line_items(line_items)
          line_items.map do |line_item|
            line_item.adjustments.eligible.tax.sum('amount')
          end.sum
        end

        def included_tax_for_order_ids(order_ids, enterprise_fee_ids)
          Spree::Adjustment.tax
            .where(order: order_ids)
            .where(included: true)
            .where(adjustable_type: 'Spree::Adjustment')
            .where(adjustable_id: enterprise_fee_adjustment_ids_for_orders(order_ids,
                                                                           enterprise_fee_ids))
            .pick("sum(amount)") || 0
        end

        def enterprise_fee_adjustment_ids_for_orders(order_ids, enterprise_fee_ids)
          enterprise_fee_adjustments_for_orders(order_ids, enterprise_fee_ids).pluck(:id)
        end

        def enterprise_fees_amount_for_orders(order_ids, enterprise_fee_ids)
          enterprise_fee_adjustments_for_orders(
            order_ids, enterprise_fee_ids
          ).pick("sum(amount)") || 0
        end

        def enterprise_fee_adjustments_for_orders(order_ids, enterprise_fee_ids)
          enterprise_fees = Spree::Adjustment.enterprise_fee
            .where(order_id: order_ids)
            .where(originator_id: enterprise_fee_ids)
          return enterprise_fees unless enterprise_fee_filters?

          enterprise_fees.where(
            originator_id: enterprise_fee_filtered_ids
          )
        end

        def distributor(query_result_row)
          first_order(query_result_row).distributor&.name
        end

        def producer(query_result_row)
          Enterprise.where(id: supplier_id(query_result_row)).pick(:name)
        end

        def producer_tax_status(query_result_row)
          Enterprise.where(id: supplier_id(query_result_row)).pick(:charges_sales_tax)
        end

        def order_cycle(query_result_row)
          first_order(query_result_row).order_cycle&.name
        end

        def enterprise_fee_name(query_result_row)
          enterprise_fee(query_result_row).name
        end

        def enterprise_fee_type(query_result_row)
          enterprise_fee(query_result_row).fee_type
        end

        def enterprise_fee_owner(query_result_row)
          enterprise_fee(query_result_row).enterprise.name
        end

        def tax_category(query_result_row)
          tax_rate(query_result_row)&.tax_category&.name
        end

        def tax_rate_name(query_result_row)
          tax_rate(query_result_row)&.name
        end

        def tax_rate_amount(query_result_row)
          tax_rate(query_result_row)&.amount
        end

        def total_excl_tax(query_result_row)
          order_ids = orders(query_result_row).map(&:id)
          enterprise_fee_id = enterprise_fee_id(query_result_row)
          amount = Spree::Adjustment.enterprise_fee
            .where(order_id: order_ids)
            .where(originator_id: enterprise_fee_id)
            .pick("sum(amount)")
          amount - tax(query_result_row, all: true, included: true)
        end

        def tax(query_result_row, all: false, included: nil)
          order_ids = orders(query_result_row).map(&:id)
          adjustment_ids = enterprise_fee_adjustemnt_ids(query_result_row)
          query = Spree::Adjustment.tax
          query = query.where(included: true) unless included.nil?
          query = query.where(originator_id: tax_rate_id(query_result_row)) unless all == true
          query = query.where(order_id: order_ids)
            .where(adjustable_type: 'Spree::Adjustment')
            .where(adjustable_id: adjustment_ids)
            .pluck("sum(amount)")

          query.first || 0
        end

        def total_incl_tax(query_result_row)
          total_excl_tax(query_result_row) + tax(query_result_row, all: false)
        end

        def enterprise_fee_adjustemnt_ids(query_result_row)
          order_ids = orders(query_result_row).map(&:id)
          enterprise_fee_id = enterprise_fee_id(query_result_row)
          Spree::Adjustment.enterprise_fee
            .where(order_id: order_ids)
            .where(originator_id: enterprise_fee_id)
            .pluck(:id)
        end

        def enterprise_fee(query_result_row)
          first_order(query_result_row).all_adjustments
            .enterprise_fee
            .find_by(originator_id: enterprise_fee_id(query_result_row))
            .originator
        end

        def tax_rate(query_result_row)
          return nil if tax_rate_id(query_result_row).nil?

          Spree::TaxRate.find(tax_rate_id(query_result_row))
        end

        def first_order(query_result_row)
          orders(query_result_row).first
        end

        def tax_rate_id(query_result_row)
          keys(query_result_row)[0]
        end

        def supplier_id(query_result_row)
          keys(query_result_row)[2]
        end

        def enterprise_fee_id(query_result_row)
          keys(query_result_row)[1]
        end

        def keys(query_result_row)
          query_result_row.first
        end

        def orders(query_result_row)
          query_result_row.second
        end
      end
    end
  end
end