Radarr/NzbDrone.Update/Program.cs

112 lines
3.7 KiB
C#
Raw Normal View History

using System;
2011-11-14 03:09:34 +00:00
using System.IO;
using System.Linq;
2011-11-13 20:31:02 +00:00
using NLog;
2011-11-14 06:01:51 +00:00
using NLog.Config;
using NLog.Targets;
2011-11-21 02:59:42 +00:00
using Ninject;
2011-11-13 20:31:02 +00:00
using NzbDrone.Common;
using NzbDrone.Update.Providers;
namespace NzbDrone.Update
{
2011-11-13 20:31:02 +00:00
public class Program
{
2011-11-13 20:31:02 +00:00
private readonly UpdateProvider _updateProvider;
private readonly ProcessProvider _processProvider;
2011-11-21 02:59:42 +00:00
private static StandardKernel _kernel;
2011-11-13 20:31:02 +00:00
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");
2011-11-14 06:01:51 +00:00
InitLoggers();
2011-11-21 02:59:42 +00:00
_kernel = new StandardKernel();
2011-11-21 03:42:45 +00:00
logger.Info("Updating NzbDrone to version {0}", _kernel.Get<EnviromentProvider>().Version);
2011-11-21 02:59:42 +00:00
_kernel.Get<Program>().Start(args);
2011-11-13 20:31:02 +00:00
}
catch (Exception e)
{
logger.FatalException("An error has occurred while applying update package.", e);
2011-11-13 20:31:02 +00:00
}
TransferUpdateLogs();
}
private static void TransferUpdateLogs()
{
try
{
2011-11-21 02:59:42 +00:00
var enviromentProvider = _kernel.Get<EnviromentProvider>();
var diskProvider = _kernel.Get<DiskProvider>();
logger.Info("Copying log files to application directory.");
diskProvider.CopyDirectory(enviromentProvider.GetSandboxLogFolder(), enviromentProvider.GetUpdateLogFolder());
}
catch (Exception e)
{
logger.FatalException("Can't copy upgrade log files to target folder", e);
}
2011-11-13 20:31:02 +00:00
}
2011-11-14 06:01:51 +00:00
private static void InitLoggers()
{
LogConfiguration.RegisterConsoleLogger(LogLevel.Trace);
LogConfiguration.RegisterUdpLogger();
2011-11-21 02:59:42 +00:00
var lastUpgradeLog = new FileTarget();
lastUpgradeLog.AutoFlush = true;
lastUpgradeLog.ConcurrentWrites = false;
2011-11-21 04:18:40 +00:00
lastUpgradeLog.FileName = Path.Combine(new EnviromentProvider().GetSandboxLogFolder(), DateTime.Now.ToString("yyyy.MM.dd-H-mm") + ".txt");
lastUpgradeLog.KeepFileOpen = false;
2011-11-21 02:59:42 +00:00
lastUpgradeLog.Layout = "${longdate} - ${logger}: ${message} ${exception:format=ToString}";
LogManager.Configuration.AddTarget(lastUpgradeLog.GetType().Name, lastUpgradeLog);
LogManager.Configuration.LoggingRules.Add(new LoggingRule("*", LogLevel.Trace, lastUpgradeLog));
2011-11-21 02:59:42 +00:00
LogConfiguration.RegisterExceptioneer();
2011-11-14 06:01:51 +00:00
LogConfiguration.Reload();
}
2011-11-13 20:31:02 +00:00
public void Start(string[] args)
{
VerfityArguments(args);
int processId = ParseProcessId(args);
2011-11-14 03:09:34 +00:00
FileInfo exeFileInfo = new FileInfo(_processProvider.GetProcessById(processId).StartPath);
string appPath = exeFileInfo.Directory.FullName;
2011-11-13 20:31:02 +00:00
logger.Info("Starting update process");
_updateProvider.Start(appPath);
}
private int ParseProcessId(string[] args)
{
int id = 0;
if (!Int32.TryParse(args[0], out id) || id <= 0)
{
throw new ArgumentOutOfRangeException("Invalid process id: " + args[0]);
}
return id;
}
private void VerfityArguments(string[] args)
{
2011-11-13 20:31:02 +00:00
if (args == null || args.Length != 2)
throw new ArgumentException("Wrong number of parameters were passed in.");
}
}
}