aswi20220x00-gitlab/Backend/Core/Seeding/Seeder.cs @ 8b2a2c15

using Core.Contexts;
using Core.Entities;
using Models.Enums;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace Core.Seeding
{
    public class Seeder
    {
        public static void SeedDevelopment(DatabaseContext context)
        {
            if (!context.Users.Any(u => u.Role == ERole.ADMINISTRATOR))
            {
                var users = AddUsers(context);
                var classes = AddClasses(context);
                DummyTags.AddDummyTags(context);
                context.SaveChanges();

                AddTagInstance(context);
                context.SaveChanges();
            }
           
        }

        private static void AddTagInstance(DatabaseContext context)
        {
            var tag = context.Tags.Where(t => context.SubTags.Any(st => st.Tag == t)).First();
            var subtag = context.SubTags.Where(st => st.Tag == tag).First();
            var annotation = context.Annotations.First();
            AnnotationTag at = new AnnotationTag()
            {
                Annotation = annotation,
                Instance = 1,
                Length = 10,
                Position = 0,
                SubTag = subtag,
                Note = "asdasdasd",
                Tag = tag
            };
            context.AnnotationTags.Add(at);
        }

        public static void SeedProduction(DatabaseContext context)
        {
            if (!context.Users.Any(u => u.Role == ERole.ADMINISTRATOR))
            {
                User admin = new User()
                {
                    Name = "Admin",
                    Surname = "Admin",
                    Role = ERole.ADMINISTRATOR,
                    Username = "admin",
                    Password = BCrypt.Net.BCrypt.HashPassword("123a456b789c")
                };
                context.Users.Add(admin);
                context.SaveChanges();
            }
        }

        private static List<Class> AddClasses(DatabaseContext context)
        {
            List<Class> classes = new List<Class>();
            Class class1 = new Class()
            {
                Name = "Class1",
                Description = "Class1 description",
                Color = "Class 1 color"
            };
            context.Classes.Add(class1);
            classes.Add(class1);
            return classes;
        }

        private static List<User> AddUsers(DatabaseContext context)
        {
            List<User> users = new List<User>();
            User annotator1 = new User()
            {
                Username = "annotator1",
                Name = "Anno",
                Surname = "Tator1",
                Password = BCrypt.Net.BCrypt.HashPassword("password"),
                Role = ERole.ANNOTATOR
            };
            context.Users.Add(annotator1);
            users.Add(annotator1);

            User annotator2 = new User()
            {
                Username = "annotator2",
                Name = "Anno",
                Surname = "Tator2",
                Password = BCrypt.Net.BCrypt.HashPassword("password"),
                Role = ERole.ANNOTATOR
            };
            context.Users.Add(annotator2);
            users.Add(annotator2);

            User annotator3 = new User()
            {
                Username = "annotator3",
                Name = "Anno",
                Surname = "Tator3",
                Password = BCrypt.Net.BCrypt.HashPassword("password"),
                Role = ERole.ANNOTATOR
            };
            context.Users.Add(annotator3);
            users.Add(annotator3);

            User annotator4 = new User()
            {
                Username = "annotator4",
                Name = "Anno",
                Surname = "Tator4",
                Password = BCrypt.Net.BCrypt.HashPassword("password"),
                Role = ERole.ANNOTATOR
            };
            context.Users.Add(annotator4);
            users.Add(annotator4);

            User annotator5 = new User()
            {
                Username = "annotator5",
                Name = "Anno",
                Surname = "Tator5",
                Password = BCrypt.Net.BCrypt.HashPassword("password"),
                Role = ERole.ANNOTATOR
            };
            context.Users.Add(annotator5);
            users.Add(annotator5);

            User admin = new User()
            {
                Username = "admin",
                Name = "Admin",
                Surname = "User",
                Password = BCrypt.Net.BCrypt.HashPassword("admin"),
                Role = ERole.ADMINISTRATOR
            };
            context.Users.Add(admin);
            users.Add(admin);

            return users;
        }
    }
}