combined tax updates with other operations

src/Api/Controllers/AccountsController.cs

@@ -467,6 +467,10 @@ namespace Bit.Api.Controllers
             {
                 license = await ApiHelpers.ReadJsonFileFromBody<UserLicense>(HttpContext, model.License);
             }
+            else if (!valid && !_globalSettings.SelfHosted)
+            {
+                throw new BadRequestException("Country is required.");
+            }
 
             if (!valid || (_globalSettings.SelfHosted && license == null))
             {
@@ -474,7 +478,8 @@ namespace Bit.Api.Controllers
             }
 
             var result = await _userService.SignUpPremiumAsync(user, model.PaymentToken,
-                model.PaymentMethodType.Value, model.AdditionalStorageGb.GetValueOrDefault(0), license);
+                model.PaymentMethodType.Value, model.AdditionalStorageGb.GetValueOrDefault(0), license,
+                model.Country, model.PostalCode);
             var profile = new ProfileResponseModel(user, null, await _userService.TwoFactorIsEnabledAsync(user));
             return new PaymentResponseModel
             {
@@ -534,6 +539,11 @@ namespace Bit.Api.Controllers
                 throw new UnauthorizedAccessException();
             }
 
+            await _paymentService.SaveTaxInfoAsync(user, new TaxInfo
+            {
+                BillingAddressCountry = model.Country,
+                BillingAddressPostalCode = model.PostalCode,
+            });
             await _userService.ReplacePaymentMethodAsync(user, model.PaymentToken, model.PaymentMethodType.Value);
         }
 

src/Api/Controllers/OrganizationsController.cs

@@ -208,9 +208,18 @@ namespace Bit.Api.Controllers
             {
                 throw new NotFoundException();
             }
-
+            
             await _organizationService.ReplacePaymentMethodAsync(orgIdGuid, model.PaymentToken,
-                model.PaymentMethodType.Value);
+                model.PaymentMethodType.Value, new TaxInfo
+                {
+                    BillingAddressLine1 = model.Line1,
+                    BillingAddressLine2 = model.Line2,
+                    BillingAddressState = model.State,
+                    BillingAddressCity = model.City,
+                    BillingAddressPostalCode = model.PostalCode,
+                    BillingAddressCountry = model.Country,
+                    TaxIdNumber = model.TaxId,
+                });
         }
 
         [HttpPost("{id}/upgrade")]

src/Core/Models/Api/Request/Accounts/PremiumRequestModel.cs

@@ -13,11 +13,17 @@ namespace Bit.Core.Models.Api
         [Range(0, 99)]
         public short? AdditionalStorageGb { get; set; }
         public IFormFile License { get; set; }
+        public string Country { get; set; }
+        public string PostalCode { get; set; }
 
         public bool Validate(GlobalSettings globalSettings)
         {
-            return (License == null && !globalSettings.SelfHosted) ||
+            if (!(License == null && !globalSettings.SelfHosted) ||
-                (License != null && globalSettings.SelfHosted);
+                (License != null && globalSettings.SelfHosted))
+            {
+                return false;
+            }
+            return globalSettings.SelfHosted || !string.IsNullOrWhiteSpace(Country);
         }
 
         public IEnumerable<ValidationResult> Validate(ValidationContext validationContext)
@@ -27,6 +33,11 @@ namespace Bit.Core.Models.Api
             {
                 yield return new ValidationResult("Payment token or license is required.");
             }
+            if (Country == "US" && string.IsNullOrWhiteSpace(PostalCode))
+            {
+                yield return new ValidationResult("Zip / postal code is required.",
+                    new string[] { nameof(PostalCode) });
+            }
         }
     }
 }

src/Core/Models/Api/Request/PaymentRequestModel.cs

