mirror of
				https://github.com/jellyfin/jellyfin.git
				synced 2025-10-31 00:01:12 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			696 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
			
		
		
	
	
			696 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
| using System;
 | |
| using System.Collections.Concurrent;
 | |
| using System.Collections.Generic;
 | |
| using System.IO;
 | |
| using System.Linq;
 | |
| using System.Threading;
 | |
| using System.Threading.Tasks;
 | |
| using MediaBrowser.Common;
 | |
| using MediaBrowser.Common.Configuration;
 | |
| using MediaBrowser.Common.Events;
 | |
| using MediaBrowser.Common.Net;
 | |
| using MediaBrowser.Common.Plugins;
 | |
| using MediaBrowser.Common.Progress;
 | |
| using MediaBrowser.Common.Security;
 | |
| using MediaBrowser.Common.Updates;
 | |
| using MediaBrowser.Model.Cryptography;
 | |
| using MediaBrowser.Model.Events;
 | |
| using MediaBrowser.Model.IO;
 | |
| using Microsoft.Extensions.Logging;
 | |
| using MediaBrowser.Model.Serialization;
 | |
| using MediaBrowser.Model.Updates;
 | |
| using MediaBrowser.Controller.Configuration;
 | |
| 
 | |
| namespace Emby.Server.Implementations.Updates
 | |
