spree_multi_vendor 2.0.1

data/app/models/spree/vendor_order_totals.rb

@@ -0,0 +1,28 @@
+module Spree
+  class VendorOrderTotals
+    include ActiveModel::Model
+
+    METHOD_NAMES = %w[
+      total ship_total subtotal included_tax_total additional_tax_total promo_total
+      pre_tax_item_amount pre_tax_ship_amount pre_tax_total
+    ].freeze
+
+    attr_accessor :vendor, :order
+
+    delegate :id, :name, to: :vendor
+
+    METHOD_NAMES.each do |method_name|
+      define_method(method_name) do
+        order.send("vendor_#{method_name}", vendor)
+      end
+
+      define_method("display_#{method_name}") do
+        order.send("display_vendor_#{method_name}", vendor)
+      end
+    end
+
+    def item_count
+      order.vendor_item_count(vendor)
+    end
+  end
+end

data/app/models/spree/vendor_user.rb

@@ -0,0 +1,16 @@
+module Spree
+  class VendorUser < Spree::Base
+    belongs_to :vendor, class_name: 'Spree::Vendor', required: Spree.version.to_f >= 3.5
+    belongs_to :user, class_name: Spree.user_class.name, required: Spree.version.to_f >= 3.5
+
+    validates :vendor_id, uniqueness: { scope: :user_id }
+
+    before_save :set_vendor_notification_email
+
+    def set_vendor_notification_email
+      return if vendor&.notification_email.present? || !vendor&.users&.empty?
+
+      vendor.update_notification_email(user.email)
+    end
+  end
+end

data/app/models/spree_multi_vendor/configuration.rb

@@ -0,0 +1,10 @@
+module SpreeMultiVendor
+  class Configuration < Spree::Preferences::Configuration
+    DEFAULT_VENDORIZED_MODELS ||= %w[product variant stock_location shipping_method].freeze
+
+   # Some example preferences are shown below, for more information visit:
+   # https://guides.spreecommerce.org/developer/core/preferences.html
+
+   preference :vendorized_models, :array, default: DEFAULT_VENDORIZED_MODELS
+  end
+end

data/app/models/spree_multi_vendor/spree/ability_decorator.rb

@@ -0,0 +1,9 @@
+module SpreeMultiVendor::Spree::AbilityDecorator
+  private
+
+  def abilities_to_register
+    super << Spree::VendorAbility
+  end
+end
+
+Spree::Ability.prepend SpreeMultiVendor::Spree::AbilityDecorator

data/app/models/spree_multi_vendor/spree/base_decorator.rb

@@ -0,0 +1,13 @@
+module SpreeMultiVendor::Spree::BaseDecorator
+  SpreeMultiVendor.vendorized_models.each do |model|
+    model.include Spree::VendorConcern
+
+    if model.method_defined?(:whitelisted_ransackable_associations)
+      if model.whitelisted_ransackable_associations
+        model.whitelisted_ransackable_associations += %w[vendor]
+      else
+        model.whitelisted_ransackable_associations = %w[vendor]
+      end
+    end
+  end
+end

data/app/models/spree_multi_vendor/spree/line_item_decorator.rb

