using System.Collections.Generic; using MediaBrowser.Common.Configuration; using MediaBrowser.Common.Security; using MediaBrowser.Model.Serialization; using Mediabrowser.Model.Entities; using Mediabrowser.PluginSecurity; using MediaBrowser.Common.Net; using System; using System.Threading; using System.Threading.Tasks; using System.Linq; namespace MediaBrowser.Common.Implementations.Security { /// /// Class PluginSecurityManager /// public class PluginSecurityManager : ISecurityManager { /// /// The _is MB supporter /// private bool? _isMBSupporter; /// /// The _is MB supporter initialized /// private bool _isMBSupporterInitialized; /// /// The _is MB supporter sync lock /// private object _isMBSupporterSyncLock = new object(); /// /// Gets a value indicating whether this instance is MB supporter. /// /// true if this instance is MB supporter; otherwise, false. public bool IsMBSupporter { get { LazyInitializer.EnsureInitialized(ref _isMBSupporter, ref _isMBSupporterInitialized, ref _isMBSupporterSyncLock, () => GetRegistrationStatus("MBSupporter").Result.IsRegistered); return _isMBSupporter.Value; } } private IHttpClient _httpClient; private IJsonSerializer _jsonSerializer; private IApplicationHost _appHost; private IEnumerable _registeredEntities; protected IEnumerable RegisteredEntities { get { return _registeredEntities ?? (_registeredEntities = _appHost.GetExports()); } } /// /// Initializes a new instance of the class. /// public PluginSecurityManager(IApplicationHost appHost, IHttpClient httpClient, IJsonSerializer jsonSerializer, IApplicationPaths appPaths) { if (httpClient == null) { throw new ArgumentNullException("httpClient"); } _appHost = appHost; _httpClient = httpClient; _jsonSerializer = jsonSerializer; MBRegistration.Init(appPaths); } /// /// Load all registration info for all entities that require registration /// /// public async Task LoadAllRegistrationInfo() { var tasks = new List(); ResetSupporterInfo(); tasks.AddRange(RegisteredEntities.Select(i => i.LoadRegistrationInfoAsync())); await Task.WhenAll(tasks); } /// /// Gets the registration status. /// /// The feature. /// The MB2 equivalent. /// Task{MBRegistrationRecord}. public async Task GetRegistrationStatus(string feature, string mb2Equivalent = null) { return await MBRegistration.GetRegistrationStatus(_httpClient, _jsonSerializer, feature, mb2Equivalent).ConfigureAwait(false); } /// /// Gets or sets the supporter key. /// /// The supporter key. public string SupporterKey { get { return MBRegistration.SupporterKey; } set { if (value != MBRegistration.SupporterKey) { MBRegistration.SupporterKey = value; // re-load registration info Task.Run(() => LoadAllRegistrationInfo()); } } } /// /// Gets or sets the legacy key. /// /// The legacy key. public string LegacyKey { get { return MBRegistration.LegacyKey; } set { if (value != MBRegistration.LegacyKey) { MBRegistration.LegacyKey = value; // re-load registration info Task.Run(() => LoadAllRegistrationInfo()); } } } /// /// Resets the supporter info. /// private void ResetSupporterInfo() { _isMBSupporter = null; _isMBSupporterInitialized = false; } } }