PIbd-21_Sagirov_M.M._Shipyard/Shipyard/ShipyardDataBaseImplement/Implements/MessageInfoStorage.cs

using ShipyardContracts.BindingModels;
using ShipyardContracts.SearchModels;
using ShipyardContracts.StoragesContracts;
using ShipyardContracts.ViewModels;
using ShipyardDataBaseImplement.Models;

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

        public List<MessageInfoViewModel> GetFilteredList(MessageInfoSearchModel model)
        {
            if (!model.ClientId.HasValue)
                return new();
            using var context = new ShipyardDataBase();
            return context.Messages
                .Where(x => x.ClientId == model.ClientId)
                .Select(x => x.GetViewModel)
                .ToList();
        }

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

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