atkhayar-dev-spree-multi-vendor 0.0.0

data/app/controllers/spree_multi_vendor/spree/api/v2/platform/products_controller_decorator.rb

@@ -0,0 +1,65 @@
+module SpreeMultiVendor::Spree::Api::V2::Platform::ProductsControllerDecorator
+  def self.prepended(base)
+    base.before_action :initiate_ts_client
+    base.after_action :create_ts_product, only: [:create]
+    base.after_action :update_ts_product, only: [:update]
+    base.after_action :delete_ts_product, only: [:destroy]
+  end
+
+  def initiate_ts_client
+    @typesense = Typesense::Client.new(
+      nodes: [
+        {
+          host: 'iraxmydlt35g8v6qp-1.a1.typesense.net',
+          port: 443,
+          protocol: 'https'
+        }
+      ],
+      api_key: 'P524ViQst3aE0xlpbVY7or8bCAWbVtqv',
+      num_retries: 10,
+      healthcheck_interval_seconds: 1,
+      retry_interval_seconds: 0.01,
+      connection_timeout_seconds: 10
+    )
+  end
+
+  def create_ts_product
+    @product_vendor = Spree::Vendor.find_by(id: params[:product][:vendor_id])
+    @product = Spree::Product.find_by(slug: params[:product][:slug])
+    document = {
+      'id' => @product.id.to_s,
+      'name' => params[:product][:name].to_s,
+      'status' => @product.status,
+      'categories' => [],
+      'tag' => ['CC'],
+      'type'=> 'product',
+      'store_id' => @product_vendor.code
+    }
+    document = @typesense.collections['dev-products'].documents.create(document)
+  end
+
+  def update_ts_product
+    product_id = params[:id]
+    @product = Spree::Product.find_by(id: product_id)
+    classifications = Spree::Classification.where(product_id: product_id)
+    categories = []
+    if !classifications.empty?
+      classifications.each do |classification|
+        taxon = Spree::Taxon.find_by(id: classification.taxon_id)
+        categories.push(taxon.name)
+      end
+    end
+
+    @typesense.collections['dev-products'].documents[product_id].update(
+      'name' => @product.name,
+      'categories' => categories,
+      'status' => @product.status
+    )
+  end
+
+  def delete_ts_product
+    @typesense.collections['dev-products'].documents[params[:id]].delete()
+  end
+end
+
+Spree::Api::V2::Platform::ProductsController.prepend SpreeMultiVendor::Spree::Api::V2::Platform::ProductsControllerDecorator

data/app/finders/spree_multi_vendor/products_finder_decorator.rb

@@ -0,0 +1,31 @@
+module SpreeMultiVendor
+  module ProductsFinderDecorator
+    def initialize(scope:, params:, current_currency: nil)
+      super
+
+      @vendors = params.dig(:filter, :vendor_ids)&.split(',')
+    end
+
+    def execute
+      products = by_vendors(super)
+
+      products.distinct
+    end
+
+    private
+
+    attr_reader :vendors
+
+    def vendors?
+      vendors.present?
+    end
+
+    def by_vendors(products)
+      return products unless vendors?
+
+      products.where(vendor_id: vendors)
+    end
+  end
+end
+
+::Spree::Products::Find.prepend(SpreeMultiVendor::ProductsFinderDecorator)

data/app/helpers/spree/admin/vendor_helper.rb

@@ -0,0 +1,15 @@
+module Spree
+  module Admin
+    module VendorHelper
+      def current_spree_vendor
+        if current_spree_user.vendors.any? && (!current_spree_user.respond_to?(:has_spree_role?) || !current_spree_user.has_spree_role?(:admin))
+          current_spree_user.vendors.first
+        end
+      end
+
+      def vendor_state_options
+        @vendor_state_options ||= Spree::Vendor.state_machines[:state].states.collect { |s| [Spree.t("vendor_states.#{s.name}"), s.value] }
+      end
+    end
+  end
+end

