DocSpace-buildtools/common/ASC.Core.Common/Billing/TariffService.cs

/*
 *
 * (c) Copyright Ascensio System Limited 2010-2018
 *
 * This program is freeware. You can redistribute it and/or modify it under the terms of the GNU 
 * General Public License (GPL) version 3 as published by the Free Software Foundation (https://www.gnu.org/copyleft/gpl.html). 
 * In accordance with Section 7(a) of the GNU GPL its Section 15 shall be amended to the effect that 
 * Ascensio System SIA expressly excludes the warranty of non-infringement of any third-party rights.
 *
 * THIS PROGRAM IS DISTRIBUTED WITHOUT ANY WARRANTY; WITHOUT EVEN THE IMPLIED WARRANTY OF MERCHANTABILITY OR
 * FITNESS FOR A PARTICULAR PURPOSE. For more details, see GNU GPL at https://www.gnu.org/copyleft/gpl.html
 *
 * You can contact Ascensio System SIA by email at sales@onlyoffice.com
 *
 * The interactive user interfaces in modified source and object code versions of ONLYOFFICE must display 
 * Appropriate Legal Notices, as required under Section 5 of the GNU GPL version 3.
 *
 * Pursuant to Section 7 § 3(b) of the GNU GPL you must retain the original ONLYOFFICE logo which contains 
 * relevant author attributions when distributing the software. If the display of the logo in its graphic 
 * form is not reasonably feasible for technical reasons, you must include the words "Powered by ONLYOFFICE" 
 * in every copy of the program you distribute. 
 * Pursuant to Section 7 § 3(e) we decline to grant you any rights under trademark law for use of our trademarks.
 *
*/


using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;

using ASC.Common;
using ASC.Common.Caching;
using ASC.Common.Logging;
using ASC.Core.Caching;
using ASC.Core.Common.EF;
using ASC.Core.Tenants;

using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.Options;

namespace ASC.Core.Billing
{
    public class TariffServiceStorage
    {
        public ICache Cache { get; }
        public ICacheNotify<TariffCacheItem> Notify { get; }

        public TariffServiceStorage(ICacheNotify<TariffCacheItem> notify)
        {
            Cache = AscCache.Memory;
            Notify = notify;
            Notify.Subscribe((i) =>
            {
                Cache.Remove(TariffService.GetTariffCacheKey(i.TenantId));
                Cache.Remove(TariffService.GetBillingUrlCacheKey(i.TenantId));
                Cache.Remove(TariffService.GetBillingPaymentCacheKey(i.TenantId, DateTime.MinValue, DateTime.MaxValue)); // clear all payments
            }, CacheNotifyAction.Remove);

            //TODO: Change code of WCF -> not supported in .NET standard/.Net Core
            /*try
            {
                var section = (ClientSection)ConfigurationManager.GetSection("system.serviceModel/client");
                if (section != null)
                {
                    billingConfigured = section.Endpoints.Cast<ChannelEndpointElement>()
                        .Any(e => e.Contract == typeof(IService).FullName);
                }
            }
            catch (Exception err)
            {
                log.Error(err);
            }*/
        }
    }

    class ConfigureTariffService : IConfigureNamedOptions<TariffService>
    {
        public ConfigureTariffService(
            IOptionsSnapshot<CachedQuotaService> quotaService,
            IOptionsSnapshot<CachedTenantService> tenantService,
            CoreBaseSettings coreBaseSettings,
            CoreSettings coreSettings,
            IConfiguration configuration,
            DbContextManager<CoreDbContext> coreDbContextManager,
            TariffServiceStorage tariffServiceStorage,
            IOptionsMonitor<ILog> iLog)
        {
            QuotaService = quotaService;
            TenantService = tenantService;
            CoreBaseSettings = coreBaseSettings;
            CoreSettings = coreSettings;
            Configuration = configuration;
            CoreDbContextManager = coreDbContextManager;
            TariffServiceStorage = tariffServiceStorage;
            ILog = iLog;
        }

