PIbd-22_Bazunov_A.I._SushiBar/SushiBar/SushiBarDatabaseImplement/Implements/MessageStorage.cs

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

namespace SushiBarDatabaseImplement.Implements;

public class MessageStorage : 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)
            return new List<MessageInfoViewModel>();
        using var context = new SushiBarDatabase();
        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;
    }
}