data/app/helpers/spree/api/vendor_helper.rb

@@ -0,0 +1,13 @@
+module Spree
+  module Api
+    module VendorHelper
+      def current_spree_vendor
+        # puts 'current'
+        # puts current_api_user
+        # if current_api_user.vendors.any? && (!current_api_user.respond_to?(:has_spree_role?) || !current_api_user.has_spree_role?(:admin))
+        #   current_api_user.vendors.first
+        # end
+      end
+    end
+  end
+end

data/app/helpers/spree_multi_vendor/spree/api/api_helpers_decorator.rb

@@ -0,0 +1,19 @@
+module SpreeMultiVendor
+  module Spree
+    module Api
+      module ApiHelpersDecorator
+        def self.prepended(base)
+          base::ATTRIBUTES.push(:vendor_attributes)
+
+          base.mattr_reader *base::ATTRIBUTES
+
+          base.user_attributes << :vendor_ids
+        end
+
+        @@vendor_attributes = [:id, :name, :slug, :state, :about_us, :contact_us]
+      end
+    end
+  end
+end
+
+# ::Spree::Helpers::Spree::Api::V2.prepend(SpreeMultiVendor::Spree::Api::ApiHelpersDecorator)

data/app/mailers/spree/vendor_mailer.rb

@@ -0,0 +1,15 @@
+module Spree
+  class VendorMailer < ::Spree::BaseMailer
+    def vendor_notification_email(order_id, vendor_id)
+      @vendor = Spree::Vendor.find(vendor_id)
+      return unless @vendor.notification_email.present?
+
+      @order = Spree::Order.find(order_id)
+      @line_items = @order.line_items.for_vendor(@vendor)
+      @subtotal = @order.vendor_subtotal(@vendor)
+      @total = @order.vendor_total(@vendor)
+      subject = "#{Spree::Store.current.name} #{Spree.t('order_mailer.vendor_notification_email.subject')} ##{@order.number}"
+      mail(to: @vendor.notification_email, from: from_address, subject: subject)
+    end
+  end
+end

data/app/models/concerns/spree/vendor_concern.rb

@@ -0,0 +1,11 @@
+module Spree
+  module VendorConcern
+    extend ActiveSupport::Concern
+
+    included do
+      belongs_to :vendor, class_name: 'Spree::Vendor', touch: true, optional: true
+
+      scope :with_vendor, ->(vendor_id) { where(vendor_id: vendor_id) }
+    end
+  end
+end

data/app/models/spree/order_commission.rb

@@ -0,0 +1,11 @@
+module Spree
+  class OrderCommission < Base
+    belongs_to :order, required: Spree.version.to_f >= 3.5
+    belongs_to :vendor, required: Spree.version.to_f >= 3.5
+
+    validates :order, :vendor, presence: true
+    validates :vendor_id, uniqueness: { scope: :order_id }
+
+    scope :for_vendor, ->(vendor) { where(vendor_id: vendor.id) }
+  end
+end

data/app/models/spree/vendor.rb

