spree-contrib/spree_drop_ship

class Spree::Supplier < Spree::Base
  extend FriendlyId
  friendly_id :name, use: :slugged

  attr_accessor :password, :password_confirmation

  #==========================================
  # Associations

  belongs_to :address, class_name: 'Spree::Address'
  accepts_nested_attributes_for :address

  if defined?(Ckeditor::Asset)
    has_many :ckeditor_pictures
    has_many :ckeditor_attachment_files
  end
  has_many   :products, through: :variants
  has_many   :stock_locations
  has_many   :shipments, through: :stock_locations
  has_many   :supplier_variants
  has_many   :users, class_name: Spree.user_class.to_s
  has_many   :variants, through: :supplier_variants

  #==========================================
  # Validations

  validates :commission_flat_rate,   presence: true
  validates :commission_percentage,  presence: true
  validates :email,                  presence: true, email: true, uniqueness: true
  validates :name,                   presence: true, uniqueness: true
  validates :url,                    format: { with: URI.regexp(%w[http https]), allow_blank: true }

  #==========================================
  # Callbacks

  after_create :assign_user
  after_create :create_stock_location
  after_create :send_welcome, if: -> { SpreeDropShip::Config[:send_supplier_email] }
  before_create :set_commission
  before_validation :check_url

  #==========================================
  # Instance Methods
  scope :active, -> { where(active: true) }

  def deleted?
    deleted_at.present?
  end

  def user_ids_string
    user_ids.join(',')
  end

  def user_ids_string=(s)
    self.user_ids = s.to_s.split(',').map(&:strip)
  end

  # Retreive the stock locations that has available
  # stock items of the given variant
  def stock_locations_with_available_stock_items(variant)
    stock_locations.select { |sl| sl.available?(variant) }
  end

  #==========================================
  # Protected Methods

  protected

  def assign_user
    if users.empty?
      if user = Spree.user_class.find_by(email: email)
        users << user
        save
      end
    end
  end

  def check_url
    unless url.blank? || url =~URI.regexp(%w[http https])
      self.url = "http://#{url}"
    end
  end

  def create_stock_location
    if stock_locations.empty?
      location = stock_locations.build(
        active:     true,
        country_id: address.try(:country_id),
        name:       name,
        state_id:   address.try(:state_id)
      )
      # It's important location is always created.  Some apps add validations that shouldn't break this.
      location.save validate: false
    end
  end

  def send_welcome
    Spree::SupplierMailer.welcome(id).deliver_later!
    # Specs raise error for not being able to set default_url_options[:host]
  rescue => ex # Errno::ECONNREFUSED => ex
    Rails.logger.error ex.message
    Rails.logger.error ex.backtrace.join("\n")
    return true # always return true so that failed email doesn't crash app.
  end

  def set_commission
    unless changes.key?(:commission_flat_rate)
      self.commission_flat_rate = SpreeDropShip::Config[:default_commission_flat_rate]
    end
    unless changes.key?(:commission_percentage)
      self.commission_percentage = SpreeDropShip::Config[:default_commission_percentage]
    end
  end
end