aswi20220x00-gitlab/Backend/Core/Services/TagService/TagServiceEF.cs @ a35cb648

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

namespace Core.Services.TagService
{
    public class TagServiceEF : ITagService
    {
        private readonly DatabaseContext databaseContext;
        private readonly ILogger logger;
        private readonly IMapper mapper;

        public TagServiceEF(DatabaseContext databaseContext, ILogger logger, IMapper mapper)
        {
            this.databaseContext = databaseContext;
            this.logger = logger;
            this.mapper = mapper;
        }

        public TagTreeResponse GetTagTree(ERole userRole)
        {
            // Prefetch from DB
            var tagCategories = databaseContext.TagCategories
                .Select(tc => tc)
                .OrderBy(tc => tc.Name)
                .ToList();

            var tags = databaseContext.Tags
                .Select(tc => tc)
                .ToList();

            var subTags = databaseContext.SubTags
                .Select(tc => tc)
                .ToList();

            var tagCategoriesDTOs = new List<TagCategoryInfo>();
            foreach (var tagCategory in tagCategories)
            {
                if (tagCategory.DisabledForAnnotators && userRole == ERole.ANNOTATOR)
                {
                    continue;
                }

                var tagCategoryDTO = mapper.Map<TagCategoryInfo>(tagCategory);
                var categoryTags = tags
                    .Where(t => t.Category == tagCategory)
                    .OrderBy(t => t.Name);

                var tagDTOs = new List<TagInfo>();

                foreach (var tag in categoryTags)
                {
                    var tagDTO = mapper.Map<TagInfo>(tag);
                    var tagSubTags = subTags
                        .Where(st => st.Tag == tag)
                        .OrderBy(st => st.Name)
                        .ToList();

                    var subTagDTOs = new List<SubTagInfo>();

                    if (tagSubTags.Count() != 0)
                    {
                        foreach (var subTag in tagSubTags)
                        {
                            subTagDTOs.Add(mapper.Map<SubTagInfo>(subTag));
                        }
                    }
                    tagDTO.SubTags = subTagDTOs;
                    tagDTOs.Add(tagDTO);
                }

                tagCategoryDTO.Tags = tagDTOs;
                tagCategoriesDTOs.Add(tagCategoryDTO);
            }

            return new TagTreeResponse()
            {
                TagCategories = tagCategoriesDTOs
            };
        }

        public void CreateCategory(CreateCategoryRequest request)
        {
            if (request.Color == "" || request.Name == "")
            {
                throw new InvalidOperationException("Category name or color empty");
            }

            CategoryNameUnusedElseThrow(request.Name);

            databaseContext.TagCategories.Add(new TagCategory()
            {
                Name = request.Name,
                Color = request.Color,
                Description = request.Description,
                DisabledForAnnotators = request.DisabledForAnnotators
            });

            databaseContext.SaveChanges();
        }

        public void DeleteCategory(Guid categoryId)
        {
            CategoryExistsElseThrow(categoryId);

            var category = databaseContext.TagCategories.First(tc => tc.Id == categoryId);
            databaseContext.TagCategories.Remove(category);
            databaseContext.SaveChanges();
        }

        public void UpdateCategory(ModifyCategoryRequest request, Guid categoryId)
        {
            CategoryExistsElseThrow(categoryId);
            var category = databaseContext.TagCategories.First(tc => tc.Id == categoryId);

            if (request.Name != null && request.Name != category.Name)
            {
                CategoryNameUnusedElseThrow(request.Name);
                category.Name = request.Name;
            }

            if (request.Description != null)
            {
                category.Description = request.Description;
            }

            if (request.Color != null)
            {
                if (request.Color == "")
                {
                    throw new InvalidOperationException("Empty color");
                }
                category.Color = request.Color;
            }

            if (request.DisabledForAnnotators != null)
            {
                category.DisabledForAnnotators = (bool)request.DisabledForAnnotators;
            }

            databaseContext.SaveChanges();
        }

        public void CreateTag(CreateTagRequest request)
        {
            CategoryExistsElseThrow(request.CategoryId);

            TagNameUnusedElseThrow(request.Name);

            if (request.Color == "")
            {
                throw new InvalidOperationException("Color empty");
            }

            var category = databaseContext.TagCategories.First(tc => tc.Id == request.CategoryId);

            databaseContext.Tags.Add(new Tag()
            {
                Name = request.Name,
                Color = request.Color,
                Description = request.Description,
                Category = category,
                SentimentEnabled = request.SentimentEnabled
            });

            databaseContext.SaveChanges();
        }

        public void DeleteTag(Guid tagId)
        {
            TagExistsElseThrow(tagId);
            var tag = databaseContext.Tags.First(t => t.Id == tagId);
            databaseContext.Tags.Remove(tag);
            databaseContext.SaveChanges();
        }

