PIbd-21_Balberova_D.N._SushiBar/SushiBar/SushiBarDatabaseImplement/Implements/MessageInfoStorage.cs

using SushiBarContracts.BindingModels;
using SushiBarContracts.SearchModels;
using SushiBarContracts.StoragesContracts;
using SushiBarContracts.ViewModels;
using SushiBarDatabaseImplement.Models;

namespace SushiBarDatabaseImplement.Implements
{
    public class MessageInfoStorage : IMessageInfoStorage
    {
        public MessageInfoViewModel? GetElement(MessageInfoSearchModel model)
        {
            if (model.MessageId == null)
                return null;
            using var context = new SushiBarDatabase();
            return context.Messages.FirstOrDefault(x => x.MessageId == model.MessageId)?.GetViewModel;
        }

        public List<MessageInfoViewModel> GetFilteredList(MessageInfoSearchModel model)
        {
            if (!model.ClientId.HasValue && (!model.PageSize.HasValue || !model.CurrentPage.HasValue))
                return new();
            using var context = new SushiBarDatabase();
            if (model.CurrentPage.HasValue && model.PageSize.HasValue && model.ClientId.HasValue)
            {
                return context.Messages
                    .Where(x => x.ClientId == model.ClientId)
                    .Skip((model.CurrentPage.Value - 1) * model.PageSize.Value)
                    .Take(model.PageSize.Value).Select(x => x.GetViewModel).ToList();
            }
            if (model.CurrentPage.HasValue && model.PageSize.HasValue)
            {
                return context.Messages
                    .Skip((model.CurrentPage.Value - 1) * model.PageSize.Value)
                    .Take(model.PageSize.Value).Select(x => x.GetViewModel).ToList();
            }
            return context.Messages
                .Where(x => x.ClientId == model.ClientId)
                .Select(x => x.GetViewModel)
                .ToList();
        }

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

        public MessageInfoViewModel? Insert(MessageInfoBindingModel model)
        {
            var newMessage = Message.Create(model);
            if (newMessage == null)
            {
                return null;
            }
            using var context = new SushiBarDatabase();
            context.Messages.Add(newMessage);
            context.SaveChanges();
            return newMessage.GetViewModel;
        }

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