[AC-1608] Send offboarding survey response to Stripe on subscription cancellation (#3734)

* Added offboarding survey response to cancellation when FF is on.

* Removed service methods to prevent unnecessary upstream registrations

* Forgot to actually remove the injected command in the services

* Rui's feedback

* Add missing summary

* Missed [FromBody]

src/Api/AdminConsole/Controllers/OrganizationsController.cs

@@ -18,6 +18,9 @@ using Bit.Core.AdminConsole.Repositories;
 using Bit.Core.Auth.Enums;
 using Bit.Core.Auth.Repositories;
 using Bit.Core.Auth.Services;
+using Bit.Core.Billing.Commands;
+using Bit.Core.Billing.Models;
+using Bit.Core.Billing.Queries;
 using Bit.Core.Context;
 using Bit.Core.Enums;
 using Bit.Core.Exceptions;
@@ -27,6 +30,9 @@ using Bit.Core.OrganizationFeatures.OrganizationSubscriptions.Interface;
 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 Microsoft.AspNetCore.Authorization;
 using Microsoft.AspNetCore.Mvc;
@@ -58,6 +64,9 @@ public class OrganizationsController : Controller
     private readonly IUpgradeOrganizationPlanCommand _upgradeOrganizationPlanCommand;
     private readonly IAddSecretsManagerSubscriptionCommand _addSecretsManagerSubscriptionCommand;
     private readonly IPushNotificationService _pushNotificationService;
+    private readonly ICancelSubscriptionCommand _cancelSubscriptionCommand;
+    private readonly IGetSubscriptionQuery _getSubscriptionQuery;
+    private readonly IReferenceEventService _referenceEventService;
 
     public OrganizationsController(
         IOrganizationRepository organizationRepository,
@@ -80,7 +89,10 @@ public class OrganizationsController : Controller
         IUpdateSecretsManagerSubscriptionCommand updateSecretsManagerSubscriptionCommand,
         IUpgradeOrganizationPlanCommand upgradeOrganizationPlanCommand,
         IAddSecretsManagerSubscriptionCommand addSecretsManagerSubscriptionCommand,
-        IPushNotificationService pushNotificationService)
+        IPushNotificationService pushNotificationService,
+        ICancelSubscriptionCommand cancelSubscriptionCommand,
+        IGetSubscriptionQuery getSubscriptionQuery,
+        IReferenceEventService referenceEventService)
     {
         _organizationRepository = organizationRepository;
         _organizationUserRepository = organizationUserRepository;
@@ -103,6 +115,9 @@ public class OrganizationsController : Controller
         _upgradeOrganizationPlanCommand = upgradeOrganizationPlanCommand;
         _addSecretsManagerSubscriptionCommand = addSecretsManagerSubscriptionCommand;
         _pushNotificationService = pushNotificationService;
+        _cancelSubscriptionCommand = cancelSubscriptionCommand;
+        _getSubscriptionQuery = getSubscriptionQuery;
+        _referenceEventService = referenceEventService;
     }
 
     [HttpGet("{id}")]
@@ -447,15 +462,48 @@ public class OrganizationsController : Controller
 
     [HttpPost("{id}/cancel")]
     [SelfHosted(NotSelfHostedOnly = true)]
-    public async Task PostCancel(string id)
+    public async Task PostCancel(Guid id, [FromBody] SubscriptionCancellationRequestModel request)
     {
-        var orgIdGuid = new Guid(id);
-        if (!await _currentContext.EditSubscription(orgIdGuid))
+        if (!await _currentContext.EditSubscription(id))
         {
             throw new NotFoundException();
         }
 
-        await _organizationService.CancelSubscriptionAsync(orgIdGuid);
+        var presentUserWithOffboardingSurvey =
+            _featureService.IsEnabled(FeatureFlagKeys.AC1607_PresentUsersWithOffboardingSurvey);
+
+        if (presentUserWithOffboardingSurvey)
+        {
+            var organization = await _organizationRepository.GetByIdAsync(id);
+
+            if (organization == null)
+            {
+                throw new NotFoundException();
+            }
+
+            var subscription = await _getSubscriptionQuery.GetSubscription(organization);
+
+            await _cancelSubscriptionCommand.CancelSubscription(subscription,
+                new OffboardingSurveyResponse
+                {
+                    UserId = _currentContext.UserId!.Value,
+                    Reason = request.Reason,
+                    Feedback = request.Feedback
+                },
+                organization.IsExpired());
+
+            await _referenceEventService.RaiseEventAsync(new ReferenceEvent(
+                ReferenceEventType.CancelSubscription,
+                organization,
+                _currentContext)
+            {
+                EndOfPeriod = organization.IsExpired()
+            });
+        }
+        else
+        {
+            await _organizationService.CancelSubscriptionAsync(id);
+        }
     }
 
     [HttpPost("{id}/reinstate")]

src/Api/Auth/Controllers/AccountsController.cs

@@ -21,6 +21,10 @@ using Bit.Core.Auth.Services;
 using Bit.Core.Auth.UserFeatures.UserKey;
 using Bit.Core.Auth.UserFeatures.UserMasterPassword.Interfaces;
 using Bit.Core.Auth.Utilities;
