solidus_marketplace 0.1.0

data/app/controllers/spree/admin/reports_controller_decorator.rb

@@ -0,0 +1,68 @@
+require 'csv'
+
+Spree::Admin::ReportsController.class_eval do
+  before_action :add_marketplace_reports, only: [:index]
+
+  def earnings
+    @supplier_earnings = get_supplier_earnings
+    respond_to do |format|
+      format.html
+      format.csv { send_data earnings_csv }
+    end
+  end
+
+  def earnings_csv
+    header1 = ["Supplier Earnings"]
+    header2 = ["Supplier", "Earnings", "Paypal Email"]
+
+    CSV.generate do |csv|
+      csv << header1
+      csv << []
+      csv << header2
+      @supplier_earnings.each do |se|
+        csv << ["#{se[:name]}", "#{se[:earnings].to_html}", "#{se[:paypal_email]}"]
+      end
+    end
+  end
+
+  private
+
+  def add_marketplace_reports
+    marketplace_reports.each do |report|
+      Spree::Admin::ReportsController.add_available_report! report
+    end
+  end
+
+  def marketplace_reports
+    [:earnings]
+  end
+
+  def get_supplier_earnings
+    grouped_supplier_earnings.each do |se|
+      se[:earnings] = se[:earnings].inject(Spree::Money.new(0)){ |e, c| c + e }
+    end
+  end
+
+  def grouped_supplier_earnings
+    params[:q] = search_params
+    @search = Spree::Order.complete.ransack(params[:q])
+    @orders = @search.result
+
+    supplier_earnings_map = @orders.map { |o| o.supplier_earnings_map }
+    grouped_suppliers_map = supplier_earnings_map.flatten.group_by { |e| e[:name] }.values
+    grouped_earnings = grouped_suppliers_map.map do |gs|
+      h = {}
+      h[:name] = nil
+      h[:paypal_email] = nil
+      h[:earnings] = []
+      gs.each do |s|
+        h[:name] = s[:name] if h[:name].nil?
+        h[:paypal_email] = s[:paypal_email] if h[:paypal_email].nil?
+        h[:earnings] << s[:earnings]
+      end
+      h
+    end
+    grouped_earnings
+  end
+
+end

data/app/controllers/spree/admin/shipments_controller.rb

@@ -0,0 +1,47 @@
+module Spree
+  module Admin
+    class ShipmentsController < Spree::Admin::ResourceController
+
+      def index
+        params[:q] ||= {}
+        # params[:q][:completed_at_null] ||= '1'
+        # @show_only_incomplete = params[:q][:completed_at_null].present?
+        # params[:q][:s] ||= @show_only_incomplete ? 'created_at desc' : 'completed_at desc'
+
+        # As date params are deleted if @show_only_incomplete, store
+        # the original date so we can restore them into the params
+        # after the search
+        created_at_gt = params[:q][:created_at_gt]
+        created_at_lt = params[:q][:created_at_lt]
+
+        if !params[:q][:created_at_gt].blank?
+          params[:q][:created_at_gt] = Time.zone.parse(params[:q][:created_at_gt]).beginning_of_day rescue ""
+        end
+
+        if !params[:q][:created_at_lt].blank?
+          params[:q][:created_at_lt] = Time.zone.parse(params[:q][:created_at_lt]).end_of_day rescue ""
+        end
+
+        @search = Spree::Shipment.accessible_by(current_ability, :index).ransack(params[:q])
+        @shipments = @search.result.
+          page(params[:page]).
+          per(params[:per_page] || Spree::Config[:orders_per_page])
+
+        # Restore dates
+        params[:q][:created_at_gt] = created_at_gt
+        params[:q][:created_at_lt] = created_at_lt
+      end
+
+      private
+
+      def find_resource
+        if parent_data.present?
+          parent.send(controller_name).find_by!(number: params[:id])
+        else
+          model_class.find_by!(number: params[:id])
+        end
+      end
+
+    end
+  end
+end

data/app/controllers/spree/admin/stock_items_controller_decorator.rb

@@ -0,0 +1,10 @@
+Spree::Admin::StockItemsController.class_eval do
+  before_action :load_supplier_stock_location, only: :index
+
+  def load_supplier_stock_location
+    if try_spree_current_user.supplier
+      @stock_locations = Spree::StockLocation.by_supplier(try_spree_current_user.supplier).accessible_by(current_ability, :read)
+      @stock_item_stock_locations = params[:stock_location_id].present? ? @stock_locations.where(id: params[:stock_location_id]) : @stock_locations
+    end
+  end
+end

data/app/controllers/spree/admin/stock_locations_controller_decorator.rb

@@ -0,0 +1,23 @@
+Spree::Admin::StockLocationsController.class_eval do
+
+  create.after :set_supplier
+
+  def index
+    @stock_locations = Spree::StockLocation.accessible_by(current_ability, :read)
+                          .order('name ASC')
+                          .ransack(params[:q])
+                          .result
+                          .page(params[:page])
+                          .per(params[:per_page])
+  end
+
+  private
+
+  def set_supplier
+    if try_spree_current_user.supplier?
+      @object.supplier = try_spree_current_user.supplier
+      @object.save
+    end
+  end
+
+end

