stripe_model_callbacks 0.1.7 → 0.1.9

data/lib/stripe_model_callbacks/models/stripe_customer.rb

@@ -5,6 +5,9 @@ class StripeCustomer < StripeModelCallbacks::ApplicationRecord
   has_many :stripe_invoices, primary_key: "stripe_id"
   has_many :stripe_invoice_items, primary_key: "stripe_id"
   has_many :stripe_orders, primary_key: "stripe_id"
+  has_many :stripe_payment_intents, foreign_key: "customer", primary_key: "stripe_id"
+  has_many :stripe_payment_methods, foreign_key: "customer", primary_key: "stripe_id"
+  has_many :stripe_setup_intents, foreign_key: "customer", primary_key: "stripe_id"
   has_many :stripe_subscriptions, primary_key: "stripe_id"
   has_many :stripe_subscription_schedules, primary_key: "stripe_id"
 
@@ -31,4 +34,13 @@ class StripeCustomer < StripeModelCallbacks::ApplicationRecord
       ]
     )
   end
+
+  def create_stripe_mock!
+    mock_customer = Stripe::Customer.create(
+      id: stripe_id,
+      source: StripeMock.create_test_helper.generate_card_token
+    )
+    assign_from_stripe(mock_customer)
+    save!
+  end
 end

data/lib/stripe_model_callbacks/models/stripe_invoice.rb

@@ -4,6 +4,7 @@ class StripeInvoice < StripeModelCallbacks::ApplicationRecord
   belongs_to :stripe_discount, optional: true
   belongs_to :stripe_subscription, optional: true, primary_key: "stripe_id"
 
+  has_many :stripe_charges, primary_key: "stripe_id"
   has_many :stripe_invoice_items, autosave: true, primary_key: "stripe_id"
 
   validates :stripe_id, uniqueness: true
@@ -16,11 +17,13 @@ class StripeInvoice < StripeModelCallbacks::ApplicationRecord
   monetize :tax_cents, allow_nil: true
   monetize :total_cents, allow_nil: true
 
+  has_one_attached :invoice_pdf_file
+
   def self.stripe_class
     Stripe::Invoice
   end
 
