Leonardo Galli
8 years ago
parent
d9d8cbacec
commit
ad95fbfd4a
@ -0,0 +1,20 @@
|
|||||||
|
using NzbDrone.Common.Messaging;
|
||||||
|
using NzbDrone.Core.Tv;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.MediaFiles.Events
|
||||||
|
{
|
||||||
|
public class MovieFolderCreatedEvent : IEvent
|
||||||
|
{
|
||||||
|
public Movie Movie { get; private set; }
|
||||||
|
public MovieFile MovieFile { get; private set; }
|
||||||
|
public string SeriesFolder { get; set; }
|
||||||
|
public string SeasonFolder { get; set; }
|
||||||
|
public string MovieFolder { get; set; }
|
||||||
|
|
||||||
|
public MovieFolderCreatedEvent(Movie movie, MovieFile episodeFile)
|
||||||
|
{
|
||||||
|
Movie = movie;
|
||||||
|
MovieFile = episodeFile;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,32 @@
|
|||||||
|
using System.Collections.Generic;
|
||||||
|
using NzbDrone.Core.Datastore;
|
||||||
|
using NzbDrone.Core.Messaging.Events;
|
||||||
|
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.MediaFiles
|
||||||
|
{
|
||||||
|
public interface IMovieFileRepository : IBasicRepository<MovieFile>
|
||||||
|
{
|
||||||
|
List<MovieFile> GetFilesByMovie(int movieId);
|
||||||
|
List<MovieFile> GetFilesWithoutMediaInfo();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public class MovieFileRepository : BasicRepository<MovieFile>, IMovieFileRepository
|
||||||
|
{
|
||||||
|
public MovieFileRepository(IMainDatabase database, IEventAggregator eventAggregator)
|
||||||
|
: base(database, eventAggregator)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<MovieFile> GetFilesByMovie(int movieId)
|
||||||
|
{
|
||||||
|
return Query.Where(c => c.MovieId == movieId).ToList();
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<MovieFile> GetFilesWithoutMediaInfo()
|
||||||
|
{
|
||||||
|
return Query.Where(c => c.MediaInfo == null).ToList();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,147 @@
|
|||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.IO;
|
||||||
|
using System.Linq;
|
||||||
|
using NLog;
|
||||||
|
using NzbDrone.Common.Disk;
|
||||||
|
using NzbDrone.Common.Exceptron;
|
||||||
|
using NzbDrone.Common.Extensions;
|
||||||
|
using NzbDrone.Common.Instrumentation.Extensions;
|
||||||
|
using NzbDrone.Core.Configuration;
|
||||||
|
using NzbDrone.Core.MediaFiles.Events;
|
||||||
|
using NzbDrone.Core.Messaging.Events;
|
||||||
|
using NzbDrone.Core.Tv;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.MediaFiles
|
||||||
|
{
|
||||||
|
public interface IUpdateMovieFileService
|
||||||
|
{
|
||||||
|
void ChangeFileDateForFile(MovieFile movieFile, Movie movie);
|
||||||
|
}
|
||||||
|
|
||||||
|
public class UpdateMovieFileService : IUpdateMovieFileService,
|
||||||
|
IHandle<SeriesScannedEvent>
|
||||||
|
{
|
||||||
|
private readonly IDiskProvider _diskProvider;
|
||||||
|
private readonly IConfigService _configService;
|
||||||
|
private readonly IMovieService _movieService;
|
||||||
|
private readonly Logger _logger;
|
||||||
|
|
||||||
|
public UpdateMovieFileService(IDiskProvider diskProvider,
|
||||||
|
IConfigService configService,
|
||||||
|
IMovieService movieService,
|
||||||
|
Logger logger)
|
||||||
|
{
|
||||||
|
_diskProvider = diskProvider;
|
||||||
|
_configService = configService;
|
||||||
|
_movieService = movieService;
|
||||||
|
_logger = logger;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void ChangeFileDateForFile(MovieFile movieFile, Movie movie)
|
||||||
|
{
|
||||||
|
ChangeFileDate(movieFile, movie);
|
||||||
|
}
|
||||||
|
|
||||||
|
private bool ChangeFileDate(MovieFile movieFile, Movie movie)
|
||||||
|
{
|
||||||
|
var movieFilePath = Path.Combine(movie.Path, movieFile.RelativePath);
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void Handle(SeriesScannedEvent message)
|
||||||
|
{
|
||||||
|
if (_configService.FileDate == FileDateType.None)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* var movies = _movieService.MoviesWithFiles(message.Series.Id);
|
||||||
|
|
||||||
|
var movieFiles = new List<MovieFile>();
|
||||||
|
var updated = new List<MovieFile>();
|
||||||
|
|
||||||
|
foreach (var group in movies.GroupBy(e => e.MovieFileId))
|
||||||
|
{
|
||||||
|
var moviesInFile = group.Select(e => e).ToList();
|
||||||
|
var movieFile = moviesInFile.First().MovieFile;
|
||||||
|
|
||||||
|
movieFiles.Add(movieFile);
|
||||||
|
|
||||||
|
if (ChangeFileDate(movieFile, message.Series, moviesInFile))
|
||||||
|
{
|
||||||
|
updated.Add(movieFile);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (updated.Any())
|
||||||
|
{
|
||||||
|
_logger.ProgressDebug("Changed file date for {0} files of {1} in {2}", updated.Count, movieFiles.Count, message.Series.Title);
|
||||||
|
}
|
||||||
|
|
||||||
|
else
|
||||||
|
{
|
||||||
|
_logger.ProgressDebug("No file dates changed for {0}", message.Series.Title);
|
||||||
|
}*/
|
||||||
|
}
|
||||||
|
|
||||||
|
private bool ChangeFileDateToLocalAirDate(string filePath, string fileDate, string fileTime)
|
||||||
|
{
|
||||||
|
DateTime airDate;
|
||||||
|
|
||||||
|
if (DateTime.TryParse(fileDate + ' ' + fileTime, out airDate))
|
||||||
|
{
|
||||||
|
// avoiding false +ve checks and set date skewing by not using UTC (Windows)
|
||||||
|
DateTime oldDateTime = _diskProvider.FileGetLastWrite(filePath);
|
||||||
|
|
||||||
|
if (!DateTime.Equals(airDate, oldDateTime))
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
_diskProvider.FileSetLastWriteTime(filePath, airDate);
|
||||||
|
_logger.Debug("Date of file [{0}] changed from '{1}' to '{2}'", filePath, oldDateTime, airDate);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
_logger.Warn(ex, "Unable to set date of file [" + filePath + "]");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
else
|
||||||
|
{
|
||||||
|
_logger.Debug("Could not create valid date to change file [{0}]", filePath);
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
private bool ChangeFileDateToUtcAirDate(string filePath, DateTime airDateUtc)
|
||||||
|
{
|
||||||
|
DateTime oldLastWrite = _diskProvider.FileGetLastWrite(filePath);
|
||||||
|
|
||||||
|
if (!DateTime.Equals(airDateUtc, oldLastWrite))
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
_diskProvider.FileSetLastWriteTime(filePath, airDateUtc);
|
||||||
|
_logger.Debug("Date of file [{0}] changed from '{1}' to '{2}'", filePath, oldLastWrite, airDateUtc);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
ex.ExceptronIgnoreOnMono();
|
||||||
|
_logger.Warn(ex, "Unable to set date of file [" + filePath + "]");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,29 @@
|
|||||||
|
using System.Linq;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using NzbDrone.Core.Qualities;
|
||||||
|
using NzbDrone.Core.Tv;
|
||||||
|
using NzbDrone.Core.MediaFiles.MediaInfo;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.Parser.Model
|
||||||
|
{
|
||||||
|
public class LocalMovie
|
||||||
|
{
|
||||||
|
public LocalMovie()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
public string Path { get; set; }
|
||||||
|
public long Size { get; set; }
|
||||||
|
public ParsedEpisodeInfo ParsedEpisodeInfo { get; set; }
|
||||||
|
public Movie Movie { get; set; }
|
||||||
|
public QualityModel Quality { get; set; }
|
||||||
|
public MediaInfoModel MediaInfo { get; set; }
|
||||||
|
public bool ExistingFile { get; set; }
|
||||||
|
|
||||||
|
|
||||||
|
public override string ToString()
|
||||||
|
{
|
||||||
|
return Path;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue