2012-08-30 00:20:48 +00:00
|
|
|
|
using System;
|
2013-04-15 01:41:39 +00:00
|
|
|
|
using System.Collections.Generic;
|
2012-08-30 00:20:48 +00:00
|
|
|
|
using System.IO;
|
2014-07-04 08:09:48 +00:00
|
|
|
|
using FluentValidation.Results;
|
2012-08-30 00:20:48 +00:00
|
|
|
|
using NLog;
|
2014-01-06 06:20:08 +00:00
|
|
|
|
using NzbDrone.Common.Disk;
|
2014-12-02 06:26:25 +00:00
|
|
|
|
using NzbDrone.Common.Extensions;
|
2014-05-11 04:37:07 +00:00
|
|
|
|
using NzbDrone.Common.Http;
|
2013-02-24 06:48:52 +00:00
|
|
|
|
using NzbDrone.Core.Configuration;
|
2014-04-19 15:09:22 +00:00
|
|
|
|
using NzbDrone.Core.Indexers;
|
2014-09-11 20:24:00 +00:00
|
|
|
|
using NzbDrone.Core.RemotePathMappings;
|
2013-03-06 21:20:33 +00:00
|
|
|
|
using NzbDrone.Core.Organizer;
|
2013-04-15 01:41:39 +00:00
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
2012-08-30 00:20:48 +00:00
|
|
|
|
|
2014-02-14 05:31:49 +00:00
|
|
|
|
namespace NzbDrone.Core.Download.Clients.Pneumatic
|
2012-08-30 00:20:48 +00:00
|
|
|
|
{
|
2014-07-04 08:09:48 +00:00
|
|
|
|
public class Pneumatic : DownloadClientBase<PneumaticSettings>
|
2012-08-30 00:20:48 +00:00
|
|
|
|
{
|
2014-09-11 23:49:41 +00:00
|
|
|
|
private readonly IHttpClient _httpClient;
|
2012-08-30 00:20:48 +00:00
|
|
|
|
|
2014-09-11 23:49:41 +00:00
|
|
|
|
public Pneumatic(IHttpClient httpClient,
|
2014-05-27 21:04:13 +00:00
|
|
|
|
IConfigService configService,
|
2014-07-04 20:27:21 +00:00
|
|
|
|
IDiskProvider diskProvider,
|
2014-09-11 20:24:00 +00:00
|
|
|
|
IRemotePathMappingService remotePathMappingService,
|
2014-04-19 15:09:22 +00:00
|
|
|
|
Logger logger)
|
2014-10-13 21:11:35 +00:00
|
|
|
|
: base(configService, diskProvider, remotePathMappingService, logger)
|
2012-08-30 00:20:48 +00:00
|
|
|
|
{
|
2014-09-11 23:49:41 +00:00
|
|
|
|
_httpClient = httpClient;
|
2012-08-30 00:20:48 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-04-19 15:09:22 +00:00
|
|
|
|
public override DownloadProtocol Protocol
|
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
|
|
|
|
return DownloadProtocol.Usenet;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override string Download(RemoteEpisode remoteEpisode)
|
2012-08-30 00:20:48 +00:00
|
|
|
|
{
|
2013-09-13 23:17:58 +00:00
|
|
|
|
var url = remoteEpisode.Release.DownloadUrl;
|
|
|
|
|
var title = remoteEpisode.Release.Title;
|
2013-07-08 03:15:15 +00:00
|
|
|
|
|
2013-08-17 23:27:18 +00:00
|
|
|
|
if (remoteEpisode.ParsedEpisodeInfo.FullSeason)
|
2012-08-30 00:20:48 +00:00
|
|
|
|
{
|
2014-04-19 15:09:22 +00:00
|
|
|
|
throw new NotSupportedException("Full season releases are not supported with Pneumatic.");
|
2013-08-17 23:27:18 +00:00
|
|
|
|
}
|
2012-11-19 02:17:00 +00:00
|
|
|
|
|
2014-07-28 21:14:34 +00:00
|
|
|
|
title = FileNameBuilder.CleanFileName(title);
|
2012-08-30 00:20:48 +00:00
|
|
|
|
|
2013-08-17 23:27:18 +00:00
|
|
|
|
//Save to the Pneumatic directory (The user will need to ensure its accessible by XBMC)
|
2014-09-01 16:24:17 +00:00
|
|
|
|
var nzbFile = Path.Combine(Settings.NzbFolder, title + ".nzb");
|
2012-08-30 00:20:48 +00:00
|
|
|
|
|
2014-09-01 16:24:17 +00:00
|
|
|
|
_logger.Debug("Downloading NZB from: {0} to: {1}", url, nzbFile);
|
2014-09-11 23:49:41 +00:00
|
|
|
|
_httpClient.DownloadFile(url, nzbFile);
|
2012-08-30 00:20:48 +00:00
|
|
|
|
|
2014-09-01 16:24:17 +00:00
|
|
|
|
_logger.Debug("NZB Download succeeded, saved to: {0}", nzbFile);
|
2013-04-07 19:01:24 +00:00
|
|
|
|
|
2014-09-01 18:14:54 +00:00
|
|
|
|
var strmFile = WriteStrmFile(title, nzbFile);
|
|
|
|
|
return GetDownloadClientId(strmFile);
|
2013-10-21 01:30:46 +00:00
|
|
|
|
|
|
|
|
|
return null;
|
2012-08-30 00:20:48 +00:00
|
|
|
|
}
|
|
|
|
|
|
2013-07-31 05:49:41 +00:00
|
|
|
|
public bool IsConfigured
|
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
2014-05-25 21:58:41 +00:00
|
|
|
|
return !string.IsNullOrWhiteSpace(Settings.NzbFolder);
|
2013-07-31 05:49:41 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2014-04-19 15:09:22 +00:00
|
|
|
|
public override IEnumerable<DownloadClientItem> GetItems()
|
2013-04-15 01:41:39 +00:00
|
|
|
|
{
|
2014-09-01 18:14:54 +00:00
|
|
|
|
foreach (var file in _diskProvider.GetFiles(Settings.StrmFolder, SearchOption.TopDirectoryOnly))
|
2014-09-01 16:24:17 +00:00
|
|
|
|
{
|
2014-09-01 18:14:54 +00:00
|
|
|
|
if (Path.GetExtension(file) != ".strm")
|
2014-09-01 16:24:17 +00:00
|
|
|
|
{
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
2014-09-01 18:14:54 +00:00
|
|
|
|
var title = FileNameBuilder.CleanFileName(Path.GetFileName(file));
|
2014-09-01 16:24:17 +00:00
|
|
|
|
|
|
|
|
|
var historyItem = new DownloadClientItem
|
|
|
|
|
{
|
|
|
|
|
DownloadClient = Definition.Name,
|
2014-09-01 18:14:54 +00:00
|
|
|
|
DownloadClientId = GetDownloadClientId(file),
|
2014-09-01 16:24:17 +00:00
|
|
|
|
Title = title,
|
|
|
|
|
|
2014-09-01 18:14:54 +00:00
|
|
|
|
TotalSize = _diskProvider.GetFileSize(file),
|
2014-09-01 16:24:17 +00:00
|
|
|
|
|
2014-10-12 22:00:03 +00:00
|
|
|
|
OutputPath = new OsPath(file)
|
2014-09-01 16:24:17 +00:00
|
|
|
|
};
|
|
|
|
|
|
2014-09-01 18:14:54 +00:00
|
|
|
|
if (_diskProvider.IsFileLocked(file))
|
2014-09-01 16:24:17 +00:00
|
|
|
|
{
|
|
|
|
|
historyItem.Status = DownloadItemStatus.Downloading;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
historyItem.Status = DownloadItemStatus.Completed;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
yield return historyItem;
|
|
|
|
|
}
|
2013-04-15 01:41:39 +00:00
|
|
|
|
}
|
2014-07-19 17:37:06 +00:00
|
|
|
|
|
|
|
|
|
public override void RemoveItem(String id)
|
2013-10-25 05:55:32 +00:00
|
|
|
|
{
|
2014-04-19 15:09:22 +00:00
|
|
|
|
throw new NotSupportedException();
|
2013-10-25 05:55:32 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-07-19 17:37:06 +00:00
|
|
|
|
public override String RetryDownload(String id)
|
2014-04-01 20:07:41 +00:00
|
|
|
|
{
|
2014-04-19 15:09:22 +00:00
|
|
|
|
throw new NotSupportedException();
|
2014-04-01 20:07:41 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-06-06 05:55:38 +00:00
|
|
|
|
public override DownloadClientStatus GetStatus()
|
|
|
|
|
{
|
|
|
|
|
var status = new DownloadClientStatus
|
|
|
|
|
{
|
|
|
|
|
IsLocalhost = true
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
return status;
|
|
|
|
|
}
|
|
|
|
|
|
2014-07-09 22:11:57 +00:00
|
|
|
|
protected override void Test(List<ValidationFailure> failures)
|
2014-02-26 05:40:47 +00:00
|
|
|
|
{
|
2014-07-04 08:09:48 +00:00
|
|
|
|
failures.AddIfNotNull(TestWrite(Settings.NzbFolder, "NzbFolder"));
|
2014-09-01 16:24:17 +00:00
|
|
|
|
failures.AddIfNotNull(TestWrite(Settings.StrmFolder, "StrmFolder"));
|
2012-08-30 00:20:48 +00:00
|
|
|
|
}
|
2014-02-26 05:40:47 +00:00
|
|
|
|
|
2014-07-04 08:09:48 +00:00
|
|
|
|
private ValidationFailure TestWrite(String folder, String propertyName)
|
2014-02-26 05:40:47 +00:00
|
|
|
|
{
|
2014-07-04 08:09:48 +00:00
|
|
|
|
if (!_diskProvider.FolderExists(folder))
|
|
|
|
|
{
|
|
|
|
|
return new ValidationFailure(propertyName, "Folder does not exist");
|
|
|
|
|
}
|
2014-06-12 21:46:08 +00:00
|
|
|
|
|
2014-07-04 08:09:48 +00:00
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
var testPath = Path.Combine(folder, "drone_test.txt");
|
|
|
|
|
_diskProvider.WriteAllText(testPath, DateTime.Now.ToString());
|
|
|
|
|
_diskProvider.DeleteFile(testPath);
|
|
|
|
|
}
|
|
|
|
|
catch (Exception ex)
|
|
|
|
|
{
|
|
|
|
|
_logger.ErrorException(ex.Message, ex);
|
|
|
|
|
return new ValidationFailure(propertyName, "Unable to write to folder");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return null;
|
2014-02-26 05:40:47 +00:00
|
|
|
|
}
|
2014-09-01 16:24:17 +00:00
|
|
|
|
|
2014-09-01 18:14:54 +00:00
|
|
|
|
private String WriteStrmFile(String title, String nzbFile)
|
2014-09-01 16:24:17 +00:00
|
|
|
|
{
|
|
|
|
|
String folder;
|
|
|
|
|
|
|
|
|
|
if (Settings.StrmFolder.IsNullOrWhiteSpace())
|
|
|
|
|
{
|
|
|
|
|
folder = _configService.DownloadedEpisodesFolder;
|
|
|
|
|
|
|
|
|
|
if (folder.IsNullOrWhiteSpace())
|
|
|
|
|
{
|
|
|
|
|
throw new DownloadClientException("Strm Folder needs to be set for Pneumatic Downloader");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
folder = Settings.StrmFolder;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var contents = String.Format("plugin://plugin.program.pneumatic/?mode=strm&type=add_file&nzb={0}&nzbname={1}", nzbFile, title);
|
2014-09-01 18:14:54 +00:00
|
|
|
|
var filename = Path.Combine(folder, title + ".strm");
|
|
|
|
|
|
|
|
|
|
_diskProvider.WriteAllText(filename, contents);
|
|
|
|
|
|
|
|
|
|
return filename;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private String GetDownloadClientId(String filename)
|
|
|
|
|
{
|
2014-10-14 21:31:35 +00:00
|
|
|
|
return Definition.Name + "_" + Path.GetFileName(filename) + "_" + _diskProvider.FileGetLastWrite(filename).Ticks;
|
2014-09-01 16:24:17 +00:00
|
|
|
|
}
|
2012-08-30 00:20:48 +00:00
|
|
|
|
}
|
|
|
|
|
}
|