@@ -0,0 +1,48 @@
+module SpreeMultiVendor::Spree::LineItemDecorator
+  def self.prepended(base)
+    base.scope :for_vendor, ->(vendor) do
+      if ::Spree::LineItem.reflect_on_association(:vendor)
+        where(vendor_id: vendor.id)
+      elsif ::Spree::Product.reflect_on_association(:vendor)
+        joins(variant: :product).where('spree_products.vendor_id' => vendor.id)
+      elsif ::Spree::Variant.reflect_on_association(:vendor)
+        joins(:variant).where('spree_variants.vendor_id' => vendor.id)
+      else
+        none
+      end
+    end
+    base.scope :not_for_vendor, ->(vendor) do
+      if ::Spree::Product.reflect_on_association(:vendor)
+        where.not(vendor_id: vendor.id)
+      elsif ::Spree::Product.reflect_on_association(:vendor)
+        joins(variant: :product).where.not('spree_products.vendor_id' => vendor.id)
+      elsif ::Spree::Variant.reflect_on_association(:vendor)
+        joins(:variant).where.not('spree_variants.vendor_id' => vendor.id)
+      else
+        none
+      end
+    end
+  end
+
+  def vendor
+    @vendor ||= if self.class.reflect_on_association(:vendor) && self[:vendor_id].present?
+                  ::Spree::Vendor.unscoped.find(self[:vendor_id])
+                elsif product.class.reflect_on_association(:vendor) && product.vendor_id.present?
+                  product.vendor
+                elsif variant.class.reflect_on_association(:vendor) && variant.vendor_id.present?
+                  variant.vendor
+                end
+  end
+
+  def vendor_id
+    @vendor_id ||= if self.class.reflect_on_association(:vendor) && self[:vendor_id].present?
+                    self[:vendor_id]
+                   elsif product.class.reflect_on_association(:vendor) && product.vendor_id.present?
+                     product.vendor_id
+                   elsif variant.class.reflect_on_association(:vendor) && variant.vendor_id.present?
+                     variant.vendor_id
+                   end
+  end
+end
+
+Spree::LineItem.prepend SpreeMultiVendor::Spree::LineItemDecorator

data/app/models/spree_multi_vendor/spree/order_decorator.rb

@@ -0,0 +1,105 @@
+module SpreeMultiVendor::Spree::OrderDecorator
+  def self.prepended(base)
+    base.has_many :commissions, class_name: 'Spree::OrderCommission'
+    base.state_machine.after_transition to: :complete, do: :generate_order_commissions
+    base.state_machine.after_transition to: :complete, do: :send_notification_mails_to_vendors
+  end
+
+  def generate_order_commissions
+    Spree::Orders::GenerateCommissions.call(order: self)
+  end
+
+  def vendor_line_items(vendor)
+    line_items.for_vendor(vendor)
+  end
+
+  def vendor_shipments(vendor)
+    shipments.for_vendor(vendor)
+  end
+
+  def vendor_ship_total(vendor)
+    vendor_shipments(vendor).sum(&:final_price)
+  end
+
+  def vendor_subtotal(vendor)
+    vendor_line_items(vendor).sum(&:total)
+  end
+
+  def vendor_promo_total(vendor)
+    vendor_line_items(vendor).sum(:promo_total) + vendor_shipments(vendor).sum(:promo_total)
+  end
+
+  def vendor_additional_tax_total(vendor)
+    vendor_line_items(vendor).sum(:additional_tax_total) + vendor_shipments(vendor).sum(:additional_tax_total)
+  end
+
+  def vendor_included_tax_total(vendor)
+    vendor_line_items(vendor).sum(:included_tax_total) + vendor_shipments(vendor).sum(:included_tax_total)
+  end
+
+  def vendor_pre_tax_item_amount(vendor)
+    vendor_line_items(vendor).sum(:pre_tax_amount)
+  end
+
+  def vendor_pre_tax_ship_amount(vendor)
+    vendor_shipments(vendor).sum(:pre_tax_amount)
+  end
+
+  def vendor_pre_tax_total(vendor)
+    vendor_line_items(vendor).sum(:pre_tax_amount) + vendor_shipments(vendor).sum(:pre_tax_amount)
+  end
+
+  def vendor_item_count(vendor)
+    vendor_line_items(vendor).sum(:quantity)
+  end
+
+  def vendor_total(vendor)
+    vendor_line_items(vendor).sum(&:total) + vendor_ship_total(vendor)
+  end
+
+  def display_order_commission
+    Spree::Money.new(commissions.sum(:amount), { currency: currency })
+  end
+
+  def vendor_commission(vendor)
+    commissions.for_vendor(vendor).sum(:amount)
+  end
+
+  # money methods
+  METHOD_NAMES = %w[
+    total ship_total subtotal included_tax_total additional_tax_total promo_total
+    pre_tax_item_amount pre_tax_ship_amount pre_tax_total commission
+  ].freeze
+
+  METHOD_NAMES.each do |method_name|
+    define_method("display_vendor_#{method_name}") do |vendor|
+      Spree::Money.new(send("vendor_#{method_name}", vendor), { currency: currency })
+    end
+  end
+
+  def send_notification_mails_to_vendors
+    vendor_ids.each do |vendor_id|
+      Spree::VendorMailer.vendor_notification_email(id, vendor_id).deliver_later
+    end
+  end
+
+  # we're leaving this on purpose so it can be easily modified to fit desired scenario
+  # eg. scenario A - vendorized products, scenario B - vendorized variants of the same product
+  def vendor_ids
+    @vendor_ids ||= line_items.map(&:vendor_id).uniq.compact
+  end
+
+  def vendor_list
+    @vendor_list ||= line_items.map(&:vendor).uniq.compact
+  end
+
+  def vendor_totals
+    return if vendor_list.none?
+
+    vendor_list.map do |vendor|
+      Spree::VendorOrderTotals.new(vendor: vendor, order: self)
+    end
+  end
+end
+
+Spree::Order.prepend SpreeMultiVendor::Spree::OrderDecorator