        public IOptionsSnapshot<CachedQuotaService> QuotaService { get; }
        public IOptionsSnapshot<CachedTenantService> TenantService { get; }
        public CoreBaseSettings CoreBaseSettings { get; }
        public CoreSettings CoreSettings { get; }
        public IConfiguration Configuration { get; }
        public DbContextManager<CoreDbContext> CoreDbContextManager { get; }
        public TariffServiceStorage TariffServiceStorage { get; }
        public IOptionsMonitor<ILog> ILog { get; }

        public void Configure(string name, TariffService options)
        {
            Configure(options);
            options.QuotaService = QuotaService.Get(name);
            options.TenantService = TenantService.Get(name);
            options.CoreDbContext = CoreDbContextManager.Get(name);
        }

        public void Configure(TariffService options)
        {
            options.Log = ILog.CurrentValue;
            options.CoreSettings = CoreSettings;
            options.Configuration = Configuration;
            options.TariffServiceStorage = TariffServiceStorage;
            options.Options = ILog;
            options.CoreBaseSettings = CoreBaseSettings;
            options.Test = Configuration["core:payment:test"] == "true";
            int.TryParse(Configuration["core:payment:delay"], out var paymentDelay);
            options.PaymentDelay = paymentDelay;
            options.Cache = TariffServiceStorage.Cache;
            options.Notify = TariffServiceStorage.Notify;

            options.QuotaService = QuotaService.Value;
            options.TenantService = TenantService.Value;
            options.CoreDbContext = CoreDbContextManager.Value;
        }
    }

    public class TariffService : ITariffService
    {
        private const int DEFAULT_TRIAL_PERIOD = 30;
        private static readonly TimeSpan DEFAULT_CACHE_EXPIRATION = TimeSpan.FromMinutes(5);
        private static readonly TimeSpan STANDALONE_CACHE_EXPIRATION = TimeSpan.FromMinutes(15);

        private readonly static bool billingConfigured = false;

        internal ICache Cache { get; set; }
        internal ICacheNotify<TariffCacheItem> Notify { get; set; }
        internal ILog Log { get; set; }
        internal IQuotaService QuotaService { get; set; }
        internal ITenantService TenantService { get; set; }
        internal bool Test { get; set; }
        internal int PaymentDelay { get; set; }

        public TimeSpan CacheExpiration { get; set; }
        public CoreBaseSettings CoreBaseSettings { get; set; }
        public CoreSettings CoreSettings { get; set; }
        public IConfiguration Configuration { get; set; }
        public CoreDbContext CoreDbContext { get; set; }
        public TariffServiceStorage TariffServiceStorage { get; set; }
        public IOptionsMonitor<ILog> Options { get; set; }

        public TariffService()
        {
            CacheExpiration = DEFAULT_CACHE_EXPIRATION;
        }

        public TariffService(
            IQuotaService quotaService,
            ITenantService tenantService,
            CoreBaseSettings coreBaseSettings,
            CoreSettings coreSettings,
            IConfiguration configuration,
            DbContextManager<CoreDbContext> coreDbContextManager,
            TariffServiceStorage tariffServiceStorage,
            IOptionsMonitor<ILog> options)
            : this()

        {
            Log = options.CurrentValue;
            QuotaService = quotaService;
            TenantService = tenantService;
            CoreSettings = coreSettings;
            Configuration = configuration;
            TariffServiceStorage = tariffServiceStorage;
            Options = options;
            CoreBaseSettings = coreBaseSettings;
            Test = configuration["core:payment:test"] == "true";
            int.TryParse(configuration["core:payment:delay"], out var paymentDelay);

            PaymentDelay = paymentDelay;

            Cache = TariffServiceStorage.Cache;
            Notify = TariffServiceStorage.Notify;
            CoreDbContext = coreDbContextManager.Value;
        }

