marechai/Marechai.Database/Seeders/Users.cs

/******************************************************************************
// MARECHAI: Master repository of computing history artifacts information
// ----------------------------------------------------------------------------
//
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// --[ License ] --------------------------------------------------------------
//
//     This program is free software: you can redistribute it and/or modify
//     it under the terms of the GNU General Public License as
//     published by the Free Software Foundation, either version 3 of the
//     License, or (at your option) any later version.
//
//     This program is distributed in the hope that it will be useful,
//     but WITHOUT ANY WARRANTY; without even the implied warranty of
//     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//     GNU General Public License for more details.
//
//     You should have received a copy of the GNU General Public License
//     along with this program.  If not, see <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright © 2003-2026 Natalia Portillo
*******************************************************************************/

using System;
using System.Collections.Generic;
using Marechai.Database.Models;
using Microsoft.AspNetCore.Identity;

namespace Marechai.Database.Seeders;

public static class Users
{
    public static void Seed(UserManager<ApplicationUser> userManager, RoleManager<ApplicationRole> roleManager)
    {
        ApplicationRole uberAdminRole = roleManager.FindByNameAsync("UberAdmin").Result;

        if(uberAdminRole is null)
        {
            Console.WriteLine("Cannot find UberAdmin role, is database properly seeded?");

            return;
        }

        IList<ApplicationUser> uberAdmins = userManager.GetUsersInRoleAsync("UberAdmin").Result;

        if(uberAdmins.Count > 1)
        {
            Console.WriteLine("Too many uberadmins, only one can exist");

            foreach(ApplicationUser user in uberAdmins)
            {
                Console.WriteLine("Removing uberadmin permissions from user {0}", user.UserName);

                userManager.RemoveFromRoleAsync(user, "UberAdmin");
            }
        }

        if(uberAdmins.Count == 1) return;

        ApplicationUser uberAdmin = userManager.FindByEmailAsync("claunia@claunia.com").Result ??
                                    userManager.FindByNameAsync("claunia").Result;

        if(uberAdmin is null)
        {
            uberAdmin = new ApplicationUser
            {
                UserName       = "claunia",
                Email          = "claunia@claunia.com",
                EmailConfirmed = true
            };

            var newPass = new byte[8];
            new Random().NextBytes(newPass);
            string newPassString = Convert.ToBase64String(newPass);

            IdentityResult result = userManager.CreateAsync(uberAdmin, newPassString).Result;

            if(result.Succeeded)
            {
                userManager.AddToRoleAsync(uberAdmin, "UberAdmin").Wait();
                Console.WriteLine("Created new claunia uberadmin with password {0}.", newPassString);
            }
            else
                Console.WriteLine("Could not create new uberadmin.");

            return;
        }

        Console.WriteLine("Giving uberadmin permissions to user claunia");
        userManager.AddToRoleAsync(uberAdmin, "UberAdmin").Wait();
    }
}