OrganizationEditModel.cs

// FIXME: Update this file to be null safe and then delete the line below
#nullable disable

using System.ComponentModel.DataAnnotations;
using System.Net;
using Bit.Core.AdminConsole.Entities;
using Bit.Core.AdminConsole.Entities.Provider;
using Bit.Core.AdminConsole.Enums.Provider;
using Bit.Core.Billing.Enums;
using Bit.Core.Billing.Models;
using Bit.Core.Entities;
using Bit.Core.Enums;
using Bit.Core.Models.Data.Organizations.OrganizationUsers;
using Bit.Core.Models.StaticStore;
using Bit.Core.Settings;
using Bit.Core.Utilities;
using Bit.Core.Vault.Entities;
using Bit.SharedWeb.Utilities;

namespace Bit.Admin.AdminConsole.Models;

public class OrganizationEditModel : OrganizationViewModel
{
    private readonly List<Plan> _plans;

    public OrganizationEditModel() { }

    public OrganizationEditModel(Provider provider, List<Plan> plans)
    {
        Provider = provider;
        BillingEmail = provider.Type == ProviderType.Reseller ? provider.BillingEmail : string.Empty;
        PlanType = Core.Billing.Enums.PlanType.TeamsMonthly;
        Plan = Core.Billing.Enums.PlanType.TeamsMonthly.GetDisplayAttribute()?.GetName();
        LicenseKey = RandomLicenseKey;
        _plans = plans;
    }

    public OrganizationEditModel(
        Organization org,
        Provider provider,
        IEnumerable<OrganizationUserUserDetails> orgUsers,
        IEnumerable<Cipher> ciphers,
        IEnumerable<Collection> collections,
        IEnumerable<Group> groups,
        IEnumerable<Policy> policies,
        BillingInfo billingInfo,
        BillingHistoryInfo billingHistoryInfo,
        IEnumerable<OrganizationConnection> connections,
        GlobalSettings globalSettings,
        List<Plan> plans,
        int secrets,
        int projects,
        int serviceAccounts,
        int occupiedSmSeats)
        : base(
            org,
            provider,
            connections,
            orgUsers,
            ciphers,
            collections,
            groups,
            policies,
            secrets,
            projects,
            serviceAccounts,
            occupiedSmSeats)
    {
        BillingInfo = billingInfo;
        BillingHistoryInfo = billingHistoryInfo;
        BraintreeMerchantId = globalSettings.Braintree.MerchantId;

        Name = org.DisplayName();
        BillingEmail = provider?.Type == ProviderType.Reseller ? provider.BillingEmail : org.BillingEmail;
        PlanType = org.PlanType;
        Plan = org.Plan;
        Seats = org.Seats;
        MaxAutoscaleSeats = org.MaxAutoscaleSeats;
        MaxCollections = org.MaxCollections;
        UsePolicies = org.UsePolicies;
        UseSso = org.UseSso;
        UseKeyConnector = org.UseKeyConnector;
        UseScim = org.UseScim;
        UseGroups = org.UseGroups;
        UseDirectory = org.UseDirectory;
        UseEvents = org.UseEvents;
        UseTotp = org.UseTotp;
        Use2fa = org.Use2fa;
        UseApi = org.UseApi;
        UseSecretsManager = org.UseSecretsManager;
        UseRiskInsights = org.UseRiskInsights;
        UseAdminSponsoredFamilies = org.UseAdminSponsoredFamilies;
        UseResetPassword = org.UseResetPassword;
        SelfHost = org.SelfHost;
        UsersGetPremium = org.UsersGetPremium;
        UseCustomPermissions = org.UseCustomPermissions;
        MaxStorageGb = org.MaxStorageGb;
        Gateway = org.Gateway;
        GatewayCustomerId = org.GatewayCustomerId;
        GatewaySubscriptionId = org.GatewaySubscriptionId;
        Enabled = org.Enabled;
        LicenseKey = org.LicenseKey;
        ExpirationDate = org.ExpirationDate;
        SmSeats = org.SmSeats;
        MaxAutoscaleSmSeats = org.MaxAutoscaleSmSeats;
        SmServiceAccounts = org.SmServiceAccounts;
        MaxAutoscaleSmServiceAccounts = org.MaxAutoscaleSmServiceAccounts;
        UseOrganizationDomains = org.UseOrganizationDomains;
        UseAutomaticUserConfirmation = org.UseAutomaticUserConfirmation;
        UseDisableSmAdsForUsers = org.UseDisableSmAdsForUsers;
        UsePhishingBlocker = org.UsePhishingBlocker;

        _plans = plans;
    }