        public Tariff GetTariff(int tenantId, bool withRequestToPaymentSystem = true)
        {
            //single tariff for all portals
            if (CoreBaseSettings.Standalone)
                tenantId = -1;

            var key = GetTariffCacheKey(tenantId);
            var tariff = Cache.Get<Tariff>(key);
            if (tariff == null)
            {
                tariff = Tariff.CreateDefault();

                var cached = GetBillingInfo(tenantId);
                if (cached != null)
                {
                    tariff.QuotaId = cached.Item1;
                    tariff.DueDate = cached.Item2;
                }

                tariff = CalculateTariff(tenantId, tariff);
                Cache.Insert(key, tariff, DateTime.UtcNow.Add(GetCacheExpiration()));

                if (billingConfigured && withRequestToPaymentSystem)
                {
                    Task.Run(() =>
                    {
                        try
                        {
                            using var client = GetBillingClient();
                            var p = client.GetLastPayment(GetPortalId(tenantId));
                            var quota = QuotaService.GetTenantQuotas().SingleOrDefault(q => q.AvangateId == p.ProductId);
                            if (quota == null)
                            {
                                throw new InvalidOperationException(string.Format("Quota with id {0} not found for portal {1}.", p.ProductId, GetPortalId(tenantId)));
                            }
                            var asynctariff = Tariff.CreateDefault();
                            asynctariff.QuotaId = quota.Id;
                            asynctariff.Autorenewal = p.Autorenewal;
                            asynctariff.DueDate = 9999 <= p.EndDate.Year ? DateTime.MaxValue : p.EndDate;

                            if (SaveBillingInfo(tenantId, Tuple.Create(asynctariff.QuotaId, asynctariff.DueDate), false))
                            {
                                asynctariff = CalculateTariff(tenantId, asynctariff);
                                ClearCache(tenantId);
                                Cache.Insert(key, asynctariff, DateTime.UtcNow.Add(GetCacheExpiration()));
                            }
                        }
                        catch (Exception error)
                        {
                            LogError(error);
                        }
                    });
                }
            }

            return tariff;
        }

        public void SetTariff(int tenantId, Tariff tariff)
        {
            if (tariff == null)
            {
                throw new ArgumentNullException("tariff");
            }

            var q = QuotaService.GetTenantQuota(tariff.QuotaId);
            if (q == null) return;
            SaveBillingInfo(tenantId, Tuple.Create(tariff.QuotaId, tariff.DueDate));
            if (q.Trial)
            {
                // reset trial date
                var tenant = TenantService.GetTenant(tenantId);
                if (tenant != null)
                {
                    tenant.VersionChanged = DateTime.UtcNow;
                    TenantService.SaveTenant(CoreSettings, tenant);
                }
            }

            ClearCache(tenantId);
        }

        internal static string GetTariffCacheKey(int tenantId)
        {
            return string.Format("{0}:{1}", tenantId, "tariff");
        }

        internal static string GetBillingUrlCacheKey(int tenantId)
        {
            return string.Format("{0}:{1}", tenantId, "billing:urls");
        }

        internal static string GetBillingPaymentCacheKey(int tenantId, DateTime from, DateTime to)
        {
            return string.Format("{0}:{1}:{2}-{3}", tenantId, "billing:payments", from.ToString("yyyyMMddHHmmss"), to.ToString("yyyyMMddHHmmss"));
        }


        public void ClearCache(int tenantId)
        {
            Notify.Publish(new TariffCacheItem { TenantId = tenantId }, CacheNotifyAction.Remove);
        }

        public IEnumerable<PaymentInfo> GetPayments(int tenantId, DateTime from, DateTime to)
        {
            from = from.Date;
            to = to.Date.AddTicks(TimeSpan.TicksPerDay - 1);
            var key = GetBillingPaymentCacheKey(tenantId, from, to);
            var payments = Cache.Get<List<PaymentInfo>>(key);
            if (payments == null)
            {
                payments = new List<PaymentInfo>();
                if (billingConfigured)
                {
                    try
                    {
                        var quotas = QuotaService.GetTenantQuotas();
                        using var client = GetBillingClient();
                        foreach (var pi in client.GetPayments(GetPortalId(tenantId), from, to))
                        {
                            var quota = quotas.SingleOrDefault(q => q.AvangateId == pi.ProductId);
                            if (quota != null)
                            {
                                pi.QuotaId = quota.Id;
                            }
                            payments.Add(pi);
                        }
                    }
                    catch (Exception error)
                    {
                        LogError(error);
                    }
                }

                Cache.Insert(key, payments, DateTime.UtcNow.Add(TimeSpan.FromMinutes(10)));
            }

            return payments;
        }

