jellyfin/MediaBrowser.Server.Implementations/FileOrganization/FileOrganizationService.cs

120 lines
4.2 KiB
C#
Raw Normal View History

2014-01-21 06:10:58 +00:00
using MediaBrowser.Common.Extensions;
2014-01-22 17:05:06 +00:00
using MediaBrowser.Common.IO;
2014-01-21 06:10:58 +00:00
using MediaBrowser.Common.ScheduledTasks;
2014-01-22 17:05:06 +00:00
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.FileOrganization;
2014-01-21 06:10:58 +00:00
using MediaBrowser.Controller.IO;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Persistence;
using MediaBrowser.Model.FileOrganization;
2014-01-21 06:10:58 +00:00
using MediaBrowser.Model.Logging;
2014-01-20 18:46:08 +00:00
using MediaBrowser.Model.Querying;
2014-01-21 06:10:58 +00:00
using System;
using System.IO;
using System.Threading;
using System.Threading.Tasks;
namespace MediaBrowser.Server.Implementations.FileOrganization
{
public class FileOrganizationService : IFileOrganizationService
{
private readonly ITaskManager _taskManager;
private readonly IFileOrganizationRepository _repo;
2014-01-21 06:10:58 +00:00
private readonly ILogger _logger;
private readonly IDirectoryWatchers _directoryWatchers;
private readonly ILibraryManager _libraryManager;
2014-01-22 17:05:06 +00:00
private readonly IServerConfigurationManager _config;
private readonly IFileSystem _fileSystem;
2014-01-22 17:05:06 +00:00
public FileOrganizationService(ITaskManager taskManager, IFileOrganizationRepository repo, ILogger logger, IDirectoryWatchers directoryWatchers, ILibraryManager libraryManager, IServerConfigurationManager config, IFileSystem fileSystem)
{
_taskManager = taskManager;
_repo = repo;
2014-01-21 06:10:58 +00:00
_logger = logger;
_directoryWatchers = directoryWatchers;
_libraryManager = libraryManager;
2014-01-22 17:05:06 +00:00
_config = config;
_fileSystem = fileSystem;
}
public void BeginProcessNewFiles()
{
2014-01-21 06:10:58 +00:00
_taskManager.CancelIfRunningAndQueue<OrganizerScheduledTask>();
}
public Task SaveResult(FileOrganizationResult result, CancellationToken cancellationToken)
{
2014-01-21 06:10:58 +00:00
if (result == null || string.IsNullOrEmpty(result.OriginalPath))
{
throw new ArgumentNullException("result");
}
2014-01-21 16:24:12 +00:00
result.Id = result.OriginalPath.GetMD5().ToString("N");
2014-01-21 06:10:58 +00:00
return _repo.SaveResult(result, cancellationToken);
}
2014-01-20 18:46:08 +00:00
public QueryResult<FileOrganizationResult> GetResults(FileOrganizationResultQuery query)
{
return _repo.GetResults(query);
}
2014-01-21 06:10:58 +00:00
2014-01-22 17:05:06 +00:00
public FileOrganizationResult GetResult(string id)
{
return _repo.GetResult(id);
}
2014-01-21 06:10:58 +00:00
public Task DeleteOriginalFile(string resultId)
{
var result = _repo.GetResult(resultId);
_logger.Info("Requested to delete {0}", result.OriginalPath);
try
{
File.Delete(result.OriginalPath);
}
catch (Exception ex)
{
_logger.ErrorException("Error deleting {0}", ex, result.OriginalPath);
}
return _repo.Delete(resultId);
}
public async Task PerformOrganization(string resultId)
{
var result = _repo.GetResult(resultId);
if (string.IsNullOrEmpty(result.TargetPath))
{
throw new ArgumentException("No target path available.");
}
2014-01-22 17:05:06 +00:00
var organizer = new EpisodeFileOrganizer(this, _config, _fileSystem, _logger, _libraryManager,
_directoryWatchers);
2014-01-21 06:10:58 +00:00
2014-01-22 17:05:06 +00:00
await organizer.OrganizeEpisodeFile(result.OriginalPath, _config.Configuration.TvFileOrganizationOptions, true)
.ConfigureAwait(false);
2014-01-21 06:10:58 +00:00
2014-01-22 17:05:06 +00:00
await _libraryManager.ValidateMediaLibrary(new Progress<double>(), CancellationToken.None)
.ConfigureAwait(false);
}
2014-01-21 06:10:58 +00:00
2014-01-22 17:05:06 +00:00
public Task ClearLog()
{
return _repo.DeleteAll();
}
2014-01-21 06:10:58 +00:00
2014-01-22 17:05:06 +00:00
public async Task PerformEpisodeOrganization(EpisodeFileOrganizationRequest request)
{
var organizer = new EpisodeFileOrganizer(this, _config, _fileSystem, _logger, _libraryManager,
_directoryWatchers);
2014-01-21 06:10:58 +00:00
2014-01-22 17:05:06 +00:00
await organizer.OrganizeWithCorrection(request, _config.Configuration.TvFileOrganizationOptions).ConfigureAwait(false);
2014-01-21 06:10:58 +00:00
await _libraryManager.ValidateMediaLibrary(new Progress<double>(), CancellationToken.None)
.ConfigureAwait(false);
}
}
}