-  def assign_from_stripe(object)
+  def assign_from_stripe(object) # rubocop:disable Metrics/AbcSize
     check_object_is_stripe_class(object)
     assign_attributes(
       stripe_charge_id: object.charge,
@@ -36,6 +39,7 @@ class StripeInvoice < StripeModelCallbacks::ApplicationRecord
 
     assign_amounts(object)
     assign_discount_item(object)
+    attach_invoice_pdf(object)
     assign_tax(object)
 
     assign_forgiven(object)
@@ -57,6 +61,7 @@ class StripeInvoice < StripeModelCallbacks::ApplicationRecord
 
 private
 
+
   def assign_amounts(object)
     assign_attributes(
       amount_due: Money.new(object.amount_due, object.currency),
@@ -114,6 +119,17 @@ private
     end
   end
 
+  def attach_invoice_pdf(object)
+    if object.respond_to?(:invoice_pdf) && object.invoice_pdf && !invoice_pdf_file.attached?
+      require "open-uri"
+
+      url = URI.parse(object.invoice_pdf)
+      filename = File.basename(url.path)
+      file = url.open
+      invoice_pdf_file.attach(io: file, filename: filename)
+    end
+  end
+
   def assign_status_transitions(object)
     return unless object.respond_to?(:status_transitions)
 

data/lib/stripe_model_callbacks/models/stripe_payment_intent.rb

@@ -0,0 +1,125 @@
+class StripePaymentIntent < StripeModelCallbacks::ApplicationRecord
+  CANCELLABLE_STATUSES = %w[requires_payment_method requires_capture requires_confirmation requires_action].freeze
+
+  belongs_to :stripe_customer, foreign_key: "customer", optional: true, primary_key: "stripe_id"
+  belongs_to :stripe_latest_charge,
+    class_name: "StripeCharge",
+    foreign_key: "latest_charge",
+    inverse_of: :latest_charge_on_stripe_payment_intent,
+    optional: true,
+    primary_key: "stripe_id"
+  belongs_to :stripe_payment_method, foreign_key: "payment_method", optional: true, primary_key: "stripe_id"
+
+  has_many :stripe_charges, foreign_key: "payment_intent", primary_key: "stripe_id"
+  has_many :stripe_refunds, foreign_key: "payment_intent", primary_key: "stripe_id"
+
+  scope :cancellable, -> { where(status: CANCELLABLE_STATUSES) }
+
+  def self.stripe_class
+    Stripe::PaymentIntent
+  end
+
+  def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
+
+    StripeModelCallbacks::AttributesAssignerService.execute!(
+      model: self,
+      stripe_model: object,
+      attributes: %w[
+        amount
+        amount_capturable
+        amount_details
+        amount_received
+        application
+        application_fee_amount
+        automatic_payment_methods
+        canceled_at
+        cancellation_reason
+        capture_method
+        client_secret
+        confirmation_method
+        created
+        currency
+        customer
+        description
+        id
+        invoice
+        last_payment_error
+        latest_charge
+        livemode
+        metadata
+        next_action
+        on_behalf_of
+        payment_method
+        payment_method_options
+        payment_method_types
+        processing
+        receipt_email
+        review
+        setup_future_usage
+        shipping
+        statement_descriptor
+        statement_descriptor_suffix
+        status
+        transfer_data
+        transfer_group
+      ]
+    )
+  end
+
+  def amount_money
+    Money.new(amount, currency)
+  end
+
+  def amount_capturable_money
+    Money.new(amount_capturable, currency)
+  end
+
+  def amount_received_money
+    Money.new(amount_received, currency)
+  end
+
+  def cancel(**args)
+    updated_payment_intent = Stripe::PaymentIntent.cancel(stripe_id, **args)
+    assign_from_stripe(updated_payment_intent)
+    save!
+  end
+
+  def cancellable?
+    CANCELLABLE_STATUSES.include?(status)
+  end
+
+  def capture(**args)
+    updated_payment_intent = Stripe::PaymentIntent.capture(stripe_id, **args)
+    assign_from_stripe(updated_payment_intent)
+    save!
+  end
+
+  def confirm(**args)
+    updated_payment_intent = Stripe::PaymentIntent.confirm(stripe_id, **args)
+    assign_from_stripe(updated_payment_intent)
+    save!
+  end
+
+  def create_stripe_mock!
+    mock_payment_intent = Stripe::PaymentIntent.create(
+      amount:,
+      amount_capturable:,
+      amount_details:,
+      amount_received:,
+      application:,
+      application_fee_amount:,
+      automatic_payment_methods:,
+      canceled_at:,
+      cancellation_reason:,
+      capture_method:,
+      client_secret:,
+      currency:,
+      customer:,
+      id: stripe_id,
+      status:
+    )
+    assign_from_stripe(mock_payment_intent)
+    save!
+  end
+end

data/lib/stripe_model_callbacks/models/stripe_payment_method.rb

@@ -0,0 +1,31 @@
+class StripePaymentMethod < StripeModelCallbacks::ApplicationRecord
+  belongs_to :stripe_customer, foreign_key: "customer", optional: true, primary_key: "stripe_id"
+
+  has_many :stripe_payment_intents, foreign_key: "payment_method", primary_key: "stripe_id"
+  has_many :stripe_setup_intents, foreign_key: "payment_method", primary_key: "stripe_id"
+
+  def self.stripe_class
+    Stripe::PaymentMethod
+  end
+
+  def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
+
+    self.stripe_id = object.id
+    self.stripe_type = object.type
+
+    StripeModelCallbacks::AttributesAssignerService.execute!(
+      model: self,
+      stripe_model: object,
+      attributes: %w[
+        billing_details card created customer livemode metadata
+      ]
+    )
+  end
+
+  def create_stripe_mock!
+    mock_payment_method = Stripe::PaymentMethod.create(customer:, id: stripe_id, type: "card")
+    assign_from_stripe(mock_payment_method)
+    save!
+  end
+end

data/lib/stripe_model_callbacks/models/stripe_plan.rb

@@ -30,6 +30,19 @@ class StripePlan < StripeModelCallbacks::ApplicationRecord
     )
   end
 
+  def create_stripe_mock!
+    mock_plan = Stripe::Plan.create(
+      id: stripe_id,
+      amount: amount_cents,
+      currency:,
+      interval:,
+      interval_count:,
+      product: stripe_product.stripe_id
+    )
+    assign_from_stripe(mock_plan)
+    save!
+  end
+
   def name
     nickname.presence || stripe_product&.name
   end

data/lib/stripe_model_callbacks/models/stripe_product.rb

@@ -34,4 +34,14 @@ class StripeProduct < StripeModelCallbacks::ApplicationRecord
       ]
     )
   end
+
+  def create_stripe_mock!
+    mock_product = Stripe::Product.create(
+      id: stripe_id,
+      name:,
+      type: "service"
+    )
+    assign_from_stripe(mock_product)
+    save!
+  end
 end