+using Bit.Core.Billing.Commands;
+using Bit.Core.Billing.Models;
+using Bit.Core.Billing.Queries;
+using Bit.Core.Context;
 using Bit.Core.Entities;
 using Bit.Core.Enums;
 using Bit.Core.Exceptions;
@@ -31,6 +35,8 @@ using Bit.Core.Repositories;
 using Bit.Core.Services;
 using Bit.Core.Settings;
 using Bit.Core.Tools.Entities;
+using Bit.Core.Tools.Enums;
+using Bit.Core.Tools.Models.Business;
 using Bit.Core.Tools.Repositories;
 using Bit.Core.Tools.Services;
 using Bit.Core.Utilities;
@@ -62,6 +68,10 @@ public class AccountsController : Controller
     private readonly ISetInitialMasterPasswordCommand _setInitialMasterPasswordCommand;
     private readonly IRotateUserKeyCommand _rotateUserKeyCommand;
     private readonly IFeatureService _featureService;
+    private readonly ICancelSubscriptionCommand _cancelSubscriptionCommand;
+    private readonly IGetSubscriptionQuery _getSubscriptionQuery;
+    private readonly IReferenceEventService _referenceEventService;
+    private readonly ICurrentContext _currentContext;
 
     private bool UseFlexibleCollections =>
         _featureService.IsEnabled(FeatureFlagKeys.FlexibleCollections);
@@ -93,6 +103,10 @@ public class AccountsController : Controller
         ISetInitialMasterPasswordCommand setInitialMasterPasswordCommand,
         IRotateUserKeyCommand rotateUserKeyCommand,
         IFeatureService featureService,
+        ICancelSubscriptionCommand cancelSubscriptionCommand,
+        IGetSubscriptionQuery getSubscriptionQuery,
+        IReferenceEventService referenceEventService,
+        ICurrentContext currentContext,
         IRotationValidator<IEnumerable<CipherWithIdRequestModel>, IEnumerable<Cipher>> cipherValidator,
         IRotationValidator<IEnumerable<FolderWithIdRequestModel>, IEnumerable<Folder>> folderValidator,
         IRotationValidator<IEnumerable<SendWithIdRequestModel>, IReadOnlyList<Send>> sendValidator,
@@ -118,6 +132,10 @@ public class AccountsController : Controller
         _setInitialMasterPasswordCommand = setInitialMasterPasswordCommand;
         _rotateUserKeyCommand = rotateUserKeyCommand;
         _featureService = featureService;
+        _cancelSubscriptionCommand = cancelSubscriptionCommand;
+        _getSubscriptionQuery = getSubscriptionQuery;
+        _referenceEventService = referenceEventService;
+        _currentContext = currentContext;
         _cipherValidator = cipherValidator;
         _folderValidator = folderValidator;
         _sendValidator = sendValidator;
@@ -805,15 +823,43 @@ public class AccountsController : Controller
 
     [HttpPost("cancel-premium")]
     [SelfHosted(NotSelfHostedOnly = true)]
-    public async Task PostCancel()
+    public async Task PostCancel([FromBody] SubscriptionCancellationRequestModel request)
     {
         var user = await _userService.GetUserByPrincipalAsync(User);
+
         if (user == null)
         {
             throw new UnauthorizedAccessException();
         }
 
-        await _userService.CancelPremiumAsync(user);
+        var presentUserWithOffboardingSurvey =
+            _featureService.IsEnabled(FeatureFlagKeys.AC1607_PresentUsersWithOffboardingSurvey);
+
+        if (presentUserWithOffboardingSurvey)
+        {
+            var subscription = await _getSubscriptionQuery.GetSubscription(user);
+
+            await _cancelSubscriptionCommand.CancelSubscription(subscription,
+                new OffboardingSurveyResponse
+                {
+                    UserId = user.Id,
+                    Reason = request.Reason,
+                    Feedback = request.Feedback
+                },
+                user.IsExpired());
+
+            await _referenceEventService.RaiseEventAsync(new ReferenceEvent(
+                ReferenceEventType.CancelSubscription,
+                user,
+                _currentContext)
+            {
+                EndOfPeriod = user.IsExpired()
+            });
+        }
+        else
+        {
+            await _userService.CancelPremiumAsync(user);
+        }
     }
 
     [HttpPost("reinstate-premium")]

src/Api/Models/Request/SubscriptionCancellationRequestModel.cs

@@ -0,0 +1,7 @@
+namespace Bit.Api.Models.Request;
+
+public class SubscriptionCancellationRequestModel
+{
+    public string Reason { get; set; }
+    public string Feedback { get; set; }
+}

src/Api/Startup.cs

@@ -171,6 +171,7 @@ public class Startup
         services.AddOrganizationSubscriptionServices();
         services.AddCoreLocalizationServices();
         services.AddBillingCommands();
+        services.AddBillingQueries();
 
         // Authorization Handlers
         services.AddAuthorizationHandlers();