SYBD_2024/ConstructionFirmDatabaseImplement/Implements/TeamStorage.cs

using ConstructionFirmDatabaseImplement.Models;
using Subd_4.BindingModels;
using Subd_4.SearchModels;
using Subd_4.StoragesContracts;
using Subd_4.ViewModels;

namespace ConstructionFirmDatabaseImplement.Implements
{
    public class TeamStorage : ITeamStorage
    {
        public List<TeamViewModel> GetFilteredList(TeamSearchModel model)
        {
            if (string.IsNullOrEmpty(model.TeamName))
            {
                return new();
            }
            using var context = new ConstructionFirmDatabase();
            return context.Teams.Where(x => x.TeamName.Contains(model.TeamName)).Select(x => x.GetViewModel).ToList();
        }

        public List<TeamViewModel> GetFullList()
        {
            using var context = new ConstructionFirmDatabase();
            return context.Teams.Select(x => x.GetViewModel).ToList();
        }

        public TeamViewModel? Delete(TeamBindingModel model)
        {
            using var context = new ConstructionFirmDatabase();
            var element = context.Teams.FirstOrDefault(x => x.Id == model.Id);
            if (element != null)
            {
                context.Teams.Remove(element);
                context.SaveChanges();
                return element.GetViewModel;
            }
            return null;
        }

        public TeamViewModel? GetElement(TeamSearchModel model)
        {
            using var context = new ConstructionFirmDatabase();
            return context.Teams.FirstOrDefault(x => (!string.IsNullOrEmpty(model.TeamName)) && x.TeamName == model.TeamName || model.Id.HasValue && x.Id == model.Id)?.GetViewModel;
        }

        public TeamViewModel? Insert(TeamBindingModel model)
        {
            var newTeam = Team.Create(model);
            if (newTeam == null)
            {
                return null;
            }
            using var context = new ConstructionFirmDatabase();
            context.Teams.Add(newTeam);
            context.SaveChanges();
            return newTeam.GetViewModel;
        }

        public TeamViewModel? Update(TeamBindingModel model)
        {
            using var context = new ConstructionFirmDatabase();
            var component = context.Teams.FirstOrDefault(x => x.Id == model.Id);
            if (component == null)
            {
                return null;
            }
            component.Update(model);
            context.SaveChanges();
            return component.GetViewModel;
        }
    }
}