openfoodnetwork/app/controllers/api/shipments_controller.rb

require 'open_food_network/scope_variant_to_hub'

module Api
  class ShipmentsController < Api::BaseController
    respond_to :json

    before_filter :find_order
    before_filter :find_and_update_shipment, only: [:ship, :ready, :add, :remove]

    def create
      variant = scoped_variant(params[:variant_id])
      quantity = params[:quantity].to_i
      @shipment = get_or_create_shipment(params[:stock_location_id])

      @order.contents.add(variant, quantity, nil, @shipment)

      @shipment.refresh_rates
      @shipment.save!

      render json: @shipment.reload, serializer: Api::ShipmentSerializer, status: :ok
    end

    def update
      authorize! :read, Spree::Shipment
      @shipment = @order.shipments.find_by_number!(params[:id])
      params[:shipment] ||= []
      unlock = params[:shipment].delete(:unlock)

      if unlock == 'yes'
        @shipment.adjustment.open
      end

      @shipment.update_attributes(params[:shipment])

      if unlock == 'yes'
        @shipment.adjustment.close
      end

      render json: @shipment.reload, serializer: Api::ShipmentSerializer, status: :ok
    end

    def ready
      authorize! :read, Spree::Shipment
      unless @shipment.ready?
        if @shipment.can_ready?
          @shipment.ready!
        else
          render(json: { error: I18n.t(:cannot_ready, scope: "spree.api.shipment") },
                 status: :unprocessable_entity) && return
        end
      end
      render json: @shipment, serializer: Api::ShipmentSerializer, status: :ok
    end

    def ship
      authorize! :read, Spree::Shipment
      unless @shipment.shipped?
        @shipment.ship!
      end
      render json: @shipment, serializer: Api::ShipmentSerializer, status: :ok
    end

    def add
      variant = scoped_variant(params[:variant_id])
      quantity = params[:quantity].to_i

      @order.contents.add(variant, quantity, nil, @shipment)

      render json: @shipment, serializer: Api::ShipmentSerializer, status: :ok
    end

    def remove
      variant = scoped_variant(params[:variant_id])
      quantity = params[:quantity].to_i

      @order.contents.remove(variant, quantity, @shipment)
      @shipment.reload if @shipment.persisted?

      render json: @shipment, serializer: Api::ShipmentSerializer, status: :ok
    end

    private

    def find_order
      @order = Spree::Order.find_by_number!(params[:order_id])
      authorize! :read, @order
    end

    def find_and_update_shipment
      @shipment = @order.shipments.find_by_number!(params[:id])
      @shipment.update_attributes(params[:shipment])
      @shipment.reload
    end

    def scoped_variant(variant_id)
      variant = Spree::Variant.find(variant_id)
      OpenFoodNetwork::ScopeVariantToHub.new(@order.distributor).scope(variant)
      variant
    end

    def get_or_create_shipment(stock_location_id)
      @order.shipment || @order.shipments.create(stock_location_id: stock_location_id)
    end
  end
end