UnrealEngineUWP/Engine/Source/Programs/Horde/Horde.Build/Users/IUserCollection.cs

// Copyright Epic Games, Inc. All Rights Reserved.

using System.Collections.Generic;
using System.Security.Claims;
using System.Threading.Tasks;
using Horde.Build.Models;
using Horde.Build.Utilities;
using MongoDB.Bson;

namespace Horde.Build.Collections
{
	using JobId = ObjectId<IJob>;
	using UserId = ObjectId<IUser>;

	/// <summary>
	/// Manages user documents
	/// </summary>
	public interface IUserCollection
	{
		/// <summary>
		/// Gets a user by unique id
		/// </summary>
		/// <param name="id">Id of the user</param>
		/// <returns>The user information</returns>
		Task<IUser?> GetUserAsync(UserId id);

		/// <summary>
		/// Gets a cached user by unique id
		/// </summary>
		/// <param name="id">Id of the user</param>
		/// <returns>The user information</returns>
		ValueTask<IUser?> GetCachedUserAsync(UserId? id);

		/// <summary>
		/// Gets a user by unique id
		/// </summary>
		/// <param name="ids">Ids of the users</param>
		/// <param name="nameRegex">Name regex to match for the users</param>
		/// <param name="index">Maximum number of results</param>
		/// <param name="count">Number of results to return</param>
		/// <returns>The user information</returns>
		Task<List<IUser>> FindUsersAsync(IEnumerable<UserId>? ids = null, string? nameRegex = null, int? index = null, int? count = null);

		/// <summary>
		/// Gets a user by login
		/// </summary>
		/// <param name="login">Login for the user</param>
		/// <returns>The user information</returns>
		Task<IUser?> FindUserByLoginAsync(string login);

		/// <summary>
		/// Find or add a user with the given claims. Claims will be updated if the user exists.
		/// </summary>
		/// <param name="login">Login id of the user</param>
		/// <param name="name">Full name of the user</param>
		/// <param name="email">Email address of the user</param>
		/// <returns>The user document</returns>
		Task<IUser> FindOrAddUserByLoginAsync(string login, string? name = null, string? email = null);

		/// <summary>
		/// Gets the claims for a user
		/// </summary>
		/// <param name="userId"></param>
		/// <returns></returns>
		Task<IUserClaims> GetClaimsAsync(UserId userId);

		/// <summary>
		/// Update the claims for a user
		/// </summary>
		/// <param name="userId"></param>
		/// <param name="claims"></param>
		/// <returns></returns>
		Task UpdateClaimsAsync(UserId userId, IEnumerable<IUserClaim> claims);

		/// <summary>
		/// Get settings for a user
		/// </summary>
		/// <param name="userId"></param>
		/// <returns></returns>
		Task<IUserSettings> GetSettingsAsync(UserId userId);

		/// <summary>
		/// Update a user
		/// </summary>
		/// <param name="userId">The user to update</param>
		/// <param name="enableExperimentalFeatures"></param>
		/// <param name="dashboardSettings">Opaque settings object for the dashboard</param>
		/// <param name="addPinnedJobIds"></param>
		/// <param name="removePinnedJobIds"></param>
		/// <returns>Updated user object</returns>
		Task UpdateSettingsAsync(UserId userId, bool? enableExperimentalFeatures = null, BsonValue? dashboardSettings = null, IEnumerable<JobId>? addPinnedJobIds = null, IEnumerable<JobId>? removePinnedJobIds = null);
	}

	/// <summary>
	/// Extension methods for the user collect
	/// </summary>
	static class UserCollectionExtensions
	{
		/// <summary>
		/// Gets a particular user info from the collection
		/// </summary>
		/// <param name="userCollection"></param>
		/// <param name="principal"></param>
		/// <returns></returns>
		public static Task<IUser?> GetUserAsync(this IUserCollection userCollection, ClaimsPrincipal principal)
		{
			UserId? userId = principal.GetUserId();
			if (userId != null)
			{
				return userCollection.GetUserAsync(userId.Value);
			}
			else
			{
				return Task.FromResult<IUser?>(null);
			}
		}
	}
}