[PM-31781] skip unpaid automations for exempt orgs

src/Billing/Services/Implementations/InvoiceCreatedHandler.cs

@@ -40,8 +40,8 @@ public async Task HandleAsync(Event parsedEvent)
                     Status: not StripeConstants.InvoiceStatus.Paid,
                     CollectionMethod: "charge_automatically",
                     BillingReason:
-                    "subscription_cycle" or
-                    "automatic_pending_invoice_item_invoice",
+                    StripeConstants.BillingReasons.SubscriptionCycle or
+                    StripeConstants.BillingReasons.AutomaticPendingInvoiceItemInvoice,
                     Parent.SubscriptionDetails.Subscription: not null
                 })
             {

src/Billing/Services/Implementations/PaymentFailedHandler.cs

@@ -50,7 +50,7 @@ invoice is
             AmountDue: > 0,
             Status: not StripeConstants.InvoiceStatus.Paid,
             CollectionMethod: "charge_automatically",
-            BillingReason: "subscription_cycle" or "automatic_pending_invoice_item_invoice",
+            BillingReason: StripeConstants.BillingReasons.SubscriptionCycle or StripeConstants.BillingReasons.AutomaticPendingInvoiceItemInvoice,
             Parent.SubscriptionDetails: not null
         };
 }

src/Billing/Services/Implementations/StripeEventUtilityService.cs

@@ -258,7 +258,7 @@ invoice is
             AmountDue: > 0,
             Status: not StripeConstants.InvoiceStatus.Paid,
             CollectionMethod: "charge_automatically",
-            BillingReason: "subscription_cycle" or "automatic_pending_invoice_item_invoice",
+            BillingReason: StripeConstants.BillingReasons.SubscriptionCycle or StripeConstants.BillingReasons.AutomaticPendingInvoiceItemInvoice,
             Parent.SubscriptionDetails: not null
         };
 

src/Billing/Services/Implementations/SubscriptionUpdatedHandler.cs

@@ -85,8 +85,21 @@ public async Task HandleAsync(Event parsedEvent)
 
         var currentPeriodEnd = subscription.GetCurrentPeriodEnd();
 
-        if (SubscriptionWentUnpaid(parsedEvent, subscription) ||
-            SubscriptionWentIncompleteExpired(parsedEvent, subscription))
+        var clearOrgBillingAutomationExemption = false;
+
+        if (SubscriptionWentUnpaid(parsedEvent, subscription))
+        {
+            if (await SkipUnpaidBillingAutomationsForExemptOrganizationAsync(subscriberId))
+            {
+                clearOrgBillingAutomationExemption = true;
+            }
+            else
+            {
+                await DisableSubscriberAsync(subscriberId, currentPeriodEnd);
+                await SetSubscriptionToCancelAsync(subscription);
+            }
+        }
+        else if (SubscriptionWentIncompleteExpired(parsedEvent, subscription))
         {
             await DisableSubscriberAsync(subscriberId, currentPeriodEnd);
             await SetSubscriptionToCancelAsync(subscription);
@@ -114,6 +127,11 @@ await subscriberId.Match(
                 }
 
                 await RemovePasswordManagerCouponIfRemovingSecretsManagerTrialAsync(parsedEvent, subscription);
+
+                if (clearOrgBillingAutomationExemption)
+                {
+                    await ClearBillingAutomationExemptionAsync(organizationId.Value);
+                }
             },
             _ => Task.CompletedTask);
     }
@@ -130,7 +148,7 @@ SubscriptionStatus.Active or
         } && currentSubscription is
         {
             Status: SubscriptionStatus.Unpaid,
-            LatestInvoice.BillingReason: BillingReasons.SubscriptionCreate or BillingReasons.SubscriptionCycle
+            LatestInvoice.BillingReason: BillingReasons.SubscriptionCycle
         };
 
     private static bool SubscriptionWentIncompleteExpired(
@@ -189,6 +207,44 @@ private Task DisableSubscriberAsync(SubscriberId subscriberId, DateTime? current
                 }
             });
 
+    private Task<bool> SkipUnpaidBillingAutomationsForExemptOrganizationAsync(SubscriberId subscriberId) =>
+        subscriberId.Match(
+            _ => Task.FromResult(false),
+            async organizationId =>
+            {
+                var organization = await _organizationRepository.GetByIdAsync(organizationId.Value);
+
+                if (organization is not { Enabled: true, ExemptFromBillingAutomation: true })
+                {
+                    return false;
+                }
+
+                _logger.LogInformation(
+                    "Skipping billing automations for exempt organization ({OrganizationId}). Exemption will be cleared after handler completion",
+                    organizationId.Value);
+
+                return true;
+            },
+            _ => Task.FromResult(false));
+
+    private async Task ClearBillingAutomationExemptionAsync(Guid organizationId)
+    {
+        var organization = await _organizationRepository.GetByIdAsync(organizationId);
+
+        if (organization == null)
+        {
+            return;
+        }
+
+        organization.ExemptFromBillingAutomation = false;
+        organization.RevisionDate = DateTime.UtcNow;
+        await _organizationRepository.ReplaceAsync(organization);
+
+        _logger.LogInformation(
+            "Exemption has been cleared for organization ({OrganizationId})",
+            organizationId);
+    }
+
     private Task EnableSubscriberAsync(SubscriberId subscriberId, DateTime? currentPeriodEnd) =>
         subscriberId.Match(
             async userId =>

src/Core/Billing/Constants/StripeConstants.cs

@@ -14,8 +14,10 @@ public static class AutomaticTaxStatus
 
     public static class BillingReasons
     {
+        public const string AutomaticPendingInvoiceItemInvoice = "automatic_pending_invoice_item_invoice";
         public const string SubscriptionCreate = "subscription_create";
         public const string SubscriptionCycle = "subscription_cycle";
+        public const string SubscriptionUpdate = "subscription_update";
     }
 
     public static class CollectionMethod