Преглед изворни кода

Create IRequiresRegistration. First part of #31

Eric Reed пре 12 година
родитељ
комит
d3b37a73cc

+ 2 - 0
MediaBrowser.Common.Implementations/BaseApplicationHost.cs

@@ -226,6 +226,8 @@ namespace MediaBrowser.Common.Implementations
 
                 Task.Run(() => ConfigureAutoRunAtStartup());
 
+                Task.Run(() => SecurityManager.LoadAllRegistrationInfo());
+
                 ConfigurationManager.ConfigurationUpdated += ConfigurationManager_ConfigurationUpdated;
             });
         }

+ 18 - 1
MediaBrowser.Common.Implementations/Security/PluginSecurityManager.cs

@@ -1,4 +1,5 @@
-using MediaBrowser.Common.Configuration;
+using System.Collections.Generic;
+using MediaBrowser.Common.Configuration;
 using MediaBrowser.Common.Security;
 using MediaBrowser.Model.Serialization;
 using Mediabrowser.Model.Entities;
@@ -7,6 +8,7 @@ using MediaBrowser.Common.Net;
 using System;
 using System.Threading;
 using System.Threading.Tasks;
+using System.Linq;
 
 namespace MediaBrowser.Common.Implementations.Security
 {
@@ -44,6 +46,7 @@ namespace MediaBrowser.Common.Implementations.Security
         private IHttpClient _httpClient;
         private IJsonSerializer _jsonSerializer;
         private IApplicationHost _appHost;
+        private IEnumerable<IRequiresRegistration> _registeredEntities; 
 
         /// <summary>
         /// Initializes a new instance of the <see cref="PluginSecurityManager" /> class.
@@ -58,9 +61,23 @@ namespace MediaBrowser.Common.Implementations.Security
             _appHost = appHost;
             _httpClient = httpClient;
             _jsonSerializer = jsonSerializer;
+            _registeredEntities = _appHost.GetExports<IRequiresRegistration>();
             MBRegistration.Init(appPaths);
         }
 
+        /// <summary>
+        /// Load all registration info for all entities that require registration
+        /// </summary>
+        /// <returns></returns>
+        public async Task LoadAllRegistrationInfo()
+        {
+            var tasks = new List<Task>();
+
+            tasks.AddRange(_registeredEntities.Select(i => i.LoadRegistrationInfoAsync()));
+            await Task.WhenAll(tasks);
+            ResetSupporterInfo();
+        }
+
         /// <summary>
         /// Gets the registration status.
         /// </summary>

+ 1 - 0
MediaBrowser.Common/MediaBrowser.Common.csproj

@@ -100,6 +100,7 @@
     <Compile Include="ScheduledTasks\IntervalTrigger.cs" />
     <Compile Include="ScheduledTasks\TaskCompletionEventArgs.cs" />
     <Compile Include="ScheduledTasks\WeeklyTrigger.cs" />
+    <Compile Include="Security\IRequiresRegistration.cs" />
     <Compile Include="Security\ISecurityManager.cs" />
     <Compile Include="Updates\IPackageManager.cs" />
   </ItemGroup>

+ 9 - 0
MediaBrowser.Common/Security/IRequiresRegistration.cs

@@ -0,0 +1,9 @@
+using System.Threading.Tasks;
+
+namespace MediaBrowser.Common.Security
+{
+    public interface IRequiresRegistration
+    {
+        Task LoadRegistrationInfoAsync();
+    }
+}

+ 6 - 0
MediaBrowser.Common/Security/ISecurityManager.cs

@@ -30,5 +30,11 @@ namespace MediaBrowser.Common.Security
         /// <param name="mb2Equivalent">The MB2 equivalent.</param>
         /// <returns>Task{MBRegistrationRecord}.</returns>
         Task<MBRegistrationRecord> GetRegistrationStatus(string feature, string mb2Equivalent = null);
+
+        /// <summary>
+        /// Load all registration info for all entities that require registration
+        /// </summary>
+        /// <returns></returns>
+        Task LoadAllRegistrationInfo();
     }
 }