| {
 | |
|     /// <summary>
 | |
|     /// Manages all install, uninstall and update operations (both plugins and system)
 | |
|     /// </summary>
 | |
|     public class InstallationManager : IInstallationManager
 | |
|     {
 | |
|         public event EventHandler<InstallationEventArgs> PackageInstalling;
 | |
|         public event EventHandler<InstallationEventArgs> PackageInstallationCompleted;
 | |
|         public event EventHandler<InstallationFailedEventArgs> PackageInstallationFailed;
 | |
|         public event EventHandler<InstallationEventArgs> PackageInstallationCancelled;
 | |
| 
 | |
|         /// <summary>
 | |
|         /// The current installations
 | |
|         /// </summary>
 | |
|         public List<Tuple<InstallationInfo, CancellationTokenSource>> CurrentInstallations { get; set; }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// The completed installations
 | |
|         /// </summary>
 | |
|         private ConcurrentBag<InstallationInfo> CompletedInstallationsInternal { get; set; }
 | |
| 
 | |
|         public IEnumerable<InstallationInfo> CompletedInstallations
 | |
|         {
 | |
|             get { return CompletedInstallationsInternal; }
 | |
|         }
 | |
| 
 | |
|         #region PluginUninstalled Event
 | |
|         /// <summary>
 | |
|         /// Occurs when [plugin uninstalled].
 | |
|         /// </summary>
 | |
|         public event EventHandler<GenericEventArgs<IPlugin>> PluginUninstalled;
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Called when [plugin uninstalled].
 | |
|         /// </summary>
 | |
|         /// <param name="plugin">The plugin.</param>
 | |
|         private void OnPluginUninstalled(IPlugin plugin)
 | |
|         {
 | |
|             EventHelper.FireEventIfNotNull(PluginUninstalled, this, new GenericEventArgs<IPlugin> { Argument = plugin }, _logger);
 | |
|         }
 | |
|         #endregion
 | |
| 
 | |
|         #region PluginUpdated Event
 | |
|         /// <summary>
 | |
|         /// Occurs when [plugin updated].
 | |
|         /// </summary>
 | |
|         public event EventHandler<GenericEventArgs<Tuple<IPlugin, PackageVersionInfo>>> PluginUpdated;
 | |
|         /// <summary>
 | |
|         /// Called when [plugin updated].
 | |
|         /// </summary>
 | |
|         /// <param name="plugin">The plugin.</param>
 | |
|         /// <param name="newVersion">The new version.</param>
 | |
|         private void OnPluginUpdated(IPlugin plugin, PackageVersionInfo newVersion)
 | |
|         {
 | |
|             _logger.LogInformation("Plugin updated: {0} {1} {2}", newVersion.name, newVersion.versionStr ?? string.Empty, newVersion.classification);
 | |
| 
 | |
|             EventHelper.FireEventIfNotNull(PluginUpdated, this, new GenericEventArgs<Tuple<IPlugin, PackageVersionInfo>> { Argument = new Tuple<IPlugin, PackageVersionInfo>(plugin, newVersion) }, _logger);
 | |
| 
 | |
|             _applicationHost.NotifyPendingRestart();
 | |
|         }
 | |
|         #endregion
 | |
| 
 | |
|         #region PluginInstalled Event
 | |
|         /// <summary>
 | |
|         /// Occurs when [plugin updated].
 | |
|         /// </summary>
 | |
|         public event EventHandler<GenericEventArgs<PackageVersionInfo>> PluginInstalled;
 | |
|         /// <summary>
 | |
|         /// Called when [plugin installed].
 | |
|         /// </summary>
 | |
|         /// <param name="package">The package.</param>
 | |
|         private void OnPluginInstalled(PackageVersionInfo package)
 | |
|         {
 | |
|             _logger.LogInformation("New plugin installed: {0} {1} {2}", package.name, package.versionStr ?? string.Empty, package.classification);
 | |
| 
 | |
|             EventHelper.FireEventIfNotNull(PluginInstalled, this, new GenericEventArgs<PackageVersionInfo> { Argument = package }, _logger);
 | |
| 
 | |
|             _applicationHost.NotifyPendingRestart();
 | |
|         }
 | |
|         #endregion
 | |
| 
 | |
|         /// <summary>
 | |
|         /// The _logger
 | |
|         /// </summary>
 | |
|         private readonly ILogger _logger;
 | |
| 
 | |
|         private readonly IApplicationPaths _appPaths;
 | |
|         private readonly IHttpClient _httpClient;
 | |
|         private readonly IJsonSerializer _jsonSerializer;
 | |
|         private readonly ISecurityManager _securityManager;
 | |
|         private readonly IServerConfigurationManager _config;
 | |
|         private readonly IFileSystem _fileSystem;
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Gets the application host.
 | |
|         /// </summary>
 | |
|         /// <value>The application host.</value>
 | |
|         private readonly IApplicationHost _applicationHost;
 | |
| 
 | |
|         private readonly ICryptoProvider _cryptographyProvider;
 | |
| 
 | |
|         // netframework or netcore
 | |
|         private readonly string _packageRuntime;
 | |
| 
 | |
|         public InstallationManager(ILogger logger, IApplicationHost appHost, IApplicationPaths appPaths, IHttpClient httpClient, IJsonSerializer jsonSerializer, ISecurityManager securityManager, IServerConfigurationManager config, IFileSystem fileSystem, ICryptoProvider cryptographyProvider, string packageRuntime)
 | |
|         {
 | |
|             if (logger == null)
 | |
|             {
 | |
|                 throw new ArgumentNullException("logger");
 | |
|             }
 | |
| 
 | |
|             CurrentInstallations = new List<Tuple<InstallationInfo, CancellationTokenSource>>();
 | |
|             CompletedInstallationsInternal = new ConcurrentBag<InstallationInfo>();
 | |
| 
 | |
|             _applicationHost = appHost;
 | |
|             _appPaths = appPaths;
 | |
|             _httpClient = httpClient;
 | |
|             _jsonSerializer = jsonSerializer;
 | |
|             _securityManager = securityManager;
 | |
|             _config = config;
 | |
|             _fileSystem = fileSystem;
 | |
|             _cryptographyProvider = cryptographyProvider;
 | |
|             _packageRuntime = packageRuntime;
 | |
|             _logger = logger;
 | |
|         }
 | |
| 
 | |
|         private Version GetPackageVersion(PackageVersionInfo version)
 | |
|         {
 | |
|             return new Version(ValueOrDefault(version.versionStr, "0.0.0.1"));
 | |
|         }
 | |
| 
 | |
|         private static string ValueOrDefault(string str, string def)
 | |
|         {
 | |
|             return string.IsNullOrEmpty(str) ? def : str;
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Gets all available packages.
 | |
|         /// </summary>
 | |
|         /// <returns>Task{List{PackageInfo}}.</returns>
 | |
|         public async Task<List<PackageInfo>> GetAvailablePackages(CancellationToken cancellationToken,
 | |
|             bool withRegistration = true,
 | |
|             string packageType = null,
 | |
|             Version applicationVersion = null)
 | |
|         {
 | |
|             if (withRegistration)
 | |
|             {
 | |
|                 var data = new Dictionary<string, string>
 | |
|                 {
 | |
|                     { "key", _securityManager.SupporterKey },
 | |
|                     { "mac", _applicationHost.SystemId },
 | |
|                     { "systemid", _applicationHost.SystemId }
 | |
|                 };
 | |
| 
 | |
|                 var options = new HttpRequestOptions
 | |
|                 {
 | |
|                     Url = "https://www.mb3admin.local/admin/service/package/retrieveall?includeAllRuntimes=true",
 | |
|                     CancellationToken = cancellationToken
 | |
|                 };
 | |
| 
 | |
|                 options.SetPostData(data);
 | |
| 
 | |
|                 using (var response = await _httpClient.SendAsync(options, "POST").ConfigureAwait(false))
 | |
|                 {
 | |
|                     using (var json = response.Content)
 | |
|                     {
 | |
|                         cancellationToken.ThrowIfCancellationRequested();
 | |
| 
 | |
|                         var packages = await _jsonSerializer.DeserializeFromStreamAsync<PackageInfo[]>(json).ConfigureAwait(false);
 | |
| 
 | |
|                         return FilterPackages(packages, packageType, applicationVersion);
 | |
|                     }
 | |
|                 }
 | |
|             }
 | |
|             else
 | |
|             {
 | |
|                 var packages = await GetAvailablePackagesWithoutRegistrationInfo(cancellationToken).ConfigureAwait(false);
 | |
| 
 | |
|                 return FilterPackages(packages, packageType, applicationVersion);
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Gets all available packages.
 | |
|         /// </summary>
 | |
|         /// <param name="cancellationToken">The cancellation token.</param>
 | |
|         /// <returns>Task{List{PackageInfo}}.</returns>
 | |
|         public async Task<List<PackageInfo>> GetAvailablePackagesWithoutRegistrationInfo(CancellationToken cancellationToken)
 | |
|         {
 | |
|             using (var response = await _httpClient.SendAsync(new HttpRequestOptions
 | |
|             {
 | |
|                 Url = "https://www.mb3admin.local/admin/service/EmbyPackages.json",
 | |
|                 CancellationToken = cancellationToken,
 | |
|                 Progress = new SimpleProgress<Double>(),
 | |
|                 CacheLength = GetCacheLength(),
 | |
|                 CacheMode = CacheMode.Unconditional
 | |
| 
 | |
|             }, "GET").ConfigureAwait(false))
 | |
|             {
 | |
|                 using (var stream = response.Content)
 | |
|                 {
 | |
|                     return FilterPackages(await _jsonSerializer.DeserializeFromStreamAsync<PackageInfo[]>(stream).ConfigureAwait(false));
 | |
|                 }
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         private PackageVersionClass GetSystemUpdateLevel()
 | |
|         {
 | |
|             return _applicationHost.SystemUpdateLevel;
 | |
|         }
 | |
| 
 | |
|         private TimeSpan GetCacheLength()
 | |
|         {
 | |
|             return TimeSpan.FromMinutes(3);
 | |
|         }
 | |
| 
 | |
|         protected List<PackageInfo> FilterPackages(IEnumerable<PackageInfo> packages)
 | |
|         {
 | |
|             var list = new List<PackageInfo>();
 | |
| 
 | |
|             foreach (var package in packages)
 | |
|             {
 | |
|                 var versions = new List<PackageVersionInfo>();
 | |
|                 foreach (var version in package.versions)
 | |
|                 {
 | |
|                     if (string.IsNullOrEmpty(version.sourceUrl))
 | |
|                     {
 | |
|                         continue;
 | |
|                     }
 | |
| 
 | |
|                     if (string.IsNullOrEmpty(version.runtimes) || version.runtimes.IndexOf(_packageRuntime, StringComparison.OrdinalIgnoreCase) == -1)
 | |
|                     {
 | |
|                         continue;
 | |
|                     }
 | |
| 
 | |
|                     versions.Add(version);
 | |
|                 }
 | |
| 
 | |
|                 package.versions = versions
 | |
|                     .OrderByDescending(GetPackageVersion)
 | |
|                     .ToArray();
 | |
| 
 | |
|                 if (package.versions.Length == 0)
 | |
|                 {
 | |
|                     continue;
 | |
|                 }
 | |
| 
 | |
|                 list.Add(package);
 | |
|             }
 | |
| 
 | |
|             // Remove packages with no versions
 | |
|             return list;
 | |
|         }
 | |
| 
 | |
|         protected List<PackageInfo> FilterPackages(IEnumerable<PackageInfo> packages, string packageType, Version applicationVersion)
 | |
|         {
 | |
|             var packagesList = FilterPackages(packages);
 | |
| 
 | |
|             var returnList = new List<PackageInfo>();
 | |
| 
 | |
|             var filterOnPackageType = !string.IsNullOrEmpty(packageType);
 | |
| 
 | |
|             foreach (var p in packagesList)
 | |
|             {
 | |
|                 if (filterOnPackageType && !string.Equals(p.type, packageType, StringComparison.OrdinalIgnoreCase))
 | |
|                 {
 | |
|                     continue;
 | |
|                 }
 | |
| 
 | |
|                 // If an app version was supplied, filter the versions for each package to only include supported versions
 | |
|                 if (applicationVersion != null)
 | |
|                 {
 | |
|                     p.versions = p.versions.Where(v => IsPackageVersionUpToDate(v, applicationVersion)).ToArray();
 | |
|                 }
 | |
| 
 | |
|                 if (p.versions.Length == 0)
 | |
|                 {
 | |
|                     continue;
 | |
|                 }
 | |
| 
 | |
|                 returnList.Add(p);
 | |
|             }
 | |
| 
 | |
|             return returnList;
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Determines whether [is package version up to date] [the specified package version info].
 | |
|         /// </summary>
 | |
|         /// <param name="packageVersionInfo">The package version info.</param>
 | |
|         /// <param name="currentServerVersion">The current server version.</param>
 | |
|         /// <returns><c>true</c> if [is package version up to date] [the specified package version info]; otherwise, <c>false</c>.</returns>
 | |
|         private bool IsPackageVersionUpToDate(PackageVersionInfo packageVersionInfo, Version currentServerVersion)
 | |
|         {
 | |
|             if (string.IsNullOrEmpty(packageVersionInfo.requiredVersionStr))
 | |
|             {
 | |
|                 return true;
 | |
|             }
 | |
| 
 | |
|             Version requiredVersion;
 | |
| 
 | |
|             return Version.TryParse(packageVersionInfo.requiredVersionStr, out requiredVersion) && currentServerVersion >= requiredVersion;
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Gets the package.
 | |
|         /// </summary>
 | |
|         /// <param name="name">The name.</param>
 | |
|         /// <param name="guid">The assembly guid</param>
 | |
|         /// <param name="classification">The classification.</param>
 | |
|         /// <param name="version">The version.</param>
 | |
|         /// <returns>Task{PackageVersionInfo}.</returns>
 | |
|         public async Task<PackageVersionInfo> GetPackage(string name, string guid, PackageVersionClass classification, Version version)
 | |
|         {
 | |
|             var packages = await GetAvailablePackages(CancellationToken.None, false).ConfigureAwait(false);
 | |
| 
 | |
|             var package = packages.FirstOrDefault(p => string.Equals(p.guid, guid ?? "none", StringComparison.OrdinalIgnoreCase))
 | |
|                             ?? packages.FirstOrDefault(p => p.name.Equals(name, StringComparison.OrdinalIgnoreCase));
 | |
| 
 | |
|             if (package == null)
 | |
|             {
 | |
|                 return null;
 | |
|             }
 | |
| 
 | |
|             return package.versions.FirstOrDefault(v => GetPackageVersion(v).Equals(version) && v.classification == classification);
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Gets the latest compatible version.
 | |
|         /// </summary>
 | |
|         /// <param name="name">The name.</param>
 | |
|         /// <param name="guid">The assembly guid if this is a plug-in</param>
 | |
|         /// <param name="currentServerVersion">The current server version.</param>
 | |
|         /// <param name="classification">The classification.</param>
 | |
|         /// <returns>Task{PackageVersionInfo}.</returns>
 | |
|         public async Task<PackageVersionInfo> GetLatestCompatibleVersion(string name, string guid, Version currentServerVersion, PackageVersionClass classification = PackageVersionClass.Release)
 | |
|         {
 | |
|             var packages = await GetAvailablePackages(CancellationToken.None, false).ConfigureAwait(false);
 | |
| 
 | |
|             return GetLatestCompatibleVersion(packages, name, guid, currentServerVersion, classification);
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Gets the latest compatible version.
 | |
|         /// </summary>
 | |
|         /// <param name="availablePackages">The available packages.</param>
 | |
|         /// <param name="name">The name.</param>
 | |
|         /// <param name="currentServerVersion">The current server version.</param>
 | |
|         /// <param name="classification">The classification.</param>
 | |
|         /// <returns>PackageVersionInfo.</returns>
 | |
|         public PackageVersionInfo GetLatestCompatibleVersion(IEnumerable<PackageInfo> availablePackages, string name, string guid, Version currentServerVersion, PackageVersionClass classification = PackageVersionClass.Release)
 | |
|         {
 | |
|             var package = availablePackages.FirstOrDefault(p => string.Equals(p.guid, guid ?? "none", StringComparison.OrdinalIgnoreCase))
 | |
|                             ?? availablePackages.FirstOrDefault(p => p.name.Equals(name, StringComparison.OrdinalIgnoreCase));
 | |
| 
 | |
|             if (package == null)
 | |
|             {
 | |
|                 return null;
 | |
|             }
 | |
| 
 | |
|             return package.versions
 | |
|                 .OrderByDescending(GetPackageVersion)
 | |
|                 .FirstOrDefault(v => v.classification <= classification && IsPackageVersionUpToDate(v, currentServerVersion));
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Gets the available plugin updates.
 | |
|         /// </summary>
 | |
|         /// <param name="applicationVersion">The current server version.</param>
 | |
|         /// <param name="withAutoUpdateEnabled">if set to <c>true</c> [with auto update enabled].</param>
 | |
|         /// <param name="cancellationToken">The cancellation token.</param>
 | |
|         /// <returns>Task{IEnumerable{PackageVersionInfo}}.</returns>
 | |
|         public async Task<IEnumerable<PackageVersionInfo>> GetAvailablePluginUpdates(Version applicationVersion, bool withAutoUpdateEnabled, CancellationToken cancellationToken)
 | |
|         {
 | |
|             var catalog = await GetAvailablePackagesWithoutRegistrationInfo(cancellationToken).ConfigureAwait(false);
 | |
| 
 | |
|             var systemUpdateLevel = GetSystemUpdateLevel();
 | |
| 
 | |
|             // Figure out what needs to be installed
 | |
|             return _applicationHost.Plugins.Select(p =>
 | |
|             {
 | |
|                 var latestPluginInfo = GetLatestCompatibleVersion(catalog, p.Name, p.Id.ToString(), applicationVersion, systemUpdateLevel);
 | |
| 
 | |
|                 return latestPluginInfo != null && GetPackageVersion(latestPluginInfo) > p.Version ? latestPluginInfo : null;
 | |
| 
 | |
|             }).Where(i => i != null)
 | |
|             .Where(p => !string.IsNullOrEmpty(p.sourceUrl) && !CompletedInstallations.Any(i => string.Equals(i.AssemblyGuid, p.guid, StringComparison.OrdinalIgnoreCase)));
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Installs the package.
 | |
|         /// </summary>
 | |
|         /// <param name="package">The package.</param>
 | |
|         /// <param name="isPlugin">if set to <c>true</c> [is plugin].</param>
 | |
|         /// <param name="progress">The progress.</param>
 | |
|         /// <param name="cancellationToken">The cancellation token.</param>
 | |
|         /// <returns>Task.</returns>
 | |
|         /// <exception cref="System.ArgumentNullException">package</exception>
 | |
|         public async Task InstallPackage(PackageVersionInfo package, bool isPlugin, IProgress<double> progress, CancellationToken cancellationToken)
 | |
|         {
 | |
|             if (package == null)
 | |
|             {
 | |
|                 throw new ArgumentNullException("package");
 | |
|             }
 | |
| 
 | |
|             if (progress == null)
 | |
|             {
 | |
|                 throw new ArgumentNullException("progress");
 | |
|             }
 | |
| 
 | |
|             var installationInfo = new InstallationInfo
 | |
|             {
 | |
|                 Id = Guid.NewGuid(),
 | |
|                 Name = package.name,
 | |
|                 AssemblyGuid = package.guid,
 | |
|                 UpdateClass = package.classification,
 | |
|                 Version = package.versionStr
 | |
|             };
 | |
| 
 | |
|             var innerCancellationTokenSource = new CancellationTokenSource();
 | |
| 
 | |
|             var tuple = new Tuple<InstallationInfo, CancellationTokenSource>(installationInfo, innerCancellationTokenSource);
 | |
| 
 | |
|             // Add it to the in-progress list
 | |
|             lock (CurrentInstallations)
 | |
|             {
 | |
|                 CurrentInstallations.Add(tuple);
 | |
|             }
 | |
| 
 | |
|             var innerProgress = new ActionableProgress<double>();
 | |
| 
 | |
|             // Whenever the progress updates, update the outer progress object and InstallationInfo
 | |
|             innerProgress.RegisterAction(percent =>
 | |
|             {
 | |
|                 progress.Report(percent);
 | |
| 
 | |
|                 installationInfo.PercentComplete = percent;
 | |
|             });
 | |
| 
 | |
|             var linkedToken = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, innerCancellationTokenSource.Token).Token;
 | |
| 
 | |
|             var installationEventArgs = new InstallationEventArgs
 | |
|             {
 | |
|                 InstallationInfo = installationInfo,
 | |
|                 PackageVersionInfo = package
 | |
|             };
 | |
| 
 | |
|             EventHelper.FireEventIfNotNull(PackageInstalling, this, installationEventArgs, _logger);
 | |
| 
 | |
|             try
 | |
|             {
 | |
|                 await InstallPackageInternal(package, isPlugin, innerProgress, linkedToken).ConfigureAwait(false);
 | |
| 
 | |
|                 lock (CurrentInstallations)
 | |
|                 {
 | |
|                     CurrentInstallations.Remove(tuple);
 | |
|                 }
 | |
| 
 | |
|                 CompletedInstallationsInternal.Add(installationInfo);
 | |
| 
 | |
|                 EventHelper.FireEventIfNotNull(PackageInstallationCompleted, this, installationEventArgs, _logger);
 | |
|             }
 | |
|             catch (OperationCanceledException)
 | |
|             {
 | |
|                 lock (CurrentInstallations)
 | |
|                 {
 | |
|                     CurrentInstallations.Remove(tuple);
 | |
|                 }
 | |
| 
 | |
|                 _logger.LogInformation("Package installation cancelled: {0} {1}", package.name, package.versionStr);
 | |
| 
 | |
|                 EventHelper.FireEventIfNotNull(PackageInstallationCancelled, this, installationEventArgs, _logger);
 | |
| 
 | |
|                 throw;
 | |
|             }
 | |
|             catch (Exception ex)
 | |
|             {
 | |
|                 _logger.LogError(ex, "Package installation failed");
 | |
| 
 | |
|                 lock (CurrentInstallations)
 | |
|                 {
 | |
|                     CurrentInstallations.Remove(tuple);
 | |
|                 }
 | |
| 
 | |
|                 EventHelper.FireEventIfNotNull(PackageInstallationFailed, this, new InstallationFailedEventArgs
 | |
|                 {
 | |
|                     InstallationInfo = installationInfo,
 | |
|                     Exception = ex
 | |
| 
 | |
|                 }, _logger);
 | |
| 
 | |
|                 throw;
 | |
|             }
 | |
|             finally
 | |
|             {
 | |
|                 // Dispose the progress object and remove the installation from the in-progress list
 | |
|                 tuple.Item2.Dispose();
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Installs the package internal.
 | |
|         /// </summary>
 | |
|         /// <param name="package">The package.</param>
 | |
|         /// <param name="isPlugin">if set to <c>true</c> [is plugin].</param>
 | |
|         /// <param name="progress">The progress.</param>
 | |
|         /// <param name="cancellationToken">The cancellation token.</param>
 | |
|         /// <returns>Task.</returns>
 | |
|         private async Task InstallPackageInternal(PackageVersionInfo package, bool isPlugin, IProgress<double> progress, CancellationToken cancellationToken)
 | |
|         {
 | |
|             IPlugin plugin = null;
 | |
| 
 | |
|             if (isPlugin)
 | |
|             {
 | |
|                 // Set last update time if we were installed before
 | |
|                 plugin = _applicationHost.Plugins.FirstOrDefault(p => string.Equals(p.Id.ToString(), package.guid, StringComparison.OrdinalIgnoreCase))
 | |
|                            ?? _applicationHost.Plugins.FirstOrDefault(p => p.Name.Equals(package.name, StringComparison.OrdinalIgnoreCase));
 | |
|             }
 | |
| 
 | |
|             string targetPath = plugin == null ? null : plugin.AssemblyFilePath;
 | |
| 
 | |
|             // Do the install
 | |
|             await PerformPackageInstallation(progress, targetPath, package, cancellationToken).ConfigureAwait(false);
 | |
| 
 | |
|             // Do plugin-specific processing
 | |
|             if (isPlugin)
 | |
|             {
 | |
|                 if (plugin != null)
 | |
|                 {
 | |
|                     OnPluginUpdated(plugin, package);
 | |
|                 }
 | |
|                 else
 | |
|                 {
 | |
|                     OnPluginInstalled(package);
 | |
|                 }
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         private async Task PerformPackageInstallation(IProgress<double> progress, string target, PackageVersionInfo package, CancellationToken cancellationToken)
 | |
|         {
 | |
|             // Target based on if it is an archive or single assembly
 | |
|             //  zip archives are assumed to contain directory structures relative to our ProgramDataPath
 | |
|             var extension = Path.GetExtension(package.targetFilename);
 | |
|             var isArchive = string.Equals(extension, ".zip", StringComparison.OrdinalIgnoreCase) || string.Equals(extension, ".rar", StringComparison.OrdinalIgnoreCase) || string.Equals(extension, ".7z", StringComparison.OrdinalIgnoreCase);
 | |
| 
 | |
|             if (target == null)
 | |
|             {
 | |
|                 target = Path.Combine(isArchive ? _appPaths.TempUpdatePath : _appPaths.PluginsPath, package.targetFilename);
 | |
|             }
 | |
| 
 | |
|             // Download to temporary file so that, if interrupted, it won't destroy the existing installation
 | |
|             var tempFile = await _httpClient.GetTempFile(new HttpRequestOptions
 | |
|             {
 | |
|                 Url = package.sourceUrl,
 | |
|                 CancellationToken = cancellationToken,
 | |
|                 Progress = progress
 | |
| 
 | |
|             }).ConfigureAwait(false);
 | |
| 
 | |
|             cancellationToken.ThrowIfCancellationRequested();
 | |
| 
 | |
|             // Validate with a checksum
 | |
|             var packageChecksum = string.IsNullOrWhiteSpace(package.checksum) ? Guid.Empty : new Guid(package.checksum);
 | |
|             if (!packageChecksum.Equals(Guid.Empty)) // support for legacy uploads for now
 | |
|             {
 | |
|                 using (var stream = _fileSystem.OpenRead(tempFile))
 | |
|                 {
 | |
|                     var check = Guid.Parse(BitConverter.ToString(_cryptographyProvider.ComputeMD5(stream)).Replace("-", String.Empty));
 | |
|                     if (check != packageChecksum)
 | |
|                     {
 | |
|                         throw new Exception(string.Format("Download validation failed for {0}.  Probably corrupted during transfer.", package.name));
 | |
|                     }
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             cancellationToken.ThrowIfCancellationRequested();
 | |
| 
 | |
|             // Success - move it to the real target 
 | |
|             try
 | |
|             {
 | |
|                 _fileSystem.CreateDirectory(_fileSystem.GetDirectoryName(target));
 | |
|                 _fileSystem.CopyFile(tempFile, target, true);
 | |
|                 //If it is an archive - write out a version file so we know what it is
 | |
|                 if (isArchive)
 | |
|                 {
 | |
|                     _fileSystem.WriteAllText(target + ".ver", package.versionStr);
 | |
|                 }
 | |
|             }
 | |
|             catch (IOException ex)
 | |
|             {
 | |
|                 _logger.LogError(ex, "Error attempting to move file from {TempFile} to {TargetFile}", tempFile, target);
 | |
|                 throw;
 | |
|             }
 | |
| 
 | |
|             try
 | |
|             {
 | |
|                 _fileSystem.DeleteFile(tempFile);
 | |
|             }
 | |
|             catch (IOException ex)
 | |
|             {
 | |
|                 // Don't fail because of this
 | |
|                 _logger.LogError(ex, "Error deleting temp file {TempFile}", tempFile);
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Uninstalls a plugin
 | |
|         /// </summary>
 | |
|         /// <param name="plugin">The plugin.</param>
 | |
|         /// <exception cref="System.ArgumentException"></exception>
 | |
|         public void UninstallPlugin(IPlugin plugin)
 | |
|         {
 | |
|             plugin.OnUninstalling();
 | |
| 
 | |
|             // Remove it the quick way for now
 | |
|             _applicationHost.RemovePlugin(plugin);
 | |
| 
 | |
|             var path = plugin.AssemblyFilePath;
 | |
|             _logger.LogInformation("Deleting plugin file {0}", path);
 | |
| 
 | |
|             // Make this case-insensitive to account for possible incorrect assembly naming
 | |
|             var file = _fileSystem.GetFilePaths(_fileSystem.GetDirectoryName(path))
 | |
|                 .FirstOrDefault(i => string.Equals(i, path, StringComparison.OrdinalIgnoreCase));
 | |
| 
 | |
|             if (!string.IsNullOrWhiteSpace(file))
 | |
|             {
 | |
|                 path = file;
 | |
|             }
 | |
| 
 | |
|             _fileSystem.DeleteFile(path);
 | |
| 
 | |
|             var list = _config.Configuration.UninstalledPlugins.ToList();
 | |
|             var filename = Path.GetFileName(path);
 | |
|             if (!list.Contains(filename, StringComparer.OrdinalIgnoreCase))
 | |
|             {
 | |
|                 list.Add(filename);
 | |
|                 _config.Configuration.UninstalledPlugins = list.ToArray();
 | |
|                 _config.SaveConfiguration();
 | |
|             }
 | |
| 
 | |
|             OnPluginUninstalled(plugin);
 | |
| 
 | |
|             _applicationHost.NotifyPendingRestart();
 | |
|         }
 | |
| 
 | |
|         /// <summary>
 | |
|         /// Releases unmanaged and - optionally - managed resources.
 | |
|         /// </summary>
 | |
|         /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
 | |
|         protected virtual void Dispose(bool dispose)
 | |
|         {
 | |
|             if (dispose)
 | |
|             {
 | |
|                 lock (CurrentInstallations)
 | |
|                 {
 | |
|                     foreach (var tuple in CurrentInstallations)
 | |
|                     {
 | |
|                         tuple.Item2.Dispose();
 | |
|                     }
 | |
| 
 | |
|                     CurrentInstallations.Clear();
 | |
|                 }
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         public void Dispose()
 | |
|         {
 | |
|             Dispose(true);
 | |
|         }
 | |
|     }
 | |
| }
 |