aswi20220x00-gitlab/Backend/Core/Seeding/Seeder.cs @ c99a6c25

using Core.Contexts;
using Core.Entities;
using Core.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))
            {
                DocumentContent content = new DocumentContent()
                {
                    Content = "Dummy content"
                };
                context.DocumentContents.Add(content);

                User user = new User()
                {
                    Username = "testuser",
                    Name = "Test",
                    Surname = "User",
                    Password = BCrypt.Net.BCrypt.HashPassword("password"),
                    Role = ERole.ANNOTATOR
                };
                context.Users.Add(user);

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

                Document document = new Document()
                {
                    Content = content,
                    Name = "Test document",
                    DateAdded = DateTime.Now,
                    Length = 10,
                    RequiredAnnotations = 3,
                    UserAdded = user
                };
                context.Documents.Add(document);
                context.SaveChanges();

                Annotation annotation = new Annotation()
                {
                    DateAssigned = DateTime.Now,
                    DateLastChanged = DateTime.Now,
                    Document = document,
                    Note = "",
                    State = EState.NEW,
                    User = user,
                    UserAssigned = user
                };
                context.Annotations.Add(annotation);

                Class clss = new Class()
                {
                    Color = "rgb(...)",
                    Description = "Dummy description",
                    Name = "Dummy class name"
                };
                context.Classes.Add(clss);

                annotation.Classes.Add(clss);

                TagCategory category = new TagCategory()
                {
                    Name = "Dummy category",
                    Color = "rgb(...)",
                    Description = "Dummy category description"
                };
                context.TagCategories.Add(category);

                Tag tag = new Tag()
                {
                    Name = "Dummy tag",
                    Category = category,
                    Description = "Dummy tag",
                    Color = "rgb(...)"
                };
                context.Tags.Add(tag);
                SubTag subTag = new SubTag()
                {
                    Description = "Dummy subtag",
                    Name = "Dummy subtag",
                    Tag = tag
                };
                context.SubTags.Add(subTag);

                AnnotationTag annotationTag = new AnnotationTag()
                {
                    Annotation = annotation,
                    Instance = 0,
                    Length = 1,
                    Position = 0,
                    Tag = tag,
                    Note = "Dummy note"
                };
                context.AnnotationTags.Add(annotationTag);

                context.SaveChanges();
            }
        }

        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();
            }
        }
    }
}