mirror of https://github.com/lidarr/Lidarr
138 lines
4.6 KiB
C#
138 lines
4.6 KiB
C#
using System.IO;
|
|
using System.Xml.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Common.Http;
|
|
|
|
namespace NzbDrone.Core.Notifications.Subsonic
|
|
{
|
|
public interface ISubsonicServerProxy
|
|
{
|
|
string GetBaseUrl(SubsonicSettings settings, string relativePath = null);
|
|
void Notify(SubsonicSettings settings, string message);
|
|
void Update(SubsonicSettings settings);
|
|
string Version(SubsonicSettings settings);
|
|
}
|
|
|
|
public class SubsonicServerProxy : ISubsonicServerProxy
|
|
{
|
|
private readonly IHttpClient _httpClient;
|
|
private readonly Logger _logger;
|
|
|
|
public SubsonicServerProxy(IHttpClient httpClient, Logger logger)
|
|
{
|
|
_httpClient = httpClient;
|
|
_logger = logger;
|
|
}
|
|
|
|
public string GetBaseUrl(SubsonicSettings settings, string relativePath = null)
|
|
{
|
|
var baseUrl = HttpRequestBuilder.BuildBaseUrl(settings.UseSsl, settings.Host, settings.Port, settings.UrlBase);
|
|
baseUrl = HttpUri.CombinePath(baseUrl, relativePath);
|
|
|
|
return baseUrl;
|
|
}
|
|
|
|
public void Notify(SubsonicSettings settings, string message)
|
|
{
|
|
var resource = "addChatMessage";
|
|
var request = GetSubsonicServerRequest(resource, HttpMethod.GET, settings);
|
|
request.AddQueryParam("message", message);
|
|
|
|
var response = _httpClient.Execute(request.Build());
|
|
|
|
_logger.Trace("Update response: {0}", response.Content);
|
|
CheckForError(response, settings);
|
|
}
|
|
|
|
public void Update(SubsonicSettings settings)
|
|
{
|
|
var resource = "startScan";
|
|
var request = GetSubsonicServerRequest(resource, HttpMethod.GET, settings);
|
|
var response = _httpClient.Execute(request.Build());
|
|
|
|
_logger.Trace("Update response: {0}", response.Content);
|
|
CheckForError(response, settings);
|
|
}
|
|
|
|
public string Version(SubsonicSettings settings)
|
|
{
|
|
var request = GetSubsonicServerRequest("ping", HttpMethod.GET, settings);
|
|
var response = _httpClient.Execute(request.Build());
|
|
|
|
_logger.Trace("Version response: {0}", response.Content);
|
|
CheckForError(response, settings);
|
|
|
|
var xDoc = XDocument.Load(new StringReader(response.Content.Replace("&", "&")));
|
|
var version = xDoc.Root?.Attribute("version")?.Value;
|
|
|
|
if (version == null)
|
|
{
|
|
throw new SubsonicException("Could not read version from Subsonic");
|
|
}
|
|
|
|
return version;
|
|
}
|
|
|
|
private HttpRequestBuilder GetSubsonicServerRequest(string resource, HttpMethod method, SubsonicSettings settings)
|
|
{
|
|
var client = new HttpRequestBuilder(GetBaseUrl(settings, "rest"));
|
|
|
|
client.Resource(resource);
|
|
|
|
if (settings.Username.IsNotNullOrWhiteSpace())
|
|
{
|
|
client.AddQueryParam("u", settings.Username)
|
|
.AddQueryParam("p", settings.Password)
|
|
.AddQueryParam("c", "Lidarr")
|
|
.AddQueryParam("v", "1.15.0");
|
|
}
|
|
|
|
client.Method = method;
|
|
|
|
return client;
|
|
}
|
|
|
|
private void CheckForError(HttpResponse response, SubsonicSettings settings)
|
|
{
|
|
_logger.Trace("Checking for error");
|
|
|
|
var xDoc = XDocument.Load(new StringReader(response.Content.Replace("&", "&")));
|
|
var status = xDoc.Root?.Attribute("status")?.Value;
|
|
|
|
if (status == null)
|
|
{
|
|
throw new SubsonicException("Invalid Response, Check Server Settings");
|
|
}
|
|
|
|
if (status != "ok")
|
|
{
|
|
var ns = xDoc.Root.GetDefaultNamespace();
|
|
var error = xDoc.Root.Element(XName.Get("error", ns.ToString()));
|
|
var errorMessage = error?.Attribute("message")?.Value;
|
|
var errorCode = error?.Attribute("code")?.Value;
|
|
|
|
if (errorCode == null)
|
|
{
|
|
throw new SubsonicException("Subsonic returned error, check settings");
|
|
}
|
|
|
|
if (errorCode == "40")
|
|
{
|
|
throw new SubsonicAuthenticationException(errorMessage);
|
|
}
|
|
|
|
throw new SubsonicException(errorMessage);
|
|
}
|
|
|
|
if (response.Content.IsNullOrWhiteSpace())
|
|
{
|
|
_logger.Trace("No response body returned, no error detected");
|
|
return;
|
|
}
|
|
|
|
_logger.Trace("No error detected");
|
|
}
|
|
}
|
|
}
|