CourseWork_Bank/Bank/BankDatabaseImplement/Implements/CashierImplements/MoneyTransferStorage.cs

using BankContracts.BindingModels.Cashier;
using BankContracts.SearchModels.Cashier;
using BankContracts.StoragesModels.Cashier;
using BankContracts.ViewModels.Cashier.ViewModels;
using BankDatabaseImplement.Models.CashierModels;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace BankDatabaseImplement.Implements.CashierImplements
{
	// Реализация интерфейса хранилища для сущности "Перевод денег"
	public class MoneyTransferStorage : IMoneyTransferStorage
    {
        public List<MoneyTransferViewModel> GetFullList()
        {
            using var context = new BankDatabase();

            return context.MoneyTransfers
                    .Include(x => x.AccountPayeer)
                    .Include(x => x.AccountSender)
                    .Include(x => x.Cashier)
                    .Select(x => x.GetViewModel)
                    .ToList();
        }

        public List<MoneyTransferViewModel> GetFilteredList(MoneyTransferSearchModel model)
        {
            using var context = new BankDatabase();

            var result = context.MoneyTransfers.Include(x => x.Cashier)
                    .Include(x => x.AccountPayeer)
                    .Include(x => x.AccountSender)
                    .ToList();

            if (model.AccountPayeeId.HasValue && model.AccountSenderId.HasValue && model.AccountPayeeId == model.AccountSenderId)
            {
                return result.Where(x => (x.AccountSenderId == model.AccountSenderId || x.AccountPayeeId == model.AccountPayeeId)
                        && x.AccountSender != null)
                    .Select(x => x.GetViewModel)
                    .ToList();
            }

            if (model.AccountPayeeId.HasValue) result = result.Where(x => x.AccountPayeeId == model.AccountPayeeId).ToList();

            if (model.AccountSenderId.HasValue) result = result.Where(x => x.AccountSenderId == model.AccountSenderId).ToList();

            if (model.CashierId.HasValue) result = result.Where(x => x.CashierId == model.CashierId).ToList();

            if (model.DateTransfer.HasValue) result = result.Where(x => x.DateTransfer >= model.DateTransfer).ToList();

            if (model.ClientId.HasValue)
            {
                result = result.Where(x => x.AccountSender != null)
                    .Where(x => (x.AccountPayeer.ClientId == model.ClientId || x.AccountSender.ClientId == model.ClientId))
                    .ToList();
            }

            return result.Select(x => x.GetViewModel).ToList();
        }

        public MoneyTransferViewModel? GetElement(MoneyTransferSearchModel model)
        {
            if (model.AccountSenderId < 0 && !model.Id.HasValue)
            {
                return null;
            }

            using var context = new BankDatabase();

            return context.MoneyTransfers
                    .Include(x => x.AccountPayeer)
                    .Include(x => x.AccountSender)
                    .Include(x => x.Cashier)
                    .FirstOrDefault(x => x.Id == model.Id)
                    ?.GetViewModel;
        }

        public MoneyTransferViewModel? Insert(MoneyTransferBindingModel model)
        {
            using var context = new BankDatabase();

            var newMoneyTransfer = MoneyTransfer.Create(context, model);

            if (newMoneyTransfer == null)
            {
                return null;
            }

            context.MoneyTransfers.Add(newMoneyTransfer);
            context.SaveChanges();

            return newMoneyTransfer.GetViewModel;
        }

        public MoneyTransferViewModel? Update(MoneyTransferBindingModel model)
        {
            using var context = new BankDatabase();
            using var transaction = context.Database.BeginTransaction();

            try
            {
                var moneyTransfer = context.MoneyTransfers.FirstOrDefault(rec => rec.Id == model.Id);

                if (moneyTransfer == null)
                {
                    return null;
                }

                moneyTransfer.Update(model);
                context.SaveChanges();
                transaction.Commit();

                return moneyTransfer.GetViewModel;
            }
            catch
            {
                transaction.Rollback();
                throw;
            }
        }

        public MoneyTransferViewModel? Delete(MoneyTransferBindingModel model)
        {
            using var context = new BankDatabase();

            var element = context.MoneyTransfers
                .Include(x => x.AccountPayeeId)
                .Include(x => x.AccountSenderId)
                .Include(x => x.Sum)
                .Include(x => x.DateTransfer)
                .FirstOrDefault(rec => rec.Id == model.Id);

            if (element != null)
            {
                context.MoneyTransfers.Remove(element);
                context.SaveChanges();

                return element.GetViewModel;
            }

            return null;
        }
    }
}