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)
    {
        using var context = new SushiBarDatabase();
        var res = context.Messages
            .Where(x => !model.ClientId.HasValue || x.ClientId == model.ClientId)
            .Select(x => x.GetViewModel)
            .ToList();
        return model is not { Page: { }, PageSize: { } } ?
            res.ToList() :
            res.Skip((model.Page.Value - 1) * model.PageSize.Value).Take(model.PageSize.Value).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 res = context.Messages.FirstOrDefault(x => x.MessageId.Equals(model.MessageId));
        if (res == null) return res?.GetViewModel;
        res.Update(model);
        context.SaveChanges();
        return res?.GetViewModel;
    }
}