spree/spree

module Spree
  module Api
    module V2
      module Platform
        class ShipmentsController < ResourceController
          include NumberResource

          before_action :load_variant, only: %i[transfer_to_location transfer_to_shipment]

          SHIPMENT_STATES = %w[ready ship cancel resume pend]

          def create
            # FIXME: support saving metadata
            result = create_service.call(
              store: current_store,
              shipment_attributes: params.require(:shipment).permit(
                :order_id, :stock_location_id, :variant_id, :quantity
              )
            )
            render_result(result, 201)
          end

          def update
            result = update_service.call(shipment: resource, shipment_attributes: permitted_resource_params)
            render_result(result)
          end

          SHIPMENT_STATES.each do |state|
            define_method state do
              result = change_state_service.call(shipment: resource, state: state)
              render_result(result)
            end
          end

          def add_item
            result = add_item_service.call(
              shipment: resource,
              variant_id: params.dig(:shipment, :variant_id),
              quantity: params.dig(:shipment, :quantity)
            )
            render_result(result)
          end

          def remove_item
            result = remove_item_service.call(
              shipment: resource,
              variant_id: params.dig(:shipment, :variant_id),
              quantity: params.dig(:shipment, :quantity)
            )

            if result.success?
              if result.value == :shipment_deleted
                head 204
              else
                render_serialized_payload { serialize_resource(result.value) }
              end
            else
              render_error_payload(result.error)
            end
          end

          def transfer_to_location
            stock_location = Spree::StockLocation.find(params.dig(:shipment, :stock_location_id))
            quantity = params.dig(:shipment, :quantity)&.to_i || 1

            unless quantity > 0
              render_error_payload("#{I18n.t('spree.api.shipment_transfer_errors_occurred')} \n #{I18n.t('spree.api.negative_quantity')}")
              return
            end

            transfer = resource.transfer_to_location(@variant, quantity, stock_location)
            if transfer.valid? && transfer.run!
              render json: { message: I18n.t('spree.api.shipment_transfer_success') }, status: 201
            else
              render_error_payload(transfer.errors)
            end
          end

          def transfer_to_shipment
            target_shipment = Spree::Shipment.find_by!(number: params.dig(:shipment, :target_shipment_number))
            quantity = params.dig(:shipment, :quantity)&.to_i || 1

            error =
              if quantity < 0 && target_shipment == resource
                "#{I18n.t('spree.api.negative_quantity')}, \n#{I18n.t('spree.api.wrong_shipment_target')}"
              elsif target_shipment == resource
                I18n.t('spree.api.wrong_shipment_target')
              elsif quantity < 0
                I18n.t('spree.api.negative_quantity')
              end

            if error
              render_error_payload("#{I18n.t('spree.api.shipment_transfer_errors_occurred')} \n#{error}")
            else
              transfer = resource.transfer_to_shipment(@variant, quantity, target_shipment)
              if transfer.valid? && transfer.run!
                render json: { message: I18n.t('spree.api.shipment_transfer_success') }, status: 201
              else
                render_error_payload(transfer.errors)
              end
            end
          end

          private

          def model_class
            Spree::Shipment
          end

          def load_variant
            @variant = current_store.variants.find_by(id: params.dig(:shipment, :variant_id))
          end

          def spree_permitted_attributes
            Spree::Shipment.json_api_permitted_attributes + [
              :selected_shipping_rate_id
            ]
          end

          def create_service
            Spree::Api::Dependencies.platform_shipment_create_service.constantize
          end

          def update_service
            Spree::Api::Dependencies.platform_shipment_update_service.constantize
          end

          def change_state_service
            Spree::Api::Dependencies.platform_shipment_change_state_service.constantize
          end

          def add_item_service
            Spree::Api::Dependencies.platform_shipment_add_item_service.constantize
          end

          def remove_item_service
            Spree::Api::Dependencies.platform_shipment_remove_item_service.constantize
          end
        end
      end
    end
  end
end