ApplicationHost.cs 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452
  1. using System.Diagnostics;
  2. using System.Net.Cache;
  3. using System.Net.Http;
  4. using System.Net.Sockets;
  5. using MediaBrowser.Api;
  6. using MediaBrowser.Common;
  7. using MediaBrowser.Common.Configuration;
  8. using MediaBrowser.Common.Constants;
  9. using MediaBrowser.Common.Extensions;
  10. using MediaBrowser.Common.Implementations;
  11. using MediaBrowser.Common.Implementations.ScheduledTasks;
  12. using MediaBrowser.Common.IO;
  13. using MediaBrowser.Common.Net;
  14. using MediaBrowser.Controller;
  15. using MediaBrowser.Controller.Configuration;
  16. using MediaBrowser.Controller.Drawing;
  17. using MediaBrowser.Controller.Entities;
  18. using MediaBrowser.Controller.IO;
  19. using MediaBrowser.Controller.Library;
  20. using MediaBrowser.Controller.Localization;
  21. using MediaBrowser.Controller.MediaInfo;
  22. using MediaBrowser.Controller.Persistence;
  23. using MediaBrowser.Controller.Plugins;
  24. using MediaBrowser.Controller.Providers;
  25. using MediaBrowser.Controller.Resolvers;
  26. using MediaBrowser.Controller.Sorting;
  27. using MediaBrowser.Controller.Updates;
  28. using MediaBrowser.Controller.Weather;
  29. using MediaBrowser.IsoMounter;
  30. using MediaBrowser.Model.IO;
  31. using MediaBrowser.Model.MediaInfo;
  32. using MediaBrowser.Model.System;
  33. using MediaBrowser.Model.Updates;
  34. using MediaBrowser.Server.Implementations;
  35. using MediaBrowser.Server.Implementations.BdInfo;
  36. using MediaBrowser.Server.Implementations.Configuration;
  37. using MediaBrowser.Server.Implementations.HttpServer;
  38. using MediaBrowser.Server.Implementations.IO;
  39. using MediaBrowser.Server.Implementations.Library;
  40. using MediaBrowser.Server.Implementations.Providers;
  41. using MediaBrowser.Server.Implementations.ServerManager;
  42. using MediaBrowser.Server.Implementations.Udp;
  43. using MediaBrowser.Server.Implementations.Updates;
  44. using MediaBrowser.Server.Implementations.WebSocket;
  45. using MediaBrowser.ServerApplication.Implementations;
  46. using MediaBrowser.WebDashboard.Api;
  47. using System;
  48. using System.Collections.Generic;
  49. using System.IO;
  50. using System.Linq;
  51. using System.Reflection;
  52. using System.Threading;
  53. using System.Threading.Tasks;
  54. namespace MediaBrowser.ServerApplication
  55. {
  56. /// <summary>
  57. /// Class CompositionRoot
  58. /// </summary>
  59. public class ApplicationHost : BaseApplicationHost<ServerApplicationPaths>, IServerApplicationHost
  60. {
  61. private const int UdpServerPort = 7359;
  62. /// <summary>
  63. /// Gets the server kernel.
  64. /// </summary>
  65. /// <value>The server kernel.</value>
  66. protected Kernel ServerKernel { get; set; }
  67. /// <summary>
  68. /// Gets the server configuration manager.
  69. /// </summary>
  70. /// <value>The server configuration manager.</value>
  71. public IServerConfigurationManager ServerConfigurationManager
  72. {
  73. get { return (IServerConfigurationManager)ConfigurationManager; }
  74. }
  75. /// <summary>
  76. /// Gets the name of the log file prefix.
  77. /// </summary>
  78. /// <value>The name of the log file prefix.</value>
  79. protected override string LogFilePrefixName
  80. {
  81. get { return "Server"; }
  82. }
  83. /// <summary>
  84. /// Gets the configuration manager.
  85. /// </summary>
  86. /// <returns>IConfigurationManager.</returns>
  87. protected override IConfigurationManager GetConfigurationManager()
  88. {
  89. return new ServerConfigurationManager(ApplicationPaths, LogManager, XmlSerializer);
  90. }
  91. /// <summary>
  92. /// Gets or sets the installation manager.
  93. /// </summary>
  94. /// <value>The installation manager.</value>
  95. private IInstallationManager InstallationManager { get; set; }
  96. /// <summary>
  97. /// Gets or sets the server manager.
  98. /// </summary>
  99. /// <value>The server manager.</value>
  100. private IServerManager ServerManager { get; set; }
  101. /// <summary>
  102. /// Gets or sets the user manager.
  103. /// </summary>
  104. /// <value>The user manager.</value>
  105. public IUserManager UserManager { get; set; }
  106. /// <summary>
  107. /// Gets or sets the library manager.
  108. /// </summary>
  109. /// <value>The library manager.</value>
  110. internal ILibraryManager LibraryManager { get; set; }
  111. /// <summary>
  112. /// Gets or sets the directory watchers.
  113. /// </summary>
  114. /// <value>The directory watchers.</value>
  115. private IDirectoryWatchers DirectoryWatchers { get; set; }
  116. /// <summary>
  117. /// Gets or sets the provider manager.
  118. /// </summary>
  119. /// <value>The provider manager.</value>
  120. private IProviderManager ProviderManager { get; set; }
  121. /// <summary>
  122. /// Gets or sets the zip client.
  123. /// </summary>
  124. /// <value>The zip client.</value>
  125. private IZipClient ZipClient { get; set; }
  126. /// <summary>
  127. /// Gets or sets the HTTP server.
  128. /// </summary>
  129. /// <value>The HTTP server.</value>
  130. private IHttpServer HttpServer { get; set; }
  131. /// <summary>
  132. /// Gets or sets the UDP server.
  133. /// </summary>
  134. /// <value>The UDP server.</value>
  135. private UdpServer UdpServer { get; set; }
  136. /// <summary>
  137. /// The full path to our startmenu shortcut
  138. /// </summary>
  139. protected override string ProductShortcutPath
  140. {
  141. get { return Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.StartMenu), "Media Browser 3", "Media Browser Server.lnk"); }
  142. }
  143. /// <summary>
  144. /// Runs the startup tasks.
  145. /// </summary>
  146. /// <returns>Task.</returns>
  147. protected override async Task RunStartupTasks()
  148. {
  149. // Do these before allowing the base method to run, which will invoke startup scheduled tasks
  150. await ServerKernel.LoadRepositories(ServerConfigurationManager).ConfigureAwait(false);
  151. await base.RunStartupTasks().ConfigureAwait(false);
  152. DirectoryWatchers.Start();
  153. Parallel.ForEach(GetExports<IServerEntryPoint>(), entryPoint => entryPoint.Run());
  154. }
  155. /// <summary>
  156. /// Registers resources that classes will depend on
  157. /// </summary>
  158. /// <returns>Task.</returns>
  159. protected override async Task RegisterResources()
  160. {
  161. ServerKernel = new Kernel(ServerConfigurationManager);
  162. await base.RegisterResources().ConfigureAwait(false);
  163. RegisterSingleInstance<IHttpResultFactory>(new HttpResultFactory(LogManager));
  164. RegisterSingleInstance<IServerApplicationHost>(this);
  165. RegisterSingleInstance<IServerApplicationPaths>(ApplicationPaths);
  166. RegisterSingleInstance(ServerKernel);
  167. RegisterSingleInstance(ServerConfigurationManager);
  168. RegisterSingleInstance<IWebSocketServer>(() => new AlchemyServer(Logger));
  169. RegisterSingleInstance<IIsoManager>(new PismoIsoManager(Logger));
  170. RegisterSingleInstance<IBlurayExaminer>(new BdInfoExaminer());
  171. ZipClient = new DotNetZipClient();
  172. RegisterSingleInstance(ZipClient);
  173. HttpServer = ServerFactory.CreateServer(this, ProtobufSerializer, Logger, "Media Browser", "index.html");
  174. RegisterSingleInstance(HttpServer, false);
  175. ServerManager = new ServerManager(this, JsonSerializer, Logger, ServerConfigurationManager, ServerKernel);
  176. RegisterSingleInstance(ServerManager);
  177. UserManager = new UserManager(ServerKernel, Logger, ServerConfigurationManager);
  178. RegisterSingleInstance(UserManager);
  179. LibraryManager = new LibraryManager(ServerKernel, Logger, TaskManager, UserManager, ServerConfigurationManager);
  180. RegisterSingleInstance(LibraryManager);
  181. InstallationManager = new InstallationManager(HttpClient, PackageManager, JsonSerializer, Logger, this);
  182. RegisterSingleInstance(InstallationManager);
  183. DirectoryWatchers = new DirectoryWatchers(LogManager, TaskManager, LibraryManager, ServerConfigurationManager);
  184. RegisterSingleInstance(DirectoryWatchers);
  185. ProviderManager = new ProviderManager(HttpClient, ServerConfigurationManager, DirectoryWatchers, LogManager);
  186. RegisterSingleInstance(ProviderManager);
  187. SetKernelProperties();
  188. SetStaticProperties();
  189. }
  190. /// <summary>
  191. /// Sets the kernel properties.
  192. /// </summary>
  193. private void SetKernelProperties()
  194. {
  195. ServerKernel.FFMpegManager = new FFMpegManager(ServerKernel, ZipClient, JsonSerializer, ProtobufSerializer, LogManager, ApplicationPaths);
  196. ServerKernel.ImageManager = new ImageManager(ServerKernel, ProtobufSerializer, LogManager.GetLogger("ImageManager"), ApplicationPaths);
  197. Parallel.Invoke(
  198. () => ServerKernel.UserDataRepositories = GetExports<IUserDataRepository>(),
  199. () => ServerKernel.UserRepositories = GetExports<IUserRepository>(),
  200. () => ServerKernel.DisplayPreferencesRepositories = GetExports<IDisplayPreferencesRepository>(),
  201. () => ServerKernel.ItemRepositories = GetExports<IItemRepository>(),
  202. () => ServerKernel.WeatherProviders = GetExports<IWeatherProvider>(),
  203. () => ServerKernel.ImageEnhancers = GetExports<IImageEnhancer>().OrderBy(e => e.Priority).ToArray(),
  204. () => ServerKernel.StringFiles = GetExports<LocalizedStringData>()
  205. );
  206. }
  207. /// <summary>
  208. /// Dirty hacks
  209. /// </summary>
  210. private void SetStaticProperties()
  211. {
  212. // For now there's no real way to inject these properly
  213. BaseItem.Logger = LogManager.GetLogger("BaseItem");
  214. BaseItem.ConfigurationManager = ServerConfigurationManager;
  215. BaseItem.LibraryManager = LibraryManager;
  216. BaseItem.ProviderManager = ProviderManager;
  217. User.XmlSerializer = XmlSerializer;
  218. User.UserManager = UserManager;
  219. Ratings.ConfigurationManager = ServerConfigurationManager;
  220. LocalizedStrings.ApplicationPaths = ApplicationPaths;
  221. }
  222. /// <summary>
  223. /// Finds the parts.
  224. /// </summary>
  225. protected override void FindParts()
  226. {
  227. base.FindParts();
  228. //if (IsFirstRun)
  229. {
  230. RegisterServerWithAdministratorAccess();
  231. }
  232. Parallel.Invoke(
  233. () =>
  234. {
  235. HttpServer.Init(GetExports<IRestfulService>(false));
  236. ServerManager.AddWebSocketListeners(GetExports<IWebSocketListener>(false));
  237. ServerManager.Start();
  238. },
  239. () => LibraryManager.AddParts(GetExports<IResolverIgnoreRule>(), GetExports<IVirtualFolderCreator>(), GetExports<IItemResolver>(), GetExports<IIntroProvider>(), GetExports<IBaseItemComparer>()),
  240. () => ProviderManager.AddMetadataProviders(GetExports<BaseMetadataProvider>().OrderBy(e => e.Priority).ToArray())
  241. );
  242. UdpServer = new UdpServer(Logger, NetworkManager, ServerConfigurationManager);
  243. try
  244. {
  245. UdpServer.Start(UdpServerPort);
  246. }
  247. catch (SocketException ex)
  248. {
  249. Logger.ErrorException("Failed to start UDP Server", ex);
  250. }
  251. }
  252. /// <summary>
  253. /// Restarts this instance.
  254. /// </summary>
  255. public override void Restart()
  256. {
  257. App.Instance.Restart();
  258. }
  259. /// <summary>
  260. /// Gets or sets a value indicating whether this instance can self update.
  261. /// </summary>
  262. /// <value><c>true</c> if this instance can self update; otherwise, <c>false</c>.</value>
  263. public override bool CanSelfUpdate
  264. {
  265. get { return ConfigurationManager.CommonConfiguration.EnableAutoUpdate; }
  266. }
  267. /// <summary>
  268. /// Releases unmanaged and - optionally - managed resources.
  269. /// </summary>
  270. /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
  271. protected override void Dispose(bool dispose)
  272. {
  273. if (dispose)
  274. {
  275. if (UdpServer != null)
  276. {
  277. UdpServer.Dispose();
  278. }
  279. }
  280. base.Dispose(dispose);
  281. }
  282. /// <summary>
  283. /// Checks for update.
  284. /// </summary>
  285. /// <param name="cancellationToken">The cancellation token.</param>
  286. /// <param name="progress">The progress.</param>
  287. /// <returns>Task{CheckForUpdateResult}.</returns>
  288. public async override Task<CheckForUpdateResult> CheckForApplicationUpdate(CancellationToken cancellationToken, IProgress<double> progress)
  289. {
  290. var availablePackages = await PackageManager.GetAvailablePackages(CancellationToken.None).ConfigureAwait(false);
  291. var version = InstallationManager.GetLatestCompatibleVersion(availablePackages, Constants.MBServerPkgName, ConfigurationManager.CommonConfiguration.SystemUpdateLevel);
  292. return version != null ? new CheckForUpdateResult { AvailableVersion = version.version, IsUpdateAvailable = version.version > ApplicationVersion, Package = version } :
  293. new CheckForUpdateResult { AvailableVersion = ApplicationVersion, IsUpdateAvailable = false };
  294. }
  295. /// <summary>
  296. /// Gets the composable part assemblies.
  297. /// </summary>
  298. /// <returns>IEnumerable{Assembly}.</returns>
  299. protected override IEnumerable<Assembly> GetComposablePartAssemblies()
  300. {
  301. // Gets all plugin assemblies by first reading all bytes of the .dll and calling Assembly.Load against that
  302. // This will prevent the .dll file from getting locked, and allow us to replace it when needed
  303. foreach (var pluginAssembly in Directory
  304. .EnumerateFiles(ApplicationPaths.PluginsPath, "*.dll", SearchOption.TopDirectoryOnly)
  305. .Select(LoadAssembly).Where(a => a != null))
  306. {
  307. yield return pluginAssembly;
  308. }
  309. // Include composable parts in the Api assembly
  310. yield return typeof(ApiEntryPoint).Assembly;
  311. // Include composable parts in the Dashboard assembly
  312. yield return typeof(DashboardInfo).Assembly;
  313. // Include composable parts in the Model assembly
  314. yield return typeof(SystemInfo).Assembly;
  315. // Include composable parts in the Common assembly
  316. yield return typeof(IApplicationHost).Assembly;
  317. // Include composable parts in the Controller assembly
  318. yield return typeof(Kernel).Assembly;
  319. // Common implementations
  320. yield return typeof(TaskManager).Assembly;
  321. // Server implementations
  322. yield return typeof(ServerApplicationPaths).Assembly;
  323. // Include composable parts in the running assembly
  324. yield return GetType().Assembly;
  325. }
  326. private readonly Guid _systemId = Environment.MachineName.GetMD5();
  327. /// <summary>
  328. /// Gets the system status.
  329. /// </summary>
  330. /// <returns>SystemInfo.</returns>
  331. public virtual SystemInfo GetSystemInfo()
  332. {
  333. return new SystemInfo
  334. {
  335. HasPendingRestart = HasPendingRestart,
  336. Version = ApplicationVersion.ToString(),
  337. IsNetworkDeployed = CanSelfUpdate,
  338. WebSocketPortNumber = ServerManager.WebSocketPortNumber,
  339. SupportsNativeWebSocket = ServerManager.SupportsNativeWebSocket,
  340. FailedPluginAssemblies = FailedAssemblies.ToArray(),
  341. InProgressInstallations = InstallationManager.CurrentInstallations.Select(i => i.Item1).ToArray(),
  342. CompletedInstallations = InstallationManager.CompletedInstallations.ToArray(),
  343. Id = _systemId
  344. };
  345. }
  346. /// <summary>
  347. /// Shuts down.
  348. /// </summary>
  349. public override void Shutdown()
  350. {
  351. App.Instance.Dispatcher.Invoke(App.Instance.Shutdown);
  352. }
  353. /// <summary>
  354. /// Registers the server with administrator access.
  355. /// </summary>
  356. private void RegisterServerWithAdministratorAccess()
  357. {
  358. Logger.Info("Requesting administrative access to authorize http server");
  359. // Create a temp file path to extract the bat file to
  360. var tmpFile = Path.Combine(ConfigurationManager.CommonApplicationPaths.TempDirectory, Guid.NewGuid() + ".bat");
  361. // Extract the bat file
  362. using (var stream = Assembly.GetExecutingAssembly().GetManifestResourceStream("MediaBrowser.ServerApplication.RegisterServer.bat"))
  363. {
  364. using (var fileStream = File.Create(tmpFile))
  365. {
  366. stream.CopyTo(fileStream);
  367. }
  368. }
  369. var startInfo = new ProcessStartInfo
  370. {
  371. FileName = tmpFile,
  372. Arguments = string.Format("{0} {1} {2} {3}", ServerConfigurationManager.Configuration.HttpServerPortNumber,
  373. ServerKernel.HttpServerUrlPrefix,
  374. UdpServerPort,
  375. ServerConfigurationManager.Configuration.LegacyWebSocketPortNumber),
  376. CreateNoWindow = true,
  377. WindowStyle = ProcessWindowStyle.Hidden,
  378. Verb = "runas",
  379. ErrorDialog = false
  380. };
  381. using (var process = Process.Start(startInfo))
  382. {
  383. process.WaitForExit();
  384. }
  385. }
  386. }
  387. }