DbInitializer.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Identity;
using Microsoft.EntityFrameworkCore.Internal;
using SchoolEquipmentManager.Logic;
using SchoolEquipmentManager.Models;

namespace SchoolEquipmentManager
{
    public class DbInitializer
    {
        private AppContext _dbContext;
        private UserManager<ApplicationUser> _userManager;
        private RoleManager<IdentityRole> _roleManager;
        private IHostingEnvironment _hostingEnv;
        private IMessageService _messageService;

        public DbInitializer(AppContext dbContext, UserManager<ApplicationUser> userManager, RoleManager<IdentityRole> roleManager, IHostingEnvironment env, IMessageService messageService)
        {
            _dbContext = dbContext;
            _userManager = userManager;
            _roleManager = roleManager;
            _hostingEnv = env;
            _messageService = messageService;
        }

        public void Initialize()
        {
            _dbContext.Database.EnsureCreated();

            CreateRoles().Wait();

            if (_dbContext.Teachers.Any())
                return;

            if (_hostingEnv.IsDevelopment())
                InitDevelopmentDb();
            else
                InitProductionDb();

            _dbContext.SaveChanges();
        }

        private void InitDevelopmentDb()
        {
            #region Locations

            var location1 = new Location()
            {
                Name = "Sala 8",
            };
            _dbContext.Locations.Add(location1);

            var location2 = new Location()
            {
                Name = "Sala 10",
            };
            _dbContext.Locations.Add(location2);

            #endregion

            #region Templates

            var template1 = new ItemTemplate()
            {
                Name = "Komputer",
                Description = "Taki tam komputerek"
            };
            _dbContext.ItemTemplates.Add(template1);

            var template2 = new ItemTemplate()
            {
                Name = "Laptop",
                Description = "Raz działa, a raz nie"
            };
            _dbContext.ItemTemplates.Add(template2);

            #endregion

            #region Teachers

            var teacher1 = new Teacher()
            {
                Name = "Karol",
                Surname = "Wojtyła",
                BarCode = "KAR-WOJ-592479",
            };
            _dbContext.Teachers.Add(teacher1);

            var teacher2 = new Teacher()
            {
                Name = "Jarosław",
                Surname = "Polskezbaw",
                BarCode = "JAR-POL-582307",
            };
            _dbContext.Teachers.Add(teacher2);

            var teacher3 = new Teacher()
            {
                Name = "Szymon",
                Surname = "Jankowski",
                BarCode = "SZY-JAN-021370",
                Messages = new List<Message>()
                {
                    new Message()
                    {
                        Date = DateTime.Now,
                        Title = "Witaj w systemie!",
                        Body = "To jest testowa wiadomość.",
                        Read = false,
                    },
                    new Message()
                    {
                        Date = DateTime.Now,
                        Title = "Druga wiadomość",
                        Body = "To jest już druga testowa wiadomość.",
                        Read = false,
                    }
                }
            };
            _dbContext.Teachers.Add(teacher3);

            {
                var user = new ApplicationUser()
                {
                    UserName = "admin",
                    Email = "szymskipl@gmail.com",
                    Teacher = teacher3,
                };

                var task = _userManager.CreateAsync(user, "admin123");
                task.Wait();

                _userManager.AddToRoleAsync(user, Roles.Administrator).Wait();

                _messageService.SendMessage(teacher3, "SystemInstalled");
            }

            {
                var user = new ApplicationUser()
                {
                    UserName = "karol",
                    Teacher = teacher1,
                };

                var task = _userManager.CreateAsync(user, "123456");
                task.Wait();

                _userManager.AddToRoleAsync(user, Roles.Moderator).Wait();
            }

            #endregion

            var rnd = new Random();
            for (int i = 0; i < 20; i++)
            {
                var events = new List<BorrowEvent>();

                if (rnd.Next(0, 2) == 0)
                {
                    events.Add(new BorrowEvent()
                    {
                        Teacher = rnd.Next(0, 1) == 0 ? teacher1 : teacher2,
                        Date = DateTime.Now - TimeSpan.FromMinutes(45) - TimeSpan.FromDays(rnd.Next(0, 7)),
                        Type = "borrow",
                    });
                }

                var item = new Item()
                {
                    ShortId = rnd.Next(0, 2) == 0 ? null : $"{rnd.Next(100000000, 999999999)}.{rnd.Next(10000, 99999)}",
                    Template = rnd.Next(0, 2) == 0 ? template1 : template2,
                    Notes = "",
                    Location = (rnd.Next(0, 2) == 0 ? location1 : location2),
                    Events = events,
                };
                _dbContext.Items.Add(item);
            }
        }

        private void InitProductionDb()
        {
            var adminTeacher = new Teacher()
            {
                Name = "Główny",
                Surname = "Administrator",
                BarCode = "",
                Messages = new List<Message>()
            };
            _dbContext.Teachers.Add(adminTeacher);

            var adminUser = new ApplicationUser()
            {
                UserName = "admin",
                Email = "",
                Teacher = adminTeacher,
            };

            var task = _userManager.CreateAsync(adminUser, "admin123");
            task.Wait();

            _userManager.AddToRoleAsync(adminUser, Roles.Administrator).Wait();

            _messageService.SendMessage(adminTeacher, "SystemInstalled");
        }

        public async Task CreateRoles()
        {
            if (!await _roleManager.RoleExistsAsync(Roles.Administrator))
                await _roleManager.CreateAsync(new IdentityRole(Roles.Administrator));
            if (!await _roleManager.RoleExistsAsync(Roles.Moderator))
                await _roleManager.CreateAsync(new IdentityRole(Roles.Moderator));
            if (!await _roleManager.RoleExistsAsync(Roles.Scanner))
                await _roleManager.CreateAsync(new IdentityRole(Roles.Scanner));
        }
    }
}