2015-03-08 05:44:31 +01:00
|
|
|
|
using MediaBrowser.Common.Configuration;
|
|
|
|
|
using MediaBrowser.Common.Extensions;
|
|
|
|
|
using MediaBrowser.Common.IO;
|
2015-03-31 20:50:08 +02:00
|
|
|
|
using MediaBrowser.Controller;
|
2015-03-08 05:44:31 +01:00
|
|
|
|
using MediaBrowser.Controller.Sync;
|
|
|
|
|
using MediaBrowser.Model.Logging;
|
|
|
|
|
using MediaBrowser.Model.Serialization;
|
|
|
|
|
using MediaBrowser.Model.Sync;
|
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Threading;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
|
|
|
|
|
namespace MediaBrowser.Server.Implementations.Sync
|
|
|
|
|
{
|
|
|
|
|
public class TargetDataProvider : ISyncDataProvider
|
|
|
|
|
{
|
|
|
|
|
private readonly SyncTarget _target;
|
|
|
|
|
private readonly IServerSyncProvider _provider;
|
|
|
|
|
|
|
|
|
|
private readonly SemaphoreSlim _dataLock = new SemaphoreSlim(1, 1);
|
|
|
|
|
private List<LocalItem> _items;
|
|
|
|
|
|
|
|
|
|
private readonly ILogger _logger;
|
|
|
|
|
private readonly IJsonSerializer _json;
|
|
|
|
|
private readonly IFileSystem _fileSystem;
|
|
|
|
|
private readonly IApplicationPaths _appPaths;
|
2015-03-31 20:50:08 +02:00
|
|
|
|
private readonly IServerApplicationHost _appHost;
|
2015-03-08 05:44:31 +01:00
|
|
|
|
|
|
|
|
|
private readonly SemaphoreSlim _cacheFileLock = new SemaphoreSlim(1, 1);
|
|
|
|
|
|
2015-03-31 20:50:08 +02:00
|
|
|
|
public TargetDataProvider(IServerSyncProvider provider, SyncTarget target, IServerApplicationHost appHost, ILogger logger, IJsonSerializer json, IFileSystem fileSystem, IApplicationPaths appPaths)
|
2015-03-08 05:44:31 +01:00
|
|
|
|
{
|
|
|
|
|
_logger = logger;
|
|
|
|
|
_json = json;
|
|
|
|
|
_provider = provider;
|
|
|
|
|
_target = target;
|
|
|
|
|
_fileSystem = fileSystem;
|
|
|
|
|
_appPaths = appPaths;
|
2015-03-31 20:50:08 +02:00
|
|
|
|
_appHost = appHost;
|
2015-03-08 05:44:31 +01:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private string GetRemotePath()
|
|
|
|
|
{
|
|
|
|
|
var parts = new List<string>
|
|
|
|
|
{
|
2015-03-31 20:50:08 +02:00
|
|
|
|
_appHost.FriendlyName,
|
2015-03-08 05:44:31 +01:00
|
|
|
|
"data.json"
|
|
|
|
|
};
|
|
|
|
|
|
2015-03-31 20:50:08 +02:00
|
|
|
|
parts = parts.Select(i => GetValidFilename(_provider, i)).ToList();
|
|
|
|
|
|
2015-03-08 05:44:31 +01:00
|
|
|
|
return _provider.GetFullPath(parts, _target);
|
|
|
|
|
}
|
|
|
|
|
|
2015-03-31 20:50:08 +02:00
|
|
|
|
private string GetValidFilename(IServerSyncProvider provider, string filename)
|
|
|
|
|
{
|
|
|
|
|
// We can always add this method to the sync provider if it's really needed
|
|
|
|
|
return _fileSystem.GetValidFilename(filename);
|
|
|
|
|
}
|
|
|
|
|
|
2015-03-08 05:44:31 +01:00
|
|
|
|
private async Task EnsureData(CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
if (_items == null)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
2015-04-04 03:24:49 +02:00
|
|
|
|
var path = GetRemotePath();
|
|
|
|
|
|
|
|
|
|
_logger.Debug("Getting {0} from {1}", path, _provider.Name);
|
|
|
|
|
|
|
|
|
|
using (var stream = await _provider.GetFile(path, _target, new Progress<double>(), cancellationToken))
|
2015-03-08 05:44:31 +01:00
|
|
|
|
{
|
|
|
|
|
_items = _json.DeserializeFromStream<List<LocalItem>>(stream);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
catch (FileNotFoundException)
|
|
|
|
|
{
|
|
|
|
|
_items = new List<LocalItem>();
|
|
|
|
|
}
|
|
|
|
|
catch (DirectoryNotFoundException)
|
|
|
|
|
{
|
|
|
|
|
_items = new List<LocalItem>();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private async Task SaveData(CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
using (var stream = new MemoryStream())
|
|
|
|
|
{
|
|
|
|
|
_json.SerializeToStream(_items, stream);
|
|
|
|
|
|
|
|
|
|
// Save to sync provider
|
|
|
|
|
stream.Position = 0;
|
|
|
|
|
await _provider.SendFile(stream, GetRemotePath(), _target, new Progress<double>(), cancellationToken).ConfigureAwait(false);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private async Task<T> GetData<T>(Func<List<LocalItem>, T> dataFactory)
|
|
|
|
|
{
|
|
|
|
|
await _dataLock.WaitAsync().ConfigureAwait(false);
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
await EnsureData(CancellationToken.None).ConfigureAwait(false);
|
|
|
|
|
|
|
|
|
|
return dataFactory(_items);
|
|
|
|
|
}
|
|
|
|
|
finally
|
|
|
|
|
{
|
|
|
|
|
_dataLock.Release();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private async Task UpdateData(Func<List<LocalItem>, List<LocalItem>> action)
|
|
|
|
|
{
|
|
|
|
|
await _dataLock.WaitAsync().ConfigureAwait(false);
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
await EnsureData(CancellationToken.None).ConfigureAwait(false);
|
|
|
|
|
|
|
|
|
|
_items = action(_items);
|
|
|
|
|
|
|
|
|
|
await SaveData(CancellationToken.None).ConfigureAwait(false);
|
|
|
|
|
}
|
|
|
|
|
finally
|
|
|
|
|
{
|
|
|
|
|
_dataLock.Release();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Task<List<string>> GetServerItemIds(SyncTarget target, string serverId)
|
|
|
|
|
{
|
|
|
|
|
return GetData(items => items.Where(i => string.Equals(i.ServerId, serverId, StringComparison.OrdinalIgnoreCase)).Select(i => i.ItemId).ToList());
|
|
|
|
|
}
|
|
|
|
|
|
2015-03-31 20:50:08 +02:00
|
|
|
|
public Task<List<string>> GetSyncJobItemIds(SyncTarget target, string serverId)
|
|
|
|
|
{
|
|
|
|
|
return GetData(items => items.Where(i => string.Equals(i.ServerId, serverId, StringComparison.OrdinalIgnoreCase)).Select(i => i.SyncJobItemId).Where(i => !string.IsNullOrWhiteSpace(i)).ToList());
|
|
|
|
|
}
|
|
|
|
|
|
2015-03-08 05:44:31 +01:00
|
|
|
|
public Task AddOrUpdate(SyncTarget target, LocalItem item)
|
|
|
|
|
{
|
|
|
|
|
return UpdateData(items =>
|
|
|
|
|
{
|
|
|
|
|
var list = items.Where(i => !string.Equals(i.Id, item.Id, StringComparison.OrdinalIgnoreCase))
|
|
|
|
|
.ToList();
|
|
|
|
|
|
|
|
|
|
list.Add(item);
|
|
|
|
|
|
|
|
|
|
return list;
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Task Delete(SyncTarget target, string id)
|
|
|
|
|
{
|
|
|
|
|
return UpdateData(items => items.Where(i => !string.Equals(i.Id, id, StringComparison.OrdinalIgnoreCase)).ToList());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Task<LocalItem> Get(SyncTarget target, string id)
|
|
|
|
|
{
|
|
|
|
|
return GetData(items => items.FirstOrDefault(i => string.Equals(i.Id, id, StringComparison.OrdinalIgnoreCase)));
|
|
|
|
|
}
|
|
|
|
|
|
2015-04-04 03:24:49 +02:00
|
|
|
|
public Task<List<LocalItem>> GetItems(SyncTarget target, string serverId, string itemId)
|
2015-03-08 05:44:31 +01:00
|
|
|
|
{
|
2015-04-04 03:24:49 +02:00
|
|
|
|
return GetData(items => items.Where(i => string.Equals(i.ServerId, serverId, StringComparison.OrdinalIgnoreCase) && string.Equals(i.ItemId, itemId, StringComparison.OrdinalIgnoreCase)).ToList());
|
2015-03-08 05:44:31 +01:00
|
|
|
|
}
|
|
|
|
|
|
2015-04-04 03:24:49 +02:00
|
|
|
|
public Task<List<LocalItem>> GetItemsBySyncJobItemId(SyncTarget target, string serverId, string syncJobItemId)
|
2015-03-08 05:44:31 +01:00
|
|
|
|
{
|
2015-04-04 03:24:49 +02:00
|
|
|
|
return GetData(items => items.Where(i => string.Equals(i.ServerId, serverId, StringComparison.OrdinalIgnoreCase) && string.Equals(i.SyncJobItemId, syncJobItemId, StringComparison.OrdinalIgnoreCase)).ToList());
|
2015-03-31 20:50:08 +02:00
|
|
|
|
}
|
2015-03-08 05:44:31 +01:00
|
|
|
|
}
|
|
|
|
|
}
|