ProjectLighthouse/ProjectLighthouse/Database.cs

using System.Threading.Tasks;
using Microsoft.AspNetCore.Http;
using Microsoft.EntityFrameworkCore;
using ProjectLighthouse.Helpers;
using ProjectLighthouse.Types;

namespace ProjectLighthouse {
    public class Database : DbContext {
        public DbSet<User> Users { get; set; }
        public DbSet<Location> Locations { get; set; }
        public DbSet<Slot> Slots { get; set; }
        public DbSet<QueuedLevel> QueuedLevels { get; set; }
        public DbSet<HeartedLevel> HeartedLevels { get; set; }
        public DbSet<Comment> Comments { get; set; }
        public DbSet<Token> Tokens { get; set; }

        public DbSet<LastMatch> LastMatches { get; set; }

        protected override void OnConfiguring(DbContextOptionsBuilder options) => options.UseMySql(
            ServerSettings.DbConnectionString,
            MySqlServerVersion.LatestSupportedServerVersion
        );

        public async Task<User> CreateUser(string username) {
            Location l = new(); // store to get id after submitting
            this.Locations.Add(l); // add to table
            await this.SaveChangesAsync(); // saving to the database returns the id and sets it on this entity

            User user = new() {
                Username = username,
                LocationId = l.Id,
                Biography = username + " hasn't introduced themselves yet.",
            };
            this.Users.Add(user);

            await this.SaveChangesAsync();

            return user;

        }
        
        #nullable enable
        public async Task<Token?> AuthenticateUser(LoginData loginData) {
            // TODO: don't use psn name to authenticate
            User user = await this.Users.FirstOrDefaultAsync(u => u.Username == loginData.Username) 
                        ?? await this.CreateUser(loginData.Username);

            Token token = new() {
                UserToken = HashHelper.GenerateAuthToken(),
                UserId = user.UserId,
            };

            this.Tokens.Add(token);
            await this.SaveChangesAsync();

            return token;
        }

        public async Task<User?> UserFromAuthToken(string authToken) {
            Token? token = await Tokens.FirstOrDefaultAsync(t => t.UserToken == authToken);
            if(token == null) return null;
            return await Users
                .Include(u => u.Location)
                .FirstOrDefaultAsync(u => u.UserId == token.UserId);
        }

        public async Task<User?> UserFromRequest(HttpRequest request) {
            if(!request.Cookies.TryGetValue("MM_AUTH", out string? mmAuth) || mmAuth == null) {
                return null;
            }
            
            return await UserFromAuthToken(mmAuth);
        }
        #nullable disable
    }
}