bitwarden/src/Core/Billing/Models/Sales/OrganizationSale.cs

using Bit.Core.AdminConsole.Entities;
using Bit.Core.Billing.Constants;
using Bit.Core.Billing.Tax.Models;
using Bit.Core.Models.Business;

namespace Bit.Core.Billing.Models.Sales;

#nullable enable

public class OrganizationSale
{
    private OrganizationSale() { }

    public void Deconstruct(
        out Organization organization,
        out CustomerSetup? customerSetup,
        out SubscriptionSetup subscriptionSetup)
    {
        organization = Organization;
        customerSetup = CustomerSetup;
        subscriptionSetup = SubscriptionSetup;
    }

    public required Organization Organization { get; init; }
    public CustomerSetup? CustomerSetup { get; init; }
    public required SubscriptionSetup SubscriptionSetup { get; init; }

    public static OrganizationSale From(
        Organization organization,
        OrganizationSignup signup)
    {
        var customerSetup = string.IsNullOrEmpty(organization.GatewayCustomerId) ? GetCustomerSetup(signup) : null;

        var subscriptionSetup = GetSubscriptionSetup(signup);

        subscriptionSetup.SkipTrial = signup.SkipTrial;

        return new OrganizationSale
        {
            Organization = organization,
            CustomerSetup = customerSetup,
            SubscriptionSetup = subscriptionSetup
        };
    }

    public static OrganizationSale From(
        Organization organization,
        OrganizationUpgrade upgrade) => new()
        {
            Organization = organization,
            SubscriptionSetup = GetSubscriptionSetup(upgrade)
        };

    private static CustomerSetup GetCustomerSetup(OrganizationSignup signup)
    {
        var customerSetup = new CustomerSetup
        {
            Coupon = signup.IsFromProvider
            // TODO: Remove when last of the legacy providers has been migrated.
            ? StripeConstants.CouponIDs.LegacyMSPDiscount
            : signup.IsFromSecretsManagerTrial
                ? StripeConstants.CouponIDs.SecretsManagerStandalone
                : null
        };

        if (!signup.PaymentMethodType.HasValue)
        {
            return customerSetup;
        }

        customerSetup.TokenizedPaymentSource = new TokenizedPaymentSource(
            signup.PaymentMethodType!.Value,
            signup.PaymentToken);

        customerSetup.TaxInformation = new TaxInformation(
            signup.TaxInfo.BillingAddressCountry,
            signup.TaxInfo.BillingAddressPostalCode,
            signup.TaxInfo.TaxIdNumber,
            signup.TaxInfo.TaxIdType,
            signup.TaxInfo.BillingAddressLine1,
            signup.TaxInfo.BillingAddressLine2,
            signup.TaxInfo.BillingAddressCity,
            signup.TaxInfo.BillingAddressState);

        return customerSetup;
    }

    private static CustomerSetup GetCustomerSetup(OrganizationUpgrade upgrade)
    {
        return new CustomerSetup
        {
            TaxInformation = new TaxInformation(
                upgrade.TaxInfo.BillingAddressCountry,
                upgrade.TaxInfo.BillingAddressPostalCode,
                upgrade.TaxInfo.TaxIdNumber,
                upgrade.TaxInfo.TaxIdType,
                upgrade.TaxInfo.BillingAddressLine1,
                upgrade.TaxInfo.BillingAddressLine2,
                upgrade.TaxInfo.BillingAddressCity,
                upgrade.TaxInfo.BillingAddressState)
        };
    }

    private static SubscriptionSetup GetSubscriptionSetup(OrganizationSignup signup)
    {
        var passwordManagerOptions = new SubscriptionSetup.PasswordManager
        {
            Seats = signup.AdditionalSeats,
            Storage = signup.AdditionalStorageGb,
            PremiumAccess = signup.PremiumAccessAddon
        };

        var secretsManagerOptions = signup.UseSecretsManager
            ? new SubscriptionSetup.SecretsManager
            {
                Seats = signup.AdditionalSmSeats ?? 0,
                ServiceAccounts = signup.AdditionalServiceAccounts
            }
            : null;

        var subscriptionSetup = new SubscriptionSetup
        {
            PlanType = signup.Plan,
            PasswordManagerOptions = passwordManagerOptions,
            SecretsManagerOptions = secretsManagerOptions,
            SkipTrial = signup.SkipTrial
        };

        // Set trial source based on initiation path
        if (!string.IsNullOrEmpty(signup.InitiationPath))
        {
            subscriptionSetup.TrialSource = signup.InitiationPath.Contains("trial from marketing website")
                ? "marketing-initiated"
                : "product-initiated";
        }

        return subscriptionSetup;
    }

    private static SubscriptionSetup GetSubscriptionSetup(OrganizationUpgrade upgrade)
    {
        var passwordManagerOptions = new SubscriptionSetup.PasswordManager
        {
            Seats = upgrade.AdditionalSeats,
            Storage = upgrade.AdditionalStorageGb,
            PremiumAccess = upgrade.PremiumAccessAddon
        };

        var secretsManagerOptions = upgrade.UseSecretsManager
            ? new SubscriptionSetup.SecretsManager
            {
                Seats = upgrade.AdditionalSmSeats ?? 0,
                ServiceAccounts = upgrade.AdditionalServiceAccounts
            }
            : null;

        return new SubscriptionSetup
        {
            PlanType = upgrade.Plan,
            PasswordManagerOptions = passwordManagerOptions,
            SecretsManagerOptions = secretsManagerOptions,
            SkipTrial = true // Upgrades should always skip trial
        };
    }
}