PIbd-21_Kurs_Work_Hotel/Hotel/HotelDataBaseImplement/Implements/HeadwaiterStorage.cs

using HotelContracts.BindingModels;
using HotelContracts.SearchModels;
using HotelContracts.StoragesContracts;
using HotelContracts.ViewModels;
using HotelDataBaseImplement.Models;
using Microsoft.EntityFrameworkCore;

namespace HotelDataBaseImplement.Implements
{
	public class HeadwaiterStorage : IAdministratorStorage
	{
		public List<AdministratorViewModel> GetFullList()
		{
			using var context = new HotelDataBase();

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

		public List<AdministratorViewModel> GetFilteredList(AdministratorSearchModel model)
		{
			if (string.IsNullOrEmpty(model.AdministratorFIO))
			{
				return null;
			}

			using var context = new HotelDataBase();

			return context.Administrators
				.Include(x => x.ConferenceBookings)
				.Include(x => x.Dinners)
				.Include(x => x.Rooms)
				.Where(x => x.AdministratorLogin.Contains(model.AdministratorLogin) && x.AdministratorPassword == model.AdministratorPassword)
				.Select(x => x.GetViewModel)
				.ToList();
		}

		public AdministratorViewModel? GetElement(AdministratorSearchModel model)
		{
			using var context = new HotelDataBase();

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

			if(!string.IsNullOrEmpty(model.AdministratorEmail) && !string.IsNullOrEmpty(model.AdministratorPassword))
			{
				return context.Administrators
					.Include(x => x.ConferenceBookings)
					.Include(x => x.Dinners)
					.Include(x => x.Rooms)
					.FirstOrDefault(x => x.AdministratorEmail.Equals(model.AdministratorEmail) && x.AdministratorPassword.Equals(model.AdministratorPassword))?
					.GetViewModel;
			}

			if (!string.IsNullOrEmpty(model.AdministratorEmail))
			{
				return context.Administrators
					.Include(x => x.ConferenceBookings)
					.Include(x => x.Dinners)
					.Include(x => x.Rooms)
					.FirstOrDefault(x => x.AdministratorEmail.Equals(model.AdministratorEmail))?
					.GetViewModel;
			}
			return null;
		}

		public AdministratorViewModel? Insert(AdministratorBindingModel model)
		{
			var newAdministrator = Administrator.Create(model);


			if (string.IsNullOrEmpty(model.AdministratorFIO))
			{
				return new();
			}

			using var context = new HotelDataBase();

			context.Administrators.Add(newAdministrator);
			context.SaveChanges();

			return newAdministrator.GetViewModel;
		}

		public AdministratorViewModel? Update(AdministratorBindingModel model)
		{
			using var context = new HotelDataBase();

			var administrator = context.Administrators
				.FirstOrDefault(x => x.Id == model.Id);

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

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

			return administrator.GetViewModel;
		}

		public AdministratorViewModel? Delete(AdministratorBindingModel model)
		{
			using var context = new HotelDataBase();

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

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

				return element.GetViewModel;
			}

			return null;
		}
	}
}