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

284 lines
9.9 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.Controller.Configuration;
using MediaBrowser.Controller.FileOrganization;
2014-01-21 06:10:58 +00:00
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Net;
using MediaBrowser.Controller.Persistence;
using MediaBrowser.Controller.Providers;
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.Collections.Concurrent;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Model.IO;
using MediaBrowser.Controller.Session;
using MediaBrowser.Model.Events;
using MediaBrowser.Common.Events;
2017-05-26 06:48:54 +00:00
2016-10-25 19:02:04 +00:00
using MediaBrowser.Controller.IO;
2016-10-23 19:47:34 +00:00
using MediaBrowser.Model.Tasks;
namespace Emby.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 ILibraryMonitor _libraryMonitor;
2014-01-21 06:10:58 +00:00
private readonly ILibraryManager _libraryManager;
2014-01-22 17:05:06 +00:00
private readonly IServerConfigurationManager _config;
private readonly IFileSystem _fileSystem;
private readonly IProviderManager _providerManager;
private readonly ConcurrentDictionary<string, bool> _inProgressItemIds = new ConcurrentDictionary<string, bool>();
public event EventHandler<GenericEventArgs<FileOrganizationResult>> ItemAdded;
public event EventHandler<GenericEventArgs<FileOrganizationResult>> ItemUpdated;
public event EventHandler<GenericEventArgs<FileOrganizationResult>> ItemRemoved;
public event EventHandler LogReset;
public FileOrganizationService(ITaskManager taskManager, IFileOrganizationRepository repo, ILogger logger, ILibraryMonitor libraryMonitor, ILibraryManager libraryManager, IServerConfigurationManager config, IFileSystem fileSystem, IProviderManager providerManager)
{
_taskManager = taskManager;
_repo = repo;
2014-01-21 06:10:58 +00:00
_logger = logger;
_libraryMonitor = libraryMonitor;
2014-01-21 06:10:58 +00:00
_libraryManager = libraryManager;
2014-01-22 17:05:06 +00:00
_config = config;
_fileSystem = fileSystem;
_providerManager = providerManager;
}
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)
{
var results = _repo.GetResults(query);
foreach (var result in results.Items)
{
result.IsInProgress = _inProgressItemIds.ContainsKey(result.Id);
}
return results;
}
2014-01-21 06:10:58 +00:00
2014-01-22 17:05:06 +00:00
public FileOrganizationResult GetResult(string id)
{
var result = _repo.GetResult(id);
if (result != null)
{
result.IsInProgress = _inProgressItemIds.ContainsKey(result.Id);
}
return result;
2014-01-22 17:05:06 +00:00
}
2014-02-19 16:24:06 +00:00
public FileOrganizationResult GetResultBySourcePath(string path)
{
if (string.IsNullOrEmpty(path))
{
throw new ArgumentNullException("path");
}
var id = path.GetMD5().ToString("N");
return GetResult(id);
}
public async Task DeleteOriginalFile(string resultId)
2014-01-21 06:10:58 +00:00
{
var result = _repo.GetResult(resultId);
_logger.Info("Requested to delete {0}", result.OriginalPath);
if (!AddToInProgressList(result, false))
{
throw new Exception("Path is currently processed otherwise. Please try again later.");
}
2014-01-21 06:10:58 +00:00
try
{
_fileSystem.DeleteFile(result.OriginalPath);
2014-01-21 06:10:58 +00:00
}
catch (Exception ex)
{
_logger.ErrorException("Error deleting {0}", ex, result.OriginalPath);
}
finally
{
RemoveFromInprogressList(result);
}
2014-01-21 06:10:58 +00:00
await _repo.Delete(resultId);
EventHelper.FireEventIfNotNull(ItemRemoved, this, new GenericEventArgs<FileOrganizationResult>(result), _logger);
2014-01-21 06:10:58 +00:00
}
2016-05-14 16:06:01 +00:00
private AutoOrganizeOptions GetAutoOrganizeOptions()
2014-07-31 02:09:23 +00:00
{
return _config.GetAutoOrganizeOptions();
2014-07-31 02:09:23 +00:00
}
2014-01-21 06:10:58 +00:00
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,
_libraryMonitor, _providerManager);
2014-01-21 06:10:58 +00:00
var organizeResult = await organizer.OrganizeEpisodeFile(result.OriginalPath, GetAutoOrganizeOptions(), true, CancellationToken.None)
2014-01-22 17:05:06 +00:00
.ConfigureAwait(false);
if (organizeResult.Status != FileSortingStatus.Success)
{
throw new Exception(result.StatusMessage);
}
2014-01-22 17:05:06 +00:00
}
2014-01-21 06:10:58 +00:00
public async Task ClearLog()
2014-01-22 17:05:06 +00:00
{
await _repo.DeleteAll();
EventHelper.FireEventIfNotNull(LogReset, this, EventArgs.Empty, _logger);
2014-01-22 17:05:06 +00:00
}
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,
_libraryMonitor, _providerManager);
2014-01-21 06:10:58 +00:00
var result = await organizer.OrganizeWithCorrection(request, GetAutoOrganizeOptions(), CancellationToken.None).ConfigureAwait(false);
if (result.Status != FileSortingStatus.Success)
{
throw new Exception(result.StatusMessage);
}
}
public QueryResult<SmartMatchInfo> GetSmartMatchInfos(FileOrganizationResultQuery query)
{
if (query == null)
{
throw new ArgumentNullException("query");
}
2016-05-14 16:06:01 +00:00
var options = GetAutoOrganizeOptions();
2016-02-07 05:15:26 +00:00
var items = options.SmartMatchInfos.Skip(query.StartIndex ?? 0).Take(query.Limit ?? Int32.MaxValue).ToArray();
return new QueryResult<SmartMatchInfo>()
{
2016-02-07 05:15:26 +00:00
Items = items,
TotalRecordCount = options.SmartMatchInfos.Length
};
}
public void DeleteSmartMatchEntry(string itemName, string matchString)
{
if (string.IsNullOrEmpty(itemName))
{
throw new ArgumentNullException("itemName");
}
if (string.IsNullOrEmpty(matchString))
{
throw new ArgumentNullException("matchString");
}
2016-05-14 16:06:01 +00:00
var options = GetAutoOrganizeOptions();
SmartMatchInfo info = options.SmartMatchInfos.FirstOrDefault(i => string.Equals(i.ItemName, itemName));
if (info != null && info.MatchStrings.Contains(matchString))
{
2016-02-07 05:15:26 +00:00
var list = info.MatchStrings.ToList();
list.Remove(matchString);
info.MatchStrings = list.ToArray();
if (info.MatchStrings.Length == 0)
{
2016-02-07 05:15:26 +00:00
var infos = options.SmartMatchInfos.ToList();
infos.Remove(info);
options.SmartMatchInfos = infos.ToArray();
}
_config.SaveAutoOrganizeOptions(options);
}
2014-01-21 06:10:58 +00:00
}
/// <summary>
/// Attempts to add a an item to the list of currently processed items.
/// </summary>
/// <param name="result">The result item.</param>
/// <param name="isNewItem">Passing true will notify the client to reload all items, otherwise only a single item will be refreshed.</param>
/// <returns>True if the item was added, False if the item is already contained in the list.</returns>
public bool AddToInProgressList(FileOrganizationResult result, bool isNewItem)
{
if (string.IsNullOrWhiteSpace(result.Id))
{
result.Id = result.OriginalPath.GetMD5().ToString("N");
}
if (!_inProgressItemIds.TryAdd(result.Id, false))
{
return false;
}
result.IsInProgress = true;
if (isNewItem)
{
EventHelper.FireEventIfNotNull(ItemAdded, this, new GenericEventArgs<FileOrganizationResult>(result), _logger);
}
else
{
EventHelper.FireEventIfNotNull(ItemUpdated, this, new GenericEventArgs<FileOrganizationResult>(result), _logger);
}
return true;
}
/// <summary>
/// Removes an item from the list of currently processed items.
/// </summary>
/// <param name="result">The result item.</param>
/// <returns>True if the item was removed, False if the item was not contained in the list.</returns>
public bool RemoveFromInprogressList(FileOrganizationResult result)
{
bool itemValue;
var retval = _inProgressItemIds.TryRemove(result.Id, out itemValue);
result.IsInProgress = false;
EventHelper.FireEventIfNotNull(ItemUpdated, this, new GenericEventArgs<FileOrganizationResult>(result), _logger);
return retval;
}
}
}