mirror of https://github.com/lidarr/Lidarr
62 lines
2.1 KiB
C#
62 lines
2.1 KiB
C#
using System;
|
|
using System.IO;
|
|
using NLog;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.DecisionEngine;
|
|
using NzbDrone.Core.Download;
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
namespace NzbDrone.Core.MediaFiles.TrackImport.Specifications
|
|
{
|
|
public class NotUnpackingSpecification : IImportDecisionEngineSpecification<LocalTrack>
|
|
{
|
|
private readonly IDiskProvider _diskProvider;
|
|
private readonly IConfigService _configService;
|
|
private readonly Logger _logger;
|
|
|
|
public NotUnpackingSpecification(IDiskProvider diskProvider, IConfigService configService, Logger logger)
|
|
{
|
|
_diskProvider = diskProvider;
|
|
_configService = configService;
|
|
_logger = logger;
|
|
}
|
|
|
|
public Decision IsSatisfiedBy(LocalTrack item, DownloadClientItem downloadClientItem)
|
|
{
|
|
if (item.ExistingFile)
|
|
{
|
|
_logger.Debug("{0} is in artist folder, skipping unpacking check", item.Path);
|
|
return Decision.Accept();
|
|
}
|
|
|
|
foreach (var workingFolder in _configService.DownloadClientWorkingFolders.Split('|'))
|
|
{
|
|
var parent = Directory.GetParent(item.Path);
|
|
while (parent != null)
|
|
{
|
|
if (parent.Name.StartsWith(workingFolder))
|
|
{
|
|
if (OsInfo.IsNotWindows)
|
|
{
|
|
_logger.Debug("{0} is still being unpacked", item.Path);
|
|
return Decision.Reject("File is still being unpacked");
|
|
}
|
|
|
|
if (_diskProvider.FileGetLastWrite(item.Path) > DateTime.UtcNow.AddMinutes(-5))
|
|
{
|
|
_logger.Debug("{0} appears to be unpacking still", item.Path);
|
|
return Decision.Reject("File is still being unpacked");
|
|
}
|
|
}
|
|
|
|
parent = parent.Parent;
|
|
}
|
|
}
|
|
|
|
return Decision.Accept();
|
|
}
|
|
}
|
|
}
|