CourseWork_Bank/Bank/BankDataBaseImplement/Implemets/AdditionsStorage.cs

using BankContracts.BindingModels;
using BankContracts.SearchModels;
using BankContracts.StoragesContracts;
using BankContracts.ViewModels;
using BankDataBaseImplement.Models;
using Microsoft.EntityFrameworkCore;

namespace BankDataBaseImplement.Implemets
{
    public class AdditionsStorage : IAdditionsPlanStorage
    {
        public AdditionsPlanViewModel? Delete(AdditionsBindingModel model)
        {
            using var context = new BankDataBase();

            var element = context.Additions
                .FirstOrDefault(rec => rec.Id == model.Id);

            if (element != null)
            {
                context.Additions.Remove(element);
                context.SaveChanges();

                return element.GetViewModel;
            }

            return null;
        }

        public AdditionsPlanViewModel? GetElement(AdditionsSearchModel model)
        {
            if (!model.Id.HasValue)
            {
                return null;
            }

            using var context = new BankDataBase();

            return context.Additions
                .Include(x => x.Members)
                .ThenInclude(x => x.Member)
                .ThenInclude(x => x.ConferenceMember)
                .ThenInclude(x => x.Conference)
                .Include(x => x.Rooms)
                .Include(x => x.Organiser)
                .FirstOrDefault(x => (model.Id.HasValue && x.Id == model.Id))?
                .GetViewModel;
        }

        public List<AdditionsPlanViewModel> GetFilteredList(AdditionsSearchModel model)
        {
            if (string.IsNullOrEmpty(model.AdditionsName))
            {
                return new();
            }

            using var context = new BankDataBase();

            return context.Additions
                .Include(x => x.Members)
                .ThenInclude(x => x.Member)
                .ThenInclude(x => x.ConferenceMember)
                .ThenInclude(x => x.Conference)
                .Include(x => x.Rooms)
                .Include(x => x.Organiser)
                .Where(x => x.MealPlanName.Contains(model.AdditionsName))
                .Select(x => x.GetViewModel)
                .ToList();
        }

        public List<AdditionsPlanViewModel> GetFullList()
        {
            using var context = new BankDataBase();
            return context.Additions
                .Include(x => x.Members)
                .ThenInclude(x => x.Member)
                .ThenInclude(x => x.ConferenceMember)
                .ThenInclude(x => x.Conference)
                .Include(x => x.Rooms)
                .Include(x => x.Organiser)
                .Select(x => x.GetViewModel)
                .ToList();
        }

        public AdditionsPlanViewModel? Insert(AdditionsBindingModel model)
        {
            using var context = new BankDataBase();
            var newMealPlan = Additions.Create(context,model);

            if (newMealPlan == null)
            {
                return null;
            }
            
            context.Additions.Add(newMealPlan);
            context.SaveChanges();
            return newMealPlan.GetViewModel;
        }

        public AdditionsPlanViewModel? Update(AdditionsBindingModel model)
        {
            using var context = new BankDataBase();
            var mealPlan = context.Additions.FirstOrDefault(x => x.Id == model.Id);
            if (mealPlan == null)
            {
                return null;
            }
            mealPlan.Update(model);
            context.SaveChanges();
            return mealPlan.GetViewModel;
        }
    }
}