You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.9 KiB
58 lines
1.9 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using NLog;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Core.Movies;
|
|
|
|
namespace NzbDrone.Core.MediaFiles
|
|
{
|
|
public interface IMediaFileTableCleanupService
|
|
{
|
|
void Clean(Movie movie, List<string> filesOnDisk);
|
|
}
|
|
|
|
public class MediaFileTableCleanupService : IMediaFileTableCleanupService
|
|
{
|
|
private readonly IMediaFileService _mediaFileService;
|
|
private readonly IMovieService _movieService;
|
|
private readonly Logger _logger;
|
|
|
|
public MediaFileTableCleanupService(IMediaFileService mediaFileService,
|
|
IMovieService movieService,
|
|
Logger logger)
|
|
{
|
|
_mediaFileService = mediaFileService;
|
|
_movieService = movieService;
|
|
_logger = logger;
|
|
}
|
|
|
|
public void Clean(Movie movie, List<string> filesOnDisk)
|
|
{
|
|
var movieFiles = _mediaFileService.GetFilesByMovie(movie.Id);
|
|
|
|
var filesOnDiskKeys = new HashSet<string>(filesOnDisk, PathEqualityComparer.Instance);
|
|
|
|
foreach (var movieFile in movieFiles)
|
|
{
|
|
var movieFilePath = Path.Combine(movie.Path, movieFile.RelativePath);
|
|
|
|
try
|
|
{
|
|
if (!filesOnDiskKeys.Contains(movieFilePath))
|
|
{
|
|
_logger.Debug("File [{0}] no longer exists on disk, removing from db", movieFilePath);
|
|
_mediaFileService.Delete(movieFile, DeleteMediaFileReason.MissingFromDisk);
|
|
continue;
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
var errorMessage = string.Format("Unable to cleanup MovieFile in DB: {0}", movieFile.Id);
|
|
_logger.Error(ex, errorMessage);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|