400 lines
14 KiB
C#
400 lines
14 KiB
C#
/*
|
||
*
|
||
* (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.Net;
|
||
using System.Threading;
|
||
using System.Web;
|
||
|
||
using ASC.Common;
|
||
using ASC.Common.Notify.Engine;
|
||
using ASC.Core.Billing;
|
||
using ASC.Core.Caching;
|
||
using ASC.Core.Tenants;
|
||
|
||
using Microsoft.AspNetCore.Http;
|
||
using Microsoft.Extensions.Options;
|
||
|
||
namespace ASC.Core
|
||
{
|
||
class ConfigureTenantManager : IConfigureNamedOptions<TenantManager>
|
||
{
|
||
public IOptionsSnapshot<CachedTenantService> TenantService { get; }
|
||
public IOptionsSnapshot<CachedQuotaService> QuotaService { get; }
|
||
public IOptionsSnapshot<TariffService> TariffService { get; }
|
||
public IHttpContextAccessor HttpContextAccessor { get; }
|
||
public CoreBaseSettings CoreBaseSettings { get; }
|
||
public CoreSettings CoreSettings { get; }
|
||
|
||
public ConfigureTenantManager(
|
||
IOptionsSnapshot<CachedTenantService> tenantService,
|
||
IOptionsSnapshot<CachedQuotaService> quotaService,
|
||
IOptionsSnapshot<TariffService> tariffService,
|
||
IHttpContextAccessor httpContextAccessor,
|
||
CoreBaseSettings coreBaseSettings,
|
||
CoreSettings coreSettings
|
||
)
|
||
{
|
||
TenantService = tenantService;
|
||
QuotaService = quotaService;
|
||
TariffService = tariffService;
|
||
HttpContextAccessor = httpContextAccessor;
|
||
CoreBaseSettings = coreBaseSettings;
|
||
CoreSettings = coreSettings;
|
||
}
|
||
|
||
public void Configure(string name, TenantManager options)
|
||
{
|
||
Configure(options);
|
||
|
||
options.TenantService = TenantService.Get(name);
|
||
options.QuotaService = QuotaService.Get(name);
|
||
options.TariffService = TariffService.Get(name);
|
||
}
|
||
|
||
public void Configure(TenantManager options)
|
||
{
|
||
options.HttpContextAccessor = HttpContextAccessor;
|
||
options.CoreBaseSettings = CoreBaseSettings;
|
||
options.CoreSettings = CoreSettings;
|
||
|
||
options.TenantService = TenantService.Value;
|
||
options.QuotaService = QuotaService.Value;
|
||
options.TariffService = TariffService.Value;
|
||
}
|
||
}
|
||
|
||
public class TenantManager
|
||
{
|
||
private class TenantHolder
|
||
{
|
||
public Tenant Tenant;
|
||
}
|
||
|
||
private static readonly AsyncLocal<TenantHolder> currentTenant = new AsyncLocal<TenantHolder>();
|
||
|
||
public Tenant CurrentTenant
|
||
{
|
||
get
|
||
{
|
||
return currentTenant.Value?.Tenant;
|
||
}
|
||
set
|
||
{
|
||
var holder = currentTenant.Value;
|
||
if (holder != null)
|
||
{
|
||
holder.Tenant = null;
|
||
}
|
||
|
||
if (value != null)
|
||
{
|
||
currentTenant.Value = new TenantHolder { Tenant = value };
|
||
}
|
||
}
|
||
}
|
||
|
||
public const string CURRENT_TENANT = "CURRENT_TENANT";
|
||
internal ITenantService TenantService { get; set; }
|
||
internal IQuotaService QuotaService { get; set; }
|
||
internal ITariffService TariffService { get; set; }
|
||
|
||
private static List<string> thisCompAddresses = new List<string>();
|
||
|
||
internal IHttpContextAccessor HttpContextAccessor { get; set; }
|
||
internal CoreBaseSettings CoreBaseSettings { get; set; }
|
||
internal CoreSettings CoreSettings { get; set; }
|
||
|
||
static TenantManager()
|
||
{
|
||
thisCompAddresses.Add("localhost");
|
||
thisCompAddresses.Add(Dns.GetHostName().ToLowerInvariant());
|
||
thisCompAddresses.AddRange(Dns.GetHostAddresses("localhost").Select(a => a.ToString()));
|
||
try
|
||
{
|
||
thisCompAddresses.AddRange(Dns.GetHostAddresses(Dns.GetHostName()).Select(a => a.ToString()));
|
||
}
|
||
catch
|
||
{
|
||
// ignore
|
||
}
|
||
}
|
||
|
||
public TenantManager()
|
||
{
|
||
|
||
}
|
||
|
||
public TenantManager(
|
||
ITenantService tenantService,
|
||
IQuotaService quotaService,
|
||
ITariffService tariffService,
|
||
CoreBaseSettings coreBaseSettings,
|
||
CoreSettings coreSettings)
|
||
{
|
||
TenantService = tenantService;
|
||
QuotaService = quotaService;
|
||
TariffService = tariffService;
|
||
CoreBaseSettings = coreBaseSettings;
|
||
CoreSettings = coreSettings;
|
||
}
|
||
|
||
public TenantManager(
|
||
ITenantService tenantService,
|
||
IQuotaService quotaService,
|
||
ITariffService tariffService,
|
||
IHttpContextAccessor httpContextAccessor,
|
||
CoreBaseSettings coreBaseSettings,
|
||
CoreSettings coreSettings) : this(tenantService, quotaService, tariffService, coreBaseSettings, coreSettings)
|
||
{
|
||
HttpContextAccessor = httpContextAccessor;
|
||
}
|
||
|
||
|
||
public List<Tenant> GetTenants(bool active = true)
|
||
{
|
||
return TenantService.GetTenants(default, active).ToList();
|
||
}
|
||
|
||
public Tenant GetTenant(int tenantId)
|
||
{
|
||
return TenantService.GetTenant(tenantId);
|
||
}
|
||
|
||
public Tenant GetTenant(string domain)
|
||
{
|
||
if (string.IsNullOrEmpty(domain)) return null;
|
||
|
||
Tenant t = null;
|
||
if (thisCompAddresses.Contains(domain, StringComparer.InvariantCultureIgnoreCase))
|
||
{
|
||
t = TenantService.GetTenant("localhost");
|
||
}
|
||
var isAlias = false;
|
||
if (t == null)
|
||
{
|
||
var baseUrl = CoreSettings.BaseDomain;
|
||
if (!string.IsNullOrEmpty(baseUrl) && domain.EndsWith("." + baseUrl, StringComparison.InvariantCultureIgnoreCase))
|
||
{
|
||
isAlias = true;
|
||
t = TenantService.GetTenant(domain.Substring(0, domain.Length - baseUrl.Length - 1));
|
||
}
|
||
}
|
||
if (t == null)
|
||
{
|
||
t = TenantService.GetTenant(domain);
|
||
}
|
||
if (t == null && CoreBaseSettings.Standalone && !isAlias)
|
||
{
|
||
t = TenantService.GetTenantForStandaloneWithoutAlias(domain);
|
||
}
|
||
return t;
|
||
}
|
||
|
||
public Tenant SetTenantVersion(Tenant tenant, int version)
|
||
{
|
||
if (tenant == null) throw new ArgumentNullException("tenant");
|
||
if (tenant.Version != version)
|
||
{
|
||
tenant.Version = version;
|
||
SaveTenant(tenant);
|
||
}
|
||
else
|
||
{
|
||
throw new ArgumentException("This is current version already");
|
||
}
|
||
return tenant;
|
||
}
|
||
|
||
public Tenant SaveTenant(Tenant tenant)
|
||
{
|
||
var newTenant = TenantService.SaveTenant(CoreSettings, tenant);
|
||
if (CallContext.GetData(CURRENT_TENANT) is Tenant) SetCurrentTenant(newTenant);
|
||
|
||
return newTenant;
|
||
}
|
||
|
||
public void RemoveTenant(int tenantId, bool auto = false)
|
||
{
|
||
TenantService.RemoveTenant(tenantId, auto);
|
||
}
|
||
|
||
public Tenant GetCurrentTenant(HttpContext context)
|
||
{
|
||
return GetCurrentTenant(true, context);
|
||
}
|
||
|
||
public Tenant GetCurrentTenant(bool throwIfNotFound, HttpContext context)
|
||
{
|
||
Tenant tenant = null;
|
||
if (CurrentTenant != null)
|
||
{
|
||
return CurrentTenant;
|
||
}
|
||
|
||
if (context != null)
|
||
{
|
||
tenant = context.Items[CURRENT_TENANT] as Tenant;
|
||
if (tenant == null && context.Request != null)
|
||
{
|
||
tenant = GetTenant(context.Request.GetUrlRewriter().Host);
|
||
context.Items[CURRENT_TENANT] = tenant;
|
||
}
|
||
}
|
||
if (tenant == null)
|
||
{
|
||
tenant = CallContext.GetData(CURRENT_TENANT) as Tenant;
|
||
}
|
||
if (tenant == null && throwIfNotFound)
|
||
{
|
||
throw new Exception("Could not resolve current tenant :-(.");
|
||
}
|
||
|
||
CurrentTenant = tenant;
|
||
|
||
return tenant;
|
||
}
|
||
|
||
public Tenant GetCurrentTenant()
|
||
{
|
||
return GetCurrentTenant(true);
|
||
}
|
||
|
||
public Tenant GetCurrentTenant(bool throwIfNotFound)
|
||
{
|
||
return GetCurrentTenant(throwIfNotFound, HttpContextAccessor?.HttpContext);
|
||
}
|
||
|
||
public void SetCurrentTenant(Tenant tenant)
|
||
{
|
||
if (tenant != null)
|
||
{
|
||
CurrentTenant = tenant;
|
||
if (HttpContextAccessor?.HttpContext != null)
|
||
{
|
||
HttpContextAccessor.HttpContext.Items[CURRENT_TENANT] = tenant;
|
||
}
|
||
Thread.CurrentThread.CurrentCulture = tenant.GetCulture();
|
||
Thread.CurrentThread.CurrentUICulture = tenant.GetCulture();
|
||
}
|
||
}
|
||
|
||
public Tenant SetCurrentTenant(int tenantId)
|
||
{
|
||
var result = GetTenant(tenantId);
|
||
SetCurrentTenant(result);
|
||
return result;
|
||
}
|
||
|
||
public Tenant SetCurrentTenant(string domain)
|
||
{
|
||
var result = GetTenant(domain);
|
||
SetCurrentTenant(result);
|
||
return result;
|
||
}
|
||
|
||
public void CheckTenantAddress(string address)
|
||
{
|
||
TenantService.ValidateDomain(address);
|
||
}
|
||
|
||
public IEnumerable<TenantVersion> GetTenantVersions()
|
||
{
|
||
return TenantService.GetTenantVersions();
|
||
}
|
||
|
||
|
||
public IEnumerable<TenantQuota> GetTenantQuotas()
|
||
{
|
||
return GetTenantQuotas(false);
|
||
}
|
||
|
||
public IEnumerable<TenantQuota> GetTenantQuotas(bool all)
|
||
{
|
||
return QuotaService.GetTenantQuotas().Where(q => q.Id < 0 && (all || q.Visible)).OrderByDescending(q => q.Id).ToList();
|
||
}
|
||
|
||
public TenantQuota GetTenantQuota(int tenant)
|
||
{
|
||
// если в tenants_quota есть строка, с данным идентификатором портала, то в качестве квоты берется именно она
|
||
var q = QuotaService.GetTenantQuota(tenant) ?? QuotaService.GetTenantQuota(Tenant.DEFAULT_TENANT) ?? TenantQuota.Default;
|
||
if (q.Id != tenant && TariffService != null)
|
||
{
|
||
var tariffQuota = QuotaService.GetTenantQuota(TariffService.GetTariff(tenant).QuotaId);
|
||
if (tariffQuota != null)
|
||
{
|
||
return tariffQuota;
|
||
}
|
||
}
|
||
return q;
|
||
}
|
||
|
||
public IDictionary<string, IEnumerable<Tuple<string, decimal>>> GetProductPriceInfo(bool all = true)
|
||
{
|
||
var productIds = GetTenantQuotas(all)
|
||
.Select(p => p.AvangateId)
|
||
.Where(id => !string.IsNullOrEmpty(id))
|
||
.Distinct()
|
||
.ToArray();
|
||
return TariffService.GetProductPriceInfo(productIds);
|
||
}
|
||
|
||
public TenantQuota SaveTenantQuota(TenantQuota quota)
|
||
{
|
||
quota = QuotaService.SaveTenantQuota(quota);
|
||
return quota;
|
||
}
|
||
|
||
public void SetTenantQuotaRow(TenantQuotaRow row, bool exchange)
|
||
{
|
||
QuotaService.SetTenantQuotaRow(row, exchange);
|
||
}
|
||
|
||
public List<TenantQuotaRow> FindTenantQuotaRows(TenantQuotaRowQuery query)
|
||
{
|
||
return QuotaService.FindTenantQuotaRows(query).ToList();
|
||
}
|
||
}
|
||
|
||
public static class TenantManagerConfigExtension
|
||
{
|
||
public static DIHelper AddTenantManagerService(this DIHelper services)
|
||
{
|
||
services.TryAddScoped<TenantManager>();
|
||
services.TryAddScoped<IConfigureOptions<TenantManager>, ConfigureTenantManager>();
|
||
|
||
return services
|
||
.AddTenantService()
|
||
.AddQuotaService()
|
||
.AddTariffService()
|
||
.AddCoreBaseSettingsService()
|
||
.AddCoreSettingsService();
|
||
}
|
||
}
|
||
}
|