        public Uri GetShoppingUri(int? tenant, int quotaId, string affiliateId, string currency = null, string language = null, string customerId = null)
        {
            var quota = QuotaService.GetTenantQuota(quotaId);
            if (quota == null) return null;

            var key = tenant.HasValue
                          ? GetBillingUrlCacheKey(tenant.Value)
                          : string.Format("notenant{0}", !string.IsNullOrEmpty(affiliateId) ? "_" + affiliateId : "");
            key += quota.Visible ? "" : "0";
            if (!(Cache.Get<Dictionary<string, Tuple<Uri, Uri>>>(key) is IDictionary<string, Tuple<Uri, Uri>> urls))
            {
                urls = new Dictionary<string, Tuple<Uri, Uri>>();
                if (billingConfigured)
                {
                    try
                    {
                        var products = QuotaService.GetTenantQuotas()
                                                   .Where(q => !string.IsNullOrEmpty(q.AvangateId) && q.Visible == quota.Visible)
                                                   .Select(q => q.AvangateId)
                                                   .ToArray();

                        using var client = GetBillingClient();
                        urls = tenant.HasValue ?
                                   client.GetPaymentUrls(GetPortalId(tenant.Value), products, GetAffiliateId(tenant.Value), GetCampaign(tenant.Value), "__Currency__", "__Language__", "__CustomerID__") :
                                   client.GetPaymentUrls(null, products, !string.IsNullOrEmpty(affiliateId) ? affiliateId : null, null, "__Currency__", "__Language__", "__CustomerID__");
                    }
                    catch (Exception error)
                    {
                        Log.Error(error);
                    }
                }
                Cache.Insert(key, urls, DateTime.UtcNow.Add(TimeSpan.FromMinutes(10)));
            }

            ResetCacheExpiration();

            if (!string.IsNullOrEmpty(quota.AvangateId) && urls.TryGetValue(quota.AvangateId, out var tuple))
            {
                var result = tuple.Item2;

                var tariff = tenant.HasValue ? GetTariff(tenant.Value) : null;
                if (result == null || tariff == null || tariff.QuotaId == quotaId || tariff.State >= TariffState.Delay)
                {
                    result = tuple.Item1;
                }

                result = new Uri(result.ToString()
                                       .Replace("__Currency__", currency ?? "")
                                       .Replace("__Language__", (language ?? "").ToLower())
                                       .Replace("__CustomerID__", customerId ?? ""));
                return result;
            }
            return null;
        }

        public IDictionary<string, IEnumerable<Tuple<string, decimal>>> GetProductPriceInfo(params string[] productIds)
        {
            if (productIds == null)
            {
                throw new ArgumentNullException("productIds");
            }
            try
            {
                var key = "biling-prices" + string.Join(",", productIds);
                var result = Cache.Get<IDictionary<string, IEnumerable<Tuple<string, decimal>>>>(key);
                if (result == null)
                {
                    using (var client = GetBillingClient())
                    {
                        result = client.GetProductPriceInfo(productIds);
                    }
                    Cache.Insert(key, result, DateTime.Now.AddHours(1));
                }
                return result;
            }
            catch (Exception error)
            {
                LogError(error);
                return productIds
                    .Select(p => new { ProductId = p, Prices = Enumerable.Empty<Tuple<string, decimal>>() })
                    .ToDictionary(e => e.ProductId, e => e.Prices);
            }
        }

        public Invoice GetInvoice(string paymentId)
        {
            var result = new Invoice();

            if (billingConfigured)
            {
                try
                {
                    using var client = GetBillingClient();
                    result = client.GetInvoice(paymentId);
                }
                catch (Exception error)
                {
                    LogError(error);
                }
            }
            return result;
        }


