PIbd-23_Mochalov_D.V._LawFirm/LawFirm/LawFirmDatabaseImplement/Implements/MessageInfoStorage .cs

using LawFirmContracts.BindingModels;
using LawFirmContracts.SearchModels;
using LawFirmContracts.StorageContracts;
using LawFirmContracts.ViewModels;
using LawFirmDatabaseImplement.Models;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace LawFirmDatabaseImplement.Implements
{
	public class MessageInfoStorage : IMessageInfoStorage
	{

		public MessageInfoViewModel? GetElement(MessageInfoSearchModel model)
		{
			using var context = new LawFirmDatabase();
			if (model.MessageId != null)
			{
				return context.Messages.FirstOrDefault(x => x.MessageId == model.MessageId)?.GetViewModel;
			}
			return null;
		}

		public List<MessageInfoViewModel> GetFilteredList(MessageInfoSearchModel model)
		{
			using var context = new LawFirmDatabase();
			var messages = context.Messages
				.Where(x => (model.ClientId == null) || (x.ClientId == model.ClientId))
				.Select(x => x.GetViewModel);

			if (!(model.Page.HasValue && model.PageSize.HasValue))
			{
				return messages.ToList();
			}
			return messages.Skip((model.Page.Value - 1) * model.PageSize.Value).Take(model.PageSize.Value).ToList();

		}

		public List<MessageInfoViewModel> GetFullList()
		{
			using var context = new LawFirmDatabase();
			return context.Messages
				.Select(x => x.GetViewModel)
				.ToList();
		}

		public MessageInfoViewModel? Insert(MessageInfoBindingModel model)
		{
			using var context = new LawFirmDatabase();
			var newMessage = MessageInfo.Create(model);
			if (newMessage == null || context.Messages.Any(x => x.MessageId.Equals(model.MessageId)))
			{
				return null;
			}
			context.Messages.Add(newMessage);
			context.SaveChanges();
			return newMessage.GetViewModel;
		}

        public MessageInfoViewModel? Update(MessageInfoBindingModel model)
        {
            using var context = new LawFirmDatabase();
			var message = context.Messages.Include(x => x.Client).FirstOrDefault(x => x.MessageId.Equals(model.MessageId));
            if (message != null)
            {
                message.Update(model);
                context.SaveChanges();
            }
            return message?.GetViewModel;
        }
    }
}