data/lib/stripe_model_callbacks/models/stripe_recipient.rb

@@ -11,7 +11,7 @@ class StripeRecipient < StripeModelCallbacks::ApplicationRecord
 
     StripeModelCallbacks::AttributesAssignerService.execute!(
       model: self, stripe_model: object,
-      attributes: %w[active_account description email name migrated_to verified]
+      attributes: %w[active_account description email livemode name migrated_to verified]
     )
   end
 end

data/lib/stripe_model_callbacks/models/stripe_refund.rb

@@ -1,5 +1,6 @@
 class StripeRefund < StripeModelCallbacks::ApplicationRecord
   belongs_to :stripe_charge, optional: true, primary_key: "stripe_id"
+  belongs_to :stripe_payment_intent, foreign_key: "payment_intent", optional: true, primary_key: "stripe_id"
 
   monetize :amount_cents, allow_nil: true
 
@@ -12,6 +13,7 @@ class StripeRefund < StripeModelCallbacks::ApplicationRecord
     assign_attributes(
       amount: Money.new(object.amount, object.currency),
       stripe_charge_id: object.charge,
+      stripe_id: object.id,
       created: Time.zone.at(object.created),
       livemode: object.try(:livemode) == true,
       metadata: JSON.generate(object.metadata)
@@ -20,7 +22,7 @@ class StripeRefund < StripeModelCallbacks::ApplicationRecord
     StripeModelCallbacks::AttributesAssignerService.execute!(
       model: self,
       stripe_model: object,
-      attributes: %w[balance_transaction currency reason receipt_number status]
+      attributes: %w[balance_transaction currency payment_intent reason receipt_number status]
     )
 
     self.failure_reason = object.failure_reason if object.respond_to?(:failure_reason)

data/lib/stripe_model_callbacks/models/stripe_setup_intent.rb

@@ -0,0 +1,21 @@
+class StripeSetupIntent < StripeModelCallbacks::ApplicationRecord
+  belongs_to :stripe_customer, foreign_key: "customer", optional: true, primary_key: "stripe_id"
+  belongs_to :stripe_payment_method, foreign_key: "payment_method", optional: true, primary_key: "stripe_id"
+
+  def self.stripe_class
+    Stripe::SetupIntent
+  end
+
+  def assign_from_stripe(object)
+    check_object_is_stripe_class(object)
+
+    StripeModelCallbacks::AttributesAssignerService.execute!(
+      model: self,
+      stripe_model: object,
+      attributes: %w[
+        application cancellation_reason client_secret created customer description flow_directions last_setup_error latest_attempt livemode mandate metadata
+        next_action on_behalf_of payment_method payment_method_options payment_method_types single_use_mandate status usage
+      ]
+    )
+  end
+end

data/lib/stripe_model_callbacks/models/stripe_source.rb

@@ -22,7 +22,8 @@ class StripeSource < StripeModelCallbacks::ApplicationRecord
     assign_ach_credit_transfer(object)
 
     StripeModelCallbacks::AttributesAssignerService.execute!(
-      model: self, stripe_model: object,
+      model: self,
+      stripe_model: object,
       attributes: %w[
         client_secret created currency flow livemode metadata statement_descriptor
         status usage

data/lib/stripe_model_callbacks/models/stripe_subscription.rb

@@ -47,9 +47,8 @@ class StripeSubscription < StripeModelCallbacks::ApplicationRecord
     raise "Cannot reactivate unless cancel at period end has been set" unless cancel_at_period_end?
 
     # https://stripe.com/docs/subscriptions/canceling-pausing
-    items = []
-    stripe_subscription_items.each do |item|
-      items << {
+    items = stripe_subscription_items.map do |item|
+      {
         id: item.stripe_id,
         plan: item.stripe_plan_id,
         quantity: item.quantity
@@ -64,11 +63,30 @@ class StripeSubscription < StripeModelCallbacks::ApplicationRecord
   end
 
   def cancel!(args = {})
-    to_stripe.delete(args)
+    if to_stripe.respond_to?(:cancel)
+      to_stripe.cancel(args)
+    elsif to_stripe.respond_to?(:delete)
+      to_stripe.delete(args)
+    else
+      raise "Couldn't figure out how to cancel the Stripe subscription"
+    end
+
     reload_from_stripe!
     self
   end
 
+  def create_stripe_mock!
+    cancel_at_period_end = cancel_at_period_end?
+
+    mock_subscription = Stripe::Subscription.create(
+      customer: stripe_customer.stripe_id,
+      plan: stripe_plan.stripe_id
+    )
+    assign_from_stripe(mock_subscription)
+    save!
+    cancel!(at_period_end: true) if cancel_at_period_end
+  end
+
 private
 
   def assign_default_tax_rates(object)
@@ -77,6 +95,7 @@ private
     found_ids = []
 
     object.default_tax_rates.each do |default_tax_rate|
+      default_tax_rate = Stripe::TaxRate.retrieve(default_tax_rate) if default_tax_rate.is_a?(String)
       tax_rate = StripeModelCallbacks::TaxRate::UpdatedService.execute!(object: default_tax_rate)
 
       if new_record?

data/lib/stripe_model_callbacks/models/stripe_subscription_item.rb

@@ -15,18 +15,28 @@ class StripeSubscriptionItem < StripeModelCallbacks::ApplicationRecord
     self.stripe_plan_id = object.plan.id if object.try(:plan).respond_to?(:id)
 
     assign_price_from_stripe(object)
+    assign_deleted_from_stripe(object)
 
     StripeModelCallbacks::AttributesAssignerService.execute!(
       model: self, stripe_model: object,
-      attributes: %w[id created metadata quantity]
+      attributes: %w[id created deleted metadata quantity]
     )
   end
 
+  def create_stripe_mock!
+    mock_item = Stripe::SubscriptionItem.create(
+      subscription: stripe_subscription.stripe_id,
+      plan: stripe_plan.stripe_id,
+      quantity:
+    )
+    assign_from_stripe(mock_item)
+  end
+
   def update_quantity_on_stripe!(new_quantity)
     sub_object = stripe_subscription.to_stripe
 
     new_items = [{
-      id: id,
+      id:,
       plan: stripe_plan_id,
       quantity: new_quantity
     }]
@@ -51,4 +61,20 @@ private
     # Set stripe ID on the subscription item
     self.stripe_price_id = object.price.id
   end
+
+  def assign_deleted_from_stripe(object)
+    return if stripe_attribute_missing?(object, "deleted")
+
+    self.deleted = object.deleted == true
+  end
+
+  def stripe_attribute_missing?(object, attribute)
+    return false unless object.respond_to?(:to_hash)
+
+    values = object.instance_variable_get(:@values)
+    return !values.key?(attribute.to_sym) && !values.key?(attribute.to_s) if values.is_a?(Hash)
+
+    stripe_values = object.to_hash
+    !stripe_values.key?(attribute.to_sym) && !stripe_values.key?(attribute.to_s)
+  end
 end

data/lib/stripe_model_callbacks/models/stripe_subscription_schedule.rb

@@ -39,7 +39,7 @@ class StripeSubscriptionSchedule < StripeModelCallbacks::ApplicationRecord
     assign_subscription_schedule_phases(object)
   end
 
-  def cancel_on_stripe
+  def cancel_on_stripe # rubocop:disable Naming/PredicateMethod
     to_stripe.cancel
     update!(canceled_at: Time.zone.now) if respond_to?(:canceled_at)
     reload_from_stripe!
@@ -76,7 +76,7 @@ private
   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
+      __send__(:"#{timestamp}=", Time.zone.at(object_timestamp)) if object_timestamp
     end
   end
 

data/lib/stripe_model_callbacks/models/stripe_subscription_schedule_phase.rb

@@ -49,7 +49,7 @@ private
   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
+      __send__(:"#{timestamp}=", Time.zone.at(object_timestamp)) if object_timestamp
     end
   end
 

data/lib/stripe_model_callbacks/models/stripe_tax_rate.rb

@@ -5,6 +5,12 @@ class StripeTaxRate < StripeModelCallbacks::ApplicationRecord
     Stripe::TaxRate
   end
 
+  def create_stripe_mock!
+    mock_tax_rate = Stripe::TaxRate.create(description:, display_name:, id: stripe_id, percentage:)
+    assign_from_stripe(mock_tax_rate)
+    save!
+  end
+
   def assign_from_stripe(object)
     check_object_is_stripe_class(object)
     assign_attributes(

data/lib/stripe_model_callbacks/version.rb

@@ -1,3 +1,3 @@
 module StripeModelCallbacks
-  VERSION = "0.1.7".freeze
+  VERSION = "0.1.9".freeze
 end

data/lib/stripe_model_callbacks.rb

@@ -1,6 +1,6 @@
+require "active_record_auditable"
 require "auto_autoloader"
 require "money-rails"
-require "public_activity"
 require "service_pattern"
 require "stripe"
 require "stripe_event"