        public string GetButton(int tariffId, string partnerId)
        {
            return CoreDbContext.Buttons
                .Where(r => r.TariffId == tariffId && r.PartnerId == partnerId)
                .Select(r => r.ButtonUrl)
                .SingleOrDefault();
        }

        public void SaveButton(int tariffId, string partnerId, string buttonUrl)
        {
            var efButton = new DbButton()
            {
                TariffId = tariffId,
                PartnerId = partnerId,
                ButtonUrl = buttonUrl
            };

            CoreDbContext.AddOrUpdate(r => r.Buttons, efButton);
            CoreDbContext.SaveChanges();
        }


        private Tuple<int, DateTime> GetBillingInfo(int tenant)
        {
            var r = CoreDbContext.Tariffs
                .Where(r => r.Tenant == tenant)
                .OrderByDescending(r => r.Id)
                .SingleOrDefault();

            return r != null ? Tuple.Create(r.Tariff, r.Stamp.Year < 9999 ? r.Stamp : DateTime.MaxValue) : null;
        }

        private bool SaveBillingInfo(int tenant, Tuple<int, DateTime> bi, bool renewal = true)
        {
            var inserted = false;
            if (!Equals(bi, GetBillingInfo(tenant)))
            {
                using var tx = CoreDbContext.Database.BeginTransaction();

                // last record is not the same
                var count = CoreDbContext.Tariffs
                    .Count(r => r.Tenant == tenant && r.Tariff == bi.Item1 && r.Stamp == bi.Item2);

                if (bi.Item2 == DateTime.MaxValue || renewal || count == 0)
                {
                    var efTariff = new DbTariff
                    {
                        Tenant = tenant,
                        Tariff = bi.Item1,
                        Stamp = bi.Item2
                    };

                    CoreDbContext.Tariffs.Add(efTariff);
                    CoreDbContext.SaveChanges();

                    Cache.Remove(GetTariffCacheKey(tenant));
                    inserted = true;
                }
                tx.Commit();
            }

            if (inserted)
            {
                var t = TenantService.GetTenant(tenant);
                if (t != null)
                {
                    // update tenant.LastModified to flush cache in documents
                    TenantService.SaveTenant(CoreSettings, t);
                }
            }
            return inserted;
        }

        public void DeleteDefaultBillingInfo()
        {
            const int tenant = Tenant.DEFAULT_TENANT;

            var tariffs = CoreDbContext.Tariffs.Where(r => r.Tenant == tenant).ToList();

            foreach (var t in tariffs)
            {
                t.Tenant = -2;
            }

            CoreDbContext.SaveChanges();

            ClearCache(tenant);
        }