        public void UpdateTag(ModifyTagRequest request, Guid tagId)
        {
            TagExistsElseThrow(tagId);
            var tag = databaseContext.Tags.First(t => t.Id == tagId);

            if (request.Name != null && request.Name != tag.Name)
            {
                TagNameUnusedElseThrow(request.Name);
                tag.Name = request.Name;
            }

            if (request.Color != null)
            {
                if (request.Color == "")
                {
                    throw new InvalidOperationException("Color empty");
                }
                tag.Color = request.Color;
            }

            if (request.Description != null)
            {
                tag.Description = request.Description;
            }

            if (request.SentimentEnabled != null)
            {
                tag.SentimentEnabled = (bool)request.SentimentEnabled;
            }

            databaseContext.SaveChanges();
        }

        public void CreateSubTag(CreateSubTagRequest request)
        {
            if (request.Name == "")
            {
                throw new InvalidOperationException("Subtag name empty");
            }

            SubTagNameUnusedElseThrow(request.Name);

            TagExistsElseThrow(request.TagId);
            var tag = databaseContext.Tags.First(t => t.Id == request.TagId);

            databaseContext.SubTags.Add(new Entities.SubTag()
            {
                Name = request.Name,
                Description = request.Description,
                Tag = tag,
                SentimentEnabled = request.SentimentEnabled
            });

            databaseContext.SaveChanges();
        }

        public void DeleteSubTag(Guid subtagId)
        {
            SubTagExistsElseThrow(subtagId);
            var subtag = databaseContext.SubTags.First(st => st.Id == subtagId);
            databaseContext.SubTags.Remove(subtag);
            databaseContext.SaveChanges();
        }

        public void UpdateSubTag(ModifySubTagRequest request, Guid subtagId)
        {
            SubTagExistsElseThrow(subtagId);
            var subtag = databaseContext.SubTags.First(st => st.Id == subtagId);

            if (request.Name != null && request.Name != subtag.Name)
            {
                SubTagNameUnusedElseThrow(request.Name);
                subtag.Name = request.Name;
            }

            if (request.Description != null)
            {
                subtag.Description = request.Description;
            }

            if (request.SentimentEnabled != null)
            {
                subtag.SentimentEnabled = (bool)request.SentimentEnabled;
            }

            databaseContext.SaveChanges();
        }

        public void AddNoteToTagInstance(Guid annotationId, Guid occurrenceId, User user, AddNoteToTagOccurenceRequest request, bool isFinal)
        {
            // We need to differentiate whether or not the annotation is final
            Annotation annotation = null;
            try
            {
                if (!isFinal)
                {
                    annotation = databaseContext.Annotations
                        .Where(a => !(a is FinalAnnotation))
                        .Include(a => a.User).First(a => a.Id == annotationId);
                }
                else
                {
                    annotation = databaseContext.FinalAnnotations.Include(a => a.User).First(a => a.Id == annotationId);
                }

            }
            catch (Exception)
            {
                throw new InvalidOperationException("Annotation not found");
            }

            if (user.Role < ERole.ADMINISTRATOR && annotation.User.Id != user.Id)
            {
                throw new UnauthorizedAccessException("User does not have access to this annotation");
            }

            AnnotationTagGeneric occurence = null;
            try
            {
                if (!isFinal)
                {
                    occurence = databaseContext.AnnotationTags.Include(at => at.Annotation)
                    .First(at => at.Annotation.Id == annotationId && at.Id == occurrenceId);
                }
                else
                {
                    occurence = databaseContext.FinalAnnotationTags.Include(at => at.Annotation)
                    .First(at => at.Annotation.Id == annotationId && at.Id == occurrenceId);
                }
            }
            catch (Exception)
            {
                throw new InvalidOperationException("The annotation does not have a tag with specified id");
            }

            occurence.Note = request.Note;
            databaseContext.SaveChanges();
        }

        private void CategoryExistsElseThrow(Guid categoryId)
        {
            if (!databaseContext.TagCategories.Any(tc => tc.Id == categoryId))
            {
                throw new InvalidOperationException("Category does not exist");
            }
        }

        private void CategoryNameUnusedElseThrow(string name)
        {
            if (databaseContext.TagCategories.Any(tc => tc.Name == name))
            {
                throw new InvalidOperationException("Category name already used");
            }
        }

        private void TagExistsElseThrow(Guid tagId)
        {
            if (!databaseContext.Tags.Any(t => t.Id == tagId))
            {
                throw new InvalidOperationException("Tag does not exist");
            }
        }

        private void TagNameUnusedElseThrow(string name)
        {
            if (databaseContext.Tags.Any(t => t.Name == name))
            {
                throw new InvalidOperationException("Tag name already used");
            }
        }

        private void SubTagExistsElseThrow(Guid subtagId)
        {
            if (!databaseContext.SubTags.Any(st => st.Id == subtagId))
            {
                throw new InvalidOperationException("Subtag does not exist");
            }
        }

        private void SubTagNameUnusedElseThrow(string name)
        {
            if (databaseContext.SubTags.Any(st => st.Name == name))
            {
                throw new InvalidOperationException("Subtag name already used");
            }
        }
    }
}