data/app/models/spree_multi_vendor/spree/shipment_decorator.rb

@@ -0,0 +1,8 @@
+module SpreeMultiVendor::Spree::ShipmentDecorator
+  def self.prepended(base)
+    base.scope :for_vendor, ->(vendor) { joins(stock_location: :vendor).where('spree_stock_locations.vendor_id' => vendor.id) }
+    base.scope :not_for_vendor, ->(vendor) { joins(stock_location: :vendor).where.not('spree_stock_locations.vendor_id' => vendor.id) }
+  end
+end
+
+Spree::Shipment.prepend SpreeMultiVendor::Spree::ShipmentDecorator

data/app/models/spree_multi_vendor/spree/stock/package_decorator.rb

@@ -0,0 +1,14 @@
+module SpreeMultiVendor::Spree::Stock::PackageDecorator
+  def shipping_methods
+    vendor = Spree::StockLocation.method_defined?(:vendor) ? stock_location.vendor : nil
+
+    # return only shipping methods for this Vendor
+    if vendor && Spree::ShippingMethod.method_defined?(:vendor)
+      vendor.shipping_methods.to_a
+    else
+      shipping_categories.map(&:shipping_methods).reduce(:&).to_a
+    end
+  end
+end
+
+Spree::Stock::Package.prepend SpreeMultiVendor::Spree::Stock::PackageDecorator

data/app/models/spree_multi_vendor/spree/user_decorator.rb

@@ -0,0 +1,8 @@
+module SpreeMultiVendor::Spree::UserDecorator
+  def self.prepended(base)
+    base.has_many :vendor_users, class_name: 'Spree::VendorUser'
+    base.has_many :vendors, through: :vendor_users, class_name: 'Spree::Vendor'
+  end
+end
+
+Spree.user_class.prepend SpreeMultiVendor::Spree::UserDecorator

data/app/models/spree_multi_vendor/spree/variant_decorator.rb

@@ -0,0 +1,41 @@
+module SpreeMultiVendor::Spree::VariantDecorator
+  def self.prepended(base)
+    base.before_create :assign_vendor
+
+    base.scope :for_vendor_user, ->(user) { where(vendor_id: user.vendor_ids) }
+  end
+
+  def vendor
+    @vendor ||= if self.class.reflect_on_association(:vendor) && self[:vendor_id].present?
+                  product.vendor
+                  ::Spree::Vendor.unscoped.find(self[:vendor_id])
+                elsif Spree::Product.reflect_on_association(:vendor) && product.vendor_id.present?
+                  product.vendor
+                end
+  end
+
+  def vendor_id
+    @vendor_id ||= if self.class.reflect_on_association(:vendor) && self[:vendor_id].present?
+                     self[:vendor_id]
+                   elsif Spree::Product.reflect_on_association(:vendor) && product.vendor_id.present?
+                     product.vendor_id
+                   end
+  end
+
+  private
+
+  def assign_vendor
+    return if !product.class.reflect_on_association(:vendor) || product.vendor.blank?
+    return if !self.class.reflect_on_association(:vendor) || self[:vendor_id].present?
+
+    self.vendor = product.vendor
+  end
+
+  def create_stock_items
+    Spree::StockLocation.where(propagate_all_variants: true, vendor_id: vendor_id).each do |stock_location|
+      stock_location.propagate_variant(self)
+    end
+  end
+end
+
+Spree::Variant.prepend SpreeMultiVendor::Spree::VariantDecorator

