using MediaBrowser.Common.Events; using MediaBrowser.Common.Extensions; using MediaBrowser.Common.Kernel; using MediaBrowser.Controller.Entities; using MediaBrowser.Model.Connectivity; using System; using System.Collections.Concurrent; using System.Collections.Generic; using System.Linq; using System.Threading; using System.Threading.Tasks; using MediaBrowser.Model.Logging; namespace MediaBrowser.Controller.Library { /// /// Class UserManager /// public class UserManager : BaseManager { /// /// The _active connections /// private readonly ConcurrentBag _activeConnections = new ConcurrentBag(); /// /// Gets all connections. /// /// All connections. public IEnumerable AllConnections { get { return _activeConnections.Where(c => Kernel.GetUserById(c.UserId) != null).OrderByDescending(c => c.LastActivityDate); } } /// /// Gets the active connections. /// /// The active connections. public IEnumerable ActiveConnections { get { return AllConnections.Where(c => (DateTime.UtcNow - c.LastActivityDate).TotalMinutes <= 10); } } /// /// The _logger /// private readonly ILogger _logger; /// /// Initializes a new instance of the class. /// /// The kernel. /// The logger. public UserManager(Kernel kernel, ILogger logger) : base(kernel) { _logger = logger; } #region UserUpdated Event /// /// Occurs when [user updated]. /// public event EventHandler> UserUpdated; /// /// Called when [user updated]. /// /// The user. internal void OnUserUpdated(User user) { EventHelper.QueueEventIfNotNull(UserUpdated, this, new GenericEventArgs { Argument = user }, _logger); // Notify connected ui's Kernel.TcpManager.SendWebSocketMessage("UserUpdated", new DtoBuilder(_logger).GetDtoUser(user)); } #endregion #region UserDeleted Event /// /// Occurs when [user deleted]. /// public event EventHandler> UserDeleted; /// /// Called when [user deleted]. /// /// The user. internal void OnUserDeleted(User user) { EventHelper.QueueEventIfNotNull(UserDeleted, this, new GenericEventArgs { Argument = user }, _logger); // Notify connected ui's Kernel.TcpManager.SendWebSocketMessage("UserDeleted", user.Id.ToString()); } #endregion /// /// Authenticates a User and returns a result indicating whether or not it succeeded /// /// The user. /// The password. /// Task{System.Boolean}. /// user public async Task AuthenticateUser(User user, string password) { if (user == null) { throw new ArgumentNullException("user"); } password = password ?? string.Empty; var existingPassword = string.IsNullOrEmpty(user.Password) ? string.Empty.GetMD5().ToString() : user.Password; var success = password.GetMD5().ToString().Equals(existingPassword); // Update LastActivityDate and LastLoginDate, then save if (success) { user.LastActivityDate = user.LastLoginDate = DateTime.UtcNow; await UpdateUser(user).ConfigureAwait(false); } _logger.Info("Authentication request for {0} {1}.", user.Name, (success ? "has succeeded" : "has been denied")); return success; } /// /// Logs the user activity. /// /// The user. /// Type of the client. /// Name of the device. /// Task. /// user public Task LogUserActivity(User user, ClientType clientType, string deviceName) { if (user == null) { throw new ArgumentNullException("user"); } var activityDate = DateTime.UtcNow; user.LastActivityDate = activityDate; LogConnection(user.Id, clientType, deviceName, activityDate); // Save this directly. No need to fire off all the events for this. return Kernel.UserRepository.SaveUser(user, CancellationToken.None); } /// /// Updates the now playing item id. /// /// The user. /// Type of the client. /// Name of the device. /// The item. /// The current position ticks. public void UpdateNowPlayingItemId(User user, ClientType clientType, string deviceName, BaseItem item, long? currentPositionTicks = null) { var conn = GetConnection(user.Id, clientType, deviceName); conn.NowPlayingPositionTicks = currentPositionTicks; conn.NowPlayingItem = DtoBuilder.GetBaseItemInfo(item); } /// /// Removes the now playing item id. /// /// The user. /// Type of the client. /// Name of the device. /// The item. public void RemoveNowPlayingItemId(User user, ClientType clientType, string deviceName, BaseItem item) { var conn = GetConnection(user.Id, clientType, deviceName); if (conn.NowPlayingItem != null && conn.NowPlayingItem.Id.Equals(item.Id.ToString())) { conn.NowPlayingItem = null; conn.NowPlayingPositionTicks = null; } } /// /// Logs the connection. /// /// The user id. /// Type of the client. /// Name of the device. /// The last activity date. private void LogConnection(Guid userId, ClientType clientType, string deviceName, DateTime lastActivityDate) { GetConnection(userId, clientType, deviceName).LastActivityDate = lastActivityDate; } /// /// Gets the connection. /// /// The user id. /// Type of the client. /// Name of the device. /// ClientConnectionInfo. private ClientConnectionInfo GetConnection(Guid userId, ClientType clientType, string deviceName) { var conn = _activeConnections.FirstOrDefault(c => c.UserId == userId && c.ClientType == clientType && string.Equals(deviceName, c.DeviceName, StringComparison.OrdinalIgnoreCase)); if (conn == null) { conn = new ClientConnectionInfo { UserId = userId, ClientType = clientType, DeviceName = deviceName }; _activeConnections.Add(conn); } return conn; } /// /// Loads the users from the repository /// /// IEnumerable{User}. internal IEnumerable LoadUsers() { var users = Kernel.UserRepository.RetrieveAllUsers().ToList(); // There always has to be at least one user. if (users.Count == 0) { var name = Environment.UserName; var user = InstantiateNewUser(name); var task = Kernel.UserRepository.SaveUser(user, CancellationToken.None); // Hate having to block threads Task.WaitAll(task); users.Add(user); } return users; } /// /// Refreshes metadata for each user /// /// The cancellation token. /// if set to true [force]. /// Task. public Task RefreshUsersMetadata(CancellationToken cancellationToken, bool force = false) { var tasks = Kernel.Users.Select(user => user.RefreshMetadata(cancellationToken, forceRefresh: force)).ToList(); return Task.WhenAll(tasks); } /// /// Renames the user. /// /// The user. /// The new name. /// Task. /// user /// public async Task RenameUser(User user, string newName) { if (user == null) { throw new ArgumentNullException("user"); } if (string.IsNullOrEmpty(newName)) { throw new ArgumentNullException("newName"); } if (Kernel.Users.Any(u => u.Id != user.Id && u.Name.Equals(newName, StringComparison.OrdinalIgnoreCase))) { throw new ArgumentException(string.Format("A user with the name '{0}' already exists.", newName)); } if (user.Name.Equals(newName, StringComparison.Ordinal)) { throw new ArgumentException("The new and old names must be different."); } await user.Rename(newName); OnUserUpdated(user); } /// /// Updates the user. /// /// The user. /// user /// public async Task UpdateUser(User user) { if (user == null) { throw new ArgumentNullException("user"); } if (user.Id == Guid.Empty || !Kernel.Users.Any(u => u.Id.Equals(user.Id))) { throw new ArgumentException(string.Format("User with name '{0}' and Id {1} does not exist.", user.Name, user.Id)); } user.DateModified = DateTime.UtcNow; await Kernel.UserRepository.SaveUser(user, CancellationToken.None).ConfigureAwait(false); OnUserUpdated(user); } /// /// Creates the user. /// /// The name. /// User. /// name /// public async Task CreateUser(string name) { if (string.IsNullOrEmpty(name)) { throw new ArgumentNullException("name"); } if (Kernel.Users.Any(u => u.Name.Equals(name, StringComparison.OrdinalIgnoreCase))) { throw new ArgumentException(string.Format("A user with the name '{0}' already exists.", name)); } var user = InstantiateNewUser(name); var list = Kernel.Users.ToList(); list.Add(user); Kernel.Users = list; await Kernel.UserRepository.SaveUser(user, CancellationToken.None).ConfigureAwait(false); return user; } /// /// Deletes the user. /// /// The user. /// Task. /// user /// public async Task DeleteUser(User user) { if (user == null) { throw new ArgumentNullException("user"); } if (Kernel.Users.FirstOrDefault(u => u.Id == user.Id) == null) { throw new ArgumentException(string.Format("The user cannot be deleted because there is no user with the Name {0} and Id {1}.", user.Name, user.Id)); } if (Kernel.Users.Count() == 1) { throw new ArgumentException(string.Format("The user '{0}' be deleted because there must be at least one user in the system.", user.Name)); } await Kernel.UserRepository.DeleteUser(user, CancellationToken.None).ConfigureAwait(false); OnUserDeleted(user); // Force this to be lazy loaded again Kernel.Users = null; } /// /// Instantiates the new user. /// /// The name. /// User. private User InstantiateNewUser(string name) { return new User { Name = name, Id = ("MBUser" + name).GetMD5(), DateCreated = DateTime.UtcNow, DateModified = DateTime.UtcNow }; } } }