    public BillingInfo BillingInfo { get; set; }
    public BillingHistoryInfo BillingHistoryInfo { get; set; }
    public string RandomLicenseKey => CoreHelpers.SecureRandomString(20);
    public string FourteenDayExpirationDate => DateTime.Now.AddDays(14).ToString("yyyy-MM-ddTHH:mm");
    public string BraintreeMerchantId { get; set; }

    [Required]
    [Display(Name = "Organization Name")]
    public string Name { get; set; }
    [Display(Name = "Billing Email")]
    public string BillingEmail { get; set; }
    [Required]
    [Display(Name = "Plan")]
    public PlanType? PlanType { get; set; }
    [Required]
    [Display(Name = "Plan Name")]
    public string Plan { get; set; }
    [Display(Name = "Seats")]
    public int? Seats { get; set; }
    [Display(Name = "Max. Autoscale Seats")]
    public int? MaxAutoscaleSeats { get; set; }
    [Display(Name = "Max. Collections")]
    public short? MaxCollections { get; set; }
    [Display(Name = "Policies")]
    public bool UsePolicies { get; set; }
    [Display(Name = "SSO")]
    public bool UseSso { get; set; }
    [Display(Name = "Key Connector with Customer Encryption")]
    public bool UseKeyConnector { get; set; }
    [Display(Name = "Groups")]
    public bool UseGroups { get; set; }
    [Display(Name = "Directory")]
    public bool UseDirectory { get; set; }
    [Display(Name = "Events")]
    public bool UseEvents { get; set; }
    [Display(Name = "TOTP")]
    public bool UseTotp { get; set; }
    [Display(Name = "2FA")]
    public bool Use2fa { get; set; }
    [Display(Name = "API")]
    public bool UseApi { get; set; }
    [Display(Name = "Reset Password")]
    public bool UseResetPassword { get; set; }
    [Display(Name = "SCIM")]
    public bool UseScim { get; set; }
    [Display(Name = "Secrets Manager")]
    public new bool UseSecretsManager { get; set; }
    [Display(Name = "Risk Insights")]
    public new bool UseRiskInsights { get; set; }
    [Display(Name = "Phishing Blocker")]
    public new bool UsePhishingBlocker { get; set; }
    [Display(Name = "Admin Sponsored Families")]
    public bool UseAdminSponsoredFamilies { get; set; }
    [Display(Name = "Self Host")]
    public bool SelfHost { get; set; }
    [Display(Name = "Users Get Premium")]
    public bool UsersGetPremium { get; set; }
    [Display(Name = "Custom Permissions")]
    public bool UseCustomPermissions { get; set; }
    [Display(Name = "Max. Storage GB")]
    public short? MaxStorageGb { get; set; }
    [Display(Name = "Gateway")]
    public GatewayType? Gateway { get; set; }
    [Display(Name = "Gateway Customer Id")]
    public string GatewayCustomerId { get; set; }
    [Display(Name = "Gateway Subscription Id")]
    public string GatewaySubscriptionId { get; set; }
    [Display(Name = "Enabled")]
    public bool Enabled { get; set; }
    [Display(Name = "License Key")]
    public string LicenseKey { get; set; }
    [Display(Name = "Expiration Date")]
    public DateTime? ExpirationDate { get; set; }
    public bool SalesAssistedTrialStarted { get; set; }
    [Display(Name = "Seats")]
    public int? SmSeats { get; set; }
    [Display(Name = "Max Autoscale Seats")]
    public int? MaxAutoscaleSmSeats { get; set; }
    [Display(Name = "Machine Accounts")]
    public int? SmServiceAccounts { get; set; }
    [Display(Name = "Max Autoscale Machine Accounts")]
    public int? MaxAutoscaleSmServiceAccounts { get; set; }
    [Display(Name = "Use Organization Domains")]
    public bool UseOrganizationDomains { get; set; }
    [Display(Name = "Disable SM Ads For Users")]
    public new bool UseDisableSmAdsForUsers { get; set; }

