You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.6 KiB
66 lines
2.6 KiB
3 years ago
|
using System;
|
||
|
|
using System.Collections.Generic;
|
||
|
|
using System.Linq;
|
||
|
|
using System.Runtime.InteropServices;
|
||
|
|
using NLog;
|
||
|
|
using NzbDrone.Common.Cloud;
|
||
|
|
using NzbDrone.Common.EnvironmentInfo;
|
||
|
|
using NzbDrone.Common.Http;
|
||
|
|
using NzbDrone.Common.Serializer;
|
||
|
|
using NzbDrone.Core.Configuration;
|
||
|
|
|
||
|
|
namespace NzbDrone.Core.HealthCheck
|
||
|
|
{
|
||
|
|
public interface IServerSideNotificationService
|
||
|
|
{
|
||
|
|
public List<HealthCheck> GetServerChecks();
|
||
|
|
}
|
||
|
|
|
||
|
|
public class ServerSideNotificationService : IServerSideNotificationService
|
||
|
|
{
|
||
|
|
private readonly IHttpClient _client;
|
||
|
|
private readonly IConfigFileProvider _configFileProvider;
|
||
|
|
private readonly IHttpRequestBuilderFactory _cloudRequestBuilder;
|
||
|
|
private readonly Logger _logger;
|
||
|
|
|
||
|
|
public ServerSideNotificationService(IHttpClient client, IConfigFileProvider configFileProvider, IReadarrCloudRequestBuilder cloudRequestBuilder, Logger logger)
|
||
|
|
{
|
||
|
|
_client = client;
|
||
|
|
_configFileProvider = configFileProvider;
|
||
|
|
_cloudRequestBuilder = cloudRequestBuilder.Services;
|
||
|
|
_logger = logger;
|
||
|
|
}
|
||
|
|
|
||
|
|
public List<HealthCheck> GetServerChecks()
|
||
|
|
{
|
||
|
|
var request = _cloudRequestBuilder.Create()
|
||
|
|
.Resource("/notification")
|
||
|
|
.AddQueryParam("version", BuildInfo.Version)
|
||
|
|
.AddQueryParam("os", OsInfo.Os.ToString().ToLowerInvariant())
|
||
|
|
.AddQueryParam("arch", RuntimeInformation.OSArchitecture)
|
||
|
|
.AddQueryParam("runtime", PlatformInfo.Platform.ToString().ToLowerInvariant())
|
||
|
|
.AddQueryParam("branch", _configFileProvider.Branch)
|
||
|
|
.Build();
|
||
|
|
try
|
||
|
|
{
|
||
|
|
_logger.Trace("Getting server side health notifications");
|
||
|
|
var response = _client.Execute(request);
|
||
|
|
var result = Json.Deserialize<List<ServerNotificationResponse>>(response.Content);
|
||
|
|
return result.Select(x => new HealthCheck(GetType(), x.Type, x.Message, x.WikiUrl)).ToList();
|
||
|
|
}
|
||
|
|
catch (Exception ex)
|
||
|
|
{
|
||
|
|
_logger.Error(ex, "Failed to retrieve server notifications");
|
||
|
|
return new List<HealthCheck>();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public class ServerNotificationResponse
|
||
|
|
{
|
||
|
|
public HealthCheckResult Type { get; set; }
|
||
|
|
public string Message { get; set; }
|
||
|
|
public string WikiUrl { get; set; }
|
||
|
|
}
|
||
|
|
}
|