2014-07-23 23:43:54 +00:00
|
|
|
|
using System.IO;
|
|
|
|
|
using NLog;
|
2014-04-10 17:58:50 +00:00
|
|
|
|
using NzbDrone.Common.Disk;
|
|
|
|
|
using NzbDrone.Core.MediaFiles.Events;
|
|
|
|
|
using NzbDrone.Core.Messaging.Events;
|
|
|
|
|
using NzbDrone.Core.Tv;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
2014-12-14 18:50:38 +00:00
|
|
|
|
using NzbDrone.Core.Configuration;
|
2014-04-10 17:58:50 +00:00
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.MediaFiles.MediaInfo
|
|
|
|
|
{
|
2019-02-22 18:45:02 +00:00
|
|
|
|
public interface IUpdateMediaInfo
|
|
|
|
|
{
|
2021-11-24 00:33:45 +00:00
|
|
|
|
bool Update(EpisodeFile episodeFile, Series series);
|
2019-02-22 18:45:02 +00:00
|
|
|
|
}
|
|
|
|
|
|
2021-11-24 00:33:45 +00:00
|
|
|
|
public class UpdateMediaInfoService : IUpdateMediaInfo, IHandle<SeriesScannedEvent>
|
2014-04-10 17:58:50 +00:00
|
|
|
|
{
|
|
|
|
|
private readonly IDiskProvider _diskProvider;
|
|
|
|
|
private readonly IMediaFileService _mediaFileService;
|
|
|
|
|
private readonly IVideoFileInfoReader _videoFileInfoReader;
|
2014-12-14 18:50:38 +00:00
|
|
|
|
private readonly IConfigService _configService;
|
2014-04-10 17:58:50 +00:00
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
|
|
public UpdateMediaInfoService(IDiskProvider diskProvider,
|
|
|
|
|
IMediaFileService mediaFileService,
|
|
|
|
|
IVideoFileInfoReader videoFileInfoReader,
|
2014-12-14 18:50:38 +00:00
|
|
|
|
IConfigService configService,
|
2014-04-10 17:58:50 +00:00
|
|
|
|
Logger logger)
|
|
|
|
|
{
|
|
|
|
|
_diskProvider = diskProvider;
|
|
|
|
|
_mediaFileService = mediaFileService;
|
|
|
|
|
_videoFileInfoReader = videoFileInfoReader;
|
2014-12-14 18:50:38 +00:00
|
|
|
|
_configService = configService;
|
2014-04-10 17:58:50 +00:00
|
|
|
|
_logger = logger;
|
|
|
|
|
}
|
|
|
|
|
|
2019-02-22 18:45:02 +00:00
|
|
|
|
public void Handle(SeriesScannedEvent message)
|
2014-04-10 17:58:50 +00:00
|
|
|
|
{
|
2019-02-22 18:45:02 +00:00
|
|
|
|
if (!_configService.EnableMediaInfo)
|
2014-04-10 17:58:50 +00:00
|
|
|
|
{
|
2019-02-22 18:45:02 +00:00
|
|
|
|
_logger.Debug("MediaInfo is disabled");
|
|
|
|
|
return;
|
|
|
|
|
}
|
2014-04-10 17:58:50 +00:00
|
|
|
|
|
2019-02-22 18:45:02 +00:00
|
|
|
|
var allMediaFiles = _mediaFileService.GetFilesBySeries(message.Series.Id);
|
|
|
|
|
var filteredMediaFiles = allMediaFiles.Where(c =>
|
|
|
|
|
c.MediaInfo == null ||
|
|
|
|
|
c.MediaInfo.SchemaRevision < VideoFileInfoReader.MINIMUM_MEDIA_INFO_SCHEMA_REVISION).ToList();
|
2014-04-10 17:58:50 +00:00
|
|
|
|
|
2019-02-22 18:45:02 +00:00
|
|
|
|
foreach (var mediaFile in filteredMediaFiles)
|
|
|
|
|
{
|
|
|
|
|
UpdateMediaInfo(mediaFile, message.Series);
|
2014-04-10 17:58:50 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2021-11-24 00:33:45 +00:00
|
|
|
|
public bool Update(EpisodeFile episodeFile, Series series)
|
2014-04-10 17:58:50 +00:00
|
|
|
|
{
|
2014-12-14 18:50:38 +00:00
|
|
|
|
if (!_configService.EnableMediaInfo)
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("MediaInfo is disabled");
|
2021-11-24 00:33:45 +00:00
|
|
|
|
return false;
|
2014-12-14 18:50:38 +00:00
|
|
|
|
}
|
2021-11-24 00:33:45 +00:00
|
|
|
|
|
|
|
|
|
return UpdateMediaInfo(episodeFile, series);
|
2019-02-22 18:45:02 +00:00
|
|
|
|
}
|
2014-12-14 18:50:38 +00:00
|
|
|
|
|
2021-11-24 00:33:45 +00:00
|
|
|
|
private bool UpdateMediaInfo(EpisodeFile episodeFile, Series series)
|
2019-02-22 18:45:02 +00:00
|
|
|
|
{
|
|
|
|
|
var path = Path.Combine(series.Path, episodeFile.RelativePath);
|
2014-04-10 17:58:50 +00:00
|
|
|
|
|
2019-02-22 18:45:02 +00:00
|
|
|
|
if (!_diskProvider.FileExists(path))
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Can't update MediaInfo because '{0}' does not exist", path);
|
2021-11-24 00:33:45 +00:00
|
|
|
|
return false;
|
2019-02-22 18:45:02 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var updatedMediaInfo = _videoFileInfoReader.GetMediaInfo(path);
|
|
|
|
|
|
2021-11-24 00:33:45 +00:00
|
|
|
|
if (updatedMediaInfo == null) return false;
|
|
|
|
|
|
|
|
|
|
episodeFile.MediaInfo = updatedMediaInfo;
|
|
|
|
|
_mediaFileService.Update(episodeFile);
|
|
|
|
|
_logger.Debug("Updated MediaInfo for '{0}'", path);
|
|
|
|
|
|
|
|
|
|
return true;
|
|
|
|
|
|
2014-04-10 17:58:50 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|