@@ -3,7 +3,7 @@ using Bit.Core.Enums;
 
 namespace Bit.Core.Models.Api
 {
-    public class PaymentRequestModel
+    public class PaymentRequestModel : OrganizationTaxInfoUpdateRequestModel
     {
         [Required]
         public PaymentMethodType? PaymentMethodType { get; set; }

src/Core/Services/IOrganizationService.cs

@@ -10,7 +10,8 @@ namespace Bit.Core.Services
 {
     public interface IOrganizationService
     {
-        Task ReplacePaymentMethodAsync(Guid organizationId, string paymentToken, PaymentMethodType paymentMethodType);
+        Task ReplacePaymentMethodAsync(Guid organizationId, string paymentToken, PaymentMethodType paymentMethodType,
+            TaxInfo taxInfo);
         Task CancelSubscriptionAsync(Guid organizationId, bool? endOfPeriod = null);
         Task ReinstateSubscriptionAsync(Guid organizationId);
         Task<Tuple<bool, string>> UpgradePlanAsync(Guid organizationId, OrganizationUpgrade upgrade);

src/Core/Services/IPaymentService.cs

@@ -14,7 +14,7 @@ namespace Bit.Core.Services
         Task<string> UpgradeFreeOrganizationAsync(Organization org, Models.StaticStore.Plan plan,
            short additionalStorageGb, short additionalSeats, bool premiumAccessAddon);
         Task<string> PurchasePremiumAsync(User user, PaymentMethodType paymentMethodType, string paymentToken,
-            short additionalStorageGb);
+            short additionalStorageGb, string country, string postalCode);
         Task<string> AdjustStorageAsync(IStorableSubscriber storableSubscriber, int additionalStorage, string storagePlanId);
         Task CancelSubscriptionAsync(ISubscriber subscriber, bool endOfPeriod = false,
             bool skipInAppPurchaseCheck = false);

src/Core/Services/IUserService.cs

@@ -46,7 +46,8 @@ namespace Bit.Core.Services
         Task<IdentityResult> DeleteAsync(User user, string token);
         Task SendDeleteConfirmationAsync(string email);
         Task<Tuple<bool, string>> SignUpPremiumAsync(User user, string paymentToken,
-            PaymentMethodType paymentMethodType, short additionalStorageGb, UserLicense license);
+            PaymentMethodType paymentMethodType, short additionalStorageGb, UserLicense license,
+            string country, string postalCode);
         Task IapCheckAsync(User user, PaymentMethodType paymentMethodType);
         Task UpdateLicenseAsync(User user, UserLicense license);
         Task<string> AdjustStorageAsync(User user, short storageAdjustmentGb);

src/Core/Services/Implementations/OrganizationService.cs

@@ -75,7 +75,7 @@ namespace Bit.Core.Services
         }
 
         public async Task ReplacePaymentMethodAsync(Guid organizationId, string paymentToken,
-            PaymentMethodType paymentMethodType)
+            PaymentMethodType paymentMethodType, TaxInfo taxInfo)
         {
             var organization = await GetOrgById(organizationId);
             if (organization == null)
@@ -83,6 +83,7 @@ namespace Bit.Core.Services
                 throw new NotFoundException();
             }
 
+            await _paymentService.SaveTaxInfoAsync(organization, taxInfo);
             var updated = await _paymentService.UpdatePaymentMethodAsync(organization,
                 paymentMethodType, paymentToken);
             if (updated)

src/Core/Services/Implementations/StripePaymentService.cs

@@ -345,7 +345,7 @@ namespace Bit.Core.Services
         }
 
         public async Task<string> PurchasePremiumAsync(User user, PaymentMethodType paymentMethodType,
-            string paymentToken, short additionalStorageGb)
+            string paymentToken, short additionalStorageGb, string country, string postalCode)
         {
             if (paymentMethodType != PaymentMethodType.Credit && string.IsNullOrWhiteSpace(paymentToken))
             {
@@ -463,10 +463,28 @@ namespace Bit.Core.Services
                     InvoiceSettings = new CustomerInvoiceSettingsOptions
                     {
                         DefaultPaymentMethod = stipeCustomerPaymentMethodId
-                    }
+                    },
+                    Address = new AddressOptions
+                    {
+                        Line1 = string.Empty,
+                        Country = country,
+                        PostalCode = postalCode,
+                    },
                 });
                 createdStripeCustomer = true;
             }
+            else if (customer != null)
+            {
+                await customerService.UpdateAsync(customer.Id, new CustomerUpdateOptions
+                {
+                    Address = new AddressOptions
+                    {
+                        Line1 = string.Empty,
+                        Country = country,
+                        PostalCode = postalCode,
+                    }
+                });
+            }
 
             if (customer == null)
             {

src/Core/Services/Implementations/UserService.cs

@@ -703,7 +703,8 @@ namespace Bit.Core.Services
         }
 
         public async Task<Tuple<bool, string>> SignUpPremiumAsync(User user, string paymentToken,
-            PaymentMethodType paymentMethodType, short additionalStorageGb, UserLicense license)
+            PaymentMethodType paymentMethodType, short additionalStorageGb, UserLicense license,
+            string country, string postalCode)
         {
             if (user.Premium)
             {
@@ -742,7 +743,7 @@ namespace Bit.Core.Services
             else
             {
                 paymentIntentClientSecret = await _paymentService.PurchasePremiumAsync(user, paymentMethodType,
-                    paymentToken, additionalStorageGb);
+                    paymentToken, additionalStorageGb, country, postalCode);
             }
 
             user.Premium = true;