@@ -0,0 +1,77 @@
+module Spree
+  class Vendor < Spree::Base
+    extend FriendlyId
+
+    acts_as_paranoid
+    acts_as_list column: :priority
+    friendly_id :slug, use: %i[slugged history]
+    validates :name,
+              presence: true,
+              uniqueness: { case_sensitive: false }
+
+    validates :slug, uniqueness: true
+    # validates :slug, uniqueness: { scope: :store_id, allow_nil: true, case_sensitive: true }
+    validates_associated :image if Spree.version.to_f >= 3.6
+
+    validates :notification_email, email: true, allow_blank: true
+
+    with_options dependent: :destroy do
+      has_one :image, as: :viewable, dependent: :destroy, class_name: 'Spree::VendorImage'
+      has_many :commissions, class_name: 'Spree::OrderCommission'
+      has_many :vendor_users
+
+      SpreeMultiVendor::Config[:vendorized_models].uniq.compact.each do |model|
+        has_many model.pluralize.to_sym
+      end
+    end
+
+    has_many :users, through: :vendor_users, class_name: Spree.user_class.to_s
+
+    after_create :create_stock_location
+    after_update :update_stock_location_names
+
+    state_machine :state, initial: :pending do
+      event :activate do
+        transition to: :active
+      end
+
+      event :block do
+        transition to: :blocked
+      end
+    end
+
+    scope :active, -> { where(state: 'active') }
+
+    self.whitelisted_ransackable_attributes = %w[name state]
+
+    def update_notification_email(email)
+      update(notification_email: email)
+    end
+
+    # Spree Globalize support
+    # https://github.com/spree-contrib/spree_multi_vendor/issues/104
+    if defined?(SpreeGlobalize)
+      attr_accessor :translations_attributes
+      translates :name,
+                 :about_us,
+                 :contact_us,
+                 :slug, fallbacks_for_empty_translations: true
+    end
+
+    private
+
+    def create_stock_location
+      stock_locations.where(name: name, country: Spree::Store.default.default_country).first_or_create!
+    end
+
+    def should_generate_new_friendly_id?
+      slug.blank? || name_changed?
+    end
+
+    def update_stock_location_names
+      if (Spree.version.to_f < 3.5 && self.name_changed?) || (Spree.version.to_f >= 3.5 && saved_changes&.include?(:name))
+        stock_locations.update_all({ name: name })
+      end
+    end
+  end
+end

data/app/models/spree/vendor_ability.rb

