171 lines
5.6 KiB
C#
171 lines
5.6 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using Newtonsoft.Json;
|
|
using NLog;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.Model;
|
|
using NzbDrone.Core.Tv;
|
|
|
|
namespace NzbDrone.Core.Download.Clients.Nzbget
|
|
{
|
|
public class NzbgetProvider : IDownloadClient
|
|
{
|
|
private static readonly Logger logger = LogManager.GetCurrentClassLogger();
|
|
private readonly IConfigService _configService;
|
|
private readonly HttpProvider _httpProvider;
|
|
|
|
public NzbgetProvider(IConfigService configService, HttpProvider httpProvider)
|
|
{
|
|
_configService = configService;
|
|
_httpProvider = httpProvider;
|
|
}
|
|
|
|
public NzbgetProvider()
|
|
{
|
|
}
|
|
|
|
public virtual bool IsInQueue(EpisodeParseResult newParseResult)
|
|
{
|
|
try
|
|
{
|
|
var queue = GetQueue().Where(c => c.ParseResult != null);
|
|
|
|
var matchigTitle = queue.Where(q => String.Equals(q.ParseResult.CleanTitle, newParseResult.Series.CleanTitle, StringComparison.InvariantCultureIgnoreCase));
|
|
|
|
var matchingTitleWithQuality = matchigTitle.Where(q => q.ParseResult.Quality >= newParseResult.Quality);
|
|
|
|
|
|
if (newParseResult.Series.SeriesTypes == SeriesTypes.Daily)
|
|
{
|
|
return matchingTitleWithQuality.Any(q => q.ParseResult.AirDate.Value.Date == newParseResult.AirDate.Value.Date);
|
|
}
|
|
|
|
var matchingSeason = matchingTitleWithQuality.Where(q => q.ParseResult.SeasonNumber == newParseResult.SeasonNumber);
|
|
|
|
if (newParseResult.FullSeason)
|
|
{
|
|
return matchingSeason.Any();
|
|
}
|
|
|
|
return matchingSeason.Any(q => q.ParseResult.EpisodeNumbers != null && q.ParseResult.EpisodeNumbers.Any(e => newParseResult.EpisodeNumbers.Contains(e)));
|
|
}
|
|
|
|
catch (Exception ex)
|
|
{
|
|
logger.WarnException("Unable to connect to Nzbget to check queue.", ex);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public virtual bool DownloadNzb(string url, string title, bool recentlyAired)
|
|
{
|
|
try
|
|
{
|
|
string cat = _configService.NzbgetTvCategory;
|
|
int priority = recentlyAired ? (int)_configService.NzbgetRecentTvPriority : (int)_configService.NzbgetBacklogTvPriority;
|
|
|
|
var command = new JsonRequest
|
|
{
|
|
Method = "appendurl",
|
|
Params = new object[]{ title, cat, priority, false, url }
|
|
};
|
|
|
|
logger.Info("Adding report [{0}] to the queue.", title);
|
|
var response = PostCommand(JsonConvert.SerializeObject(command));
|
|
|
|
CheckForError(response);
|
|
|
|
var success = JsonConvert.DeserializeObject<EnqueueResponse>(response).Result;
|
|
logger.Debug("Queue Response: [{0}]", success);
|
|
|
|
return true;
|
|
}
|
|
|
|
catch (WebException ex)
|
|
{
|
|
logger.Error("Error communicating with Nzbget: " + ex.Message);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public virtual List<QueueItem> GetQueue()
|
|
{
|
|
var command = new JsonRequest
|
|
{
|
|
Method = "listgroups",
|
|
Params = null
|
|
};
|
|
|
|
var response = PostCommand(JsonConvert.SerializeObject(command));
|
|
|
|
CheckForError(response);
|
|
|
|
return JsonConvert.DeserializeObject<Queue>(response).QueueItems;
|
|
}
|
|
|
|
public virtual VersionModel GetVersion(string host = null, int port = 0, string username = null, string password = null)
|
|
{
|
|
//Get saved values if any of these are defaults
|
|
if (host == null)
|
|
host = _configService.NzbgetHost;
|
|
|
|
if (port == 0)
|
|
port = _configService.NzbgetPort;
|
|
|
|
if (username == null)
|
|
username = _configService.NzbgetUsername;
|
|
|
|
if (password == null)
|
|
password = _configService.NzbgetPassword;
|
|
|
|
var command = new JsonRequest
|
|
{
|
|
Method = "version",
|
|
Params = null
|
|
};
|
|
|
|
var address = String.Format(@"{0}:{1}", host, port);
|
|
var response = _httpProvider.PostCommand(address, username, password, JsonConvert.SerializeObject(command));
|
|
|
|
CheckForError(response);
|
|
|
|
return JsonConvert.DeserializeObject<VersionModel>(response);
|
|
}
|
|
|
|
public virtual string Test(string host, int port, string username, string password)
|
|
{
|
|
try
|
|
{
|
|
var version = GetVersion(host, port, username, password);
|
|
return version.Result;
|
|
}
|
|
catch(Exception ex)
|
|
{
|
|
logger.DebugException("Failed to Test Nzbget", ex);
|
|
}
|
|
|
|
return String.Empty;
|
|
}
|
|
|
|
private string PostCommand(string command)
|
|
{
|
|
var url = String.Format(@"{0}:{1}",
|
|
_configService.NzbgetHost,
|
|
_configService.NzbgetPort);
|
|
|
|
return _httpProvider.PostCommand(url, _configService.NzbgetUsername, _configService.NzbgetPassword, command);
|
|
}
|
|
|
|
private void CheckForError(string response)
|
|
{
|
|
var result = JsonConvert.DeserializeObject<JsonError>(response);
|
|
|
|
if (result.Error != null)
|
|
throw new ApplicationException(result.Error.ToString());
|
|
}
|
|
}
|
|
} |