mirror of
https://github.com/discourse/discourse.git
synced 2026-08-14 13:58:53 +08:00
Previously, `checkout.session.completed` selected the Discourse user from Stripe's checkout email, which could record a subscription against the wrong account. This change sends a signed user reference through Stripe Pricing Tables and uses it as the trusted webhook binding while preserving the existing checkout email validation. --------- Co-authored-by: discourse-patch-triage[bot] <272280883+discourse-patch-triage[bot]@users.noreply.github.com>
176 lines
5.7 KiB
Ruby
Vendored
176 lines
5.7 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
module DiscourseSubscriptions
|
|
class HooksController < ::ApplicationController
|
|
include DiscourseSubscriptions::Group
|
|
include DiscourseSubscriptions::Stripe
|
|
|
|
requires_plugin PLUGIN_NAME
|
|
|
|
layout false
|
|
|
|
skip_before_action :check_xhr
|
|
skip_before_action :redirect_to_login_if_required
|
|
skip_before_action :verify_authenticity_token, only: [:create]
|
|
|
|
def create
|
|
begin
|
|
payload = request.body.read
|
|
sig_header = request.env["HTTP_STRIPE_SIGNATURE"]
|
|
webhook_secret = SiteSetting.discourse_subscriptions_webhook_secret
|
|
|
|
return head :forbidden if webhook_secret.blank?
|
|
|
|
event = ::Stripe::Webhook.construct_event(payload, sig_header, webhook_secret)
|
|
rescue JSON::ParserError => e
|
|
return render_json_error e.message
|
|
rescue ::Stripe::SignatureVerificationError => e
|
|
return render_json_error e.message
|
|
end
|
|
|
|
case event[:type]
|
|
when "checkout.session.completed", "checkout.session.async_payment_succeeded"
|
|
checkout_session = event[:data][:object]
|
|
|
|
return head :ok if checkout_session[:status] != "complete"
|
|
return head :ok if checkout_session[:payment_status] != "paid"
|
|
|
|
if SiteSetting.discourse_subscriptions_enable_verbose_logging
|
|
Rails.logger.warn("#{event[:type]} data: #{checkout_session}")
|
|
end
|
|
|
|
user = trusted_checkout_session_user(checkout_session)
|
|
|
|
return render_json_error "user not found" if !user
|
|
|
|
email = checkout_session[:customer_email]
|
|
return render_json_error "email not found" if !email
|
|
|
|
if !checkout_session_email_belongs_to_user?(email, user)
|
|
return render_json_error "user not found"
|
|
end
|
|
|
|
if checkout_session[:customer].nil?
|
|
customer = ::Stripe::Customer.create({ email: user.email }, stripe_request_opts)
|
|
customer_id = customer[:id]
|
|
else
|
|
customer_id = checkout_session[:customer]
|
|
end
|
|
|
|
if SiteSetting.discourse_subscriptions_enable_verbose_logging
|
|
Rails.logger.warn("Processing checkout session for user: #{user.email} (id: #{user.id})")
|
|
end
|
|
|
|
discourse_customer = Customer.create(user_id: user.id, customer_id: customer_id)
|
|
|
|
subscription = checkout_session[:subscription]
|
|
|
|
if subscription.present?
|
|
Subscription.create(customer_id: discourse_customer.id, external_id: subscription)
|
|
end
|
|
|
|
line_items =
|
|
::Stripe::Checkout::Session.list_line_items(
|
|
checkout_session[:id],
|
|
{ limit: 1 },
|
|
stripe_request_opts,
|
|
)
|
|
item = line_items[:data].first
|
|
|
|
group = plan_group(item[:price])
|
|
group.add(user) unless group.nil?
|
|
|
|
if SiteSetting.discourse_subscriptions_enable_verbose_logging
|
|
Rails.logger.warn("Line item with group name meta data: #{item[:price]}")
|
|
if group.nil?
|
|
Rails.logger.warn("Group not found or not listed in metadata!")
|
|
else
|
|
Rails.logger.warn("Group: #{group.name}")
|
|
end
|
|
end
|
|
|
|
discourse_customer.product_id = item[:price][:product]
|
|
discourse_customer.save!
|
|
|
|
if !subscription.nil?
|
|
::Stripe::Subscription.update(
|
|
subscription,
|
|
{ metadata: { user_id: user.id, username: user.username } },
|
|
stripe_request_opts,
|
|
)
|
|
end
|
|
when "customer.subscription.created"
|
|
when "customer.subscription.updated"
|
|
subscription = event[:data][:object]
|
|
status = subscription[:status]
|
|
return head :ok if !%w[complete active].include?(status)
|
|
|
|
customer = find_active_customer(subscription[:customer], subscription[:plan][:product])
|
|
|
|
return render_json_error "customer not found" if !customer
|
|
|
|
update_status(customer.id, subscription[:id], status)
|
|
|
|
user = ::User.find_by(id: customer.user_id)
|
|
return render_json_error "user not found" if !user
|
|
|
|
if group = plan_group(subscription[:plan])
|
|
group.add(user)
|
|
end
|
|
when "customer.subscription.deleted"
|
|
subscription = event[:data][:object]
|
|
|
|
customer = find_active_customer(subscription[:customer], subscription[:plan][:product])
|
|
|
|
return render_json_error "customer not found" if !customer
|
|
|
|
update_status(customer.id, subscription[:id], subscription[:status])
|
|
|
|
user = ::User.find(customer.user_id)
|
|
return render_json_error "user not found" if !user
|
|
|
|
if group = plan_group(subscription[:plan])
|
|
group.remove(user)
|
|
end
|
|
end
|
|
|
|
head :ok
|
|
end
|
|
|
|
private
|
|
|
|
def checkout_session_email_belongs_to_user?(email, user)
|
|
::UserEmail.exists?(user_id: user.id, email: ::Email.downcase(email))
|
|
end
|
|
|
|
def trusted_checkout_session_user(checkout_session)
|
|
client_reference_id =
|
|
checkout_session[:client_reference_id] || checkout_session["client_reference_id"]
|
|
|
|
return if client_reference_id.blank?
|
|
|
|
::User.find_signed(
|
|
client_reference_id,
|
|
purpose: DiscourseSubscriptions::CHECKOUT_SESSION_USER_REFERENCE_PURPOSE,
|
|
)
|
|
end
|
|
|
|
def update_status(customer_id, subscription_id, status)
|
|
discourse_subscription =
|
|
Subscription.find_by(customer_id: customer_id, external_id: subscription_id)
|
|
discourse_subscription.update(status: status) if discourse_subscription
|
|
end
|
|
|
|
def find_active_customer(customer_id, product_id)
|
|
Customer
|
|
.joins(:subscriptions)
|
|
.where(customer_id: customer_id, product_id: product_id)
|
|
.where(
|
|
Subscription.arel_table[:status].eq(nil).or(
|
|
Subscription.arel_table[:status].not_eq("canceled"),
|
|
),
|
|
)
|
|
.first
|
|
end
|
|
end
|
|
end
|