DatabaseCollectionExtensions.cs

Go to the documentation of this file.

using System;
using System.Linq;
using System.Linq.Expressions;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.EntityFrameworkCore;
 
using Tgstation.Server.Host.Database;
using Tgstation.Server.Host.Models;
 
namespace Tgstation.Server.Host.Extensions
{
    static class DatabaseCollectionExtensions
    {
        public static Task<TResult> GetTgsUser<TResult>(
            this IDatabaseCollection<User> databaseCollection,
            Expression<Func<User, TResult>> selector,
            CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(databaseCollection);
            ArgumentNullException.ThrowIfNull(selector);
 
            return databaseCollection
                .AsQueryable()
                .Where(x => x.CanonicalName == User.CanonicalizeName(User.TgsSystemUserName))
                .Select(selector)
                .FirstAsync(cancellationToken);
        }
    }
}