From f045d06a9c26eb890136bbafa01b6973d076d7ac Mon Sep 17 00:00:00 2001 From: cyprain-okeke <108260115+cyprain-okeke@users.noreply.github.com> Date: Wed, 26 Jun 2024 16:34:16 +0100 Subject: [PATCH] [AC-2361] Refactor StripeController (#4136) * Changes ensures provider_id is handled and stored for Braintree. Signed-off-by: Cy Okeke * refactoring of the stripeController class Signed-off-by: Cy Okeke * Move the constant variables to utility class Signed-off-by: Cy Okeke * Adding comments to the methods Signed-off-by: Cy Okeke * Add more comments to describe the method Signed-off-by: Cy Okeke * Add the providerId changes Signed-off-by: Cy Okeke * Add the missing providerId Signed-off-by: Cy Okeke * Fix the IsSponsoredSubscription bug Signed-off-by: Cy Okeke --------- Signed-off-by: Cy Okeke --- src/Billing/Controllers/StripeController.cs | 1193 +---------------- src/Billing/Services/IStripeEventProcessor.cs | 12 + .../Services/IStripeEventUtilityService.cs | 67 + src/Billing/Services/IStripeWebhookHandler.cs | 67 + .../Implementations/ChargeRefundedHandler.cs | 98 ++ .../Implementations/ChargeSucceededHandler.cs | 67 + .../Implementations/CustomerUpdatedHandler.cs | 60 + .../Implementations/InvoiceCreatedHandler.cs | 35 + .../InvoiceFinalizedHandler.cs | 23 + .../Implementations/PaymentFailedHandler.cs | 52 + .../PaymentMethodAttachedHandler.cs | 96 ++ .../PaymentSucceededHandler.cs | 171 +++ .../Implementations/StripeEventProcessor.cs | 89 ++ .../StripeEventUtilityService.cs | 401 ++++++ .../SubscriptionDeletedHandler.cs | 49 + .../SubscriptionUpdatedHandler.cs | 176 +++ .../Implementations/UpcomingInvoiceHandler.cs | 215 +++ src/Billing/Startup.cs | 15 + 18 files changed, 1705 insertions(+), 1181 deletions(-) create mode 100644 src/Billing/Services/IStripeEventProcessor.cs create mode 100644 src/Billing/Services/IStripeEventUtilityService.cs create mode 100644 src/Billing/Services/IStripeWebhookHandler.cs create mode 100644 src/Billing/Services/Implementations/ChargeRefundedHandler.cs create mode 100644 src/Billing/Services/Implementations/ChargeSucceededHandler.cs create mode 100644 src/Billing/Services/Implementations/CustomerUpdatedHandler.cs create mode 100644 src/Billing/Services/Implementations/InvoiceCreatedHandler.cs create mode 100644 src/Billing/Services/Implementations/InvoiceFinalizedHandler.cs create mode 100644 src/Billing/Services/Implementations/PaymentFailedHandler.cs create mode 100644 src/Billing/Services/Implementations/PaymentMethodAttachedHandler.cs create mode 100644 src/Billing/Services/Implementations/PaymentSucceededHandler.cs create mode 100644 src/Billing/Services/Implementations/StripeEventProcessor.cs create mode 100644 src/Billing/Services/Implementations/StripeEventUtilityService.cs create mode 100644 src/Billing/Services/Implementations/SubscriptionDeletedHandler.cs create mode 100644 src/Billing/Services/Implementations/SubscriptionUpdatedHandler.cs create mode 100644 src/Billing/Services/Implementations/UpcomingInvoiceHandler.cs diff --git a/src/Billing/Controllers/StripeController.cs b/src/Billing/Controllers/StripeController.cs index b03f6633df..9bca9f024b 100644 --- a/src/Billing/Controllers/StripeController.cs +++ b/src/Billing/Controllers/StripeController.cs @@ -1,115 +1,35 @@ -using Bit.Billing.Constants; -using Bit.Billing.Models; +using Bit.Billing.Models; using Bit.Billing.Services; -using Bit.Core; -using Bit.Core.AdminConsole.Entities; -using Bit.Core.AdminConsole.Repositories; -using Bit.Core.Billing.Constants; -using Bit.Core.Billing.Enums; -using Bit.Core.Context; -using Bit.Core.Enums; -using Bit.Core.OrganizationFeatures.OrganizationSponsorships.FamiliesForEnterprise.Interfaces; -using Bit.Core.Repositories; -using Bit.Core.Services; -using Bit.Core.Settings; -using Bit.Core.Tools.Enums; -using Bit.Core.Tools.Models.Business; -using Bit.Core.Tools.Services; using Bit.Core.Utilities; -using Braintree; -using Braintree.Exceptions; using Microsoft.AspNetCore.Mvc; -using Microsoft.Data.SqlClient; using Microsoft.Extensions.Options; using Stripe; -using Customer = Stripe.Customer; using Event = Stripe.Event; using JsonSerializer = System.Text.Json.JsonSerializer; -using Subscription = Stripe.Subscription; -using TaxRate = Bit.Core.Entities.TaxRate; -using Transaction = Bit.Core.Entities.Transaction; -using TransactionType = Bit.Core.Enums.TransactionType; namespace Bit.Billing.Controllers; [Route("stripe")] public class StripeController : Controller { - private const string PremiumPlanId = "premium-annually"; - private const string PremiumPlanIdAppStore = "premium-annually-app"; - private readonly BillingSettings _billingSettings; private readonly IWebHostEnvironment _hostingEnvironment; - private readonly IOrganizationService _organizationService; - private readonly IValidateSponsorshipCommand _validateSponsorshipCommand; - private readonly IOrganizationSponsorshipRenewCommand _organizationSponsorshipRenewCommand; - private readonly IOrganizationRepository _organizationRepository; - private readonly ITransactionRepository _transactionRepository; - private readonly IUserService _userService; - private readonly IMailService _mailService; private readonly ILogger _logger; - private readonly BraintreeGateway _btGateway; - private readonly IReferenceEventService _referenceEventService; - private readonly ITaxRateRepository _taxRateRepository; - private readonly IUserRepository _userRepository; - private readonly ICurrentContext _currentContext; - private readonly GlobalSettings _globalSettings; private readonly IStripeEventService _stripeEventService; - private readonly IStripeFacade _stripeFacade; - private readonly IFeatureService _featureService; - private readonly IProviderRepository _providerRepository; - private readonly IProviderEventService _providerEventService; + private readonly IStripeEventProcessor _stripeEventProcessor; public StripeController( - GlobalSettings globalSettings, IOptions billingSettings, IWebHostEnvironment hostingEnvironment, - IOrganizationService organizationService, - IValidateSponsorshipCommand validateSponsorshipCommand, - IOrganizationSponsorshipRenewCommand organizationSponsorshipRenewCommand, - IOrganizationRepository organizationRepository, - ITransactionRepository transactionRepository, - IUserService userService, - IMailService mailService, - IReferenceEventService referenceEventService, ILogger logger, - ITaxRateRepository taxRateRepository, - IUserRepository userRepository, - ICurrentContext currentContext, IStripeEventService stripeEventService, - IStripeFacade stripeFacade, - IFeatureService featureService, - IProviderRepository providerRepository, - IProviderEventService providerEventService) + IStripeEventProcessor stripeEventProcessor) { _billingSettings = billingSettings?.Value; _hostingEnvironment = hostingEnvironment; - _organizationService = organizationService; - _validateSponsorshipCommand = validateSponsorshipCommand; - _organizationSponsorshipRenewCommand = organizationSponsorshipRenewCommand; - _organizationRepository = organizationRepository; - _transactionRepository = transactionRepository; - _userService = userService; - _mailService = mailService; - _referenceEventService = referenceEventService; - _taxRateRepository = taxRateRepository; - _userRepository = userRepository; _logger = logger; - _btGateway = new BraintreeGateway - { - Environment = globalSettings.Braintree.Production ? - Braintree.Environment.PRODUCTION : Braintree.Environment.SANDBOX, - MerchantId = globalSettings.Braintree.MerchantId, - PublicKey = globalSettings.Braintree.PublicKey, - PrivateKey = globalSettings.Braintree.PrivateKey - }; - _currentContext = currentContext; - _globalSettings = globalSettings; _stripeEventService = stripeEventService; - _stripeFacade = stripeFacade; - _featureService = featureService; - _providerRepository = providerRepository; - _providerEventService = providerEventService; + _stripeEventProcessor = stripeEventProcessor; } [HttpPost("webhook")] @@ -155,1107 +75,18 @@ public class StripeController : Controller return new OkResult(); } - switch (parsedEvent.Type) - { - case HandledStripeWebhook.SubscriptionDeleted: - { - await HandleCustomerSubscriptionDeletedEventAsync(parsedEvent); - return Ok(); - } - case HandledStripeWebhook.SubscriptionUpdated: - { - await HandleCustomerSubscriptionUpdatedEventAsync(parsedEvent); - return Ok(); - } - case HandledStripeWebhook.UpcomingInvoice: - { - await HandleUpcomingInvoiceEventAsync(parsedEvent); - return Ok(); - } - case HandledStripeWebhook.ChargeSucceeded: - { - await HandleChargeSucceededEventAsync(parsedEvent); - return Ok(); - } - case HandledStripeWebhook.ChargeRefunded: - { - await HandleChargeRefundedEventAsync(parsedEvent); - return Ok(); - } - case HandledStripeWebhook.PaymentSucceeded: - { - await HandlePaymentSucceededEventAsync(parsedEvent); - return Ok(); - } - case HandledStripeWebhook.PaymentFailed: - { - await HandlePaymentFailedEventAsync(parsedEvent); - return Ok(); - } - case HandledStripeWebhook.InvoiceCreated: - { - await HandleInvoiceCreatedEventAsync(parsedEvent); - return Ok(); - } - case HandledStripeWebhook.PaymentMethodAttached: - { - await HandlePaymentMethodAttachedAsync(parsedEvent); - return Ok(); - } - case HandledStripeWebhook.CustomerUpdated: - { - await HandleCustomerUpdatedEventAsync(parsedEvent); - return Ok(); - } - case HandledStripeWebhook.InvoiceFinalized: - { - await HandleInvoiceFinalizedEventAsync(parsedEvent); - return Ok(); - } - default: - { - _logger.LogWarning("Unsupported event received. {EventType}", parsedEvent.Type); - return Ok(); - } - } + await _stripeEventProcessor.ProcessEventAsync(parsedEvent); + return Ok(); } /// - /// Handles the event type from Stripe. + /// Selects the appropriate Stripe webhook secret based on the API version specified in the webhook body. /// - /// - private async Task HandleCustomerSubscriptionUpdatedEventAsync(Event parsedEvent) - { - var subscription = await _stripeEventService.GetSubscription(parsedEvent, true, ["customer", "discounts"]); - var (organizationId, userId, providerId) = GetIdsFromMetadata(subscription.Metadata); - - switch (subscription.Status) - { - case StripeSubscriptionStatus.Unpaid or StripeSubscriptionStatus.IncompleteExpired - when organizationId.HasValue: - { - await _organizationService.DisableAsync(organizationId.Value, subscription.CurrentPeriodEnd); - break; - } - case StripeSubscriptionStatus.Unpaid or StripeSubscriptionStatus.IncompleteExpired: - { - if (!userId.HasValue) - { - break; - } - - if (subscription.Status is StripeSubscriptionStatus.Unpaid && - subscription.Items.Any(i => i.Price.Id is PremiumPlanId or PremiumPlanIdAppStore)) - { - await CancelSubscription(subscription.Id); - await VoidOpenInvoices(subscription.Id); - } - - await _userService.DisablePremiumAsync(userId.Value, subscription.CurrentPeriodEnd); - - break; - } - case StripeSubscriptionStatus.Active when organizationId.HasValue: - { - await _organizationService.EnableAsync(organizationId.Value); - break; - } - case StripeSubscriptionStatus.Active: - { - if (userId.HasValue) - { - await _userService.EnablePremiumAsync(userId.Value, subscription.CurrentPeriodEnd); - } - - break; - } - } - - if (organizationId.HasValue) - { - await _organizationService.UpdateExpirationDateAsync(organizationId.Value, subscription.CurrentPeriodEnd); - if (IsSponsoredSubscription(subscription)) - { - await _organizationSponsorshipRenewCommand.UpdateExpirationDateAsync(organizationId.Value, subscription.CurrentPeriodEnd); - } - - await RemovePasswordManagerCouponIfRemovingSecretsManagerTrialAsync(parsedEvent, subscription); - } - else if (userId.HasValue) - { - await _userService.UpdatePremiumExpirationAsync(userId.Value, subscription.CurrentPeriodEnd); - } - } - - /// - /// Removes the Password Manager coupon if the organization is removing the Secrets Manager trial. - /// Only applies to organizations that have a subscription from the Secrets Manager trial. - /// - /// - /// - private async Task RemovePasswordManagerCouponIfRemovingSecretsManagerTrialAsync(Event parsedEvent, - Subscription subscription) - { - if (parsedEvent.Data.PreviousAttributes?.items is null) - { - return; - } - - var previousSubscription = parsedEvent.Data - .PreviousAttributes - .ToObject() as Subscription; - - // This being false doesn't necessarily mean that the organization doesn't subscribe to Secrets Manager. - // If there are changes to any subscription item, Stripe sends every item in the subscription, both - // changed and unchanged. - var previousSubscriptionHasSecretsManager = previousSubscription?.Items is not null && - previousSubscription.Items.Any(previousItem => - StaticStore.Plans.Any(p => - p.SecretsManager is not null && - p.SecretsManager.StripeSeatPlanId == - previousItem.Plan.Id)); - - var currentSubscriptionHasSecretsManager = subscription.Items.Any(i => - StaticStore.Plans.Any(p => - p.SecretsManager is not null && - p.SecretsManager.StripeSeatPlanId == i.Plan.Id)); - - if (!previousSubscriptionHasSecretsManager || currentSubscriptionHasSecretsManager) - { - return; - } - - var customerHasSecretsManagerTrial = subscription.Customer - ?.Discount - ?.Coupon - ?.Id == "sm-standalone"; - - var subscriptionHasSecretsManagerTrial = subscription.Discount - ?.Coupon - ?.Id == "sm-standalone"; - - if (customerHasSecretsManagerTrial) - { - await _stripeFacade.DeleteCustomerDiscount(subscription.CustomerId); - } - - if (subscriptionHasSecretsManagerTrial) - { - await _stripeFacade.DeleteSubscriptionDiscount(subscription.Id); - } - } - - /// - /// Handles the event type from Stripe. - /// - /// - private async Task HandleCustomerSubscriptionDeletedEventAsync(Event parsedEvent) - { - var subscription = await _stripeEventService.GetSubscription(parsedEvent, true); - var (organizationId, userId, providerId) = GetIdsFromMetadata(subscription.Metadata); - var subCanceled = subscription.Status == StripeSubscriptionStatus.Canceled; - - if (!subCanceled) - { - return; - } - - if (organizationId.HasValue) - { - await _organizationService.DisableAsync(organizationId.Value, subscription.CurrentPeriodEnd); - } - else if (userId.HasValue) - { - await _userService.DisablePremiumAsync(userId.Value, subscription.CurrentPeriodEnd); - } - } - - /// - /// Handles the event type from Stripe. - /// - /// - private async Task HandleCustomerUpdatedEventAsync(Event parsedEvent) - { - var customer = await _stripeEventService.GetCustomer(parsedEvent, true, ["subscriptions"]); - if (customer.Subscriptions == null || !customer.Subscriptions.Any()) - { - return; - } - - var subscription = customer.Subscriptions.First(); - - var (organizationId, _, providerId) = GetIdsFromMetadata(subscription.Metadata); - - if (!organizationId.HasValue) - { - return; - } - - var organization = await _organizationRepository.GetByIdAsync(organizationId.Value); - organization.BillingEmail = customer.Email; - await _organizationRepository.ReplaceAsync(organization); - - await _referenceEventService.RaiseEventAsync( - new ReferenceEvent(ReferenceEventType.OrganizationEditedInStripe, organization, _currentContext)); - } - - /// - /// Handles the event type from Stripe. - /// - /// - private async Task HandleInvoiceCreatedEventAsync(Event parsedEvent) - { - var invoice = await _stripeEventService.GetInvoice(parsedEvent, true); - - if (ShouldAttemptToPayInvoice(invoice)) - { - await AttemptToPayInvoiceAsync(invoice); - } - - await _providerEventService.TryRecordInvoiceLineItems(parsedEvent); - } - - private async Task HandleInvoiceFinalizedEventAsync(Event parsedEvent) - { - await _providerEventService.TryRecordInvoiceLineItems(parsedEvent); - } - - /// - /// Handles the event type from Stripe. - /// - /// - private async Task HandlePaymentSucceededEventAsync(Event parsedEvent) - { - var invoice = await _stripeEventService.GetInvoice(parsedEvent, true); - if (!invoice.Paid || invoice.BillingReason != "subscription_create") - { - return; - } - - var subscription = await _stripeFacade.GetSubscription(invoice.SubscriptionId); - if (subscription?.Status != StripeSubscriptionStatus.Active) - { - return; - } - - if (DateTime.UtcNow - invoice.Created < TimeSpan.FromMinutes(1)) - { - await Task.Delay(5000); - } - - var (organizationId, userId, providerId) = GetIdsFromMetadata(subscription.Metadata); - - if (providerId.HasValue) - { - var provider = await _providerRepository.GetByIdAsync(providerId.Value); - - if (provider == null) - { - _logger.LogError( - "Received invoice.payment_succeeded webhook ({EventID}) for Provider ({ProviderID}) that does not exist", - parsedEvent.Id, - providerId.Value); - - return; - } - - var teamsMonthly = StaticStore.GetPlan(PlanType.TeamsMonthly); - - var enterpriseMonthly = StaticStore.GetPlan(PlanType.EnterpriseMonthly); - - var teamsMonthlyLineItem = - subscription.Items.Data.FirstOrDefault(item => - item.Plan.Id == teamsMonthly.PasswordManager.StripeSeatPlanId); - - var enterpriseMonthlyLineItem = - subscription.Items.Data.FirstOrDefault(item => - item.Plan.Id == enterpriseMonthly.PasswordManager.StripeSeatPlanId); - - if (teamsMonthlyLineItem == null || enterpriseMonthlyLineItem == null) - { - _logger.LogError("invoice.payment_succeeded webhook ({EventID}) for Provider ({ProviderID}) indicates missing subscription line items", - parsedEvent.Id, - provider.Id); - - return; - } - - await _referenceEventService.RaiseEventAsync(new ReferenceEvent - { - Type = ReferenceEventType.Rebilled, - Source = ReferenceEventSource.Provider, - Id = provider.Id, - PlanType = PlanType.TeamsMonthly, - Seats = (int)teamsMonthlyLineItem.Quantity - }); - - await _referenceEventService.RaiseEventAsync(new ReferenceEvent - { - Type = ReferenceEventType.Rebilled, - Source = ReferenceEventSource.Provider, - Id = provider.Id, - PlanType = PlanType.EnterpriseMonthly, - Seats = (int)enterpriseMonthlyLineItem.Quantity - }); - } - else if (organizationId.HasValue) - { - if (!subscription.Items.Any(i => - StaticStore.Plans.Any(p => p.PasswordManager.StripePlanId == i.Plan.Id))) - { - return; - } - - await _organizationService.EnableAsync(organizationId.Value, subscription.CurrentPeriodEnd); - var organization = await _organizationRepository.GetByIdAsync(organizationId.Value); - - await _referenceEventService.RaiseEventAsync( - new ReferenceEvent(ReferenceEventType.Rebilled, organization, _currentContext) - { - PlanName = organization?.Plan, - PlanType = organization?.PlanType, - Seats = organization?.Seats, - Storage = organization?.MaxStorageGb, - }); - } - else if (userId.HasValue) - { - if (subscription.Items.All(i => i.Plan.Id != PremiumPlanId)) - { - return; - } - - await _userService.EnablePremiumAsync(userId.Value, subscription.CurrentPeriodEnd); - - var user = await _userRepository.GetByIdAsync(userId.Value); - await _referenceEventService.RaiseEventAsync( - new ReferenceEvent(ReferenceEventType.Rebilled, user, _currentContext) - { - PlanName = PremiumPlanId, - Storage = user?.MaxStorageGb, - }); - } - } - - /// - /// Handles the event type from Stripe. - /// - /// - private async Task HandleChargeRefundedEventAsync(Event parsedEvent) - { - var charge = await _stripeEventService.GetCharge(parsedEvent, true, ["refunds"]); - var parentTransaction = await _transactionRepository.GetByGatewayIdAsync(GatewayType.Stripe, charge.Id); - if (parentTransaction == null) - { - // Attempt to create a transaction for the charge if it doesn't exist - var (organizationId, userId, providerId) = await GetEntityIdsFromChargeAsync(charge); - var tx = FromChargeToTransaction(charge, organizationId, userId, providerId); - try - { - parentTransaction = await _transactionRepository.CreateAsync(tx); - } - catch (SqlException e) when (e.Number == 547) // FK constraint violation - { - _logger.LogWarning( - "Charge refund could not create transaction as entity may have been deleted. {ChargeId}", - charge.Id); - return; - } - } - - var amountRefunded = charge.AmountRefunded / 100M; - - if (parentTransaction.Refunded.GetValueOrDefault() || - parentTransaction.RefundedAmount.GetValueOrDefault() >= amountRefunded) - { - _logger.LogWarning( - "Charge refund amount doesn't match parent transaction's amount or parent has already been refunded. {ChargeId}", - charge.Id); - return; - } - - parentTransaction.RefundedAmount = amountRefunded; - if (charge.Refunded) - { - parentTransaction.Refunded = true; - } - - await _transactionRepository.ReplaceAsync(parentTransaction); - - foreach (var refund in charge.Refunds) - { - var refundTransaction = await _transactionRepository.GetByGatewayIdAsync( - GatewayType.Stripe, refund.Id); - if (refundTransaction != null) - { - continue; - } - - await _transactionRepository.CreateAsync(new Transaction - { - Amount = refund.Amount / 100M, - CreationDate = refund.Created, - OrganizationId = parentTransaction.OrganizationId, - UserId = parentTransaction.UserId, - ProviderId = parentTransaction.ProviderId, - Type = TransactionType.Refund, - Gateway = GatewayType.Stripe, - GatewayId = refund.Id, - PaymentMethodType = parentTransaction.PaymentMethodType, - Details = parentTransaction.Details - }); - } - } - - /// - /// Handles the event type from Stripe. - /// - /// - private async Task HandleChargeSucceededEventAsync(Event parsedEvent) - { - var charge = await _stripeEventService.GetCharge(parsedEvent); - var existingTransaction = await _transactionRepository.GetByGatewayIdAsync(GatewayType.Stripe, charge.Id); - if (existingTransaction is not null) - { - _logger.LogInformation("Charge success already processed. {ChargeId}", charge.Id); - return; - } - - var (organizationId, userId, providerId) = await GetEntityIdsFromChargeAsync(charge); - if (!organizationId.HasValue && !userId.HasValue && !providerId.HasValue) - { - _logger.LogWarning("Charge success has no subscriber ids. {ChargeId}", charge.Id); - return; - } - - var transaction = FromChargeToTransaction(charge, organizationId, userId, providerId); - if (!transaction.PaymentMethodType.HasValue) - { - _logger.LogWarning("Charge success from unsupported source/method. {ChargeId}", charge.Id); - return; - } - - try - { - await _transactionRepository.CreateAsync(transaction); - } - catch (SqlException e) when (e.Number == 547) - { - _logger.LogWarning( - "Charge success could not create transaction as entity may have been deleted. {ChargeId}", - charge.Id); - } - } - - /// - /// Handles the event type from Stripe. - /// - /// - /// - private async Task HandleUpcomingInvoiceEventAsync(Event parsedEvent) - { - var invoice = await _stripeEventService.GetInvoice(parsedEvent); - if (string.IsNullOrEmpty(invoice.SubscriptionId)) - { - _logger.LogWarning("Received 'invoice.upcoming' Event with ID '{eventId}' that did not include a Subscription ID", parsedEvent.Id); - return; - } - - var subscription = await _stripeFacade.GetSubscription(invoice.SubscriptionId); - - if (subscription == null) - { - throw new Exception( - $"Received null Subscription from Stripe for ID '{invoice.SubscriptionId}' while processing Event with ID '{parsedEvent.Id}'"); - } - - var pm5766AutomaticTaxIsEnabled = _featureService.IsEnabled(FeatureFlagKeys.PM5766AutomaticTax); - if (pm5766AutomaticTaxIsEnabled) - { - var customerGetOptions = new CustomerGetOptions(); - customerGetOptions.AddExpand("tax"); - var customer = await _stripeFacade.GetCustomer(subscription.CustomerId, customerGetOptions); - if (!subscription.AutomaticTax.Enabled && - customer.Tax?.AutomaticTax == StripeConstants.AutomaticTaxStatus.Supported) - { - subscription = await _stripeFacade.UpdateSubscription(subscription.Id, - new SubscriptionUpdateOptions - { - DefaultTaxRates = [], - AutomaticTax = new SubscriptionAutomaticTaxOptions { Enabled = true } - }); - } - } - - var updatedSubscription = pm5766AutomaticTaxIsEnabled - ? subscription - : await VerifyCorrectTaxRateForCharge(invoice, subscription); - - var (organizationId, userId, providerId) = GetIdsFromMetadata(updatedSubscription.Metadata); - - var invoiceLineItemDescriptions = invoice.Lines.Select(i => i.Description).ToList(); - - if (organizationId.HasValue) - { - if (IsSponsoredSubscription(updatedSubscription)) - { - await _validateSponsorshipCommand.ValidateSponsorshipAsync(organizationId.Value); - } - - var organization = await _organizationRepository.GetByIdAsync(organizationId.Value); - - if (organization == null || !OrgPlanForInvoiceNotifications(organization)) - { - return; - } - - await SendEmails(new List { organization.BillingEmail }); - - /* - * TODO: https://bitwarden.atlassian.net/browse/PM-4862 - * Disabling this as part of a hot fix. It needs to check whether the organization - * belongs to a Reseller provider and only send an email to the organization owners if it does. - * It also requires a new email template as the current one contains too much billing information. - */ - - // var ownerEmails = await _organizationRepository.GetOwnerEmailAddressesById(organization.Id); - - // await SendEmails(ownerEmails); - } - else if (userId.HasValue) - { - var user = await _userService.GetUserByIdAsync(userId.Value); - - if (user?.Premium == true) - { - await SendEmails(new List { user.Email }); - } - } - else if (providerId.HasValue) - { - var provider = await _providerRepository.GetByIdAsync(providerId.Value); - - if (provider == null) - { - _logger.LogError( - "Received invoice.Upcoming webhook ({EventID}) for Provider ({ProviderID}) that does not exist", - parsedEvent.Id, - providerId.Value); - - return; - } - - await SendEmails(new List { provider.BillingEmail }); - - } - - return; - - /* - * Sends emails to the given email addresses. - */ - async Task SendEmails(IEnumerable emails) - { - var validEmails = emails.Where(e => !string.IsNullOrEmpty(e)); - - if (invoice.NextPaymentAttempt.HasValue) - { - await _mailService.SendInvoiceUpcoming( - validEmails, - invoice.AmountDue / 100M, - invoice.NextPaymentAttempt.Value, - invoiceLineItemDescriptions, - true); - } - } - } - - /// - /// Gets the organization or user ID from the metadata of a Stripe Charge object. - /// - /// - /// - private async Task<(Guid?, Guid?, Guid?)> GetEntityIdsFromChargeAsync(Charge charge) - { - Guid? organizationId = null; - Guid? userId = null; - Guid? providerId = null; - - if (charge.InvoiceId != null) - { - var invoice = await _stripeFacade.GetInvoice(charge.InvoiceId); - if (invoice?.SubscriptionId != null) - { - var subscription = await _stripeFacade.GetSubscription(invoice.SubscriptionId); - (organizationId, userId, providerId) = GetIdsFromMetadata(subscription?.Metadata); - } - } - - if (organizationId.HasValue || userId.HasValue || providerId.HasValue) - { - return (organizationId, userId, providerId); - } - - var subscriptions = await _stripeFacade.ListSubscriptions(new SubscriptionListOptions - { - Customer = charge.CustomerId - }); - - foreach (var subscription in subscriptions) - { - if (subscription.Status is StripeSubscriptionStatus.Canceled or StripeSubscriptionStatus.IncompleteExpired) - { - continue; - } - - (organizationId, userId, providerId) = GetIdsFromMetadata(subscription.Metadata); - - if (organizationId.HasValue || userId.HasValue || providerId.HasValue) - { - return (organizationId, userId, providerId); - } - } - - return (null, null, null); - } - - /// - /// Converts a Stripe Charge object to a Bitwarden Transaction object. - /// - /// - /// - /// - /// /// - /// - private static Transaction FromChargeToTransaction(Charge charge, Guid? organizationId, Guid? userId, Guid? providerId) - { - var transaction = new Transaction - { - Amount = charge.Amount / 100M, - CreationDate = charge.Created, - OrganizationId = organizationId, - UserId = userId, - ProviderId = providerId, - Type = TransactionType.Charge, - Gateway = GatewayType.Stripe, - GatewayId = charge.Id - }; - - switch (charge.Source) - { - case Card card: - { - transaction.PaymentMethodType = PaymentMethodType.Card; - transaction.Details = $"{card.Brand}, *{card.Last4}"; - break; - } - case BankAccount bankAccount: - { - transaction.PaymentMethodType = PaymentMethodType.BankAccount; - transaction.Details = $"{bankAccount.BankName}, *{bankAccount.Last4}"; - break; - } - case Source { Card: not null } source: - { - transaction.PaymentMethodType = PaymentMethodType.Card; - transaction.Details = $"{source.Card.Brand}, *{source.Card.Last4}"; - break; - } - case Source { AchDebit: not null } source: - { - transaction.PaymentMethodType = PaymentMethodType.BankAccount; - transaction.Details = $"{source.AchDebit.BankName}, *{source.AchDebit.Last4}"; - break; - } - case Source source: - { - if (source.AchCreditTransfer == null) - { - break; - } - - var achCreditTransfer = source.AchCreditTransfer; - - transaction.PaymentMethodType = PaymentMethodType.BankAccount; - transaction.Details = $"ACH => {achCreditTransfer.BankName}, {achCreditTransfer.AccountNumber}"; - - break; - } - default: - { - if (charge.PaymentMethodDetails == null) - { - break; - } - - if (charge.PaymentMethodDetails.Card != null) - { - var card = charge.PaymentMethodDetails.Card; - transaction.PaymentMethodType = PaymentMethodType.Card; - transaction.Details = $"{card.Brand?.ToUpperInvariant()}, *{card.Last4}"; - } - else if (charge.PaymentMethodDetails.AchDebit != null) - { - var achDebit = charge.PaymentMethodDetails.AchDebit; - transaction.PaymentMethodType = PaymentMethodType.BankAccount; - transaction.Details = $"{achDebit.BankName}, *{achDebit.Last4}"; - } - else if (charge.PaymentMethodDetails.AchCreditTransfer != null) - { - var achCreditTransfer = charge.PaymentMethodDetails.AchCreditTransfer; - transaction.PaymentMethodType = PaymentMethodType.BankAccount; - transaction.Details = $"ACH => {achCreditTransfer.BankName}, {achCreditTransfer.AccountNumber}"; - } - - break; - } - } - - return transaction; - } - - /// - /// Handles the event type from Stripe. - /// - /// - private async Task HandlePaymentMethodAttachedAsync(Event parsedEvent) - { - var paymentMethod = await _stripeEventService.GetPaymentMethod(parsedEvent); - if (paymentMethod is null) - { - _logger.LogWarning("Attempted to handle the event payment_method.attached but paymentMethod was null"); - return; - } - - var subscriptionListOptions = new SubscriptionListOptions - { - Customer = paymentMethod.CustomerId, - Status = StripeSubscriptionStatus.Unpaid, - Expand = ["data.latest_invoice"] - }; - - StripeList unpaidSubscriptions; - try - { - unpaidSubscriptions = await _stripeFacade.ListSubscriptions(subscriptionListOptions); - } - catch (Exception e) - { - _logger.LogError(e, - "Attempted to get unpaid invoices for customer {CustomerId} but encountered an error while calling Stripe", - paymentMethod.CustomerId); - - return; - } - - foreach (var unpaidSubscription in unpaidSubscriptions) - { - await AttemptToPayOpenSubscriptionAsync(unpaidSubscription); - } - } - - private async Task AttemptToPayOpenSubscriptionAsync(Subscription unpaidSubscription) - { - var latestInvoice = unpaidSubscription.LatestInvoice; - - if (unpaidSubscription.LatestInvoice is null) - { - _logger.LogWarning( - "Attempted to pay unpaid subscription {SubscriptionId} but latest invoice didn't exist", - unpaidSubscription.Id); - - return; - } - - if (latestInvoice.Status != StripeInvoiceStatus.Open) - { - _logger.LogWarning( - "Attempted to pay unpaid subscription {SubscriptionId} but latest invoice wasn't \"open\"", - unpaidSubscription.Id); - - return; - } - - try - { - await AttemptToPayInvoiceAsync(latestInvoice, true); - } - catch (Exception e) - { - _logger.LogError(e, - "Attempted to pay open invoice {InvoiceId} on unpaid subscription {SubscriptionId} but encountered an error", - latestInvoice.Id, unpaidSubscription.Id); - throw; - } - } - - /// - /// Gets the organizationId, userId, or providerId from the metadata of a Stripe Subscription object. - /// - /// - /// - private static Tuple GetIdsFromMetadata(Dictionary metadata) - { - if (metadata == null || metadata.Count == 0) - { - return new Tuple(null, null, null); - } - - metadata.TryGetValue("organizationId", out var orgIdString); - metadata.TryGetValue("userId", out var userIdString); - metadata.TryGetValue("providerId", out var providerIdString); - - orgIdString ??= metadata.FirstOrDefault(x => - x.Key.Equals("organizationId", StringComparison.OrdinalIgnoreCase)).Value; - - userIdString ??= metadata.FirstOrDefault(x => - x.Key.Equals("userId", StringComparison.OrdinalIgnoreCase)).Value; - - providerIdString ??= metadata.FirstOrDefault(x => - x.Key.Equals("providerId", StringComparison.OrdinalIgnoreCase)).Value; - - Guid? organizationId = string.IsNullOrWhiteSpace(orgIdString) ? null : new Guid(orgIdString); - Guid? userId = string.IsNullOrWhiteSpace(userIdString) ? null : new Guid(userIdString); - Guid? providerId = string.IsNullOrWhiteSpace(providerIdString) ? null : new Guid(providerIdString); - - return new Tuple(organizationId, userId, providerId); - } - - private static bool OrgPlanForInvoiceNotifications(Organization org) => StaticStore.GetPlan(org.PlanType).IsAnnual; - - private async Task AttemptToPayInvoiceAsync(Invoice invoice, bool attemptToPayWithStripe = false) - { - var customer = await _stripeFacade.GetCustomer(invoice.CustomerId); - - if (customer?.Metadata?.ContainsKey("btCustomerId") ?? false) - { - return await AttemptToPayInvoiceWithBraintreeAsync(invoice, customer); - } - - if (attemptToPayWithStripe) - { - return await AttemptToPayInvoiceWithStripeAsync(invoice); - } - - return false; - } - - private async Task AttemptToPayInvoiceWithBraintreeAsync(Invoice invoice, Customer customer) - { - _logger.LogDebug("Attempting to pay invoice with Braintree"); - if (!customer?.Metadata?.ContainsKey("btCustomerId") ?? true) - { - _logger.LogWarning( - "Attempted to pay invoice with Braintree but btCustomerId wasn't on Stripe customer metadata"); - return false; - } - - var subscription = await _stripeFacade.GetSubscription(invoice.SubscriptionId); - var (organizationId, userId, providerId) = GetIdsFromMetadata(subscription?.Metadata); - if (!organizationId.HasValue && !userId.HasValue) - { - _logger.LogWarning( - "Attempted to pay invoice with Braintree but Stripe subscription metadata didn't contain either a organizationId or userId"); - return false; - } - - var orgTransaction = organizationId.HasValue; - var btObjIdField = orgTransaction ? "organization_id" : "user_id"; - var btObjId = organizationId ?? userId.Value; - var btInvoiceAmount = invoice.AmountDue / 100M; - - var existingTransactions = orgTransaction ? - await _transactionRepository.GetManyByOrganizationIdAsync(organizationId.Value) : - await _transactionRepository.GetManyByUserIdAsync(userId.Value); - var duplicateTimeSpan = TimeSpan.FromHours(24); - var now = DateTime.UtcNow; - var duplicateTransaction = existingTransactions? - .FirstOrDefault(t => (now - t.CreationDate) < duplicateTimeSpan); - if (duplicateTransaction != null) - { - _logger.LogWarning("There is already a recent PayPal transaction ({0}). " + - "Do not charge again to prevent possible duplicate.", duplicateTransaction.GatewayId); - return false; - } - - Result transactionResult; - try - { - transactionResult = await _btGateway.Transaction.SaleAsync( - new Braintree.TransactionRequest - { - Amount = btInvoiceAmount, - CustomerId = customer.Metadata["btCustomerId"], - Options = new Braintree.TransactionOptionsRequest - { - SubmitForSettlement = true, - PayPal = new Braintree.TransactionOptionsPayPalRequest - { - CustomField = - $"{btObjIdField}:{btObjId},region:{_globalSettings.BaseServiceUri.CloudRegion}" - } - }, - CustomFields = new Dictionary - { - [btObjIdField] = btObjId.ToString(), - ["region"] = _globalSettings.BaseServiceUri.CloudRegion - } - }); - } - catch (NotFoundException e) - { - _logger.LogError(e, - "Attempted to make a payment with Braintree, but customer did not exist for the given btCustomerId present on the Stripe metadata"); - throw; - } - - if (!transactionResult.IsSuccess()) - { - if (invoice.AttemptCount < 4) - { - await _mailService.SendPaymentFailedAsync(customer.Email, btInvoiceAmount, true); - } - return false; - } - - try - { - await _stripeFacade.UpdateInvoice(invoice.Id, new InvoiceUpdateOptions - { - Metadata = new Dictionary - { - ["btTransactionId"] = transactionResult.Target.Id, - ["btPayPalTransactionId"] = - transactionResult.Target.PayPalDetails?.AuthorizationId - } - }); - await _stripeFacade.PayInvoice(invoice.Id, new InvoicePayOptions { PaidOutOfBand = true }); - } - catch (Exception e) - { - await _btGateway.Transaction.RefundAsync(transactionResult.Target.Id); - if (e.Message.Contains("Invoice is already paid")) - { - await _stripeFacade.UpdateInvoice(invoice.Id, new InvoiceUpdateOptions - { - Metadata = invoice.Metadata - }); - } - else - { - throw; - } - } - - return true; - } - - private async Task AttemptToPayInvoiceWithStripeAsync(Invoice invoice) - { - try - { - await _stripeFacade.PayInvoice(invoice.Id); - return true; - } - catch (Exception e) - { - _logger.LogWarning( - e, - "Exception occurred while trying to pay Stripe invoice with Id: {InvoiceId}", - invoice.Id); - - throw; - } - } - - private static bool ShouldAttemptToPayInvoice(Invoice invoice) => - invoice is - { - AmountDue: > 0, - Paid: false, - CollectionMethod: "charge_automatically", - BillingReason: "subscription_cycle" or "automatic_pending_invoice_item_invoice", - SubscriptionId: not null - }; - - private async Task VerifyCorrectTaxRateForCharge(Invoice invoice, Subscription subscription) - { - if (string.IsNullOrWhiteSpace(invoice?.CustomerAddress?.Country) || - string.IsNullOrWhiteSpace(invoice?.CustomerAddress?.PostalCode)) - { - return subscription; - } - - var localBitwardenTaxRates = await _taxRateRepository.GetByLocationAsync( - new TaxRate() - { - Country = invoice.CustomerAddress.Country, - PostalCode = invoice.CustomerAddress.PostalCode - } - ); - - if (!localBitwardenTaxRates.Any()) - { - return subscription; - } - - var stripeTaxRate = await _stripeFacade.GetTaxRate(localBitwardenTaxRates.First().Id); - if (stripeTaxRate == null || subscription.DefaultTaxRates.Any(x => x == stripeTaxRate)) - { - return subscription; - } - - subscription.DefaultTaxRates = [stripeTaxRate]; - - var subscriptionOptions = new SubscriptionUpdateOptions { DefaultTaxRates = [stripeTaxRate.Id] }; - subscription = await _stripeFacade.UpdateSubscription(subscription.Id, subscriptionOptions); - - return subscription; - } - - private static bool IsSponsoredSubscription(Subscription subscription) => - StaticStore.SponsoredPlans - .Any(p => subscription.Items - .Any(i => i.Plan.Id == p.StripePlanId)); - - /// - /// Handles the event type from Stripe. - /// - /// - private async Task HandlePaymentFailedEventAsync(Event parsedEvent) - { - var invoice = await _stripeEventService.GetInvoice(parsedEvent, true); - if (invoice.Paid || invoice.AttemptCount <= 1 || !ShouldAttemptToPayInvoice(invoice)) - { - return; - } - - var subscription = await _stripeFacade.GetSubscription(invoice.SubscriptionId); - // attempt count 4 = 11 days after initial failure - if (invoice.AttemptCount <= 3 || - !subscription.Items.Any(i => i.Price.Id is PremiumPlanId or PremiumPlanIdAppStore)) - { - await AttemptToPayInvoiceAsync(invoice); - } - } - - private async Task CancelSubscription(string subscriptionId) => - await _stripeFacade.CancelSubscription(subscriptionId, new SubscriptionCancelOptions()); - - private async Task VoidOpenInvoices(string subscriptionId) - { - var options = new InvoiceListOptions - { - Status = StripeInvoiceStatus.Open, - Subscription = subscriptionId - }; - var invoices = await _stripeFacade.ListInvoices(options); - foreach (var invoice in invoices) - { - await _stripeFacade.VoidInvoice(invoice.Id); - } - } - + /// The body of the webhook request received from Stripe. + /// + /// The Stripe webhook secret corresponding to the API version found in the webhook body. + /// Returns null if the API version is unrecognized. + /// private string PickStripeWebhookSecret(string webhookBody) { var versionContainer = JsonSerializer.Deserialize(webhookBody); diff --git a/src/Billing/Services/IStripeEventProcessor.cs b/src/Billing/Services/IStripeEventProcessor.cs new file mode 100644 index 0000000000..6924585c0e --- /dev/null +++ b/src/Billing/Services/IStripeEventProcessor.cs @@ -0,0 +1,12 @@ +using Event = Stripe.Event; +namespace Bit.Billing.Services; + +public interface IStripeEventProcessor +{ + /// + /// Processes the specified Stripe event asynchronously. + /// + /// The Stripe event to be processed. + /// A task representing the asynchronous operation. + Task ProcessEventAsync(Event parsedEvent); +} diff --git a/src/Billing/Services/IStripeEventUtilityService.cs b/src/Billing/Services/IStripeEventUtilityService.cs new file mode 100644 index 0000000000..a5f536ad11 --- /dev/null +++ b/src/Billing/Services/IStripeEventUtilityService.cs @@ -0,0 +1,67 @@ +using Stripe; +using Transaction = Bit.Core.Entities.Transaction; +namespace Bit.Billing.Services; + +public interface IStripeEventUtilityService +{ + /// + /// Gets the organization or user ID from the metadata of a Stripe Charge object. + /// + /// + /// + Task<(Guid?, Guid?, Guid?)> GetEntityIdsFromChargeAsync(Charge charge); + + /// + /// Gets the organizationId, userId, or providerId from the metadata of a Stripe Subscription object. + /// + /// + /// + Tuple GetIdsFromMetadata(Dictionary metadata); + + /// + /// Determines whether the specified subscription is a sponsored subscription. + /// + /// The subscription to be evaluated. + /// + /// A boolean value indicating whether the subscription is a sponsored subscription. + /// Returns true if the subscription matches any of the sponsored plans; otherwise, false. + /// + bool IsSponsoredSubscription(Subscription subscription); + + /// + /// Converts a Stripe Charge object to a Bitwarden Transaction object. + /// + /// + /// + /// + /// /// + /// + Transaction FromChargeToTransaction(Charge charge, Guid? organizationId, Guid? userId, Guid? providerId); + + /// + /// Attempts to pay the specified invoice. If a customer is eligible, the invoice is paid using Braintree or Stripe. + /// + /// The invoice to be paid. + /// Indicates whether to attempt payment with Stripe. Defaults to false. + /// A task representing the asynchronous operation. The task result contains a boolean value indicating whether the invoice payment attempt was successful. + Task AttemptToPayInvoiceAsync(Invoice invoice, bool attemptToPayWithStripe = false); + + + /// + /// Determines whether an invoice should be attempted to be paid based on certain criteria. + /// + /// The invoice to be evaluated. + /// A boolean value indicating whether the invoice should be attempted to be paid. + bool ShouldAttemptToPayInvoice(Invoice invoice); + + /// + /// The ID for the premium annual plan. + /// + const string PremiumPlanId = "premium-annually"; + + /// + /// The ID for the premium annual plan via the App Store. + /// + const string PremiumPlanIdAppStore = "premium-annually-app"; + +} diff --git a/src/Billing/Services/IStripeWebhookHandler.cs b/src/Billing/Services/IStripeWebhookHandler.cs new file mode 100644 index 0000000000..59be435489 --- /dev/null +++ b/src/Billing/Services/IStripeWebhookHandler.cs @@ -0,0 +1,67 @@ +using Event = Stripe.Event; +namespace Bit.Billing.Services; + +public interface IStripeWebhookHandler +{ + /// + /// Handles the specified Stripe event asynchronously. + /// + /// The Stripe event to be handled. + /// A task representing the asynchronous operation. + Task HandleAsync(Event parsedEvent); +} + +/// +/// Defines the contract for handling Stripe subscription deleted events. +/// +public interface ISubscriptionDeletedHandler : IStripeWebhookHandler; + +/// +/// Defines the contract for handling Stripe subscription updated events. +/// +public interface ISubscriptionUpdatedHandler : IStripeWebhookHandler; + +/// +/// Defines the contract for handling Stripe upcoming invoice events. +/// +public interface IUpcomingInvoiceHandler : IStripeWebhookHandler; + +/// +/// Defines the contract for handling Stripe charge succeeded events. +/// +public interface IChargeSucceededHandler : IStripeWebhookHandler; + +/// +/// Defines the contract for handling Stripe charge refunded events. +/// +public interface IChargeRefundedHandler : IStripeWebhookHandler; + +/// +/// Defines the contract for handling Stripe payment succeeded events. +/// +public interface IPaymentSucceededHandler : IStripeWebhookHandler; + +/// +/// Defines the contract for handling Stripe payment failed events. +/// +public interface IPaymentFailedHandler : IStripeWebhookHandler; + +/// +/// Defines the contract for handling Stripe invoice created events. +/// +public interface IInvoiceCreatedHandler : IStripeWebhookHandler; + +/// +/// Defines the contract for handling Stripe payment method attached events. +/// +public interface IPaymentMethodAttachedHandler : IStripeWebhookHandler; + +/// +/// Defines the contract for handling Stripe customer updated events. +/// +public interface ICustomerUpdatedHandler : IStripeWebhookHandler; + +/// +/// Defines the contract for handling Stripe Invoice Finalized events. +/// +public interface IInvoiceFinalizedHandler : IStripeWebhookHandler; diff --git a/src/Billing/Services/Implementations/ChargeRefundedHandler.cs b/src/Billing/Services/Implementations/ChargeRefundedHandler.cs new file mode 100644 index 0000000000..905491b6c5 --- /dev/null +++ b/src/Billing/Services/Implementations/ChargeRefundedHandler.cs @@ -0,0 +1,98 @@ +using Bit.Billing.Constants; +using Bit.Core.Enums; +using Bit.Core.Repositories; +using Microsoft.Data.SqlClient; +using Event = Stripe.Event; +using Transaction = Bit.Core.Entities.Transaction; +using TransactionType = Bit.Core.Enums.TransactionType; +namespace Bit.Billing.Services.Implementations; + +public class ChargeRefundedHandler : IChargeRefundedHandler +{ + private readonly ILogger _logger; + private readonly IStripeEventService _stripeEventService; + private readonly ITransactionRepository _transactionRepository; + private readonly IStripeEventUtilityService _stripeEventUtilityService; + + public ChargeRefundedHandler( + ILogger logger, + IStripeEventService stripeEventService, + ITransactionRepository transactionRepository, + IStripeEventUtilityService stripeEventUtilityService) + { + _logger = logger; + _stripeEventService = stripeEventService; + _transactionRepository = transactionRepository; + _stripeEventUtilityService = stripeEventUtilityService; + } + + /// + /// Handles the event type from Stripe. + /// + /// + public async Task HandleAsync(Event parsedEvent) + { + var charge = await _stripeEventService.GetCharge(parsedEvent, true, ["refunds"]); + var parentTransaction = await _transactionRepository.GetByGatewayIdAsync(GatewayType.Stripe, charge.Id); + if (parentTransaction == null) + { + // Attempt to create a transaction for the charge if it doesn't exist + var (organizationId, userId, providerId) = await _stripeEventUtilityService.GetEntityIdsFromChargeAsync(charge); + var tx = _stripeEventUtilityService.FromChargeToTransaction(charge, organizationId, userId, providerId); + try + { + parentTransaction = await _transactionRepository.CreateAsync(tx); + } + catch (SqlException e) when (e.Number == 547) // FK constraint violation + { + _logger.LogWarning( + "Charge refund could not create transaction as entity may have been deleted. {ChargeId}", + charge.Id); + return; + } + } + + var amountRefunded = charge.AmountRefunded / 100M; + + if (parentTransaction.Refunded.GetValueOrDefault() || + parentTransaction.RefundedAmount.GetValueOrDefault() >= amountRefunded) + { + _logger.LogWarning( + "Charge refund amount doesn't match parent transaction's amount or parent has already been refunded. {ChargeId}", + charge.Id); + return; + } + + parentTransaction.RefundedAmount = amountRefunded; + if (charge.Refunded) + { + parentTransaction.Refunded = true; + } + + await _transactionRepository.ReplaceAsync(parentTransaction); + + foreach (var refund in charge.Refunds) + { + var refundTransaction = await _transactionRepository.GetByGatewayIdAsync( + GatewayType.Stripe, refund.Id); + if (refundTransaction != null) + { + continue; + } + + await _transactionRepository.CreateAsync(new Transaction + { + Amount = refund.Amount / 100M, + CreationDate = refund.Created, + OrganizationId = parentTransaction.OrganizationId, + UserId = parentTransaction.UserId, + ProviderId = parentTransaction.ProviderId, + Type = TransactionType.Refund, + Gateway = GatewayType.Stripe, + GatewayId = refund.Id, + PaymentMethodType = parentTransaction.PaymentMethodType, + Details = parentTransaction.Details + }); + } + } +} diff --git a/src/Billing/Services/Implementations/ChargeSucceededHandler.cs b/src/Billing/Services/Implementations/ChargeSucceededHandler.cs new file mode 100644 index 0000000000..bd8ea7def2 --- /dev/null +++ b/src/Billing/Services/Implementations/ChargeSucceededHandler.cs @@ -0,0 +1,67 @@ +using Bit.Billing.Constants; +using Bit.Core.Enums; +using Bit.Core.Repositories; +using Microsoft.Data.SqlClient; +using Event = Stripe.Event; + +namespace Bit.Billing.Services.Implementations; + +public class ChargeSucceededHandler : IChargeSucceededHandler +{ + private readonly ILogger _logger; + private readonly IStripeEventService _stripeEventService; + private readonly ITransactionRepository _transactionRepository; + private readonly IStripeEventUtilityService _stripeEventUtilityService; + + public ChargeSucceededHandler( + ILogger logger, + IStripeEventService stripeEventService, + ITransactionRepository transactionRepository, + IStripeEventUtilityService stripeEventUtilityService) + { + _logger = logger; + _stripeEventService = stripeEventService; + _transactionRepository = transactionRepository; + _stripeEventUtilityService = stripeEventUtilityService; + } + + /// + /// Handles the event type from Stripe. + /// + /// + public async Task HandleAsync(Event parsedEvent) + { + var charge = await _stripeEventService.GetCharge(parsedEvent); + var existingTransaction = await _transactionRepository.GetByGatewayIdAsync(GatewayType.Stripe, charge.Id); + if (existingTransaction is not null) + { + _logger.LogInformation("Charge success already processed. {ChargeId}", charge.Id); + return; + } + + var (organizationId, userId, providerId) = await _stripeEventUtilityService.GetEntityIdsFromChargeAsync(charge); + if (!organizationId.HasValue && !userId.HasValue && !providerId.HasValue) + { + _logger.LogWarning("Charge success has no subscriber ids. {ChargeId}", charge.Id); + return; + } + + var transaction = _stripeEventUtilityService.FromChargeToTransaction(charge, organizationId, userId, providerId); + if (!transaction.PaymentMethodType.HasValue) + { + _logger.LogWarning("Charge success from unsupported source/method. {ChargeId}", charge.Id); + return; + } + + try + { + await _transactionRepository.CreateAsync(transaction); + } + catch (SqlException e) when (e.Number == 547) + { + _logger.LogWarning( + "Charge success could not create transaction as entity may have been deleted. {ChargeId}", + charge.Id); + } + } +} diff --git a/src/Billing/Services/Implementations/CustomerUpdatedHandler.cs b/src/Billing/Services/Implementations/CustomerUpdatedHandler.cs new file mode 100644 index 0000000000..ec70697c01 --- /dev/null +++ b/src/Billing/Services/Implementations/CustomerUpdatedHandler.cs @@ -0,0 +1,60 @@ +using Bit.Core.Context; +using Bit.Core.Repositories; +using Bit.Core.Tools.Enums; +using Bit.Core.Tools.Models.Business; +using Bit.Core.Tools.Services; +using Event = Stripe.Event; + +namespace Bit.Billing.Services.Implementations; + +public class CustomerUpdatedHandler : ICustomerUpdatedHandler +{ + private readonly IOrganizationRepository _organizationRepository; + private readonly IReferenceEventService _referenceEventService; + private readonly ICurrentContext _currentContext; + private readonly IStripeEventService _stripeEventService; + private readonly IStripeEventUtilityService _stripeEventUtilityService; + + public CustomerUpdatedHandler( + IOrganizationRepository organizationRepository, + IReferenceEventService referenceEventService, + ICurrentContext currentContext, + IStripeEventService stripeEventService, + IStripeEventUtilityService stripeEventUtilityService) + { + _organizationRepository = organizationRepository; + _referenceEventService = referenceEventService; + _currentContext = currentContext; + _stripeEventService = stripeEventService; + _stripeEventUtilityService = stripeEventUtilityService; + } + + /// + /// Handles the event type from Stripe. + /// + /// + public async Task HandleAsync(Event parsedEvent) + { + var customer = await _stripeEventService.GetCustomer(parsedEvent, true, ["subscriptions"]); + if (customer.Subscriptions == null || !customer.Subscriptions.Any()) + { + return; + } + + var subscription = customer.Subscriptions.First(); + + var (organizationId, _, providerId) = _stripeEventUtilityService.GetIdsFromMetadata(subscription.Metadata); + + if (!organizationId.HasValue) + { + return; + } + + var organization = await _organizationRepository.GetByIdAsync(organizationId.Value); + organization.BillingEmail = customer.Email; + await _organizationRepository.ReplaceAsync(organization); + + await _referenceEventService.RaiseEventAsync( + new ReferenceEvent(ReferenceEventType.OrganizationEditedInStripe, organization, _currentContext)); + } +} diff --git a/src/Billing/Services/Implementations/InvoiceCreatedHandler.cs b/src/Billing/Services/Implementations/InvoiceCreatedHandler.cs new file mode 100644 index 0000000000..4c84cca96b --- /dev/null +++ b/src/Billing/Services/Implementations/InvoiceCreatedHandler.cs @@ -0,0 +1,35 @@ +using Event = Stripe.Event; + +namespace Bit.Billing.Services.Implementations; + +public class InvoiceCreatedHandler : IInvoiceCreatedHandler +{ + private readonly IStripeEventService _stripeEventService; + private readonly IStripeEventUtilityService _stripeEventUtilityService; + private readonly IProviderEventService _providerEventService; + + public InvoiceCreatedHandler( + IStripeEventService stripeEventService, + IStripeEventUtilityService stripeEventUtilityService, + IProviderEventService providerEventService) + { + _stripeEventService = stripeEventService; + _stripeEventUtilityService = stripeEventUtilityService; + _providerEventService = providerEventService; + } + + /// + /// Handles the event type from Stripe. + /// + /// + public async Task HandleAsync(Event parsedEvent) + { + var invoice = await _stripeEventService.GetInvoice(parsedEvent, true); + if (_stripeEventUtilityService.ShouldAttemptToPayInvoice(invoice)) + { + await _stripeEventUtilityService.AttemptToPayInvoiceAsync(invoice); + } + + await _providerEventService.TryRecordInvoiceLineItems(parsedEvent); + } +} diff --git a/src/Billing/Services/Implementations/InvoiceFinalizedHandler.cs b/src/Billing/Services/Implementations/InvoiceFinalizedHandler.cs new file mode 100644 index 0000000000..a67d6d301e --- /dev/null +++ b/src/Billing/Services/Implementations/InvoiceFinalizedHandler.cs @@ -0,0 +1,23 @@ +using Event = Stripe.Event; + +namespace Bit.Billing.Services.Implementations; + +public class InvoiceFinalizedHandler : IInvoiceFinalizedHandler +{ + + private readonly IProviderEventService _providerEventService; + + public InvoiceFinalizedHandler(IProviderEventService providerEventService) + { + _providerEventService = providerEventService; + } + + /// + /// Handles the event type from Stripe. + /// + /// + public async Task HandleAsync(Event parsedEvent) + { + await _providerEventService.TryRecordInvoiceLineItems(parsedEvent); + } +} diff --git a/src/Billing/Services/Implementations/PaymentFailedHandler.cs b/src/Billing/Services/Implementations/PaymentFailedHandler.cs new file mode 100644 index 0000000000..acf6ca70c7 --- /dev/null +++ b/src/Billing/Services/Implementations/PaymentFailedHandler.cs @@ -0,0 +1,52 @@ +using Stripe; +using Event = Stripe.Event; + +namespace Bit.Billing.Services.Implementations; + +public class PaymentFailedHandler : IPaymentFailedHandler +{ + private readonly IStripeEventService _stripeEventService; + private readonly IStripeFacade _stripeFacade; + private readonly IStripeEventUtilityService _stripeEventUtilityService; + + public PaymentFailedHandler( + IStripeEventService stripeEventService, + IStripeFacade stripeFacade, + IStripeEventUtilityService stripeEventUtilityService) + { + _stripeEventService = stripeEventService; + _stripeFacade = stripeFacade; + _stripeEventUtilityService = stripeEventUtilityService; + } + + /// + /// Handles the event type from Stripe. + /// + /// + public async Task HandleAsync(Event parsedEvent) + { + var invoice = await _stripeEventService.GetInvoice(parsedEvent, true); + if (invoice.Paid || invoice.AttemptCount <= 1 || !ShouldAttemptToPayInvoice(invoice)) + { + return; + } + + var subscription = await _stripeFacade.GetSubscription(invoice.SubscriptionId); + // attempt count 4 = 11 days after initial failure + if (invoice.AttemptCount <= 3 || + !subscription.Items.Any(i => i.Price.Id is IStripeEventUtilityService.PremiumPlanId or IStripeEventUtilityService.PremiumPlanIdAppStore)) + { + await _stripeEventUtilityService.AttemptToPayInvoiceAsync(invoice); + } + } + + private static bool ShouldAttemptToPayInvoice(Invoice invoice) => + invoice is + { + AmountDue: > 0, + Paid: false, + CollectionMethod: "charge_automatically", + BillingReason: "subscription_cycle" or "automatic_pending_invoice_item_invoice", + SubscriptionId: not null + }; +} diff --git a/src/Billing/Services/Implementations/PaymentMethodAttachedHandler.cs b/src/Billing/Services/Implementations/PaymentMethodAttachedHandler.cs new file mode 100644 index 0000000000..6092e001ce --- /dev/null +++ b/src/Billing/Services/Implementations/PaymentMethodAttachedHandler.cs @@ -0,0 +1,96 @@ +using Bit.Billing.Constants; +using Stripe; +using Event = Stripe.Event; + +namespace Bit.Billing.Services.Implementations; + +public class PaymentMethodAttachedHandler : IPaymentMethodAttachedHandler +{ + private readonly ILogger _logger; + private readonly IStripeEventService _stripeEventService; + private readonly IStripeFacade _stripeFacade; + private readonly IStripeEventUtilityService _stripeEventUtilityService; + + public PaymentMethodAttachedHandler( + ILogger logger, + IStripeEventService stripeEventService, + IStripeFacade stripeFacade, + IStripeEventUtilityService stripeEventUtilityService) + { + _logger = logger; + _stripeEventService = stripeEventService; + _stripeFacade = stripeFacade; + _stripeEventUtilityService = stripeEventUtilityService; + } + + public async Task HandleAsync(Event parsedEvent) + { + var paymentMethod = await _stripeEventService.GetPaymentMethod(parsedEvent); + if (paymentMethod is null) + { + _logger.LogWarning("Attempted to handle the event payment_method.attached but paymentMethod was null"); + return; + } + + var subscriptionListOptions = new SubscriptionListOptions + { + Customer = paymentMethod.CustomerId, + Status = StripeSubscriptionStatus.Unpaid, + Expand = ["data.latest_invoice"] + }; + + StripeList unpaidSubscriptions; + try + { + unpaidSubscriptions = await _stripeFacade.ListSubscriptions(subscriptionListOptions); + } + catch (Exception e) + { + _logger.LogError(e, + "Attempted to get unpaid invoices for customer {CustomerId} but encountered an error while calling Stripe", + paymentMethod.CustomerId); + + return; + } + + foreach (var unpaidSubscription in unpaidSubscriptions) + { + await AttemptToPayOpenSubscriptionAsync(unpaidSubscription); + } + } + + private async Task AttemptToPayOpenSubscriptionAsync(Subscription unpaidSubscription) + { + var latestInvoice = unpaidSubscription.LatestInvoice; + + if (unpaidSubscription.LatestInvoice is null) + { + _logger.LogWarning( + "Attempted to pay unpaid subscription {SubscriptionId} but latest invoice didn't exist", + unpaidSubscription.Id); + + return; + } + + if (latestInvoice.Status != StripeInvoiceStatus.Open) + { + _logger.LogWarning( + "Attempted to pay unpaid subscription {SubscriptionId} but latest invoice wasn't \"open\"", + unpaidSubscription.Id); + + return; + } + + try + { + await _stripeEventUtilityService.AttemptToPayInvoiceAsync(latestInvoice, true); + } + catch (Exception e) + { + _logger.LogError(e, + "Attempted to pay open invoice {InvoiceId} on unpaid subscription {SubscriptionId} but encountered an error", + latestInvoice.Id, unpaidSubscription.Id); + throw; + } + } +} diff --git a/src/Billing/Services/Implementations/PaymentSucceededHandler.cs b/src/Billing/Services/Implementations/PaymentSucceededHandler.cs new file mode 100644 index 0000000000..6aa8aa2b9f --- /dev/null +++ b/src/Billing/Services/Implementations/PaymentSucceededHandler.cs @@ -0,0 +1,171 @@ +using Bit.Billing.Constants; +using Bit.Core.AdminConsole.Repositories; +using Bit.Core.Billing.Enums; +using Bit.Core.Context; +using Bit.Core.Repositories; +using Bit.Core.Services; +using Bit.Core.Tools.Enums; +using Bit.Core.Tools.Models.Business; +using Bit.Core.Tools.Services; +using Bit.Core.Utilities; +using Event = Stripe.Event; + +namespace Bit.Billing.Services.Implementations; + +public class PaymentSucceededHandler : IPaymentSucceededHandler +{ + private readonly ILogger _logger; + private readonly IStripeEventService _stripeEventService; + private readonly IOrganizationService _organizationService; + private readonly IUserService _userService; + private readonly IStripeFacade _stripeFacade; + private readonly IProviderRepository _providerRepository; + private readonly IOrganizationRepository _organizationRepository; + private readonly IReferenceEventService _referenceEventService; + private readonly ICurrentContext _currentContext; + private readonly IUserRepository _userRepository; + private readonly IStripeEventUtilityService _stripeEventUtilityService; + + public PaymentSucceededHandler( + ILogger logger, + IStripeEventService stripeEventService, + IStripeFacade stripeFacade, + IProviderRepository providerRepository, + IOrganizationRepository organizationRepository, + IReferenceEventService referenceEventService, + ICurrentContext currentContext, + IUserRepository userRepository, + IStripeEventUtilityService stripeEventUtilityService, + IUserService userService, + IOrganizationService organizationService) + { + _logger = logger; + _stripeEventService = stripeEventService; + _stripeFacade = stripeFacade; + _providerRepository = providerRepository; + _organizationRepository = organizationRepository; + _referenceEventService = referenceEventService; + _currentContext = currentContext; + _userRepository = userRepository; + _stripeEventUtilityService = stripeEventUtilityService; + _userService = userService; + _organizationService = organizationService; + } + + /// + /// Handles the event type from Stripe. + /// + /// + public async Task HandleAsync(Event parsedEvent) + { + var invoice = await _stripeEventService.GetInvoice(parsedEvent, true); + if (!invoice.Paid || invoice.BillingReason != "subscription_create") + { + return; + } + + var subscription = await _stripeFacade.GetSubscription(invoice.SubscriptionId); + if (subscription?.Status != StripeSubscriptionStatus.Active) + { + return; + } + + if (DateTime.UtcNow - invoice.Created < TimeSpan.FromMinutes(1)) + { + await Task.Delay(5000); + } + + var (organizationId, userId, providerId) = _stripeEventUtilityService.GetIdsFromMetadata(subscription.Metadata); + + if (providerId.HasValue) + { + var provider = await _providerRepository.GetByIdAsync(providerId.Value); + + if (provider == null) + { + _logger.LogError( + "Received invoice.payment_succeeded webhook ({EventID}) for Provider ({ProviderID}) that does not exist", + parsedEvent.Id, + providerId.Value); + + return; + } + + var teamsMonthly = StaticStore.GetPlan(PlanType.TeamsMonthly); + + var enterpriseMonthly = StaticStore.GetPlan(PlanType.EnterpriseMonthly); + + var teamsMonthlyLineItem = + subscription.Items.Data.FirstOrDefault(item => + item.Plan.Id == teamsMonthly.PasswordManager.StripeSeatPlanId); + + var enterpriseMonthlyLineItem = + subscription.Items.Data.FirstOrDefault(item => + item.Plan.Id == enterpriseMonthly.PasswordManager.StripeSeatPlanId); + + if (teamsMonthlyLineItem == null || enterpriseMonthlyLineItem == null) + { + _logger.LogError("invoice.payment_succeeded webhook ({EventID}) for Provider ({ProviderID}) indicates missing subscription line items", + parsedEvent.Id, + provider.Id); + + return; + } + + await _referenceEventService.RaiseEventAsync(new ReferenceEvent + { + Type = ReferenceEventType.Rebilled, + Source = ReferenceEventSource.Provider, + Id = provider.Id, + PlanType = PlanType.TeamsMonthly, + Seats = (int)teamsMonthlyLineItem.Quantity + }); + + await _referenceEventService.RaiseEventAsync(new ReferenceEvent + { + Type = ReferenceEventType.Rebilled, + Source = ReferenceEventSource.Provider, + Id = provider.Id, + PlanType = PlanType.EnterpriseMonthly, + Seats = (int)enterpriseMonthlyLineItem.Quantity + }); + } + else if (organizationId.HasValue) + { + if (!subscription.Items.Any(i => + StaticStore.Plans.Any(p => p.PasswordManager.StripePlanId == i.Plan.Id))) + { + return; + } + + await _organizationService.EnableAsync(organizationId.Value, subscription.CurrentPeriodEnd); + var organization = await _organizationRepository.GetByIdAsync(organizationId.Value); + + await _referenceEventService.RaiseEventAsync( + new ReferenceEvent(ReferenceEventType.Rebilled, organization, _currentContext) + { + PlanName = organization?.Plan, + PlanType = organization?.PlanType, + Seats = organization?.Seats, + Storage = organization?.MaxStorageGb, + }); + } + else if (userId.HasValue) + { + if (subscription.Items.All(i => i.Plan.Id != IStripeEventUtilityService.PremiumPlanId)) + { + return; + } + + await _userService.EnablePremiumAsync(userId.Value, subscription.CurrentPeriodEnd); + + var user = await _userRepository.GetByIdAsync(userId.Value); + await _referenceEventService.RaiseEventAsync( + new ReferenceEvent(ReferenceEventType.Rebilled, user, _currentContext) + { + PlanName = IStripeEventUtilityService.PremiumPlanId, + Storage = user?.MaxStorageGb, + }); + } + } +} diff --git a/src/Billing/Services/Implementations/StripeEventProcessor.cs b/src/Billing/Services/Implementations/StripeEventProcessor.cs new file mode 100644 index 0000000000..db4e3929f9 --- /dev/null +++ b/src/Billing/Services/Implementations/StripeEventProcessor.cs @@ -0,0 +1,89 @@ +using Bit.Billing.Constants; +using Event = Stripe.Event; + +namespace Bit.Billing.Services.Implementations; + +public class StripeEventProcessor : IStripeEventProcessor +{ + private readonly ILogger _logger; + private readonly ISubscriptionDeletedHandler _subscriptionDeletedHandler; + private readonly ISubscriptionUpdatedHandler _subscriptionUpdatedHandler; + private readonly IUpcomingInvoiceHandler _upcomingInvoiceHandler; + private readonly IChargeSucceededHandler _chargeSucceededHandler; + private readonly IChargeRefundedHandler _chargeRefundedHandler; + private readonly IPaymentSucceededHandler _paymentSucceededHandler; + private readonly IPaymentFailedHandler _paymentFailedHandler; + private readonly IInvoiceCreatedHandler _invoiceCreatedHandler; + private readonly IPaymentMethodAttachedHandler _paymentMethodAttachedHandler; + private readonly ICustomerUpdatedHandler _customerUpdatedHandler; + + public StripeEventProcessor( + ILogger logger, + ISubscriptionDeletedHandler subscriptionDeletedHandler, + ISubscriptionUpdatedHandler subscriptionUpdatedHandler, + IUpcomingInvoiceHandler upcomingInvoiceHandler, + IChargeSucceededHandler chargeSucceededHandler, + IChargeRefundedHandler chargeRefundedHandler, + IPaymentSucceededHandler paymentSucceededHandler, + IPaymentFailedHandler paymentFailedHandler, + IInvoiceCreatedHandler invoiceCreatedHandler, + IPaymentMethodAttachedHandler paymentMethodAttachedHandler, + ICustomerUpdatedHandler customerUpdatedHandler) + { + _logger = logger; + _subscriptionDeletedHandler = subscriptionDeletedHandler; + _subscriptionUpdatedHandler = subscriptionUpdatedHandler; + _upcomingInvoiceHandler = upcomingInvoiceHandler; + _chargeSucceededHandler = chargeSucceededHandler; + _chargeRefundedHandler = chargeRefundedHandler; + _paymentSucceededHandler = paymentSucceededHandler; + _paymentFailedHandler = paymentFailedHandler; + _invoiceCreatedHandler = invoiceCreatedHandler; + _paymentMethodAttachedHandler = paymentMethodAttachedHandler; + _customerUpdatedHandler = customerUpdatedHandler; + } + + public async Task ProcessEventAsync(Event parsedEvent) + { + switch (parsedEvent.Type) + { + case HandledStripeWebhook.SubscriptionDeleted: + await _subscriptionDeletedHandler.HandleAsync(parsedEvent); + break; + case HandledStripeWebhook.SubscriptionUpdated: + await _subscriptionUpdatedHandler.HandleAsync(parsedEvent); + break; + case HandledStripeWebhook.UpcomingInvoice: + await _upcomingInvoiceHandler.HandleAsync(parsedEvent); + break; + case HandledStripeWebhook.ChargeSucceeded: + await _chargeSucceededHandler.HandleAsync(parsedEvent); + break; + case HandledStripeWebhook.ChargeRefunded: + await _chargeRefundedHandler.HandleAsync(parsedEvent); + break; + case HandledStripeWebhook.PaymentSucceeded: + await _paymentSucceededHandler.HandleAsync(parsedEvent); + break; + case HandledStripeWebhook.PaymentFailed: + await _paymentFailedHandler.HandleAsync(parsedEvent); + break; + case HandledStripeWebhook.InvoiceCreated: + await _invoiceCreatedHandler.HandleAsync(parsedEvent); + break; + case HandledStripeWebhook.PaymentMethodAttached: + await _paymentMethodAttachedHandler.HandleAsync(parsedEvent); + break; + case HandledStripeWebhook.CustomerUpdated: + await _customerUpdatedHandler.HandleAsync(parsedEvent); + break; + case HandledStripeWebhook.InvoiceFinalized: + await _customerUpdatedHandler.HandleAsync(parsedEvent); + break; + default: + _logger.LogWarning("Unsupported event received. {EventType}", parsedEvent.Type); + break; + } + } + +} diff --git a/src/Billing/Services/Implementations/StripeEventUtilityService.cs b/src/Billing/Services/Implementations/StripeEventUtilityService.cs new file mode 100644 index 0000000000..f656dbcc11 --- /dev/null +++ b/src/Billing/Services/Implementations/StripeEventUtilityService.cs @@ -0,0 +1,401 @@ +using Bit.Billing.Constants; +using Bit.Core.Enums; +using Bit.Core.Exceptions; +using Bit.Core.Repositories; +using Bit.Core.Services; +using Bit.Core.Settings; +using Bit.Core.Utilities; +using Braintree; +using Stripe; +using Customer = Stripe.Customer; +using Subscription = Stripe.Subscription; +using Transaction = Bit.Core.Entities.Transaction; +using TransactionType = Bit.Core.Enums.TransactionType; + +namespace Bit.Billing.Services.Implementations; + +public class StripeEventUtilityService : IStripeEventUtilityService +{ + private readonly IStripeFacade _stripeFacade; + private readonly ILogger _logger; + private readonly ITransactionRepository _transactionRepository; + private readonly IMailService _mailService; + private readonly BraintreeGateway _btGateway; + private readonly GlobalSettings _globalSettings; + + public StripeEventUtilityService( + IStripeFacade stripeFacade, + ILogger logger, + ITransactionRepository transactionRepository, + IMailService mailService, + GlobalSettings globalSettings) + { + _stripeFacade = stripeFacade; + _logger = logger; + _transactionRepository = transactionRepository; + _mailService = mailService; + _btGateway = new BraintreeGateway + { + Environment = globalSettings.Braintree.Production ? + Braintree.Environment.PRODUCTION : Braintree.Environment.SANDBOX, + MerchantId = globalSettings.Braintree.MerchantId, + PublicKey = globalSettings.Braintree.PublicKey, + PrivateKey = globalSettings.Braintree.PrivateKey + }; + _globalSettings = globalSettings; + } + + /// + /// Gets the organizationId, userId, or providerId from the metadata of a Stripe Subscription object. + /// + /// + /// + public Tuple GetIdsFromMetadata(Dictionary metadata) + { + if (metadata == null || metadata.Count == 0) + { + return new Tuple(null, null, null); + } + + metadata.TryGetValue("organizationId", out var orgIdString); + metadata.TryGetValue("userId", out var userIdString); + metadata.TryGetValue("providerId", out var providerIdString); + + orgIdString ??= metadata.FirstOrDefault(x => + x.Key.Equals("organizationId", StringComparison.OrdinalIgnoreCase)).Value; + + userIdString ??= metadata.FirstOrDefault(x => + x.Key.Equals("userId", StringComparison.OrdinalIgnoreCase)).Value; + + providerIdString ??= metadata.FirstOrDefault(x => + x.Key.Equals("providerId", StringComparison.OrdinalIgnoreCase)).Value; + + Guid? organizationId = string.IsNullOrWhiteSpace(orgIdString) ? null : new Guid(orgIdString); + Guid? userId = string.IsNullOrWhiteSpace(userIdString) ? null : new Guid(userIdString); + Guid? providerId = string.IsNullOrWhiteSpace(providerIdString) ? null : new Guid(providerIdString); + + return new Tuple(organizationId, userId, providerId); + } + + /// + /// Gets the organization or user ID from the metadata of a Stripe Charge object. + /// + /// + /// + public async Task<(Guid?, Guid?, Guid?)> GetEntityIdsFromChargeAsync(Charge charge) + { + Guid? organizationId = null; + Guid? userId = null; + Guid? providerId = null; + + if (charge.InvoiceId != null) + { + var invoice = await _stripeFacade.GetInvoice(charge.InvoiceId); + if (invoice?.SubscriptionId != null) + { + var subscription = await _stripeFacade.GetSubscription(invoice.SubscriptionId); + (organizationId, userId, providerId) = GetIdsFromMetadata(subscription?.Metadata); + } + } + + if (organizationId.HasValue || userId.HasValue || providerId.HasValue) + { + return (organizationId, userId, providerId); + } + + var subscriptions = await _stripeFacade.ListSubscriptions(new SubscriptionListOptions + { + Customer = charge.CustomerId + }); + + foreach (var subscription in subscriptions) + { + if (subscription.Status is StripeSubscriptionStatus.Canceled or StripeSubscriptionStatus.IncompleteExpired) + { + continue; + } + + (organizationId, userId, providerId) = GetIdsFromMetadata(subscription.Metadata); + + if (organizationId.HasValue || userId.HasValue || providerId.HasValue) + { + return (organizationId, userId, providerId); + } + } + + return (null, null, null); + } + + public bool IsSponsoredSubscription(Subscription subscription) => + StaticStore.SponsoredPlans + .Any(p => subscription.Items + .Any(i => i.Plan.Id == p.StripePlanId)); + + /// + /// Converts a Stripe Charge object to a Bitwarden Transaction object. + /// + /// + /// + /// + /// /// + /// + public Transaction FromChargeToTransaction(Charge charge, Guid? organizationId, Guid? userId, Guid? providerId) + { + var transaction = new Transaction + { + Amount = charge.Amount / 100M, + CreationDate = charge.Created, + OrganizationId = organizationId, + UserId = userId, + ProviderId = providerId, + Type = TransactionType.Charge, + Gateway = GatewayType.Stripe, + GatewayId = charge.Id + }; + + switch (charge.Source) + { + case Card card: + { + transaction.PaymentMethodType = PaymentMethodType.Card; + transaction.Details = $"{card.Brand}, *{card.Last4}"; + break; + } + case BankAccount bankAccount: + { + transaction.PaymentMethodType = PaymentMethodType.BankAccount; + transaction.Details = $"{bankAccount.BankName}, *{bankAccount.Last4}"; + break; + } + case Source { Card: not null } source: + { + transaction.PaymentMethodType = PaymentMethodType.Card; + transaction.Details = $"{source.Card.Brand}, *{source.Card.Last4}"; + break; + } + case Source { AchDebit: not null } source: + { + transaction.PaymentMethodType = PaymentMethodType.BankAccount; + transaction.Details = $"{source.AchDebit.BankName}, *{source.AchDebit.Last4}"; + break; + } + case Source source: + { + if (source.AchCreditTransfer == null) + { + break; + } + + var achCreditTransfer = source.AchCreditTransfer; + + transaction.PaymentMethodType = PaymentMethodType.BankAccount; + transaction.Details = $"ACH => {achCreditTransfer.BankName}, {achCreditTransfer.AccountNumber}"; + + break; + } + default: + { + if (charge.PaymentMethodDetails == null) + { + break; + } + + if (charge.PaymentMethodDetails.Card != null) + { + var card = charge.PaymentMethodDetails.Card; + transaction.PaymentMethodType = PaymentMethodType.Card; + transaction.Details = $"{card.Brand?.ToUpperInvariant()}, *{card.Last4}"; + } + else if (charge.PaymentMethodDetails.AchDebit != null) + { + var achDebit = charge.PaymentMethodDetails.AchDebit; + transaction.PaymentMethodType = PaymentMethodType.BankAccount; + transaction.Details = $"{achDebit.BankName}, *{achDebit.Last4}"; + } + else if (charge.PaymentMethodDetails.AchCreditTransfer != null) + { + var achCreditTransfer = charge.PaymentMethodDetails.AchCreditTransfer; + transaction.PaymentMethodType = PaymentMethodType.BankAccount; + transaction.Details = $"ACH => {achCreditTransfer.BankName}, {achCreditTransfer.AccountNumber}"; + } + + break; + } + } + + return transaction; + } + + public async Task AttemptToPayInvoiceAsync(Invoice invoice, bool attemptToPayWithStripe = false) + { + var customer = await _stripeFacade.GetCustomer(invoice.CustomerId); + + if (customer?.Metadata?.ContainsKey("btCustomerId") ?? false) + { + return await AttemptToPayInvoiceWithBraintreeAsync(invoice, customer); + } + + if (attemptToPayWithStripe) + { + return await AttemptToPayInvoiceWithStripeAsync(invoice); + } + + return false; + } + + public bool ShouldAttemptToPayInvoice(Invoice invoice) => + invoice is + { + AmountDue: > 0, + Paid: false, + CollectionMethod: "charge_automatically", + BillingReason: "subscription_cycle" or "automatic_pending_invoice_item_invoice", + SubscriptionId: not null + }; + + private async Task AttemptToPayInvoiceWithBraintreeAsync(Invoice invoice, Customer customer) + { + _logger.LogDebug("Attempting to pay invoice with Braintree"); + if (!customer?.Metadata?.ContainsKey("btCustomerId") ?? true) + { + _logger.LogWarning( + "Attempted to pay invoice with Braintree but btCustomerId wasn't on Stripe customer metadata"); + return false; + } + + var subscription = await _stripeFacade.GetSubscription(invoice.SubscriptionId); + var (organizationId, userId, providerId) = GetIdsFromMetadata(subscription?.Metadata); + if (!organizationId.HasValue && !userId.HasValue && !providerId.HasValue) + { + _logger.LogWarning( + "Attempted to pay invoice with Braintree but Stripe subscription metadata didn't contain either a organizationId or userId or "); + return false; + } + + var orgTransaction = organizationId.HasValue; + string btObjIdField; + Guid btObjId; + if (organizationId.HasValue) + { + btObjIdField = "organization_id"; + btObjId = organizationId.Value; + } + else if (userId.HasValue) + { + btObjIdField = "user_id"; + btObjId = userId.Value; + } + else + { + btObjIdField = "provider_id"; + btObjId = providerId.Value; + } + var btInvoiceAmount = invoice.AmountDue / 100M; + + var existingTransactions = organizationId.HasValue + ? await _transactionRepository.GetManyByOrganizationIdAsync(organizationId.Value) + : userId.HasValue + ? await _transactionRepository.GetManyByUserIdAsync(userId.Value) + : await _transactionRepository.GetManyByProviderIdAsync(providerId.Value); + + var duplicateTimeSpan = TimeSpan.FromHours(24); + var now = DateTime.UtcNow; + var duplicateTransaction = existingTransactions? + .FirstOrDefault(t => (now - t.CreationDate) < duplicateTimeSpan); + if (duplicateTransaction != null) + { + _logger.LogWarning("There is already a recent PayPal transaction ({0}). " + + "Do not charge again to prevent possible duplicate.", duplicateTransaction.GatewayId); + return false; + } + + Result transactionResult; + try + { + transactionResult = await _btGateway.Transaction.SaleAsync( + new Braintree.TransactionRequest + { + Amount = btInvoiceAmount, + CustomerId = customer.Metadata["btCustomerId"], + Options = new Braintree.TransactionOptionsRequest + { + SubmitForSettlement = true, + PayPal = new Braintree.TransactionOptionsPayPalRequest + { + CustomField = + $"{btObjIdField}:{btObjId},region:{_globalSettings.BaseServiceUri.CloudRegion}" + } + }, + CustomFields = new Dictionary + { + [btObjIdField] = btObjId.ToString(), + ["region"] = _globalSettings.BaseServiceUri.CloudRegion + } + }); + } + catch (NotFoundException e) + { + _logger.LogError(e, + "Attempted to make a payment with Braintree, but customer did not exist for the given btCustomerId present on the Stripe metadata"); + throw; + } + + if (!transactionResult.IsSuccess()) + { + if (invoice.AttemptCount < 4) + { + await _mailService.SendPaymentFailedAsync(customer.Email, btInvoiceAmount, true); + } + return false; + } + + try + { + await _stripeFacade.UpdateInvoice(invoice.Id, new InvoiceUpdateOptions + { + Metadata = new Dictionary + { + ["btTransactionId"] = transactionResult.Target.Id, + ["btPayPalTransactionId"] = + transactionResult.Target.PayPalDetails?.AuthorizationId + } + }); + await _stripeFacade.PayInvoice(invoice.Id, new InvoicePayOptions { PaidOutOfBand = true }); + } + catch (Exception e) + { + await _btGateway.Transaction.RefundAsync(transactionResult.Target.Id); + if (e.Message.Contains("Invoice is already paid")) + { + await _stripeFacade.UpdateInvoice(invoice.Id, new InvoiceUpdateOptions + { + Metadata = invoice.Metadata + }); + } + else + { + throw; + } + } + + return true; + } + + private async Task AttemptToPayInvoiceWithStripeAsync(Invoice invoice) + { + try + { + await _stripeFacade.PayInvoice(invoice.Id); + return true; + } + catch (Exception e) + { + _logger.LogWarning( + e, + "Exception occurred while trying to pay Stripe invoice with Id: {InvoiceId}", + invoice.Id); + + throw; + } + } +} diff --git a/src/Billing/Services/Implementations/SubscriptionDeletedHandler.cs b/src/Billing/Services/Implementations/SubscriptionDeletedHandler.cs new file mode 100644 index 0000000000..c495e8dd79 --- /dev/null +++ b/src/Billing/Services/Implementations/SubscriptionDeletedHandler.cs @@ -0,0 +1,49 @@ +using Bit.Billing.Constants; +using Bit.Core.Services; +using Event = Stripe.Event; +namespace Bit.Billing.Services.Implementations; + +public class SubscriptionDeletedHandler : ISubscriptionDeletedHandler +{ + private readonly IStripeEventService _stripeEventService; + private readonly IOrganizationService _organizationService; + private readonly IUserService _userService; + private readonly IStripeEventUtilityService _stripeEventUtilityService; + + public SubscriptionDeletedHandler( + IStripeEventService stripeEventService, + IOrganizationService organizationService, + IUserService userService, + IStripeEventUtilityService stripeEventUtilityService) + { + _stripeEventService = stripeEventService; + _organizationService = organizationService; + _userService = userService; + _stripeEventUtilityService = stripeEventUtilityService; + } + + /// + /// Handles the event type from Stripe. + /// + /// + public async Task HandleAsync(Event parsedEvent) + { + var subscription = await _stripeEventService.GetSubscription(parsedEvent, true); + var (organizationId, userId, providerId) = _stripeEventUtilityService.GetIdsFromMetadata(subscription.Metadata); + var subCanceled = subscription.Status == StripeSubscriptionStatus.Canceled; + + if (!subCanceled) + { + return; + } + + if (organizationId.HasValue) + { + await _organizationService.DisableAsync(organizationId.Value, subscription.CurrentPeriodEnd); + } + else if (userId.HasValue) + { + await _userService.DisablePremiumAsync(userId.Value, subscription.CurrentPeriodEnd); + } + } +} diff --git a/src/Billing/Services/Implementations/SubscriptionUpdatedHandler.cs b/src/Billing/Services/Implementations/SubscriptionUpdatedHandler.cs new file mode 100644 index 0000000000..4b4c9dcf4a --- /dev/null +++ b/src/Billing/Services/Implementations/SubscriptionUpdatedHandler.cs @@ -0,0 +1,176 @@ +using Bit.Billing.Constants; +using Bit.Core.OrganizationFeatures.OrganizationSponsorships.FamiliesForEnterprise.Interfaces; +using Bit.Core.Services; +using Bit.Core.Utilities; +using Stripe; +using Event = Stripe.Event; + +namespace Bit.Billing.Services.Implementations; + +public class SubscriptionUpdatedHandler : ISubscriptionUpdatedHandler +{ + private readonly IStripeEventService _stripeEventService; + private readonly IStripeEventUtilityService _stripeEventUtilityService; + private readonly IOrganizationService _organizationService; + private readonly IStripeFacade _stripeFacade; + private readonly IOrganizationSponsorshipRenewCommand _organizationSponsorshipRenewCommand; + private readonly IUserService _userService; + + public SubscriptionUpdatedHandler( + IStripeEventService stripeEventService, + IStripeEventUtilityService stripeEventUtilityService, + IOrganizationService organizationService, + IStripeFacade stripeFacade, + IOrganizationSponsorshipRenewCommand organizationSponsorshipRenewCommand, + IUserService userService) + { + _stripeEventService = stripeEventService; + _stripeEventUtilityService = stripeEventUtilityService; + _organizationService = organizationService; + _stripeFacade = stripeFacade; + _organizationSponsorshipRenewCommand = organizationSponsorshipRenewCommand; + _userService = userService; + } + + /// + /// Handles the event type from Stripe. + /// + /// + public async Task HandleAsync(Event parsedEvent) + { + var subscription = await _stripeEventService.GetSubscription(parsedEvent, true, ["customer", "discounts"]); + var (organizationId, userId, providerId) = _stripeEventUtilityService.GetIdsFromMetadata(subscription.Metadata); + + switch (subscription.Status) + { + case StripeSubscriptionStatus.Unpaid or StripeSubscriptionStatus.IncompleteExpired + when organizationId.HasValue: + { + await _organizationService.DisableAsync(organizationId.Value, subscription.CurrentPeriodEnd); + break; + } + case StripeSubscriptionStatus.Unpaid or StripeSubscriptionStatus.IncompleteExpired: + { + if (!userId.HasValue) + { + break; + } + + if (subscription.Status is StripeSubscriptionStatus.Unpaid && + subscription.Items.Any(i => i.Price.Id is IStripeEventUtilityService.PremiumPlanId or IStripeEventUtilityService.PremiumPlanIdAppStore)) + { + await CancelSubscription(subscription.Id); + await VoidOpenInvoices(subscription.Id); + } + + await _userService.DisablePremiumAsync(userId.Value, subscription.CurrentPeriodEnd); + + break; + } + case StripeSubscriptionStatus.Active when organizationId.HasValue: + { + await _organizationService.EnableAsync(organizationId.Value); + break; + } + case StripeSubscriptionStatus.Active: + { + if (userId.HasValue) + { + await _userService.EnablePremiumAsync(userId.Value, subscription.CurrentPeriodEnd); + } + + break; + } + } + + if (organizationId.HasValue) + { + await _organizationService.UpdateExpirationDateAsync(organizationId.Value, subscription.CurrentPeriodEnd); + if (_stripeEventUtilityService.IsSponsoredSubscription(subscription)) + { + await _organizationSponsorshipRenewCommand.UpdateExpirationDateAsync(organizationId.Value, subscription.CurrentPeriodEnd); + } + + await RemovePasswordManagerCouponIfRemovingSecretsManagerTrialAsync(parsedEvent, subscription); + } + else if (userId.HasValue) + { + await _userService.UpdatePremiumExpirationAsync(userId.Value, subscription.CurrentPeriodEnd); + } + } + + private async Task CancelSubscription(string subscriptionId) => + await _stripeFacade.CancelSubscription(subscriptionId, new SubscriptionCancelOptions()); + + private async Task VoidOpenInvoices(string subscriptionId) + { + var options = new InvoiceListOptions + { + Status = StripeInvoiceStatus.Open, + Subscription = subscriptionId + }; + var invoices = await _stripeFacade.ListInvoices(options); + foreach (var invoice in invoices) + { + await _stripeFacade.VoidInvoice(invoice.Id); + } + } + + /// + /// Removes the Password Manager coupon if the organization is removing the Secrets Manager trial. + /// Only applies to organizations that have a subscription from the Secrets Manager trial. + /// + /// + /// + private async Task RemovePasswordManagerCouponIfRemovingSecretsManagerTrialAsync(Event parsedEvent, + Subscription subscription) + { + if (parsedEvent.Data.PreviousAttributes?.items is null) + { + return; + } + + var previousSubscription = parsedEvent.Data + .PreviousAttributes + .ToObject() as Subscription; + + // This being false doesn't necessarily mean that the organization doesn't subscribe to Secrets Manager. + // If there are changes to any subscription item, Stripe sends every item in the subscription, both + // changed and unchanged. + var previousSubscriptionHasSecretsManager = previousSubscription?.Items is not null && + previousSubscription.Items.Any(previousItem => + StaticStore.Plans.Any(p => + p.SecretsManager is not null && + p.SecretsManager.StripeSeatPlanId == + previousItem.Plan.Id)); + + var currentSubscriptionHasSecretsManager = subscription.Items.Any(i => + StaticStore.Plans.Any(p => + p.SecretsManager is not null && + p.SecretsManager.StripeSeatPlanId == i.Plan.Id)); + + if (!previousSubscriptionHasSecretsManager || currentSubscriptionHasSecretsManager) + { + return; + } + + var customerHasSecretsManagerTrial = subscription.Customer + ?.Discount + ?.Coupon + ?.Id == "sm-standalone"; + + var subscriptionHasSecretsManagerTrial = subscription.Discount + ?.Coupon + ?.Id == "sm-standalone"; + + if (customerHasSecretsManagerTrial) + { + await _stripeFacade.DeleteCustomerDiscount(subscription.CustomerId); + } + + if (subscriptionHasSecretsManagerTrial) + { + await _stripeFacade.DeleteSubscriptionDiscount(subscription.Id); + } + } +} diff --git a/src/Billing/Services/Implementations/UpcomingInvoiceHandler.cs b/src/Billing/Services/Implementations/UpcomingInvoiceHandler.cs new file mode 100644 index 0000000000..48b6910974 --- /dev/null +++ b/src/Billing/Services/Implementations/UpcomingInvoiceHandler.cs @@ -0,0 +1,215 @@ +using Bit.Billing.Constants; +using Bit.Core; +using Bit.Core.AdminConsole.Entities; +using Bit.Core.AdminConsole.Repositories; +using Bit.Core.Billing.Constants; +using Bit.Core.OrganizationFeatures.OrganizationSponsorships.FamiliesForEnterprise.Interfaces; +using Bit.Core.Repositories; +using Bit.Core.Services; +using Bit.Core.Utilities; +using Stripe; +using Event = Stripe.Event; +using TaxRate = Bit.Core.Entities.TaxRate; + +namespace Bit.Billing.Services.Implementations; + +public class UpcomingInvoiceHandler : IUpcomingInvoiceHandler +{ + private readonly ILogger _logger; + private readonly IStripeEventService _stripeEventService; + private readonly IUserService _userService; + private readonly IStripeFacade _stripeFacade; + private readonly IFeatureService _featureService; + private readonly IMailService _mailService; + private readonly IProviderRepository _providerRepository; + private readonly IValidateSponsorshipCommand _validateSponsorshipCommand; + private readonly IOrganizationRepository _organizationRepository; + private readonly IStripeEventUtilityService _stripeEventUtilityService; + private readonly ITaxRateRepository _taxRateRepository; + + public UpcomingInvoiceHandler( + ILogger logger, + IStripeEventService stripeEventService, + IUserService userService, + IStripeFacade stripeFacade, + IFeatureService featureService, + IMailService mailService, + IProviderRepository providerRepository, + IValidateSponsorshipCommand validateSponsorshipCommand, + IOrganizationRepository organizationRepository, + IStripeEventUtilityService stripeEventUtilityService, + ITaxRateRepository taxRateRepository) + { + _logger = logger; + _stripeEventService = stripeEventService; + _userService = userService; + _stripeFacade = stripeFacade; + _featureService = featureService; + _mailService = mailService; + _providerRepository = providerRepository; + _validateSponsorshipCommand = validateSponsorshipCommand; + _organizationRepository = organizationRepository; + _stripeEventUtilityService = stripeEventUtilityService; + _taxRateRepository = taxRateRepository; + } + + /// + /// Handles the event type from Stripe. + /// + /// + /// + public async Task HandleAsync(Event parsedEvent) + { + var invoice = await _stripeEventService.GetInvoice(parsedEvent); + if (string.IsNullOrEmpty(invoice.SubscriptionId)) + { + _logger.LogWarning("Received 'invoice.upcoming' Event with ID '{eventId}' that did not include a Subscription ID", parsedEvent.Id); + return; + } + + var subscription = await _stripeFacade.GetSubscription(invoice.SubscriptionId); + + if (subscription == null) + { + throw new Exception( + $"Received null Subscription from Stripe for ID '{invoice.SubscriptionId}' while processing Event with ID '{parsedEvent.Id}'"); + } + + var pm5766AutomaticTaxIsEnabled = _featureService.IsEnabled(FeatureFlagKeys.PM5766AutomaticTax); + if (pm5766AutomaticTaxIsEnabled) + { + var customerGetOptions = new CustomerGetOptions(); + customerGetOptions.AddExpand("tax"); + var customer = await _stripeFacade.GetCustomer(subscription.CustomerId, customerGetOptions); + if (!subscription.AutomaticTax.Enabled && + customer.Tax?.AutomaticTax == StripeConstants.AutomaticTaxStatus.Supported) + { + subscription = await _stripeFacade.UpdateSubscription(subscription.Id, + new SubscriptionUpdateOptions + { + DefaultTaxRates = [], + AutomaticTax = new SubscriptionAutomaticTaxOptions { Enabled = true } + }); + } + } + + var updatedSubscription = pm5766AutomaticTaxIsEnabled + ? subscription + : await VerifyCorrectTaxRateForChargeAsync(invoice, subscription); + + var (organizationId, userId, providerId) = _stripeEventUtilityService.GetIdsFromMetadata(updatedSubscription.Metadata); + + var invoiceLineItemDescriptions = invoice.Lines.Select(i => i.Description).ToList(); + + if (organizationId.HasValue) + { + if (_stripeEventUtilityService.IsSponsoredSubscription(updatedSubscription)) + { + await _validateSponsorshipCommand.ValidateSponsorshipAsync(organizationId.Value); + } + + var organization = await _organizationRepository.GetByIdAsync(organizationId.Value); + + if (organization == null || !OrgPlanForInvoiceNotifications(organization)) + { + return; + } + + await SendEmails(new List { organization.BillingEmail }); + + /* + * TODO: https://bitwarden.atlassian.net/browse/PM-4862 + * Disabling this as part of a hot fix. It needs to check whether the organization + * belongs to a Reseller provider and only send an email to the organization owners if it does. + * It also requires a new email template as the current one contains too much billing information. + */ + + // var ownerEmails = await _organizationRepository.GetOwnerEmailAddressesById(organization.Id); + + // await SendEmails(ownerEmails); + } + else if (userId.HasValue) + { + var user = await _userService.GetUserByIdAsync(userId.Value); + + if (user?.Premium == true) + { + await SendEmails(new List { user.Email }); + } + } + else if (providerId.HasValue) + { + var provider = await _providerRepository.GetByIdAsync(providerId.Value); + + if (provider == null) + { + _logger.LogError( + "Received invoice.Upcoming webhook ({EventID}) for Provider ({ProviderID}) that does not exist", + parsedEvent.Id, + providerId.Value); + + return; + } + + await SendEmails(new List { provider.BillingEmail }); + + } + + return; + + /* + * Sends emails to the given email addresses. + */ + async Task SendEmails(IEnumerable emails) + { + var validEmails = emails.Where(e => !string.IsNullOrEmpty(e)); + + if (invoice.NextPaymentAttempt.HasValue) + { + await _mailService.SendInvoiceUpcoming( + validEmails, + invoice.AmountDue / 100M, + invoice.NextPaymentAttempt.Value, + invoiceLineItemDescriptions, + true); + } + } + } + + private async Task VerifyCorrectTaxRateForChargeAsync(Invoice invoice, Stripe.Subscription subscription) + { + if (string.IsNullOrWhiteSpace(invoice?.CustomerAddress?.Country) || + string.IsNullOrWhiteSpace(invoice?.CustomerAddress?.PostalCode)) + { + return subscription; + } + + var localBitwardenTaxRates = await _taxRateRepository.GetByLocationAsync( + new TaxRate() + { + Country = invoice.CustomerAddress.Country, + PostalCode = invoice.CustomerAddress.PostalCode + } + ); + + if (!localBitwardenTaxRates.Any()) + { + return subscription; + } + + var stripeTaxRate = await _stripeFacade.GetTaxRate(localBitwardenTaxRates.First().Id); + if (stripeTaxRate == null || subscription.DefaultTaxRates.Any(x => x == stripeTaxRate)) + { + return subscription; + } + + subscription.DefaultTaxRates = [stripeTaxRate]; + + var subscriptionOptions = new SubscriptionUpdateOptions { DefaultTaxRates = [stripeTaxRate.Id] }; + subscription = await _stripeFacade.UpdateSubscription(subscription.Id, subscriptionOptions); + + return subscription; + } + + private static bool OrgPlanForInvoiceNotifications(Organization org) => StaticStore.GetPlan(org.PlanType).IsAnnual; +} diff --git a/src/Billing/Startup.cs b/src/Billing/Startup.cs index 1bc2789a4a..369f76a93f 100644 --- a/src/Billing/Startup.cs +++ b/src/Billing/Startup.cs @@ -52,6 +52,21 @@ public class Startup // Context services.AddScoped(); + //Handlers + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + // Identity services.AddCustomIdentityServices(globalSettings); //services.AddPasswordlessIdentityServices(globalSettings);