data/app/overrides/spree/admin/orders/_form.rb

@@ -0,0 +1,13 @@
+Deface::Override.new(
+    virtual_path: 'spree/admin/orders/_form',
+    name: 'vendorize_shipments',
+    replace: "erb[loud]:contains('render partial: \"spree/admin/orders/shipment\", collection: @order.shipments.order(:created_at), locals: {order: order}')",
+    text: "<%= render partial: \"spree/admin/orders/shipment\", collection: (current_spree_vendor ? @order.shipments.for_vendor(current_spree_vendor).order(:created_at) : @order.shipments.order(:created_at)), locals: {order: order} %>"
+)
+
+Deface::Override.new(
+    virtual_path: 'spree/admin/orders/_form',
+    name: 'display_vendor_total',
+    replace: "erb[loud]:contains('order.display_total')",
+    text: "<%= current_spree_vendor ? order.display_vendor_total(current_spree_vendor).to_html : order.display_total.to_html %>"
+)

data/app/overrides/spree/admin/orders/_line_items.rb

@@ -0,0 +1,6 @@
+Deface::Override.new(
+    virtual_path: 'spree/admin/orders/_line_items',
+    name: 'vendorize_line_items',
+    replace: "erb[silent]:contains('order.line_items.each do |item|')",
+    text: "<% (current_spree_vendor ? order.line_items.for_vendor(current_spree_vendor) : order.line_items).each do |item| %>"
+)

data/app/overrides/spree/admin/orders/_line_items_edit_form.rb

@@ -0,0 +1,6 @@
+Deface::Override.new(
+    virtual_path: 'spree/admin/orders/_line_items_edit_form',
+    name: 'display_vendor_total',
+    replace: "erb[loud]:contains('order.display_total')",
+    text: "<%= current_spree_vendor ? order.display_vendor_total(current_spree_vendor).to_html : order.display_total.to_html %>"
+)

data/app/overrides/spree/admin/orders/index.rb

@@ -0,0 +1,6 @@
+Deface::Override.new(
+    virtual_path: 'spree/admin/orders/index',
+    name: 'display_vendor_total',
+    replace: "erb[loud]:contains('order.display_total.to_html')",
+    text: "<%= current_spree_vendor ? order.display_vendor_total(current_spree_vendor).to_html : order.display_total.to_html %>"
+)

data/app/overrides/spree/admin/products/form.rb

@@ -0,0 +1,29 @@
+Deface::Override.new(
+  virtual_path: 'spree/admin/products/_form',
+  name: 'Enable vendors to manage product master price',
+  replace: 'div[data-hook="admin_product_form_price"]',
+  text: <<-HTML
+          <%= f.field_container :price, class: ['form-group'] do %>
+            <%= f.label :price, raw(Spree.t(:master_price) + content_tag(:span, ' *', class: "required")) %>
+            <%= f.text_field :price, value: number_to_currency(@product.price, unit: ''), class: 'form-control', disabled: (cannot? :update, Spree::Price) %>
+            <%= f.error_message_on :price %>
+          <% end %>
+        HTML
+)
+
+Deface::Override.new(
+    virtual_path: 'spree/admin/products/_form',
+    name: 'Enable admin to menage product vendor',
+    insert_before: 'div[data-hook="admin_product_form_taxons"]',
+    text: <<-HTML
+            <% if current_spree_user.respond_to?(:has_spree_role?) && current_spree_user.has_spree_role?(:admin) %>
+              <div data-hook="admin_product_form_vendor">
+                <%= f.field_container :vendor, class: ['form-group'] do %>
+                  <%= f.label :vendor_id, Spree.t(:vendor) %>
+                  <%= f.collection_select(:vendor_id, @vendors, :id, :name, { include_blank: Spree.t('match_choices.none') }, { class: 'select2' }) %>
+                  <%= f.error_message_on :vendor %>
+                <% end %>
+              </div>
+            <% end %>
+          HTML
+)