    [Display(Name = "Automatic User Confirmation")]
    public bool UseAutomaticUserConfirmation { get; set; }
    /**
     * Creates a Plan[] object for use in Javascript
     * This is mapped manually below to provide some type safety in case the plan objects change
     * Add mappings for individual properties as you need them
     */
    public object GetPlansHelper() =>
        _plans
            .Select(p =>
            {
                var plan = new
                {
                    Type = p.Type,
                    ProductTier = p.ProductTier,
                    Name = p.Name,
                    IsAnnual = p.IsAnnual,
                    NameLocalizationKey = p.NameLocalizationKey,
                    DescriptionLocalizationKey = p.DescriptionLocalizationKey,
                    CanBeUsedByBusiness = p.CanBeUsedByBusiness,
                    TrialPeriodDays = p.TrialPeriodDays,
                    HasSelfHost = p.HasSelfHost,
                    HasPolicies = p.HasPolicies,
                    HasGroups = p.HasGroups,
                    HasDirectory = p.HasDirectory,
                    HasEvents = p.HasEvents,
                    HasTotp = p.HasTotp,
                    Has2fa = p.Has2fa,
                    HasApi = p.HasApi,
                    HasSso = p.HasSso,
                    HasOrganizationDomains = p.HasOrganizationDomains,
                    HasKeyConnector = p.HasKeyConnector,
                    HasScim = p.HasScim,
                    HasResetPassword = p.HasResetPassword,
                    UsersGetPremium = p.UsersGetPremium,
                    HasCustomPermissions = p.HasCustomPermissions,
                    UpgradeSortOrder = p.UpgradeSortOrder,
                    DisplaySortOrder = p.DisplaySortOrder,
                    LegacyYear = p.LegacyYear,
                    Disabled = p.Disabled,
                    SupportsSecretsManager = p.SupportsSecretsManager,
                    AutomaticUserConfirmation = p.AutomaticUserConfirmation,
                    PasswordManager =
                        new
                        {
                            StripePlanId = p.PasswordManager?.StripePlanId,
                            StripeSeatPlanId = p.PasswordManager?.StripeSeatPlanId,
                            StripeProviderPortalSeatPlanId = p.PasswordManager?.StripeProviderPortalSeatPlanId,
                            BasePrice = p.PasswordManager?.BasePrice,
                            SeatPrice = p.PasswordManager?.SeatPrice,
                            ProviderPortalSeatPrice = p.PasswordManager?.ProviderPortalSeatPrice,
                            AllowSeatAutoscale = p.PasswordManager?.AllowSeatAutoscale,
                            HasAdditionalSeatsOption = p.PasswordManager?.HasAdditionalSeatsOption,
                            MaxAdditionalSeats = p.PasswordManager?.MaxAdditionalSeats,
                            BaseSeats = p.PasswordManager?.BaseSeats,
                            HasPremiumAccessOption = p.PasswordManager?.HasPremiumAccessOption,
                            StripePremiumAccessPlanId = p.PasswordManager?.StripePremiumAccessPlanId,
                            PremiumAccessOptionPrice = p.PasswordManager?.PremiumAccessOptionPrice,
                            MaxSeats = p.PasswordManager?.MaxSeats,
                            BaseStorageGb = p.PasswordManager?.BaseStorageGb,
                            HasAdditionalStorageOption = p.PasswordManager?.HasAdditionalStorageOption,
                            AdditionalStoragePricePerGb = p.PasswordManager?.AdditionalStoragePricePerGb,
                            StripeStoragePlanId = p.PasswordManager?.StripeStoragePlanId,
                            MaxAdditionalStorage = p.PasswordManager?.MaxAdditionalStorage,
                            MaxCollections = p.PasswordManager?.MaxCollections
                        },
                    SecretsManager = new
                    {
                        MaxServiceAccounts = p.SecretsManager?.MaxServiceAccounts,
                        AllowServiceAccountsAutoscale = p.SecretsManager?.AllowServiceAccountsAutoscale,
                        StripeServiceAccountPlanId = p.SecretsManager?.StripeServiceAccountPlanId,
                        AdditionalPricePerServiceAccount = p.SecretsManager?.AdditionalPricePerServiceAccount,
                        BaseServiceAccount = p.SecretsManager?.BaseServiceAccount,
                        MaxAdditionalServiceAccount = p.SecretsManager?.MaxAdditionalServiceAccount,
                        HasAdditionalServiceAccountOption = p.SecretsManager?.HasAdditionalServiceAccountOption,
                        StripeSeatPlanId = p.SecretsManager?.StripeSeatPlanId,
                        HasAdditionalSeatsOption = p.SecretsManager?.HasAdditionalSeatsOption,
                        BasePrice = p.SecretsManager?.BasePrice,
                        SeatPrice = p.SecretsManager?.SeatPrice,
                        BaseSeats = p.SecretsManager?.BaseSeats,
                        MaxSeats = p.SecretsManager?.MaxSeats,
                        MaxAdditionalSeats = p.SecretsManager?.MaxAdditionalSeats,
                        AllowSeatAutoscale = p.SecretsManager?.AllowSeatAutoscale,
                        MaxProjects = p.SecretsManager?.MaxProjects
                    }
                };
                return plan;
            });