@@ -0,0 +1,155 @@
+class Spree::VendorAbility
+  include CanCan::Ability
+
+  def initialize(user)
+    @vendor_ids = user.vendors.active.ids
+
+    if @vendor_ids.any?
+      apply_classifications_permissions
+      apply_order_permissions
+      apply_image_permissions
+      apply_option_type_permissions
+      apply_price_permissions
+      apply_product_option_type_permissions
+      apply_product_permissions
+      apply_product_properties_permissions
+      apply_properties_permissions
+      apply_prototypes_permissions
+      apply_shipment_permissions
+      apply_shipping_methods_permissions
+      apply_stock_permissions
+      apply_stock_item_permissions
+      apply_stock_location_permissions
+      apply_stock_movement_permissions
+      apply_variant_permissions
+      apply_vendor_permissions
+      apply_vendor_settings_permissions
+      apply_state_changes_permissions
+    end
+  end
+
+  protected
+
+  def apply_classifications_permissions
+    can :manage, Spree::Classification, product: { vendor_id: @vendor_ids }
+  end
+
+  def apply_order_permissions
+    cannot :create, Spree::Order
+
+    order_scope = if ::Spree::Order.reflect_on_association(:vendor)
+                    { vendor_id: @vendor_ids }
+                  elsif ::Spree::LineItem.reflect_on_association(:vendor)
+                    { line_items: { vendor_id: @vendor_ids } }
+                  elsif ::Spree::Product.reflect_on_association(:vendor)
+                    { line_items: { product: { vendor_id: @vendor_ids } } }
+                  elsif ::Spree::Variant.reflect_on_association(:vendor)
+                    { line_items: { variant: { vendor_id: @vendor_ids } } }
+                  end
+
+    if order_scope.present?
+      can %i[admin show index edit update cart], Spree::Order, order_scope.merge(state: 'complete')
+    else
+      cannot_display_model(Spree::Order)
+    end
+  end
+
+  def apply_image_permissions
+    can :create, Spree::Image
+
+    can [:manage, :modify], Spree::Image, ['viewable_type = ?', 'Spree::Variant'] do |image|
+      vendor_id = image.viewable.try(:vendor_id)
+      vendor_id.present? && @vendor_ids.include?(vendor_id)
+    end
+  end
+
+  def apply_option_type_permissions
+    can :display, Spree::OptionType
+    can :display, Spree::OptionValue
+  end
+
+  def apply_price_permissions
+    can :modify, Spree::Price, variant: { vendor_id: @vendor_ids }
+  end
+
+  def apply_product_option_type_permissions
+    can :manage,  Spree::ProductOptionType, product: { vendor_id: @vendor_ids }
+    can :create,  Spree::ProductOptionType
+    can :manage,  Spree::OptionValueVariant, variant: { vendor_id: @vendor_ids }
+    can :create,  Spree::OptionValueVariant
+  end
+
+  def apply_product_permissions
+    cannot_display_model(Spree::Product)
+
+    return unless Spree::Product.reflect_on_association(:vendor)
+
+    can :manage, Spree::Product, vendor_id: @vendor_ids
+    can :create, Spree::Product
+  end
+
+  def apply_properties_permissions
+    can :display, Spree::Property
+  end
+
+  def apply_product_properties_permissions
+    can :manage,  Spree::ProductProperty, product: { vendor_id: @vendor_ids }
+    can :create,  Spree::ProductProperty
+  end
+
+  def apply_prototypes_permissions
+    can [:read, :admin], Spree::Prototype
+  end
+
+  def apply_shipment_permissions
+    can :update, Spree::Shipment, inventory_units: { line_item: { product: { vendor_id: @vendor_ids } } }
+  end
+
+  def apply_shipping_methods_permissions
+    can :manage, Spree::ShippingMethod, vendor_id: @vendor_ids
+    can :create, Spree::ShippingMethod
+  end
+
+  def apply_stock_permissions
+    can :admin, Spree::Stock
+  end
+
+  def apply_stock_item_permissions
+    can [:admin, :modify, :read], Spree::StockItem, stock_location: { vendor_id: @vendor_ids }
+  end
+
+  def apply_stock_location_permissions
+    can :manage, Spree::StockLocation, vendor_id: @vendor_ids
+    can :create, Spree::StockLocation
+  end
+
+  def apply_stock_movement_permissions
+    can :create, Spree::StockMovement
+    can :manage, Spree::StockMovement, stock_item: { stock_location: { vendor_id: @vendor_ids } }
+  end
+
+  def apply_variant_permissions
+    cannot_display_model(Spree::Variant)
+    can :manage, Spree::Variant, vendor_id: @vendor_ids
+    can :create, Spree::Variant
+  end
+
+  def apply_vendor_permissions
+    can :manage, Spree::Vendor, id: @vendor_ids
+    cannot :create, Spree::Vendor
+  end
+
+  def apply_vendor_settings_permissions
+    can :manage, :vendor_settings
+  end
+
+  def apply_state_changes_permissions
+    can [:admin, :index], Spree::StateChange do |state_change|
+      (@vendor_ids & state_change.user.vendor_ids).any?
+    end
+  end
+
+  def cannot_display_model(model)
+    Spree.version.to_f < 4.0 ? (cannot :display, model) : (cannot :read, model)
+  end
+end

data/app/models/spree/vendor_image.rb

@@ -0,0 +1,18 @@
+module Spree
+  class VendorImage < Asset
+    include Spree::Image::Configuration::ActiveStorage
+    include Rails.application.routes.url_helpers
+
+    def styles
+      self.class.styles.map do |_, size|
+        width, height = size[/(\d+)x(\d+)/].split('x')
+
+        {
+          url: polymorphic_path(attachment.variant(resize: size), only_path: true),
+          width: width,
+          height: height
+        }
+      end
+    end
+  end
+end

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, touch: true
+
+    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,107 @@
+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_pre_tax_item_amount(vendor) +
+      vendor_pre_tax_ship_amount(vendor) +
+      vendor_additional_tax_total(vendor)
+  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 unless Spree::Order.included_modules.include?(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