stripe_model_callbacks 0.1.2 → 0.1.3

data/lib/stripe_model_callbacks/models/stripe_source.rb

@@ -9,6 +9,8 @@ class StripeSource < StripeModelCallbacks::ApplicationRecord
   end
 
   def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
+
     self.stripe_type = object.type
     self.amount = Money.new(object.amount, object.currency) if object.respond_to?(:amount)
 
@@ -28,7 +30,7 @@ class StripeSource < StripeModelCallbacks::ApplicationRecord
 
 private
 
-  def assign_owner(object)
+  def assign_owner(object) # rubocop:disable Metrics/AbcSize
     assign_attributes(
       owner_address_city: object.owner.address&.city,
       owner_address_country: object.owner.address&.country,
@@ -42,7 +44,7 @@ private
     )
   end
 
-  def assign_owner_verified(object)
+  def assign_owner_verified(object) # rubocop:disable Metrics/AbcSize
     assign_attributes(
       owner_verified_address_city: object.owner.verified_address&.city,
       owner_verified_address_country: object.owner.verified_address&.country,

data/lib/stripe_model_callbacks/models/stripe_subscription.rb

@@ -4,7 +4,11 @@ class StripeSubscription < StripeModelCallbacks::ApplicationRecord
   belongs_to :stripe_plan, optional: true, primary_key: "stripe_id"
   has_many :stripe_invoices, primary_key: "stripe_id"
   has_many :stripe_discounts, primary_key: "stripe_id"
+  has_many :stripe_subscription_default_tax_rates, autosave: true, dependent: :destroy
   has_many :stripe_subscription_items, autosave: true, primary_key: "stripe_id"
+  has_many :stripe_subscription_schedules, primary_key: "stripe_id"
+  has_many :stripe_plans, through: :stripe_subscription_items
+  has_many :default_tax_rates, source: :stripe_tax_rate, through: :stripe_subscription_default_tax_rates
 
   STATES = %w[trialing active past_due canceled unpaid].freeze
 
@@ -18,13 +22,16 @@ class StripeSubscription < StripeModelCallbacks::ApplicationRecord
   end
 
   def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
     assign_attributes(
       canceled_at: object.canceled_at ? Time.zone.at(object.canceled_at) : nil,
-      stripe_customer_id: object.customer,
       ended_at: object.ended_at ? Time.zone.at(object.ended_at) : nil,
-      stripe_plan_id: object.plan&.id
+      latest_stripe_invoice_id: latest_invoice_id(object),
+      stripe_customer_id: object.customer,
+      stripe_plan_id: object.respond_to?(:plan) ? object.plan&.id : nil
     )
 
+    assign_default_tax_rates(object)
     assign_discount(object)
     assign_items(object)
     assign_periods(object)
@@ -63,6 +70,32 @@ class StripeSubscription < StripeModelCallbacks::ApplicationRecord
 
 private
 
+  def assign_default_tax_rates(object)
+    return unless object.try(:default_tax_rates)
+
+    found_ids = []
+
+    object.default_tax_rates.each do |default_tax_rate|
+      tax_rate = StripeModelCallbacks::TaxRate::UpdatedService.execute!(object: default_tax_rate)
+
+      if new_record?
+        stripe_subscription_default_tax_rates.build(stripe_subscription_id: object.id, stripe_tax_rate_id: tax_rate.id)
+      else
+        default_tax_rate = StripeSubscriptionDefaultTaxRate.find_by(stripe_subscription_id: object.id, stripe_tax_rate_id: tax_rate.id)
+        stripe_subscription_default_tax_rates.build(stripe_subscription_id: object.id, stripe_tax_rate_id: tax_rate.id) unless default_tax_rate
+        found_ids << default_tax_rate.id if default_tax_rate
+      end
+    end
+
+    clean_up_default_tax_rates_not_found(found_ids) if persisted?
+  end
+
+  def clean_up_default_tax_rates_not_found(found_ids)
+    stripe_subscription_default_tax_rates.select(&:persisted?).each do |subscription_default_tax_rate|
+      subscription_default_tax_rate.mark_for_destruction if found_ids.exclude?(subscription_default_tax_rate.stripe_tax_rate_id)
+    end
+  end
+
   def assign_discount(object)
     return if object.discount.blank?
 
@@ -80,10 +113,16 @@ private
   end
 
   def assign_periods(object)
+    start_date = if object.respond_to?(:start_date)
+      object.start_date
+    else
+      object.start
+    end
+
     assign_attributes(
       current_period_start: Time.zone.at(object.current_period_start),
       current_period_end: Time.zone.at(object.current_period_end),
-      start: Time.zone.at(object.start),
+      start_date: Time.zone.at(start_date),
       trial_start: object.trial_start ? Time.zone.at(object.trial_start) : nil,
       trial_end: object.trial_end ? Time.zone.at(object.trial_end) : nil
     )
@@ -91,11 +130,22 @@ private
 
   def find_item_by_stripe_item(item)
     stripe_subscription_items.find do |sub_item|
-      if item.plan.is_a?(String)
-        sub_item.stripe_plan_id == item.plan
+      if item.try(:plan)
+        if item.plan.is_a?(String)
+          sub_item.stripe_plan_id == item.plan
+        else
+          sub_item.stripe_plan_id == item.plan.id
+        end
       else
-        sub_item.stripe_plan_id == item.plan.id
+        sub_item.stripe_price_id == item.price.id
       end
     end
   end
+
+  def latest_invoice_id(object)
+    return unless object.respond_to?(:latest_invoice)
+    return object.latest_invoice if object.latest_invoice.is_a?(String)
+
+    object.latest_invoice&.id
+  end
 end

data/lib/stripe_model_callbacks/models/stripe_subscription_default_tax_rate.rb

@@ -0,0 +1,4 @@
+class StripeSubscriptionDefaultTaxRate < StripeModelCallbacks::ApplicationRecord
+  belongs_to :stripe_subscription
+  belongs_to :stripe_tax_rate
+end

data/lib/stripe_model_callbacks/models/stripe_subscription_item.rb

@@ -1,5 +1,6 @@
 class StripeSubscriptionItem < StripeModelCallbacks::ApplicationRecord
   belongs_to :stripe_plan, optional: true, primary_key: "stripe_id"
+  belongs_to :stripe_price, optional: true, primary_key: "stripe_id"
   belongs_to :stripe_subscription, optional: true, primary_key: "stripe_id"
 
   has_many :stripe_invoice_items, primary_key: "stripe_id"
@@ -9,8 +10,11 @@ class StripeSubscriptionItem < StripeModelCallbacks::ApplicationRecord
   end
 
   def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
     self.stripe_subscription_id = object.subscription if object.respond_to?(:subscription)
-    self.stripe_plan_id = object.plan.id if object.plan.respond_to?(:id)
+    self.stripe_plan_id = object.plan.id if object.try(:plan).respond_to?(:id)
+
+    assign_price_from_stripe(object)
 
     StripeModelCallbacks::AttributesAssignerService.execute!(
       model: self, stripe_model: object,
@@ -33,4 +37,18 @@ class StripeSubscriptionItem < StripeModelCallbacks::ApplicationRecord
     stripe_subscription.reload_from_stripe!
     nil
   end
+
+private
+
+  def assign_price_from_stripe(object)
+    # Older versions doesn't support price
+    return unless object.try(:price).respond_to?(:id)
+
+    # Make sure price is created
+    price = StripePrice.find_by(stripe_id: object.price.id)
+    StripePrice.create_from_stripe!(object.price) unless price
+
+    # Set stripe ID on the subscription item
+    self.stripe_price_id = object.price.id
+  end
 end

data/lib/stripe_model_callbacks/models/stripe_subscription_schedule.rb

@@ -0,0 +1,94 @@
+class StripeSubscriptionSchedule < StripeModelCallbacks::ApplicationRecord
+  MATCHING_STRIPE_ATTRIBUTES = %w[
+    billing created collection_method
+    default_payment_method default_source
+    end_behavior id metadata livemode
+    renewal_behavior renewal_interval
+    status
+  ].freeze
+  private_constant :MATCHING_STRIPE_ATTRIBUTES
+
+  belongs_to :stripe_customer, optional: true, primary_key: "stripe_id"
+  belongs_to :stripe_subscription, optional: true, primary_key: "stripe_id"
+
+  has_many :stripe_subscription_schedule_phases, primary_key: "stripe_id", dependent: :destroy
+
+  def self.stripe_class
+    Stripe::SubscriptionSchedule
+  end
+
+  def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
+    assign_attributes(
+      released_stripe_subscription_id: object.released_subscription,
+      stripe_customer_id: object.customer,
+      stripe_subscription_id: object.subscription
+    )
+
+    assign_billing_thresholds(object)
+    assign_current_phase(object)
+    assign_invoice_settings(object)
+    assign_timestamps(object)
+
+    StripeModelCallbacks::AttributesAssignerService.execute!(
+      model: self,
+      stripe_model: object,
+      attributes: MATCHING_STRIPE_ATTRIBUTES
+    )
+
+    assign_subscription_schedule_phases(object)
+  end
+
+  def cancel_on_stripe
+    to_stripe.cancel
+    update!(canceled_at: Time.zone.now) if respond_to?(:canceled_at)
+    reload_from_stripe!
+    true
+  end
+
+  def cancel_on_stripe!
+    raise ActiveRecord::RecordInvalid, self unless cancel_on_stripe
+  end
+
+private
+
+  def assign_billing_thresholds(object)
+    billing_thresholds = object.billing_thresholds
+
+    self.billing_thresholds_amount_gte = billing_thresholds&.amount_gte
+    self.billing_thresholds_reset_billing_cycle_anchor = billing_thresholds&.reset_billing_cycle_anchor
+  end
+
+  def assign_current_phase(object)
+    current_phase = object.current_phase
+
+    start_date = current_phase&.start_date
+    self.current_phase_start_date = Time.zone.at(start_date) if start_date
+
+    end_date = current_phase&.end_date
+    self.current_phase_end_date = Time.zone.at(end_date) if end_date
+  end
+
+  def assign_invoice_settings(object)
+    self.invoice_settings_days_until_due = object.invoice_settings&.days_until_due
+  end
+
+  def assign_timestamps(object)
+    %i[canceled_at completed_at released_at].each do |timestamp|
+      object_timestamp = object.__send__(timestamp)
+      __send__("#{timestamp}=", Time.zone.at(object_timestamp)) if object_timestamp
+    end
+  end
+
+  def assign_subscription_schedule_phases(object)
+    self.stripe_subscription_schedule_phases = new_stripe_subscription_schedule_phases(object)
+  end
+
+  def new_stripe_subscription_schedule_phases(object)
+    @new_stripe_subscription_schedule_phases ||= object.phases.collect do |phase|
+      subscription_schedule_phase = StripeSubscriptionSchedulePhase.new(stripe_subscription_schedule: self)
+      subscription_schedule_phase.assign_from_stripe(phase)
+      subscription_schedule_phase
+    end
+  end
+end

data/lib/stripe_model_callbacks/models/stripe_subscription_schedule_phase.rb

@@ -0,0 +1,67 @@
+class StripeSubscriptionSchedulePhase < StripeModelCallbacks::ApplicationRecord
+  MATCHING_STRIPE_ATTRIBUTES = %w[
+    application_fee_percent
+    collection_method
+    default_payment_method
+    prorate
+    proration_behavior
+  ].freeze
+  private_constant :MATCHING_STRIPE_ATTRIBUTES
+
+  belongs_to :stripe_subscription_schedule, primary_key: "stripe_id"
+
+  has_many :stripe_subscription_schedule_phase_plans, dependent: :destroy
+
+  def self.stripe_class
+    Stripe::SubscriptionSchedulePhase
+  end
+
+  def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
+    StripeModelCallbacks::AttributesAssignerService.execute!(
+      model: self,
+      stripe_model: object,
+      attributes: MATCHING_STRIPE_ATTRIBUTES
+    )
+
+    assign_billing_thresholds(object)
+    assign_invoice_settings(object)
+    assign_timestamps(object)
+
+    assign_subscription_schedule_phase_plans(object)
+  end
+
+private
+
+  def assign_billing_thresholds(object)
+    billing_thresholds = object.billing_thresholds
+
+    self.billing_thresholds_amount_gte = billing_thresholds&.amount_gte
+    self.billing_thresholds_reset_billing_cycle_anchor = billing_thresholds&.reset_billing_cycle_anchor
+  end
+
+  def assign_invoice_settings(object)
+    invoice_settings = object.invoice_settings
+
+    self.invoice_settings_days_until_due = invoice_settings&.days_until_due
+  end
+
+  def assign_timestamps(object)
+    %i[end_date start_date trial_end].each do |timestamp|
+      object_timestamp = object.__send__(timestamp)
+      __send__("#{timestamp}=", Time.zone.at(object_timestamp)) if object_timestamp
+    end
+  end
+
+  def assign_subscription_schedule_phase_plans(object)
+    self.stripe_subscription_schedule_phase_plans = new_stripe_subscription_schedule_phase_plans(object)
+  end
+
+  def new_stripe_subscription_schedule_phase_plans(object)
+    @new_stripe_subscription_schedule_phase_plans ||= object.plans.collect do |plan|
+      subscription_schedule_phase_plan = StripeSubscriptionSchedulePhasePlan.new(stripe_subscription_schedule_phase: self)
+      subscription_schedule_phase_plan.assign_from_stripe(plan)
+      subscription_schedule_phase_plan
+    end
+  end
+end

data/lib/stripe_model_callbacks/models/stripe_subscription_schedule_phase_plan.rb

@@ -0,0 +1,37 @@
+class StripeSubscriptionSchedulePhasePlan < StripeModelCallbacks::ApplicationRecord
+  MATCHING_STRIPE_ATTRIBUTES = %w[quantity].freeze
+  private_constant :MATCHING_STRIPE_ATTRIBUTES
+
+  belongs_to :stripe_plan, optional: true, primary_key: "stripe_id"
+  belongs_to :stripe_subscription_schedule_phase
+
+  has_one :stripe_subscription_schedule, through: :stripe_subscription_schedule_phase
+
+  def self.stripe_class
+    Stripe::SubscriptionSchedulePhasePlan
+  end
+
+  def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
+    StripeModelCallbacks::AttributesAssignerService.execute!(
+      model: self,
+      stripe_model: object,
+      attributes: MATCHING_STRIPE_ATTRIBUTES
+    )
+
+    assign_attributes(
+      stripe_plan_id: object.plan,
+      stripe_price_id: object.price
+    )
+
+    assign_billing_thresholds(object)
+  end
+
+private
+
+  def assign_billing_thresholds(object)
+    billing_thresholds = object.billing_thresholds
+
+    self.billing_thresholds_usage_gte = billing_thresholds&.usage_gte
+  end
+end

data/lib/stripe_model_callbacks/models/stripe_tax_rate.rb

@@ -0,0 +1,22 @@
+class StripeTaxRate < StripeModelCallbacks::ApplicationRecord
+  has_many :stripe_subscription_default_tax_rates, dependent: :destroy
+
+  def self.stripe_class
+    Stripe::TaxRate
+  end
+
+  def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
+    assign_attributes(
+      created: Time.zone.at(object.created),
+      inclusive: object.inclusive == true,
+      stripe_id: object.id
+    )
+
+    StripeModelCallbacks::AttributesAssignerService.execute!(
+      model: self,
+      stripe_model: object,
+      attributes: %w[display_name description jurisdiction percentage inclusive]
+    )
+  end
+end

data/lib/stripe_model_callbacks/models/stripe_transfer.rb

@@ -7,6 +7,7 @@ class StripeTransfer < StripeModelCallbacks::ApplicationRecord
   end
 
   def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
     assign_attributes(
       amount: Money.new(object.amount, object.currency),
       amount_reversed: Money.new(object.amount_reversed, object.currency),

data/lib/stripe_model_callbacks/version.rb

@@ -1,3 +1,3 @@
 module StripeModelCallbacks
-  VERSION = "0.1.2".freeze
+  VERSION = "0.1.3".freeze
 end

data/lib/stripe_model_callbacks.rb

@@ -6,6 +6,7 @@ require "stripe"
 require "stripe_event"
 require "stripe_model_callbacks/engine"
 require "stripe_model_callbacks/autoload_models"
+require "with_advisory_lock"
 
 module StripeModelCallbacks
   path = "#{File.dirname(__FILE__)}/stripe_model_callbacks"