module Spree
  module OrdersControllerDecorator
    def self.prepended(base)
      base.include Spree::BraintreeHelper
      base.helper_method [:asset_available?, :options_from_braintree_payments]
      base.before_action :process_paypal_express, only: :update
    end

    def process_paypal_express
      return false if current_order.blank?

      if params[:paypal].blank? || params[:paypal][:payment_method_nonce].blank?
        # when user goes back from checkout, paypal express payments should be invalidated  to ensure standard checkout flow
        current_order.invalidate_paypal_express_payments
        return true
      end
      payment_method = Spree::PaymentMethod.find_by_id(params[:paypal][:payment_method_id])
      return true unless payment_method

      email = params[:order][:email]
      # when user goes back from checkout, order's state should be resetted to ensure paypal checkout flow
      current_order.state = 'cart'
      payment_method.push_order_to_state(current_order, 'address', email)
      current_order.save_paypal_payment(payment_params)

      manage_paypal_addresses
      current_order.remove_phone_number_placeholder

      redirect_to checkout_state_path(current_order.state, paypal_email: email)
    end

    private

    def address_params(key)
      params[:order].require(key).permit(:firstname, :lastname, :zipcode, :city, :address1, :address2, :phone, :full_name, :country, :state)
    end

    def payment_params
      {
        braintree_nonce: params[:paypal][:payment_method_nonce],
        payment_method_id: params[:paypal][:payment_method_id],
        paypal_email: params[:order][:email],
        advanced_fraud_data: params[:device_data]
      }
    end

    def manage_paypal_addresses
      # addresses need to be cleared for restarted checkout orders
      current_order.ship_address_id = nil
      current_order.bill_address_id = nil

      %w(ship_address bill_address).each do |address_type|
        next if params[:order][address_type].blank?
        current_order.save_paypal_address(address_type, address_params(address_type))
      end
      current_order.set_billing_address
      return false if current_order.no_phone_number?

      current_order.ship_address && current_order.bill_address
    end
  end
end

::Spree::OrdersController.prepend(Spree::OrdersControllerDecorator)