ProjectLighthouse/ProjectLighthouse/Database.cs

#nullable enable
using System;
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<Comment> Comments { get; set; }
        public DbSet<Token> Tokens { 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 = "No biography provided",
                Pins = "",
                PlanetHash = "",
            };
            this.Users.Add(user);

            await this.SaveChangesAsync();

            return user;

        }
        
        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<bool> IsUserAuthenticated(string authToken) => await this.UserFromAuthToken(authToken) != null;

        public async Task<User?> UserFromAuthToken(string authToken) {
            Token? token = await Tokens.FirstOrDefaultAsync(t => t.UserToken == authToken);
            if(token == null) return null;
            return await Users.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);
        }
    }
}