pull/2789/head
parent
f87e2dd717
commit
066a9de7ae
@ -0,0 +1,118 @@
|
|||||||
|
using NzbDrone.Common.Disk;
|
||||||
|
using NzbDrone.Common.EnvironmentInfo;
|
||||||
|
using NzbDrone.Common.Extensions;
|
||||||
|
using NzbDrone.Core.Update;
|
||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.IO;
|
||||||
|
using System.Linq;
|
||||||
|
using System.Text;
|
||||||
|
using System.Text.RegularExpressions;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.Configuration
|
||||||
|
{
|
||||||
|
public interface IDeploymentInfoProvider
|
||||||
|
{
|
||||||
|
Version PackageVersion { get; }
|
||||||
|
string PackageBranch { get; }
|
||||||
|
UpdateMechanism PackageUpdateMechanism { get; }
|
||||||
|
|
||||||
|
Version ReleaseVersion { get; }
|
||||||
|
string ReleaseBranch { get; }
|
||||||
|
|
||||||
|
bool BuiltInUpdaterAllowed { get; }
|
||||||
|
UpdateMechanism DefaultUpdateMechanism { get; }
|
||||||
|
string DefaultBranch { get; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public class DeploymentInfoProvider : IDeploymentInfoProvider
|
||||||
|
{
|
||||||
|
public DeploymentInfoProvider(IAppFolderInfo appFolderInfo, IDiskProvider diskProvider)
|
||||||
|
{
|
||||||
|
var bin = appFolderInfo.StartUpFolder;
|
||||||
|
var packageInfoPath = Path.Combine(bin, "..", "package_info");
|
||||||
|
var releaseInfoPath = Path.Combine(bin, "release_info");
|
||||||
|
|
||||||
|
PackageUpdateMechanism = UpdateMechanism.BuiltIn;
|
||||||
|
DefaultBranch = "master";
|
||||||
|
|
||||||
|
if (Path.GetFileName(bin) == "bin" && diskProvider.FileExists(packageInfoPath))
|
||||||
|
{
|
||||||
|
var data = diskProvider.ReadAllText(packageInfoPath);
|
||||||
|
|
||||||
|
PackageVersion = ReadVersion(data, "PackageVersion");
|
||||||
|
PackageUpdateMechanism = ReadEnumValue(data, "UpdateMethod", UpdateMechanism.BuiltIn);
|
||||||
|
PackageBranch = ReadValue(data, "Branch", null);
|
||||||
|
|
||||||
|
ReleaseVersion = ReadVersion(data, "ReleaseVersion");
|
||||||
|
|
||||||
|
if (PackageBranch.IsNotNullOrWhiteSpace())
|
||||||
|
{
|
||||||
|
DefaultBranch = PackageBranch;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (diskProvider.FileExists(releaseInfoPath))
|
||||||
|
{
|
||||||
|
var data = diskProvider.ReadAllText(releaseInfoPath);
|
||||||
|
|
||||||
|
ReleaseVersion = ReadVersion(data, "ReleaseVersion", ReleaseVersion);
|
||||||
|
ReleaseBranch = ReadValue(data, "Branch", null);
|
||||||
|
|
||||||
|
if (ReleaseBranch.IsNotNullOrWhiteSpace())
|
||||||
|
{
|
||||||
|
DefaultBranch = ReleaseBranch;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
DefaultUpdateMechanism = PackageUpdateMechanism;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static string ReadValue(string fileData, string key, string defaultValue)
|
||||||
|
{
|
||||||
|
var match = Regex.Match(fileData, "^" + key + "=(.*)$", RegexOptions.Multiline);
|
||||||
|
if (match.Success)
|
||||||
|
{
|
||||||
|
return match.Groups[1].Value.Trim();
|
||||||
|
}
|
||||||
|
|
||||||
|
return defaultValue;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static T ReadEnumValue<T>(string fileData, string key, T defaultValue)
|
||||||
|
where T : struct
|
||||||
|
{
|
||||||
|
var value = ReadValue(fileData, key, null);
|
||||||
|
if (value != null && Enum.TryParse<T>(value, true, out var result))
|
||||||
|
{
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
return defaultValue;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static Version ReadVersion(string fileData, string key, Version defaultValue = null)
|
||||||
|
{
|
||||||
|
var value = ReadValue(fileData, key, null);
|
||||||
|
if (value != null && Version.TryParse(value, out var result))
|
||||||
|
{
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
return defaultValue;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Version PackageVersion { get; private set; }
|
||||||
|
public string PackageBranch { get; private set; }
|
||||||
|
public UpdateMechanism PackageUpdateMechanism { get; private set; }
|
||||||
|
|
||||||
|
public Version ReleaseVersion { get; set; }
|
||||||
|
public string ReleaseBranch { get; set; }
|
||||||
|
|
||||||
|
|
||||||
|
public bool BuiltInUpdaterAllowed => PackageUpdateMechanism == UpdateMechanism.BuiltIn;
|
||||||
|
public UpdateMechanism DefaultUpdateMechanism { get; private set; }
|
||||||
|
public string DefaultBranch { get; private set; }
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,62 @@
|
|||||||
|
using NLog;
|
||||||
|
using NzbDrone.Core.Configuration;
|
||||||
|
using NzbDrone.Core.Lifecycle;
|
||||||
|
using NzbDrone.Core.Messaging.Events;
|
||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Linq;
|
||||||
|
using System.Text;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.Update
|
||||||
|
{
|
||||||
|
public interface IUpdaterConfigProvider
|
||||||
|
{
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
public class UpdaterConfigProvider : IUpdaterConfigProvider, IHandle<ApplicationStartedEvent>
|
||||||
|
{
|
||||||
|
private Logger _logger;
|
||||||
|
private IConfigFileProvider _configFileProvider;
|
||||||
|
private IDeploymentInfoProvider _deploymentInfoProvider;
|
||||||
|
|
||||||
|
public UpdaterConfigProvider(IDeploymentInfoProvider deploymentInfoProvider, IConfigFileProvider configFileProvider, Logger logger)
|
||||||
|
{
|
||||||
|
_deploymentInfoProvider = deploymentInfoProvider;
|
||||||
|
_configFileProvider = configFileProvider;
|
||||||
|
_logger = logger;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Handle(ApplicationStartedEvent message)
|
||||||
|
{
|
||||||
|
var updateMechanism = _configFileProvider.UpdateMechanism;
|
||||||
|
var packageUpdateMechanism = _deploymentInfoProvider.PackageUpdateMechanism;
|
||||||
|
|
||||||
|
var externalMechanisms = Enum.GetValues(typeof(UpdateMechanism))
|
||||||
|
.Cast<UpdateMechanism>()
|
||||||
|
.Where(v => (int)v >= (int)UpdateMechanism.External)
|
||||||
|
.ToArray();
|
||||||
|
|
||||||
|
foreach (var externalMechanism in externalMechanisms)
|
||||||
|
{
|
||||||
|
if (packageUpdateMechanism != externalMechanism && updateMechanism == externalMechanism ||
|
||||||
|
packageUpdateMechanism == externalMechanism && updateMechanism == UpdateMechanism.BuiltIn)
|
||||||
|
{
|
||||||
|
_logger.Info("Update mechanism {0} not supported in the current configuration, changing to {1}.", updateMechanism, packageUpdateMechanism);
|
||||||
|
ChangeUpdateMechanism(packageUpdateMechanism);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void ChangeUpdateMechanism(UpdateMechanism updateMechanism)
|
||||||
|
{
|
||||||
|
var config = new Dictionary<string, object>
|
||||||
|
{
|
||||||
|
[nameof(_configFileProvider.UpdateMechanism)] = updateMechanism
|
||||||
|
};
|
||||||
|
_configFileProvider.SaveConfigDictionary(config);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in new issue