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;
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); }
}
///
/// Initializes a new instance of the class.
///
/// The kernel.
public UserManager(Kernel kernel)
: base(kernel)
{
}
#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 });
// Notify connected ui's
Kernel.TcpManager.SendWebSocketMessage("UserUpdated", DtoBuilder.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 });
// 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
};
}
}
}