using System; using System.IO; using NzbDrone.Common.Disk; using NzbDrone.Common.EnvironmentInfo; using NzbDrone.Core.Configuration; using NzbDrone.Core.Update; namespace NzbDrone.Core.HealthCheck.Checks { public class UpdateCheck : HealthCheckBase { private readonly IDiskProvider _diskProvider; private readonly IAppFolderInfo _appFolderInfo; private readonly ICheckUpdateService _checkUpdateService; private readonly IConfigFileProvider _configFileProvider; public UpdateCheck(IDiskProvider diskProvider, IAppFolderInfo appFolderInfo, ICheckUpdateService checkUpdateService, IConfigFileProvider configFileProvider) { _diskProvider = diskProvider; _appFolderInfo = appFolderInfo; _checkUpdateService = checkUpdateService; _configFileProvider = configFileProvider; } public override HealthCheck Check() { if (OsInfo.IsWindows || _configFileProvider.UpdateAutomatically) { try { var testPath = Path.Combine(_appFolderInfo.StartUpFolder, "drone_test.txt"); _diskProvider.WriteAllText(testPath, DateTime.Now.ToString()); _diskProvider.DeleteFile(testPath); } catch (Exception) { return new HealthCheck(GetType(), HealthCheckResult.Error, "Unable to update, running from write-protected folder"); } } if (BuildInfo.BuildDateTime < DateTime.UtcNow.AddDays(-14)) { if (_checkUpdateService.AvailableUpdate() != null) { return new HealthCheck(GetType(), HealthCheckResult.Warning, "New update is available"); } } return new HealthCheck(GetType()); } public override bool CheckOnConfigChange { get { return false; } } } }