using Bit.Core.Entities;
using Bit.Core.Enums;
using Stripe;

namespace Bit.Core.Models.Business;

public class OrganizationSubscriptionOptionsBase : Stripe.SubscriptionCreateOptions
{
    public OrganizationSubscriptionOptionsBase(Organization org, List<StaticStore.Plan> plans, TaxInfo taxInfo, int additionalSeats,
        int additionalStorageGb, bool premiumAccessAddon, int additionalSmSeats, int additionalServiceAccounts)
    {
        Items = new List<SubscriptionItemOptions>();
        Metadata = new Dictionary<string, string>
        {
            [org.GatewayIdField()] = org.Id.ToString()
        };
        foreach (var plan in plans)
        {
            AddPlanIdToSubscription(plan);

            switch (plan.BitwardenProduct)
            {
                case BitwardenProductType.PasswordManager:
                    {
                        AddPremiumAccessAddon(premiumAccessAddon, plan);
                        AddAdditionalSeatToSubscription(additionalSeats, plan);
                        AddAdditionalStorage(additionalStorageGb, plan);
                        break;
                    }
                case BitwardenProductType.SecretsManager:
                    {
                        AddAdditionalSeatToSubscription(additionalSmSeats, plan);
                        AddServiceAccount(additionalServiceAccounts, plan);
                        break;
                    }
            }
        }

        if (!string.IsNullOrWhiteSpace(taxInfo?.StripeTaxRateId))
        {
            DefaultTaxRates = new List<string> { taxInfo.StripeTaxRateId };
        }
    }

    private void AddServiceAccount(int additionalServiceAccounts, StaticStore.Plan plan)
    {
        if (additionalServiceAccounts > 0 && plan.StripeServiceAccountPlanId != null)
        {
            Items.Add(new SubscriptionItemOptions
            {
                Plan = plan.StripeServiceAccountPlanId,
                Quantity = additionalServiceAccounts
            });
        }
    }

    private void AddAdditionalStorage(int additionalStorageGb, StaticStore.Plan plan)
    {
        if (additionalStorageGb > 0)
        {
            Items.Add(new SubscriptionItemOptions
            {
                Plan = plan.StripeStoragePlanId,
                Quantity = additionalStorageGb
            });
        }
    }

    private void AddPremiumAccessAddon(bool premiumAccessAddon, StaticStore.Plan plan)
    {
        if (premiumAccessAddon && plan.StripePremiumAccessPlanId != null)
        {
            Items.Add(new SubscriptionItemOptions { Plan = plan.StripePremiumAccessPlanId, Quantity = 1 });
        }
    }

    private void AddAdditionalSeatToSubscription(int additionalSeats, StaticStore.Plan plan)
    {
        if (additionalSeats > 0 && plan.StripeSeatPlanId != null)
        {
            Items.Add(new SubscriptionItemOptions { Plan = plan.StripeSeatPlanId, Quantity = additionalSeats });
        }
    }

    private void AddPlanIdToSubscription(StaticStore.Plan plan)
    {
        if (plan.StripePlanId != null)
        {
            Items.Add(new SubscriptionItemOptions { Plan = plan.StripePlanId, Quantity = 1 });
        }
    }
}

public class OrganizationPurchaseSubscriptionOptions : OrganizationSubscriptionOptionsBase
{
    public OrganizationPurchaseSubscriptionOptions(
        Organization org, List<StaticStore.Plan> plans,
        TaxInfo taxInfo, int additionalSeats,
        int additionalStorageGb, bool premiumAccessAddon,
        int additionalSmSeats, int additionalServiceAccounts) :
        base(org, plans, taxInfo, additionalSeats, additionalStorageGb, premiumAccessAddon, additionalSmSeats, additionalServiceAccounts)
    {
        OffSession = true;
        TrialPeriodDays = plans.FirstOrDefault(x => x.BitwardenProduct == BitwardenProductType.PasswordManager)!.TrialPeriodDays;
    }
}

public class OrganizationUpgradeSubscriptionOptions : OrganizationSubscriptionOptionsBase
{
    public OrganizationUpgradeSubscriptionOptions(
        string customerId, Organization org,
        List<StaticStore.Plan> plans, OrganizationUpgrade upgrade) :
        base(org, plans, upgrade.TaxInfo, upgrade.AdditionalSeats, upgrade.AdditionalStorageGb,
        upgrade.PremiumAccessAddon, upgrade.AdditionalSmSeats.GetValueOrDefault(),
        upgrade.AdditionalServiceAccounts.GetValueOrDefault())
    {
        Customer = customerId;
    }
}