    public Organization CreateOrganization(Provider provider)
    {
        BillingEmail = provider.BillingEmail;
        return ToOrganization(new Organization());
    }

    public Organization ToOrganization(Organization existingOrganization)
    {
        existingOrganization.Name = WebUtility.HtmlEncode(Name.Trim());
        existingOrganization.BillingEmail = BillingEmail?.ToLowerInvariant()?.Trim();
        existingOrganization.PlanType = PlanType.Value;
        existingOrganization.Plan = Plan;
        existingOrganization.Seats = Seats;
        existingOrganization.MaxCollections = MaxCollections;
        existingOrganization.UsePolicies = UsePolicies;
        existingOrganization.UseSso = UseSso;
        existingOrganization.UseKeyConnector = UseKeyConnector;
        existingOrganization.UseScim = UseScim;
        existingOrganization.UseGroups = UseGroups;
        existingOrganization.UseDirectory = UseDirectory;
        existingOrganization.UseEvents = UseEvents;
        existingOrganization.UseTotp = UseTotp;
        existingOrganization.Use2fa = Use2fa;
        existingOrganization.UseApi = UseApi;
        existingOrganization.UseSecretsManager = UseSecretsManager;
        existingOrganization.UseRiskInsights = UseRiskInsights;
        existingOrganization.UseAdminSponsoredFamilies = UseAdminSponsoredFamilies;
        existingOrganization.UseResetPassword = UseResetPassword;
        existingOrganization.SelfHost = SelfHost;
        existingOrganization.UsersGetPremium = UsersGetPremium;
        existingOrganization.UseCustomPermissions = UseCustomPermissions;
        existingOrganization.MaxStorageGb = MaxStorageGb;
        existingOrganization.Gateway = Gateway;
        existingOrganization.GatewayCustomerId = GatewayCustomerId;
        existingOrganization.GatewaySubscriptionId = GatewaySubscriptionId;
        existingOrganization.Enabled = Enabled;
        existingOrganization.LicenseKey = LicenseKey;
        existingOrganization.ExpirationDate = ExpirationDate;
        existingOrganization.MaxAutoscaleSeats = MaxAutoscaleSeats;
        existingOrganization.SmSeats = SmSeats;
        existingOrganization.MaxAutoscaleSmSeats = MaxAutoscaleSmSeats;
        existingOrganization.SmServiceAccounts = SmServiceAccounts;
        existingOrganization.MaxAutoscaleSmServiceAccounts = MaxAutoscaleSmServiceAccounts;
        existingOrganization.UseOrganizationDomains = UseOrganizationDomains;
        existingOrganization.UseDisableSmAdsForUsers = UseDisableSmAdsForUsers;
        existingOrganization.UsePhishingBlocker = UsePhishingBlocker;
        return existingOrganization;
    }
}