data/app/controllers/spree/admin/suppliers_controller.rb

@@ -0,0 +1,54 @@
+class Spree::Admin::SuppliersController < Spree::Admin::ResourceController
+
+  def update
+    @object.address = Spree::Address.immutable_merge(@object.address, permitted_resource_params[:address_attributes])
+
+    if @object.update_attributes(permitted_resource_params.except(:address_attributes))
+      respond_with(@object) do |format|
+        format.html do
+          flash[:success] = flash_message_for(@object, :successfully_updated)
+          redirect_to location_after_save
+        end
+        format.js { render layout: false }
+      end
+    else
+      respond_with(@object) do |format|
+        format.html do
+          flash.now[:error] = @object.errors.full_messages.join(", ")
+          render_after_update_error
+        end
+        format.js { render layout: false }
+      end
+    end
+  end
+
+  def edit
+    @object.address = Spree::Address.build_default unless @object.address.present?
+    respond_with(@object) do |format|
+      format.html { render :layout => !request.xhr? }
+      format.js   { render :layout => false }
+    end
+  end
+
+  def new
+    @object.address = Spree::Address.build_default
+  end
+
+  private
+
+    def collection
+      params[:q] ||= {}
+      params[:q][:meta_sort] ||= "name.asc"
+      @search = Spree::Supplier.search(params[:q])
+      @collection = @search.result.page(params[:page]).per(Spree::Config[:orders_per_page])
+    end
+
+    def find_resource
+      Spree::Supplier.friendly.find(params[:id])
+    end
+
+    def location_after_save
+      spree.edit_admin_supplier_path(@object)
+    end
+
+end

data/app/controllers/spree/admin/users_controller_decorator.rb

@@ -0,0 +1,34 @@
+Spree::Admin::UsersController.class_eval do
+
+  def create
+    current_supplier = try_spree_current_user.supplier
+    @user = Spree.user_class.new(user_params)
+    @user.supplier = current_supplier
+    if @user.save
+      set_roles
+      set_stock_locations
+
+      flash[:success] = Spree.t(:created_successfully)
+      redirect_to edit_admin_user_url(@user)
+    else
+      load_roles
+      load_stock_locations
+
+      flash.now[:error] = @user.errors.full_messages.join(", ")
+      render :new, status: :unprocessable_entity
+    end
+  end
+
+  private
+
+  def build_resource
+    current_supplier = try_spree_current_user.supplier
+
+    if parent?
+      parent.send(controller_name).build
+    else
+      model_class.new(supplier: current_supplier)
+    end
+  end
+
+end

data/app/controllers/spree/api/suppliers_controller.rb

@@ -0,0 +1,16 @@
+module Spree
+  module Api
+    class SuppliersController < Spree::Api::BaseController
+      def index
+        if params[:ids]
+          @suppliers = Spree::Supplier.accessible_by(current_ability, :read).where(id: params[:ids].split(',')).order(:name)
+        else
+          @suppliers = Spree::Supplier.accessible_by(current_ability, :read).order(:name).ransack(params[:q]).result
+        end
+
+        @suppliers = paginate(@suppliers)
+        respond_with(@suppliers)
+      end
+    end
+  end
+end

data/app/controllers/spree/api/v1/stock_locations_controller_decorator.rb

@@ -0,0 +1,18 @@
+Spree::Api::StockLocationsController.class_eval do
+
+  before_action :supplier_locations, only: [:index]
+  before_action :supplier_transfers, only: [:index]
+
+  private
+
+  def supplier_locations
+    params[:q] ||= {}
+    params[:q][:supplier_id_eq] = spree_current_user.supplier_id
+  end
+
+  def supplier_transfers
+    params[:q] ||= {}
+    params[:q][:supplier_id_eq] = spree_current_user.supplier_id
+  end
+
+end

data/app/controllers/spree/base_controller_decorator.rb

@@ -0,0 +1,13 @@
+Spree::BaseController.class_eval do
+
+  prepend_before_action :redirect_supplier
+
+  private
+
+  def redirect_supplier
+    if ['/admin', '/admin/authorization_failure'].include?(request.path) && try_spree_current_user.try(:supplier)
+      redirect_to '/admin/shipments' and return false
+    end
+  end
+
+end

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

@@ -0,0 +1,22 @@
+module Spree
+  module Api
+    module ApiHelpers
+
+      @@supplier_attributes = [
+        :id,
+        :address_id,
+        :commission_flat_rate,
+        :commission_percentage,
+        :email,
+        :name,
+        :url,
+        :deleted_at,
+        :tax_id,
+        :token,
+        :slug
+      ]
+
+      mattr_reader(:supplier_attributes)
+    end
+  end
+end

data/app/mailers/spree/marketplace_order_mailer.rb

@@ -0,0 +1,13 @@
+module Spree
+  class MarketplaceOrderMailer < Spree::BaseMailer
+
+    default from: Spree::Store.default.mail_from_address
+
+    def supplier_order(shipment_id)
+      @shipment = Shipment.find shipment_id
+      @supplier = @shipment.supplier
+      mail to: @supplier.email, subject: Spree.t('marketplace_order_mailer.supplier_order.subject', name: Spree::Store.current.name, number: @shipment.number)
+    end
+
+  end
+end