data/app/overrides/spree/admin/products/new.rb

@@ -0,0 +1,16 @@
+Deface::Override.new(
+    virtual_path: 'spree/admin/products/new',
+    name: 'Enable admin to create product with assigned vendor',
+    insert_after: 'div[data-hook="new_product_shipping_category"]',
+    text: <<-HTML
+            <% if current_spree_user.respond_to?(:has_spree_role?) && current_spree_user.has_spree_role?(:admin) %>
+              <div data-hook="new_product_vendor" class="col-12 col-md-4">
+                <%= f.field_container :vendor, class: ['form-group'] do %>
+                  <%= f.label :vendor_id, Spree.t(:vendor) %>
+                  <%= f.collection_select(:vendor_id, @vendors, :id, :name, { include_blank: Spree.t('match_choices.none') }, { class: 'select2' }) %>
+                  <%= f.error_message_on :vendor %>
+                <% end %>
+              </div>
+            <% end %>
+    HTML
+)

data/app/overrides/spree/admin/shared/_main_menu.rb

@@ -0,0 +1,23 @@
+Deface::Override.new(
+  virtual_path: 'spree/admin/shared/_main_menu',
+  name: 'Display configuration tab for vendors',
+  replace: 'erb[silent]:contains("current_store")',
+  text: '<% if can?(:admin, current_store) || current_spree_user.vendors.any? %>'
+)
+Deface::Override.new(
+  virtual_path:  'spree/admin/shared/_main_menu',
+  name:          'vendors_main_menu_tabs',
+  insert_bottom: 'nav',
+  text:       <<-HTML
+                <% if current_spree_user.respond_to?(:has_spree_role?) && current_spree_user.has_spree_role?(:admin) %>
+                  <ul class="nav nav-sidebar border-bottom">
+                    <%= tab plural_resource_name(Spree::Vendor), url: admin_vendors_path, icon: 'money' %>
+                  </ul>
+                <% end %>
+                <% if defined?(current_spree_vendor) && current_spree_vendor %>
+                  <ul class="nav nav-sidebar border-bottom">
+                    <%= tab Spree::Vendor.model_name.human, url: admin_vendor_settings_path, icon: 'money' %>
+                  </ul>
+                <% end %>
+HTML
+)

data/app/overrides/spree/admin/shared/_order_summary.rb

@@ -0,0 +1,35 @@
+Deface::Override.new(
+    virtual_path: 'spree/admin/shared/_order_summary',
+    name: 'display_vendor_subtotal',
+    replace: "erb[loud]:contains('@order.display_item_total.to_html')",
+    text: "<%= current_spree_vendor ? @order.display_vendor_subtotal(current_spree_vendor).to_html : @order.display_item_total.to_html %>"
+)
+
+Deface::Override.new(
+    virtual_path: 'spree/admin/shared/_order_summary',
+    name: 'display_vendor_ship_total',
+    replace: "erb[loud]:contains('@order.display_ship_total.to_html')",
+    text: "<%= current_spree_vendor ? @order.display_vendor_ship_total(current_spree_vendor).to_html : @order.display_ship_total.to_html %>"
+)
+
+Deface::Override.new(
+    virtual_path: 'spree/admin/shared/_order_summary',
+    name: 'display_vendor_total',
+    replace: "erb[loud]:contains('@order.display_total.to_html')",
+    text: "<%= current_spree_vendor ? @order.display_vendor_total(current_spree_vendor).to_html : @order.display_total.to_html %>"
+)
+Deface::Override.new(
+    virtual_path: 'spree/admin/shared/_order_summary',
+    name: 'display_commission',
+    insert_after: 'td[id="item_total"]',
+    text: <<-HTML
+            <tr>
+              <td data-hook='admin_order_tab_commission'>
+                <strong><%= Spree.t(:commission) %></strong>
+              </td>
+              <td id='order_commission'>
+                <%= current_spree_vendor ? @order.display_vendor_commission(current_spree_vendor) : @order.display_order_commission %>
+              </td>
+          </tr>
+    HTML
+)