using MediaBrowser.Common.Events; using MediaBrowser.Controller.Configuration; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Entities.Audio; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Persistence; using MediaBrowser.Model.Dto; using MediaBrowser.Model.Entities; using Microsoft.Extensions.Logging; using System; using System.Collections.Concurrent; using System.Collections.Generic; using System.Threading; using System.Threading.Tasks; using MediaBrowser.Controller.Dto; using System.Globalization; namespace Emby.Server.Implementations.Library { /// /// Class UserDataManager /// public class UserDataManager : IUserDataManager { public event EventHandler UserDataSaved; private readonly ConcurrentDictionary _userData = new ConcurrentDictionary(StringComparer.OrdinalIgnoreCase); private readonly ILogger _logger; private readonly IServerConfigurationManager _config; private Func _userManager; public UserDataManager(ILoggerFactory loggerFactory, IServerConfigurationManager config, Func userManager) { _config = config; _logger = loggerFactory.CreateLogger(GetType().Name); _userManager = userManager; } /// /// Gets or sets the repository. /// /// The repository. public IUserDataRepository Repository { get; set; } public void SaveUserData(Guid userId, BaseItem item, UserItemData userData, UserDataSaveReason reason, CancellationToken cancellationToken) { var user = _userManager().GetUserById(userId); SaveUserData(user, item, userData, reason, cancellationToken); } public void SaveUserData(User user, BaseItem item, UserItemData userData, UserDataSaveReason reason, CancellationToken cancellationToken) { if (userData == null) { throw new ArgumentNullException("userData"); } if (item == null) { throw new ArgumentNullException("item"); } cancellationToken.ThrowIfCancellationRequested(); var keys = item.GetUserDataKeys(); var userId = user.InternalId; foreach (var key in keys) { Repository.SaveUserData(userId, key, userData, cancellationToken); } var cacheKey = GetCacheKey(userId, item.Id); _userData.AddOrUpdate(cacheKey, userData, (k, v) => userData); EventHelper.FireEventIfNotNull(UserDataSaved, this, new UserDataSaveEventArgs { Keys = keys, UserData = userData, SaveReason = reason, UserId = user.Id, Item = item }, _logger); } /// /// Save the provided user data for the given user. Batch operation. Does not fire any events or update the cache. /// /// /// /// /// public void SaveAllUserData(Guid userId, UserItemData[] userData, CancellationToken cancellationToken) { var user = _userManager().GetUserById(userId); Repository.SaveAllUserData(user.InternalId, userData, cancellationToken); } /// /// Retrieve all user data for the given user /// /// /// public List GetAllUserData(Guid userId) { var user = _userManager().GetUserById(userId); return Repository.GetAllUserData(user.InternalId); } public UserItemData GetUserData(Guid userId, Guid itemId, List keys) { var user = _userManager().GetUserById(userId); return GetUserData(user, itemId, keys); } public UserItemData GetUserData(User user, Guid itemId, List keys) { var userId = user.InternalId; var cacheKey = GetCacheKey(userId, itemId); return _userData.GetOrAdd(cacheKey, k => GetUserDataInternal(userId, keys)); } private UserItemData GetUserDataInternal(long internalUserId, List keys) { var userData = Repository.GetUserData(internalUserId, keys); if (userData != null) { return userData; } if (keys.Count > 0) { return new UserItemData { Key = keys[0] }; } return null; } /// /// Gets the internal key. /// /// System.String. private string GetCacheKey(long internalUserId, Guid itemId) { return internalUserId.ToString(CultureInfo.InvariantCulture) + "-" + itemId.ToString("N"); } public UserItemData GetUserData(User user, BaseItem item) { return GetUserData(user, item.Id, item.GetUserDataKeys()); } public UserItemData GetUserData(string userId, BaseItem item) { return GetUserData(new Guid(userId), item); } public UserItemData GetUserData(Guid userId, BaseItem item) { return GetUserData(userId, item.Id, item.GetUserDataKeys()); } public UserItemDataDto GetUserDataDto(BaseItem item, User user) { var userData = GetUserData(user, item); var dto = GetUserItemDataDto(userData); item.FillUserDataDtoValues(dto, userData, null, user, new DtoOptions()); return dto; } public UserItemDataDto GetUserDataDto(BaseItem item, BaseItemDto itemDto, User user, DtoOptions options) { var userData = GetUserData(user, item); var dto = GetUserItemDataDto(userData); item.FillUserDataDtoValues(dto, userData, itemDto, user, options); return dto; } /// /// Converts a UserItemData to a DTOUserItemData /// /// The data. /// DtoUserItemData. /// private UserItemDataDto GetUserItemDataDto(UserItemData data) { if (data == null) { throw new ArgumentNullException("data"); } return new UserItemDataDto { IsFavorite = data.IsFavorite, Likes = data.Likes, PlaybackPositionTicks = data.PlaybackPositionTicks, PlayCount = data.PlayCount, Rating = data.Rating, Played = data.Played, LastPlayedDate = data.LastPlayedDate, Key = data.Key }; } public bool UpdatePlayState(BaseItem item, UserItemData data, long? reportedPositionTicks) { var playedToCompletion = false; var runtimeTicks = item.GetRunTimeTicksForPlayState(); var positionTicks = reportedPositionTicks ?? runtimeTicks; var hasRuntime = runtimeTicks > 0; // If a position has been reported, and if we know the duration if (positionTicks > 0 && hasRuntime) { var pctIn = Decimal.Divide(positionTicks, runtimeTicks) * 100; // Don't track in very beginning if (pctIn < _config.Configuration.MinResumePct) { positionTicks = 0; } // If we're at the end, assume completed else if (pctIn > _config.Configuration.MaxResumePct || positionTicks >= runtimeTicks) { positionTicks = 0; data.Played = playedToCompletion = true; } else { // Enforce MinResumeDuration var durationSeconds = TimeSpan.FromTicks(runtimeTicks).TotalSeconds; if (durationSeconds < _config.Configuration.MinResumeDurationSeconds) { positionTicks = 0; data.Played = playedToCompletion = true; } } } else if (!hasRuntime) { // If we don't know the runtime we'll just have to assume it was fully played data.Played = playedToCompletion = true; positionTicks = 0; } if (!item.SupportsPlayedStatus) { positionTicks = 0; data.Played = false; } if (!item.SupportsPositionTicksResume) { positionTicks = 0; } data.PlaybackPositionTicks = positionTicks; return playedToCompletion; } } }