2023-05-23 10:52:39 +00:00
|
|
|
using System;
|
2017-06-25 16:23:32 +00:00
|
|
|
using System.Collections.Generic;
|
2015-12-25 09:22:00 +00:00
|
|
|
using System.IO;
|
|
|
|
using System.Linq;
|
|
|
|
using NLog;
|
|
|
|
using NzbDrone.Common.Disk;
|
|
|
|
using NzbDrone.Common.Extensions;
|
|
|
|
using NzbDrone.Core.Configuration;
|
|
|
|
using NzbDrone.Core.Extras.Files;
|
|
|
|
using NzbDrone.Core.MediaFiles;
|
2022-02-27 20:37:23 +00:00
|
|
|
using NzbDrone.Core.Parser.Model;
|
2015-12-25 09:22:00 +00:00
|
|
|
using NzbDrone.Core.Tv;
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Extras.Others
|
|
|
|
{
|
|
|
|
public class OtherExtraService : ExtraFileManager<OtherExtraFile>
|
|
|
|
{
|
2022-02-27 20:37:23 +00:00
|
|
|
private readonly IDiskProvider _diskProvider;
|
2015-12-25 09:22:00 +00:00
|
|
|
private readonly IOtherExtraFileService _otherExtraFileService;
|
2019-10-07 16:40:48 +00:00
|
|
|
private readonly IMediaFileAttributeService _mediaFileAttributeService;
|
2022-02-27 20:37:23 +00:00
|
|
|
private readonly Logger _logger;
|
2015-12-25 09:22:00 +00:00
|
|
|
|
|
|
|
public OtherExtraService(IConfigService configService,
|
2017-02-24 06:42:04 +00:00
|
|
|
IDiskProvider diskProvider,
|
2015-12-25 09:22:00 +00:00
|
|
|
IDiskTransferService diskTransferService,
|
|
|
|
IOtherExtraFileService otherExtraFileService,
|
2019-10-07 16:40:48 +00:00
|
|
|
IMediaFileAttributeService mediaFileAttributeService,
|
2015-12-25 09:22:00 +00:00
|
|
|
Logger logger)
|
2017-02-24 06:42:04 +00:00
|
|
|
: base(configService, diskProvider, diskTransferService, logger)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2022-02-27 20:37:23 +00:00
|
|
|
_diskProvider = diskProvider;
|
2015-12-25 09:22:00 +00:00
|
|
|
_otherExtraFileService = otherExtraFileService;
|
2019-10-07 16:40:48 +00:00
|
|
|
_mediaFileAttributeService = mediaFileAttributeService;
|
2022-02-27 20:37:23 +00:00
|
|
|
_logger = logger;
|
2015-12-25 09:22:00 +00:00
|
|
|
}
|
|
|
|
|
2016-12-09 06:54:15 +00:00
|
|
|
public override int Order => 2;
|
2015-12-25 09:22:00 +00:00
|
|
|
|
2020-04-28 21:30:18 +00:00
|
|
|
public override IEnumerable<ExtraFile> CreateAfterMediaCoverUpdate(Series series)
|
|
|
|
{
|
|
|
|
return Enumerable.Empty<ExtraFile>();
|
|
|
|
}
|
|
|
|
|
2015-12-25 09:22:00 +00:00
|
|
|
public override IEnumerable<ExtraFile> CreateAfterSeriesScan(Series series, List<EpisodeFile> episodeFiles)
|
|
|
|
{
|
|
|
|
return Enumerable.Empty<ExtraFile>();
|
|
|
|
}
|
|
|
|
|
|
|
|
public override IEnumerable<ExtraFile> CreateAfterEpisodeImport(Series series, EpisodeFile episodeFile)
|
|
|
|
{
|
|
|
|
return Enumerable.Empty<ExtraFile>();
|
|
|
|
}
|
|
|
|
|
2020-04-28 21:30:18 +00:00
|
|
|
public override IEnumerable<ExtraFile> CreateAfterEpisodeFolder(Series series, string seriesFolder, string seasonFolder)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
|
|
|
return Enumerable.Empty<ExtraFile>();
|
|
|
|
}
|
|
|
|
|
|
|
|
public override IEnumerable<ExtraFile> MoveFilesAfterRename(Series series, List<EpisodeFile> episodeFiles)
|
|
|
|
{
|
|
|
|
var extraFiles = _otherExtraFileService.GetFilesBySeries(series.Id);
|
|
|
|
var movedFiles = new List<OtherExtraFile>();
|
|
|
|
|
|
|
|
foreach (var episodeFile in episodeFiles)
|
|
|
|
{
|
|
|
|
var extraFilesForEpisodeFile = extraFiles.Where(m => m.EpisodeFileId == episodeFile.Id).ToList();
|
|
|
|
|
|
|
|
foreach (var extraFile in extraFilesForEpisodeFile)
|
|
|
|
{
|
2017-02-24 06:42:04 +00:00
|
|
|
movedFiles.AddIfNotNull(MoveFile(series, episodeFile, extraFile));
|
2015-12-25 09:22:00 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
_otherExtraFileService.Upsert(movedFiles);
|
|
|
|
|
|
|
|
return movedFiles;
|
|
|
|
}
|
|
|
|
|
2022-02-27 20:37:23 +00:00
|
|
|
public override bool CanImportFile(LocalEpisode localEpisode, EpisodeFile episodeFile, string path, string extension, bool readOnly)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2022-02-27 20:37:23 +00:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
public override IEnumerable<ExtraFile> ImportFiles(LocalEpisode localEpisode, EpisodeFile episodeFile, List<string> files, bool isReadOnly)
|
|
|
|
{
|
|
|
|
var importedFiles = new List<ExtraFile>();
|
|
|
|
var filteredFiles = files.Where(f => CanImportFile(localEpisode, episodeFile, f, Path.GetExtension(f), isReadOnly)).ToList();
|
|
|
|
var sourcePath = localEpisode.Path;
|
|
|
|
var sourceFolder = _diskProvider.GetParentFolder(sourcePath);
|
|
|
|
var sourceFileName = Path.GetFileNameWithoutExtension(sourcePath);
|
|
|
|
var matchingFiles = new List<string>();
|
|
|
|
var hasNfo = false;
|
|
|
|
|
|
|
|
foreach (var file in filteredFiles)
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
// Filter out duplicate NFO files
|
|
|
|
if (file.EndsWith(".nfo", StringComparison.InvariantCultureIgnoreCase))
|
|
|
|
{
|
|
|
|
if (hasNfo)
|
|
|
|
{
|
|
|
|
continue;
|
|
|
|
}
|
2015-12-25 09:22:00 +00:00
|
|
|
|
2022-02-27 20:37:23 +00:00
|
|
|
hasNfo = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Filename match
|
|
|
|
if (Path.GetFileNameWithoutExtension(file).StartsWith(sourceFileName, StringComparison.InvariantCultureIgnoreCase))
|
|
|
|
{
|
|
|
|
matchingFiles.Add(file);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Season and episode match
|
|
|
|
var fileEpisodeInfo = Parser.Parser.ParsePath(file) ?? new ParsedEpisodeInfo();
|
|
|
|
|
|
|
|
if (fileEpisodeInfo.EpisodeNumbers.Length == 0)
|
|
|
|
{
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (fileEpisodeInfo.SeasonNumber == localEpisode.FileEpisodeInfo.SeasonNumber &&
|
|
|
|
fileEpisodeInfo.EpisodeNumbers.SequenceEqual(localEpisode.FileEpisodeInfo.EpisodeNumbers))
|
|
|
|
{
|
|
|
|
matchingFiles.Add(file);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
|
|
|
_logger.Warn(ex, "Failed to import extra file: {0}", file);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-05-23 10:52:39 +00:00
|
|
|
foreach (var file in matchingFiles)
|
2022-02-27 20:37:23 +00:00
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
var extraFile = ImportFile(localEpisode.Series, episodeFile, file, isReadOnly, Path.GetExtension(file), null);
|
|
|
|
_mediaFileAttributeService.SetFilePermissions(file);
|
|
|
|
_otherExtraFileService.Upsert(extraFile);
|
|
|
|
importedFiles.Add(extraFile);
|
|
|
|
}
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
|
|
|
_logger.Warn(ex, "Failed to import extra file: {0}", file);
|
|
|
|
}
|
|
|
|
}
|
2015-12-25 09:22:00 +00:00
|
|
|
|
2022-02-27 20:37:23 +00:00
|
|
|
return importedFiles;
|
2015-12-25 09:22:00 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|