parent
feb947fb74
commit
cbe4be814c
@ -0,0 +1,14 @@
|
|||||||
|
using NzbDrone.Common.Messaging;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.MediaFiles.Commands
|
||||||
|
{
|
||||||
|
public class CleanMediaFileDb : ICommand
|
||||||
|
{
|
||||||
|
public int SeriesId { get; private set; }
|
||||||
|
|
||||||
|
public CleanMediaFileDb(int seriesId)
|
||||||
|
{
|
||||||
|
SeriesId = seriesId;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,17 @@
|
|||||||
|
using NzbDrone.Common.Messaging;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.MediaFiles.Commands
|
||||||
|
{
|
||||||
|
public class DiskScanCommand : ICommand
|
||||||
|
{
|
||||||
|
public int? SeriesId { get; private set; }
|
||||||
|
|
||||||
|
public DiskScanCommand(int seriesId = 0)
|
||||||
|
{
|
||||||
|
if (seriesId != 0)
|
||||||
|
{
|
||||||
|
SeriesId = seriesId;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -1,48 +0,0 @@
|
|||||||
using System;
|
|
||||||
using NLog;
|
|
||||||
using NzbDrone.Common;
|
|
||||||
|
|
||||||
namespace NzbDrone.Core.MediaFiles
|
|
||||||
{
|
|
||||||
public interface ICleanGhostFiles
|
|
||||||
{
|
|
||||||
void RemoveNonExistingFiles(int seriesId);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
public class GhostFileCleanupService : ICleanGhostFiles
|
|
||||||
{
|
|
||||||
private readonly IMediaFileService _mediaFileService;
|
|
||||||
private readonly IDiskProvider _diskProvider;
|
|
||||||
private readonly Logger _logger;
|
|
||||||
|
|
||||||
public GhostFileCleanupService(IMediaFileService mediaFileService, IDiskProvider diskProvider, Logger logger)
|
|
||||||
{
|
|
||||||
_mediaFileService = mediaFileService;
|
|
||||||
_diskProvider = diskProvider;
|
|
||||||
_logger = logger;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void RemoveNonExistingFiles(int seriesId)
|
|
||||||
{
|
|
||||||
var seriesFile = _mediaFileService.GetFilesBySeries(seriesId);
|
|
||||||
|
|
||||||
foreach (var episodeFile in seriesFile)
|
|
||||||
{
|
|
||||||
try
|
|
||||||
{
|
|
||||||
if (!_diskProvider.FileExists(episodeFile.Path))
|
|
||||||
{
|
|
||||||
_logger.Trace("File [{0}] no longer exists on disk. removing from db", episodeFile.Path);
|
|
||||||
_mediaFileService.Delete(episodeFile);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
catch (Exception ex)
|
|
||||||
{
|
|
||||||
var message = String.Format("Unable to cleanup EpisodeFile in DB: {0}", episodeFile.Id);
|
|
||||||
_logger.ErrorException(message, ex);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@ -0,0 +1,55 @@
|
|||||||
|
using System;
|
||||||
|
using System.Linq;
|
||||||
|
using NLog;
|
||||||
|
using NzbDrone.Common;
|
||||||
|
using NzbDrone.Common.Messaging;
|
||||||
|
using NzbDrone.Core.MediaFiles.Commands;
|
||||||
|
using NzbDrone.Core.Tv;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.MediaFiles
|
||||||
|
{
|
||||||
|
|
||||||
|
public class MediaFileTableCleanupService : IExecute<CleanMediaFileDb>
|
||||||
|
{
|
||||||
|
private readonly IMediaFileService _mediaFileService;
|
||||||
|
private readonly IDiskProvider _diskProvider;
|
||||||
|
private readonly IEpisodeService _episodeService;
|
||||||
|
private readonly Logger _logger;
|
||||||
|
|
||||||
|
public MediaFileTableCleanupService(IMediaFileService mediaFileService, IDiskProvider diskProvider, IEpisodeService episodeService, Logger logger)
|
||||||
|
{
|
||||||
|
_mediaFileService = mediaFileService;
|
||||||
|
_diskProvider = diskProvider;
|
||||||
|
_episodeService = episodeService;
|
||||||
|
_logger = logger;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Execute(CleanMediaFileDb message)
|
||||||
|
{
|
||||||
|
var seriesFile = _mediaFileService.GetFilesBySeries(message.SeriesId);
|
||||||
|
|
||||||
|
foreach (var episodeFile in seriesFile)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
if (!_diskProvider.FileExists(episodeFile.Path))
|
||||||
|
{
|
||||||
|
_logger.Trace("File [{0}] no longer exists on disk. removing from db", episodeFile.Path);
|
||||||
|
_mediaFileService.Delete(episodeFile);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!_episodeService.GetEpisodesByFileId(episodeFile.Id).Any())
|
||||||
|
{
|
||||||
|
_logger.Trace("File [{0}] is not assigned to any episodes. removing from db", episodeFile.Path);
|
||||||
|
_mediaFileService.Delete(episodeFile);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
var errorMessage = String.Format("Unable to cleanup EpisodeFile in DB: {0}", episodeFile.Id);
|
||||||
|
_logger.ErrorException(errorMessage, ex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in new issue