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

using AutoMapper;
using Core.Contexts;
using Models.Tags;
using Serilog;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

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()
        {
            // Prefetch from DB
            var tagCategories = databaseContext.TagCategories.Select(tc => tc).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)
            {
                var tagCategoryDTO = mapper.Map<TagCategoryInfo>(tagCategory);
                var categoryTags = tags.Where(t => t.Category == tagCategory);
                var tagDTOs = new List<TagInfo>();

                foreach (var tag in categoryTags)
                {
                    var tagDTO = mapper.Map<TagInfo>(tag);
                    var tagSubTags = subTags.Where(st => st.Tag == tag).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");
            }

            CategoryNameUnusedOrThrow(request.Name);

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

            databaseContext.SaveChanges();
        }

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

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

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

            if (request.Name != null)
            {
                CategoryNameUnusedOrThrow(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;
            }

            databaseContext.SaveChanges();
        }

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

            TagNameUnusedOrThrow(request.Name);

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

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

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

            databaseContext.SaveChanges();
        }

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

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

            if (request.Name != null)
            {
                TagNameUnusedOrThrow(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;
            }

            databaseContext.SaveChanges();
        }






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

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

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

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