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

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)
                .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)
            {
                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
            };
        }
    }
}