        private Tariff CalculateTariff(int tenantId, Tariff tariff)
        {
            tariff.State = TariffState.Paid;
            var q = QuotaService.GetTenantQuota(tariff.QuotaId);

            if (q == null || q.GetFeature("old"))
            {
                tariff.QuotaId = Tenant.DEFAULT_TENANT;
                q = QuotaService.GetTenantQuota(tariff.QuotaId);
            }

            var delay = 0;
            if (q != null && q.Trial)
            {
                tariff.State = TariffState.Trial;
                if (tariff.DueDate == DateTime.MinValue || tariff.DueDate == DateTime.MaxValue)
                {
                    var tenant = TenantService.GetTenant(tenantId);
                    if (tenant != null)
                    {
                        var fromDate = tenant.CreatedDateTime < tenant.VersionChanged ? tenant.VersionChanged : tenant.CreatedDateTime;
                        var trialPeriod = GetPeriod("TrialPeriod", DEFAULT_TRIAL_PERIOD);
                        if (fromDate == DateTime.MinValue) fromDate = DateTime.UtcNow.Date;
                        tariff.DueDate = trialPeriod != default ? fromDate.Date.AddDays(trialPeriod) : DateTime.MaxValue;
                    }
                    else
                    {
                        tariff.DueDate = DateTime.MaxValue;
                    }
                }
            }
            else
            {
                delay = PaymentDelay;
            }

            if (tariff.DueDate != DateTime.MinValue && tariff.DueDate.Date < DateTime.Today && delay > 0)
            {
                tariff.State = TariffState.Delay;

                tariff.DelayDueDate = tariff.DueDate.Date.AddDays(delay);
            }

            if (tariff.DueDate == DateTime.MinValue ||
                tariff.DueDate != DateTime.MaxValue && tariff.DueDate.Date.AddDays(delay) < DateTime.Today)
            {
                tariff.State = TariffState.NotPaid;

                if ((q == null || !q.Trial) && CoreBaseSettings.Standalone)
                {
                    if (q != null)
                    {
                        var defaultQuota = QuotaService.GetTenantQuota(Tenant.DEFAULT_TENANT);
                        if (defaultQuota.CountPortals != q.CountPortals)
                        {
                            defaultQuota.CountPortals = q.CountPortals;
                            QuotaService.SaveTenantQuota(defaultQuota);
                        }
                    }

                    var unlimTariff = Tariff.CreateDefault();
                    unlimTariff.LicenseDate = tariff.DueDate;

                    tariff = unlimTariff;
                }
            }

            tariff.Prolongable = tariff.DueDate == DateTime.MinValue || tariff.DueDate == DateTime.MaxValue ||
                                 tariff.State == TariffState.Trial ||
                                 new DateTime(tariff.DueDate.Year, tariff.DueDate.Month, 1) <= new DateTime(DateTime.UtcNow.Year, DateTime.UtcNow.Month, 1).AddMonths(1);

            return tariff;
        }

        private int GetPeriod(string key, int defaultValue)
        {
            var settings = TenantService.GetTenantSettings(Tenant.DEFAULT_TENANT, key);
            return settings != null ? Convert.ToInt32(Encoding.UTF8.GetString(settings)) : defaultValue;
        }

        private BillingClient GetBillingClient()
        {
            try
            {
                return new BillingClient(Test, Configuration, Options);
            }
            catch (InvalidOperationException ioe)
            {
                throw new BillingNotConfiguredException(ioe.Message, ioe);
            }
            catch (ReflectionTypeLoadException rtle)
            {
                Log.ErrorFormat("{0}{1}LoaderExceptions: {2}",
                    rtle,
                    Environment.NewLine,
                    string.Join(Environment.NewLine, rtle.LoaderExceptions.Select(e => e.ToString())));
                throw;
            }
        }

        private string GetPortalId(int tenant)
        {
            return CoreSettings.GetKey(tenant);
        }

        private string GetAffiliateId(int tenant)
        {
            return CoreSettings.GetAffiliateId(tenant);
        }

        private string GetCampaign(int tenant)
        {
            return CoreSettings.GetCampaign(tenant);
        }

        private TimeSpan GetCacheExpiration()
        {
            if (CoreBaseSettings.Standalone && CacheExpiration < STANDALONE_CACHE_EXPIRATION)
            {
                CacheExpiration = CacheExpiration.Add(TimeSpan.FromSeconds(30));
            }
            return CacheExpiration;
        }

        private void ResetCacheExpiration()
        {
            if (CoreBaseSettings.Standalone)
            {
                CacheExpiration = DEFAULT_CACHE_EXPIRATION;
            }
        }

        private void LogError(Exception error)
        {
            if (error is BillingNotFoundException)
            {
                Log.DebugFormat("Payment not found: {0}", error.Message);
            }
            else if (error is BillingNotConfiguredException)
            {
                Log.DebugFormat("Billing not configured: {0}", error.Message);
            }
            else
            {
                if (Log.IsDebugEnabled)
                {
                    Log.Error(error);
                }
                else
                {
                    Log.Error(error.Message);
                }
            }
        }
    }

    public static class TariffConfigExtension
    {
        public static DIHelper AddTariffService(this DIHelper services)
        {
            services.AddCoreDbContextService();

            services.TryAddSingleton<TariffServiceStorage>();

            services.TryAddScoped<ITariffService, TariffService>();
            services.TryAddScoped<IConfigureOptions<TariffService>, ConfigureTariffService>();

            return services;
        }
    }
}