data/app/mailers/spree/supplier_mailer.rb

@@ -0,0 +1,12 @@
+module Spree
+  class SupplierMailer < Spree::BaseMailer
+
+    default from: Spree::Store.default.mail_from_address
+
+    def welcome(supplier_id)
+      @supplier = Supplier.find supplier_id
+      mail to: @supplier.email, subject: Spree.t('supplier_mailer.welcome.subject')
+    end
+
+  end
+end

data/app/models/ckeditor/asset_decorator.rb

@@ -0,0 +1,5 @@
+if defined?(Ckeditor)
+  Ckeditor::Asset.class_eval do
+    belongs_to :supplier, class_name: 'Spree::Supplier'
+  end
+end

data/app/models/spree.rb

@@ -0,0 +1,5 @@
+module Spree
+  def self.table_name_prefix
+    'spree_'
+  end
+end

data/app/models/spree/marketplace_configuration.rb

@@ -0,0 +1,14 @@
+module Spree
+  class MarketplaceConfiguration < Preferences::Configuration
+
+    # Default flat rate to charge suppliers per order for commission.
+    preference :default_commission_flat_rate, :float, default: 0.0
+
+    # Default percentage to charge suppliers per order for commission.
+    preference :default_commission_percentage, :float, default: 0.0
+
+    # Determines whether or not to email a new supplier their welcome email.
+    preference :send_supplier_email, :boolean, default: true
+
+  end
+end

data/app/models/spree/order_decorator.rb

@@ -0,0 +1,44 @@
+Spree::Order.class_eval do
+
+  has_many :stock_locations, through: :shipments
+  has_many :suppliers, through: :stock_locations
+
+  def supplier_total(user_or_supplier)
+    supplier = user_or_supplier.is_a?(Spree::Supplier) ? user_or_supplier : user_or_supplier.supplier
+    shipments = self.shipments.by_supplier(supplier)
+    commissions = shipments.map(&:supplier_commission_total)
+    Spree::Money.new(commissions.sum)
+  end
+
+  def supplier_earnings_map
+    suppliers.map do |s|
+      {
+        name: s.name,
+        earnings: self.supplier_total(s),
+        paypal_email: s.paypal_email
+      }
+    end
+  end
+
+  # Once order is finalized we want to notify the suppliers of their drop ship orders.
+  # Here we are handling notification by emailing the suppliers.
+  # If you want to customize this you could override it as a hook for notifying a supplier with a API request instead.
+  def finalize_with_supplier!
+    finalize_without_supplier!
+    shipments.each do |shipment|
+      if SolidusMarketplace::Config[:send_supplier_email] && shipment.supplier.present?
+        begin
+          Spree::MarketplaceOrderMailer.supplier_order(shipment.id).deliver!
+        rescue => ex #Errno::ECONNREFUSED => ex
+          puts ex.message
+          puts ex.backtrace.join("\n")
+          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
+      end
+    end
+  end
+  # alias_method :finalize!, :drop_ship
+
+end

data/app/models/spree/payment_decorator.rb

@@ -0,0 +1,7 @@
+module Spree
+  Payment.class_eval do
+
+    belongs_to :payable, polymorphic: true
+
+  end
+end

data/app/models/spree/product_decorator.rb

@@ -0,0 +1,52 @@
+Spree::Product.class_eval do
+
+  has_many :suppliers, -> { readonly }, through: :master
+
+  scope :of_supplier, -> (supplier_id) { joins(:suppliers).where('spree_suppliers.id = ?', supplier_id) }
+
+  def add_supplier!(supplier_or_id)
+    supplier = supplier_or_id.is_a?(Spree::Supplier) ? supplier_or_id : Spree::Supplier.find(supplier_or_id)
+    populate_for_supplier! supplier if supplier
+  end
+
+  def add_suppliers!(supplier_ids)
+    Spree::Supplier.where(id: supplier_ids).each do |supplier|
+      populate_for_supplier! supplier
+    end
+  end
+
+  def remove_suppliers!(supplier_ids)
+    Spree::Supplier.where(id: supplier_ids).each do |supplier|
+      unpopulate_for_supplier! supplier
+    end
+  end
+
+  # Returns true if the product has one or more suppliers.
+  def supplier?
+    suppliers.present?
+  end
+
+  private
+
+  def populate_for_supplier!(supplier)
+    variants_including_master.each do |variant|
+      unless variant.suppliers.pluck(:id).include?(supplier.id)
+        variant.suppliers << supplier
+        supplier.stock_locations.each { |location| location.propagate_variant(variant) unless location.stock_item(variant) }
+      end
+    end
+  end
+
+  def unpopulate_for_supplier!(supplier)
+    variants_including_master.each do |variant|
+      if variant.suppliers.pluck(:id).include?(supplier.id)
+        variant.suppliers.delete(supplier)
+
+        supplier.stock_locations.each do |location|
+          location.unpropagate_variant(variant)
+        end
+      end
+    end
+  end
+
+end