madetech/solidus_wms

module SolidusWms
  class Engine < Rails::Engine
    require 'spree/core'

    isolate_namespace Spree
    engine_name 'solidus_wms'

    # use rspec for tests
    config.generators do |g|
      g.test_framework :rspec
    end

    initializer 'solidus_wms.environment', before: 'spree.environment' do
      Spree::WmsConfig = SolidusWms::Configuration.new
    end

    def self.activate
      Dir.glob(File.join(File.dirname(__FILE__), '../../app/**/*_decorator*.rb')) do |c|
        Rails.configuration.cache_classes ? require(c) : load(c)
      end

      Dir.glob(File.join(File.dirname(__FILE__), '../../app/overrides/*.rb')) do |c|
        Rails.configuration.cache_classes ? require(c) : load(c)
      end

      Dir.glob(File.join(File.dirname(__FILE__), '../../lib/wms/*.rb')) do |c|
        Rails.configuration.cache_classes ? require(c) : load(c)
      end

      Dir.glob(File.join(File.dirname(__FILE__), '../../lib/spree/**/*.rb')) do |c|
        Rails.configuration.cache_classes ? require(c) : load(c)
      end

      Spree::Order.include(SolidusWms::OrderExportedAt)
      Spree::Order.whitelisted_ransackable_attributes << 'exported_at'
      Spree::Api::OrdersController.include(SolidusWms::AdditionalOrdersApiActions)
      Spree::OrdersController.include(SolidusWms::AdditionalOrdersActions)

      Spree::Admin::ReportsController.prepend(SolidusWms::AdditionalAdminReports)

      Spree::Shipment.include(SolidusWms::ShipmentStatemachineTransitions)
      Spree::Shipment.prepend(SolidusWms::ShipmentDetermineState)
      Spree::Shipment.whitelisted_ransackable_attributes << 'state'
      Spree::Shipment.whitelisted_ransackable_attributes << 'created_at'
      Spree::Api::ShipmentsController.include(SolidusWms::AdditionalShipmentsApiActions)
    end

    config.to_prepare(&method(:activate).to_proc)
  end
end