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

using HotelContracts.BindingModels;
using HotelContracts.SearchModels;
using HotelContracts.StoragesContracts;
using HotelContracts.ViewModels;
using HotelDataBaseImplement.Models;
using Microsoft.EntityFrameworkCore;
using Microsoft.IdentityModel.Tokens;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace HotelDataBaseImplement.Implements
{
	public class RoomStorage : IRoomStorage
	{
		public List<RoomViewModel> GetFullList()
		{
			using var context = new HotelDataBase();
			return context.Rooms
				.Include(x => x.Dinners)
				.ThenInclude(x => x.Dinner)
				.ThenInclude(x => x.ConferenceBookingDinner)
				.ThenInclude(x => x.ConferenceBooking)
				.Include(x => x.MealPlan)
				.Include(x => x.Administrator)
				.ToList()
				.Select(x => x.GetViewModel)
				.ToList();
		}

		public List<RoomViewModel> GetFilteredList(RoomSearchModel model)
		{
			if (!model.DateFrom.HasValue && !model.DateTo.HasValue && !model.Id.HasValue && !model.AdministratorId.HasValue)
			{
				return new();
			}

			using var context = new HotelDataBase();

            if (model.DateFrom.HasValue)
            {
                return context.Rooms
                        .Include(x => x.Dinners)
						.ThenInclude(x => x.Dinner)
						.ThenInclude(x => x.ConferenceBookingDinner)
						.ThenInclude(x => x.ConferenceBooking)
						.Include(x => x.MealPlan)
						.Include(x => x.Administrator)
                        .Where(x => x.DateCreate >= model.DateFrom && x.DateCreate <= model.DateTo && x.AdministratorId == model.AdministratorId)
                        .Select(x => x.GetViewModel)
                        .ToList();
            }
            else if (model.AdministratorId.HasValue)
			{
				return context.Rooms
					.Include(x => x.Dinners)
					.ThenInclude(x => x.Dinner)
					.ThenInclude(x => x.ConferenceBookingDinner)
					.ThenInclude(x => x.ConferenceBooking)
					.Include(x => x.MealPlan)
					.Include(x => x.Administrator)
					.Where(x => x.AdministratorId == model.AdministratorId)
					.ToList()
					.Select(x => x.GetViewModel)
					.ToList();
			}

			return context.Rooms
				.Include(x => x.Dinners)
				.ThenInclude(x => x.Dinner)
				.ThenInclude(x => x.ConferenceBookingDinner)
				.ThenInclude(x => x.ConferenceBooking)
				.Include(x => x.MealPlan)
				.Include(x => x.Administrator)
				.Where(x => x.RoomNumber == model.RoomNumber)
				.ToList()
				.Select(x => x.GetViewModel)
				.ToList();
		}

		public RoomViewModel? GetElement(RoomSearchModel model)
		{
			if (!model.Id.HasValue && model.RoomNumber == null)
			{
				return null;
			}

			using var context = new HotelDataBase();

			var room = context.Rooms
				.Include(x => x.Dinners)
				.ThenInclude(x => x.Dinner)
				.ThenInclude(x => x.ConferenceBookingDinner)
				.ThenInclude(x => x.ConferenceBooking)
				.Include(x => x.MealPlan)
				.Include(x => x.Administrator)
				.FirstOrDefault(x => (model.RoomNumber != null && x.RoomNumber == model.RoomNumber) || (model.Id.HasValue && x.Id == model.Id));

			return room?.GetViewModel;
		}

		public RoomViewModel? Insert(RoomBindingModel model)
		{
			using var context = new HotelDataBase();
			var newRoom = Room.Create(context, model);

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

			context.Rooms.Add(newRoom);
			context.SaveChanges();

			return context.Rooms
				.Include(x => x.Dinners)
				.ThenInclude(x => x.Dinner)
				.ThenInclude(x => x.ConferenceBookingDinner)
				.ThenInclude(x => x.ConferenceBooking)
				.Include(x => x.MealPlan)
				.Include(x => x.Administrator)
			   .FirstOrDefault(x => x.Id == newRoom.Id)
			   ?.GetViewModel;
		}

		public RoomViewModel? Update(RoomBindingModel model)
		{
			using var context = new HotelDataBase();
			using var transaction = context.Database.BeginTransaction();
			try
			{
				var elem = context.Rooms
					.Include(x => x.Dinners)
					.ThenInclude(x => x.Dinner)
					.ThenInclude(x => x.ConferenceBookingDinner)
					.ThenInclude(x => x.ConferenceBooking)
					.ThenInclude(x => x.Conference)
					.FirstOrDefault(rec => rec.Id == model.Id);
				if (elem == null)
				{
					return null;
				}
                elem.UpdateDinners(context, model);
                context.SaveChanges();
                elem.Update(model);
				context.SaveChanges();
				transaction.Commit();
				return elem.GetViewModel;
			}
			catch
			{
				transaction.Rollback();
				throw;
			}
		}

		public RoomViewModel? Delete(RoomBindingModel model)
		{
			using var context = new HotelDataBase();

			var element = context.Rooms
				.Include(x => x.Dinners)
				.ThenInclude(x => x.Dinner)
				.ThenInclude(x => x.ConferenceBookingDinner)
				.ThenInclude(x => x.ConferenceBooking)
				.ThenInclude(x => x.Conference)
				.FirstOrDefault(rec => rec.Id == model.Id);

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

				return element.GetViewModel;
			}

			return null;
		}
	}
}