payolapayments/payola

require 'aasm'

module Payola
  class Sale < ActiveRecord::Base
    include Payola::GuidBehavior

    has_paper_trail if respond_to? :has_paper_trail

    validates_presence_of :email
    validates_presence_of :product_id
    validates_presence_of :product_type
    validates_presence_of :stripe_token
    validates_presence_of :currency

    belongs_to :product, Rails::VERSION::MAJOR > 4 ? { polymorphic: true, optional: true } : { polymorphic: true }
    belongs_to :owner, Rails::VERSION::MAJOR > 4 ? { polymorphic: true, optional: true } : { polymorphic: true }
    belongs_to :coupon, Rails::VERSION::MAJOR > 4 ? { optional: true } : {}
    belongs_to :affiliate, Rails::VERSION::MAJOR > 4 ? { optional: true } : {}

    include AASM

    aasm column: 'state', skip_validation_on_save: true do
      state :pending, initial: true
      state :processing
      state :finished
      state :errored
      state :refunded

      event :process, after: :charge_card do
        transitions from: :pending, to: :processing
      end

      event :finish, after: :instrument_finish do
        transitions from: :processing, to: :finished
      end

      event :fail, after: :instrument_fail do
        transitions from: [:pending, :processing], to: :errored
      end

      event :refund, after: :instrument_refund do
        transitions from: :finished, to: :refunded
      end
    end

    def verifier
      @verifier ||= ActiveSupport::MessageVerifier.new(Payola.secret_key_for_sale(self), digest: 'SHA256')
    end

    def verify_charge
      begin
        self.verify_charge!
      rescue RuntimeError => e
        self.error = e.message
        self.fail!
      end
    end

    def verify_charge!
      if Payola.charge_verifier.arity > 1
        Payola.charge_verifier.call(self, custom_fields)
      else
        Payola.charge_verifier.call(self)
      end
    end

    def custom_fields
      if self.signed_custom_fields.present?
        verifier.verify(self.signed_custom_fields)
      else
        nil
      end      
    end

    def redirector
      product
    end

    private

    def charge_card
      Payola::ChargeCard.call(self)
    end

    def instrument_finish
      Payola.instrument(instrument_key('finished'), self)
      Payola.instrument(instrument_key('finished', false), self)
    end

    def instrument_fail
      Payola.instrument(instrument_key('failed'), self)
      Payola.instrument(instrument_key('failed', false), self)
    end

    def instrument_refund
      Payola.instrument(instrument_key('refunded'), self)
      Payola.instrument(instrument_key('refunded', false), self)
    end

    def product_class
      product.product_class
    end

    def instrument_key(instrument_type, include_class=true)
      if include_class
        "payola.#{product_class}.sale.#{instrument_type}"
      else
        "payola.sale.#{instrument_type}"
      end
    end

  end
end