2014-04-19 15:09:22 +00:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using NLog;
|
|
|
|
|
using NzbDrone.Common;
|
|
|
|
|
using NzbDrone.Common.Disk;
|
|
|
|
|
using NzbDrone.Common.Http;
|
2014-05-27 21:04:13 +00:00
|
|
|
|
using NzbDrone.Core.Configuration;
|
2014-04-19 15:09:22 +00:00
|
|
|
|
using NzbDrone.Core.Indexers;
|
|
|
|
|
using NzbDrone.Core.Messaging.Commands;
|
|
|
|
|
using NzbDrone.Core.Organizer;
|
|
|
|
|
using NzbDrone.Core.Parser;
|
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
|
using NzbDrone.Core.MediaFiles;
|
2014-06-12 21:46:08 +00:00
|
|
|
|
using Omu.ValueInjecter;
|
2014-04-19 15:09:22 +00:00
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Download.Clients.UsenetBlackhole
|
|
|
|
|
{
|
|
|
|
|
public class UsenetBlackhole : DownloadClientBase<UsenetBlackholeSettings>, IExecute<TestUsenetBlackholeCommand>
|
|
|
|
|
{
|
|
|
|
|
private readonly IDiskProvider _diskProvider;
|
|
|
|
|
private readonly IDiskScanService _diskScanService;
|
|
|
|
|
private readonly IHttpProvider _httpProvider;
|
|
|
|
|
|
|
|
|
|
public UsenetBlackhole(IDiskProvider diskProvider,
|
|
|
|
|
IDiskScanService diskScanService,
|
|
|
|
|
IHttpProvider httpProvider,
|
2014-05-27 21:04:13 +00:00
|
|
|
|
IConfigService configService,
|
|
|
|
|
IParsingService parsingService,
|
2014-04-19 15:09:22 +00:00
|
|
|
|
Logger logger)
|
2014-05-27 21:04:13 +00:00
|
|
|
|
: base(configService, parsingService, logger)
|
2014-04-19 15:09:22 +00:00
|
|
|
|
{
|
|
|
|
|
_diskProvider = diskProvider;
|
|
|
|
|
_diskScanService = diskScanService;
|
|
|
|
|
_httpProvider = httpProvider;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override DownloadProtocol Protocol
|
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
|
|
|
|
return DownloadProtocol.Usenet;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override string Download(RemoteEpisode remoteEpisode)
|
|
|
|
|
{
|
|
|
|
|
var url = remoteEpisode.Release.DownloadUrl;
|
|
|
|
|
var title = remoteEpisode.Release.Title;
|
|
|
|
|
|
|
|
|
|
title = FileNameBuilder.CleanFilename(title);
|
|
|
|
|
|
|
|
|
|
var filename = Path.Combine(Settings.NzbFolder, title + ".nzb");
|
|
|
|
|
|
|
|
|
|
_logger.Debug("Downloading NZB from: {0} to: {1}", url, filename);
|
|
|
|
|
_httpProvider.DownloadFile(url, filename);
|
|
|
|
|
_logger.Debug("NZB Download succeeded, saved to: {0}", filename);
|
|
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override IEnumerable<DownloadClientItem> GetItems()
|
|
|
|
|
{
|
|
|
|
|
foreach (var folder in _diskProvider.GetDirectories(Settings.WatchFolder))
|
|
|
|
|
{
|
|
|
|
|
var title = FileNameBuilder.CleanFilename(Path.GetFileName(folder));
|
|
|
|
|
|
|
|
|
|
var files = _diskProvider.GetFiles(folder, SearchOption.AllDirectories);
|
|
|
|
|
|
|
|
|
|
var historyItem = new DownloadClientItem
|
|
|
|
|
{
|
|
|
|
|
DownloadClient = Definition.Name,
|
|
|
|
|
DownloadClientId = Definition.Name + "_" + Path.GetFileName(folder) + "_" + _diskProvider.FolderGetCreationTimeUtc(folder).Ticks,
|
|
|
|
|
Title = title,
|
|
|
|
|
|
|
|
|
|
TotalSize = files.Select(_diskProvider.GetFileSize).Sum(),
|
|
|
|
|
|
|
|
|
|
OutputPath = folder
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
if (files.Any(_diskProvider.IsFileLocked))
|
|
|
|
|
{
|
|
|
|
|
historyItem.Status = DownloadItemStatus.Downloading;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
historyItem.Status = DownloadItemStatus.Completed;
|
2014-05-24 10:48:55 +00:00
|
|
|
|
|
|
|
|
|
historyItem.RemainingTime = TimeSpan.Zero;
|
2014-04-19 15:09:22 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
historyItem.RemoteEpisode = GetRemoteEpisode(historyItem.Title);
|
|
|
|
|
if (historyItem.RemoteEpisode == null) continue;
|
|
|
|
|
|
|
|
|
|
yield return historyItem;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
foreach (var videoFile in _diskScanService.GetVideoFiles(Settings.WatchFolder, false))
|
|
|
|
|
{
|
|
|
|
|
var title = FileNameBuilder.CleanFilename(Path.GetFileName(videoFile));
|
|
|
|
|
|
|
|
|
|
var historyItem = new DownloadClientItem
|
|
|
|
|
{
|
|
|
|
|
DownloadClient = Definition.Name,
|
|
|
|
|
DownloadClientId = Definition.Name + "_" + Path.GetFileName(videoFile) + "_" + _diskProvider.FileGetLastWriteUtc(videoFile).Ticks,
|
|
|
|
|
Title = title,
|
|
|
|
|
|
|
|
|
|
TotalSize = _diskProvider.GetFileSize(videoFile),
|
|
|
|
|
|
|
|
|
|
OutputPath = videoFile
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
if (_diskProvider.IsFileLocked(videoFile))
|
|
|
|
|
{
|
|
|
|
|
historyItem.Status = DownloadItemStatus.Downloading;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
historyItem.Status = DownloadItemStatus.Completed;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
historyItem.RemoteEpisode = GetRemoteEpisode(historyItem.Title);
|
|
|
|
|
if (historyItem.RemoteEpisode == null) continue;
|
|
|
|
|
|
|
|
|
|
yield return historyItem;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override void RemoveItem(string id)
|
|
|
|
|
{
|
|
|
|
|
throw new NotSupportedException();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override void RetryDownload(string id)
|
|
|
|
|
{
|
|
|
|
|
throw new NotSupportedException();
|
|
|
|
|
}
|
|
|
|
|
|
2014-06-06 05:55:38 +00:00
|
|
|
|
public override DownloadClientStatus GetStatus()
|
|
|
|
|
{
|
|
|
|
|
return new DownloadClientStatus
|
|
|
|
|
{
|
|
|
|
|
IsLocalhost = true,
|
|
|
|
|
OutputRootFolders = new List<string> { Settings.WatchFolder }
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
2014-06-12 21:46:08 +00:00
|
|
|
|
public override void Test(UsenetBlackholeSettings settings)
|
|
|
|
|
{
|
|
|
|
|
PerformWriteTest(settings.NzbFolder);
|
|
|
|
|
PerformWriteTest(settings.WatchFolder);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void PerformWriteTest(string folder)
|
2014-04-19 15:09:22 +00:00
|
|
|
|
{
|
|
|
|
|
var testPath = Path.Combine(folder, "drone_test.txt");
|
|
|
|
|
_diskProvider.WriteAllText(testPath, DateTime.Now.ToString());
|
|
|
|
|
_diskProvider.DeleteFile(testPath);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void Execute(TestUsenetBlackholeCommand message)
|
|
|
|
|
{
|
2014-06-12 21:46:08 +00:00
|
|
|
|
var settings = new UsenetBlackholeSettings();
|
|
|
|
|
settings.InjectFrom(message);
|
|
|
|
|
|
|
|
|
|
Test(settings);
|
2014-04-19 15:09:22 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|