using System; using System.IO; using System.Linq; using Autofac; using NLog; using NzbDrone.Common; using NzbDrone.Update.Providers; namespace NzbDrone.Update { public class Program { private readonly UpdateProvider _updateProvider; private readonly ProcessProvider _processProvider; private static IContainer _container; private static readonly Logger logger = LogManager.GetCurrentClassLogger(); public Program(UpdateProvider updateProvider, ProcessProvider processProvider) { _updateProvider = updateProvider; _processProvider = processProvider; } public static void Main(string[] args) { try { Console.WriteLine("Starting NzbDrone Update Client"); var builder = new ContainerBuilder(); builder.RegisterAssemblyTypes(typeof(UpdateProvider).Assembly).SingleInstance(); builder.RegisterAssemblyTypes(typeof(RestProvider).Assembly).SingleInstance(); _container = builder.Build(); ReportingService.RestProvider = _container.Resolve<RestProvider>(); logger.Info("Updating NzbDrone to version {0}", _container.Resolve<EnvironmentProvider>().Version); _container.Resolve<Program>().Start(args); } catch (Exception e) { logger.FatalException("An error has occurred while applying update package.", e); } TransferUpdateLogs(); } private static void TransferUpdateLogs() { try { var environmentProvider = _container.Resolve<EnvironmentProvider>(); var diskProvider = _container.Resolve<DiskProvider>(); logger.Info("Copying log files to application directory."); diskProvider.CopyDirectory(environmentProvider.GetSandboxLogFolder(), environmentProvider.GetUpdateLogFolder()); } catch (Exception e) { logger.FatalException("Can't copy upgrade log files to target folder", e); } } public void Start(string[] args) { VerfityArguments(args); int processId = ParseProcessId(args); var exeFileInfo = new FileInfo(_processProvider.GetProcessById(processId).StartPath); string targetFolder = exeFileInfo.Directory.FullName; logger.Info("Starting update process. Target Path:{0}", targetFolder); _updateProvider.Start(targetFolder); } private int ParseProcessId(string[] args) { int id; if (!Int32.TryParse(args[0], out id) || id <= 0) { throw new ArgumentOutOfRangeException("Invalid process id: " + args[0]); } logger.Debug("NzbDrone processId:{0}", id); return id; } private void VerfityArguments(string[] args) { if (args == null || args.Length != 2) throw new ArgumentException("Wrong number of parameters were passed in."); logger.Debug("Arguments verified. [{0}] , [{1}]", args[0], args[1]); } } }