DatabaseSeeder.cs

Go to the documentation of this file.

using Microsoft.EntityFrameworkCore;
using System;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using Tgstation.Server.Api.Rights;
using Tgstation.Server.Host.Models;
using Tgstation.Server.Host.Security;

namespace Tgstation.Server.Host.Database
{
    sealed class DatabaseSeeder : IDatabaseSeeder
    {
        readonly ICryptographySuite cryptographySuite;

        public DatabaseSeeder(ICryptographySuite cryptographySuite)
        {
            this.cryptographySuite = cryptographySuite ?? throw new ArgumentNullException(nameof(cryptographySuite));
        }

        void SeedAdminUser(IDatabaseContext databaseContext)
        {
            var admin = new User
            {
                AdministrationRights = ~AdministrationRights.None,
                CreatedAt = DateTimeOffset.Now,
                InstanceManagerRights = ~InstanceManagerRights.None,
                Name = Api.Models.User.AdminName,
                CanonicalName = Api.Models.User.AdminName.ToUpperInvariant(),
                Enabled = true,
            };
            cryptographySuite.SetUserPassword(admin, Api.Models.User.DefaultAdminPassword, true);
            databaseContext.Users.Add(admin);
        }

        public async Task SeedDatabase(IDatabaseContext databaseContext, CancellationToken cancellationToken)
        {
            SeedAdminUser(databaseContext);
            await databaseContext.Save(cancellationToken).ConfigureAwait(false);
        }

        public async Task ResetAdminPassword(IDatabaseContext databaseContext, CancellationToken cancellationToken)
        {
            var admin = await databaseContext.Users.Where(x => x.CanonicalName == Api.Models.User.AdminName.ToUpperInvariant()).FirstOrDefaultAsync(cancellationToken).ConfigureAwait(false);
            if (admin == default)
                SeedAdminUser(databaseContext);
            else
            {
                admin.Enabled = true;
                cryptographySuite.SetUserPassword(admin, Api.Models.User.DefaultAdminPassword, false);
            }

            await databaseContext.Save(cancellationToken).ConfigureAwait(false);
        }
    }
}