2
0

UserDataManager.cs 9.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289
  1. #pragma warning disable CS1591
  2. using System;
  3. using System.Collections.Concurrent;
  4. using System.Collections.Generic;
  5. using System.Globalization;
  6. using System.Threading;
  7. using Jellyfin.Data.Entities;
  8. using MediaBrowser.Controller.Configuration;
  9. using MediaBrowser.Controller.Dto;
  10. using MediaBrowser.Controller.Entities;
  11. using MediaBrowser.Controller.Library;
  12. using MediaBrowser.Controller.Persistence;
  13. using MediaBrowser.Model.Dto;
  14. using MediaBrowser.Model.Entities;
  15. using Book = MediaBrowser.Controller.Entities.Book;
  16. using AudioBook = MediaBrowser.Controller.Entities.AudioBook;
  17. namespace Emby.Server.Implementations.Library
  18. {
  19. /// <summary>
  20. /// Class UserDataManager.
  21. /// </summary>
  22. public class UserDataManager : IUserDataManager
  23. {
  24. public event EventHandler<UserDataSaveEventArgs> UserDataSaved;
  25. private readonly ConcurrentDictionary<string, UserItemData> _userData =
  26. new ConcurrentDictionary<string, UserItemData>(StringComparer.OrdinalIgnoreCase);
  27. private readonly IServerConfigurationManager _config;
  28. private readonly IUserManager _userManager;
  29. private readonly IUserDataRepository _repository;
  30. public UserDataManager(
  31. IServerConfigurationManager config,
  32. IUserManager userManager,
  33. IUserDataRepository repository)
  34. {
  35. _config = config;
  36. _userManager = userManager;
  37. _repository = repository;
  38. }
  39. public void SaveUserData(Guid userId, BaseItem item, UserItemData userData, UserDataSaveReason reason, CancellationToken cancellationToken)
  40. {
  41. var user = _userManager.GetUserById(userId);
  42. SaveUserData(user, item, userData, reason, cancellationToken);
  43. }
  44. public void SaveUserData(User user, BaseItem item, UserItemData userData, UserDataSaveReason reason, CancellationToken cancellationToken)
  45. {
  46. if (userData == null)
  47. {
  48. throw new ArgumentNullException(nameof(userData));
  49. }
  50. if (item == null)
  51. {
  52. throw new ArgumentNullException(nameof(item));
  53. }
  54. cancellationToken.ThrowIfCancellationRequested();
  55. var keys = item.GetUserDataKeys();
  56. var userId = user.InternalId;
  57. foreach (var key in keys)
  58. {
  59. _repository.SaveUserData(userId, key, userData, cancellationToken);
  60. }
  61. var cacheKey = GetCacheKey(userId, item.Id);
  62. _userData.AddOrUpdate(cacheKey, userData, (k, v) => userData);
  63. UserDataSaved?.Invoke(this, new UserDataSaveEventArgs
  64. {
  65. Keys = keys,
  66. UserData = userData,
  67. SaveReason = reason,
  68. UserId = user.Id,
  69. Item = item
  70. });
  71. }
  72. /// <summary>
  73. /// Save the provided user data for the given user. Batch operation. Does not fire any events or update the cache.
  74. /// </summary>
  75. /// <param name="userId"></param>
  76. /// <param name="userData"></param>
  77. /// <param name="cancellationToken"></param>
  78. /// <returns></returns>
  79. public void SaveAllUserData(Guid userId, UserItemData[] userData, CancellationToken cancellationToken)
  80. {
  81. var user = _userManager.GetUserById(userId);
  82. _repository.SaveAllUserData(user.InternalId, userData, cancellationToken);
  83. }
  84. /// <summary>
  85. /// Retrieve all user data for the given user.
  86. /// </summary>
  87. /// <param name="userId"></param>
  88. /// <returns></returns>
  89. public List<UserItemData> GetAllUserData(Guid userId)
  90. {
  91. var user = _userManager.GetUserById(userId);
  92. return _repository.GetAllUserData(user.InternalId);
  93. }
  94. public UserItemData GetUserData(Guid userId, Guid itemId, List<string> keys)
  95. {
  96. var user = _userManager.GetUserById(userId);
  97. return GetUserData(user, itemId, keys);
  98. }
  99. public UserItemData GetUserData(User user, Guid itemId, List<string> keys)
  100. {
  101. var userId = user.InternalId;
  102. var cacheKey = GetCacheKey(userId, itemId);
  103. return _userData.GetOrAdd(cacheKey, k => GetUserDataInternal(userId, keys));
  104. }
  105. private UserItemData GetUserDataInternal(long internalUserId, List<string> keys)
  106. {
  107. var userData = _repository.GetUserData(internalUserId, keys);
  108. if (userData != null)
  109. {
  110. return userData;
  111. }
  112. if (keys.Count > 0)
  113. {
  114. return new UserItemData
  115. {
  116. Key = keys[0]
  117. };
  118. }
  119. return null;
  120. }
  121. /// <summary>
  122. /// Gets the internal key.
  123. /// </summary>
  124. /// <returns>System.String.</returns>
  125. private static string GetCacheKey(long internalUserId, Guid itemId)
  126. {
  127. return internalUserId.ToString(CultureInfo.InvariantCulture) + "-" + itemId.ToString("N", CultureInfo.InvariantCulture);
  128. }
  129. public UserItemData GetUserData(User user, BaseItem item)
  130. {
  131. return GetUserData(user, item.Id, item.GetUserDataKeys());
  132. }
  133. public UserItemData GetUserData(Guid userId, BaseItem item)
  134. {
  135. return GetUserData(userId, item.Id, item.GetUserDataKeys());
  136. }
  137. public UserItemDataDto GetUserDataDto(BaseItem item, User user)
  138. {
  139. var userData = GetUserData(user, item);
  140. var dto = GetUserItemDataDto(userData);
  141. item.FillUserDataDtoValues(dto, userData, null, user, new DtoOptions());
  142. return dto;
  143. }
  144. public UserItemDataDto GetUserDataDto(BaseItem item, BaseItemDto itemDto, User user, DtoOptions options)
  145. {
  146. var userData = GetUserData(user, item);
  147. var dto = GetUserItemDataDto(userData);
  148. item.FillUserDataDtoValues(dto, userData, itemDto, user, options);
  149. return dto;
  150. }
  151. /// <summary>
  152. /// Converts a UserItemData to a DTOUserItemData.
  153. /// </summary>
  154. /// <param name="data">The data.</param>
  155. /// <returns>DtoUserItemData.</returns>
  156. /// <exception cref="ArgumentNullException"></exception>
  157. private UserItemDataDto GetUserItemDataDto(UserItemData data)
  158. {
  159. if (data == null)
  160. {
  161. throw new ArgumentNullException(nameof(data));
  162. }
  163. return new UserItemDataDto
  164. {
  165. IsFavorite = data.IsFavorite,
  166. Likes = data.Likes,
  167. PlaybackPositionTicks = data.PlaybackPositionTicks,
  168. PlayCount = data.PlayCount,
  169. Rating = data.Rating,
  170. Played = data.Played,
  171. LastPlayedDate = data.LastPlayedDate,
  172. Key = data.Key
  173. };
  174. }
  175. public bool UpdatePlayState(BaseItem item, UserItemData data, long? reportedPositionTicks)
  176. {
  177. var playedToCompletion = false;
  178. var runtimeTicks = item.GetRunTimeTicksForPlayState();
  179. var positionTicks = reportedPositionTicks ?? runtimeTicks;
  180. var hasRuntime = runtimeTicks > 0;
  181. // If a position has been reported, and if we know the duration
  182. if (positionTicks > 0 && hasRuntime && !(item is AudioBook))
  183. {
  184. var pctIn = decimal.Divide(positionTicks, runtimeTicks) * 100;
  185. if (pctIn < _config.Configuration.MinResumePct)
  186. {
  187. // ignore progress during the beginning
  188. positionTicks = 0;
  189. }
  190. else if (pctIn > _config.Configuration.MaxResumePct || positionTicks >= runtimeTicks)
  191. {
  192. // mark as completed close to the end
  193. positionTicks = 0;
  194. data.Played = playedToCompletion = true;
  195. }
  196. else
  197. {
  198. // Enforce MinResumeDuration
  199. var durationSeconds = TimeSpan.FromTicks(runtimeTicks).TotalSeconds;
  200. if (durationSeconds < _config.Configuration.MinResumeDurationSeconds && !(item is Book))
  201. {
  202. positionTicks = 0;
  203. data.Played = playedToCompletion = true;
  204. }
  205. }
  206. }
  207. else if (positionTicks > 0 && hasRuntime && item is AudioBook)
  208. {
  209. var playbackPositionInMinutes = TimeSpan.FromTicks(positionTicks).TotalMinutes;
  210. var remainingTimeInMinutes = TimeSpan.FromTicks(runtimeTicks - positionTicks).TotalMinutes;
  211. if (playbackPositionInMinutes < _config.Configuration.MinAudiobookResume)
  212. {
  213. // ignore progress during the beginning
  214. positionTicks = 0;
  215. }
  216. else if (remainingTimeInMinutes < _config.Configuration.MaxAudiobookResume || positionTicks >= runtimeTicks)
  217. {
  218. // mark as completed close to the end
  219. positionTicks = 0;
  220. data.Played = playedToCompletion = true;
  221. }
  222. }
  223. else if (!hasRuntime)
  224. {
  225. // If we don't know the runtime we'll just have to assume it was fully played
  226. data.Played = playedToCompletion = true;
  227. positionTicks = 0;
  228. }
  229. if (!item.SupportsPlayedStatus)
  230. {
  231. positionTicks = 0;
  232. data.Played = false;
  233. }
  234. if (!item.SupportsPositionTicksResume)
  235. {
  236. positionTicks = 0;
  237. }
  238. data.PlaybackPositionTicks = positionTicks;
  239. return playedToCompletion;
  240. }
  241. }
  242. }