CourseWork_Bank/Bank/BankDataBaseImplement/Implemets/ClientStorage.cs

using BankContracts.BindingModels;
using BankContracts.SearchModels;
using BankContracts.StoragesContracts;
using BankContracts.ViewModels;
using BankDataBaseImplement.Models;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace BankDataBaseImplement.Implemets
{
    public class ClientStorage : IHeadwaiterStorage
    {
        public ClientViewModel? Delete(ClientBindingModel model)
        {
            using var context = new BankDataBase();

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

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

                return element.GetViewModel;
            }

            return null;
        }

        public ClientViewModel? GetElement(ClientSearchModel model)
        {
            using var context = new BankDataBase();

            if (model.Id.HasValue)
                return context.Headwaiters
                    .Include(x => x.ConferenceBookings)
                    .Include(x => x.Dinners)
                    .Include(x => x.Rooms)
                    .FirstOrDefault(x => x.Id == model.Id)?
                    .GetViewModel;

            if (!string.IsNullOrEmpty(model.HeadwaiterLogin) && !string.IsNullOrEmpty(model.HeadwaiterPassword))
                return context.Headwaiters
                    .Include(x => x.ConferenceBookings)
                    .Include(x => x.Dinners)
                    .Include(x => x.Rooms)
                    .FirstOrDefault(x => x.HeadwaiterLogin.Equals(model.HeadwaiterLogin) && x.HeadwaiterPassword.Equals(model.HeadwaiterPassword))?
                    .GetViewModel;

            if (!string.IsNullOrEmpty(model.HeadwaiterLogin))
                return context.Headwaiters
                    .Include(x => x.ConferenceBookings)
                    .Include(x => x.Dinners)
                    .Include(x => x.Rooms)
                    .FirstOrDefault(x => x.HeadwaiterLogin.Equals(model.HeadwaiterLogin))?
                    .GetViewModel;

            return null;
        }

        public List<ClientViewModel> GetFilteredList(ClientSearchModel model)
        {
            if (string.IsNullOrEmpty(model.HeadwaiterFIO))
            {
                return new();
            }

            using var context = new BankDataBase();

            return context.Headwaiters
                .Include(x => x.ConferenceBookings)
                .Include(x => x.Dinners)
                .Include(x => x.Rooms)
                .Where(x => x.HeadwaiterLogin.Contains(model.HeadwaiterLogin) && x.HeadwaiterPassword == model.HeadwaiterPassword)
                .Select(x => x.GetViewModel)
                .ToList();
        }

        public List<ClientViewModel> GetFullList()
        {
            using var context = new BankDataBase();

            return context.Headwaiters
                .Select(x => x.GetViewModel)
                .ToList();
        }

        public ClientViewModel? Insert(ClientBindingModel model)
        {
            var newHeadwaiter = Client.Create(model);

            if (newHeadwaiter == null)
            {
                return null;
            }

            using var context = new BankDataBase();

            context.Headwaiters.Add(newHeadwaiter);
            context.SaveChanges();

            return newHeadwaiter.GetViewModel;
        }

        public ClientViewModel? Update(ClientBindingModel model)
        {
            using var context = new BankDataBase();

            var headwaiter = context.Headwaiters
                .FirstOrDefault(x => x.Id == model.Id);

            if (headwaiter == null)
            {
                return null;
            }

            headwaiter.Update(model);
            context.SaveChanges();

            return headwaiter.GetViewModel;
        }
    }
}