SYBD_2024/ConstructionFirmDatabaseImplement/Implements/ProjectStorage.cs

using Microsoft.EntityFrameworkCore;
using Subd_4.BindingModels;
using Subd_4.SearchModels;
using Subd_4.StoragesContracts;
using Subd_4.ViewModels;

namespace ConstructionFirmDatabaseImplement.Implements
{
    public class ProjectStorage : IProjectStorage
    {
        public ProjectViewModel? Delete(ProjectBindingModel model)
        {
            using var context = new ConstructionFirmDatabase();
            var element = context.Projects
                .FirstOrDefault(p => p.Id == model.Id);

            if (element != null)
            {
                context.Projects.Remove(element);
                context.SaveChanges();
                return element.GetViewModel;
            }
            return null;
        }

        public ProjectViewModel? GetElement(ProjectSearchModel model)
        {
            using var context = new ConstructionFirmDatabase();
            return context.Projects
                .Include(x => x.MaterialProjects)
                .ThenInclude(x => x.ConstructionMaterial)
                .Include(x => x.ProjectTeams)
                .ThenInclude(x => x.Team)
                .FirstOrDefault(x => (model.Id.HasValue && x.Id == model.Id))?
                .GetViewModel;
        }

        public List<ProjectViewModel> GetFilteredList(ProjectSearchModel model)
        {
            if (!model.Id.HasValue)
            {
                return new();
            }
            using var context = new ConstructionFirmDatabase();
            return context.Projects
                .Include(x => x.MaterialProjects)
                .ThenInclude(x => x.ConstructionMaterial)
                .Include(x => x.ProjectTeams)
                .ThenInclude(x => x.Team)
                .Where(x => x.Id == model.Id).ToList()
                .Select(x => x.GetViewModel).ToList();
        }

        public List<ProjectViewModel> GetFullList()
        {
            using var context = new ConstructionFirmDatabase();
            return context.Projects
                .Include(x => x.MaterialProjects)
                .ThenInclude(x => x.ConstructionMaterial)
                .Include(x => x.ProjectTeams)
                .ThenInclude(x => x.Team)
                .ToList()
                .Select(x => x.GetViewModel).ToList();
        }

        public ProjectViewModel? Insert(ProjectBindingModel model)
        {
            using var context = new ConstructionFirmDatabase();
            var newProject = Models.Project.Create(context, model);
            if (newProject == null)
            {
                return null;
            }
            context.Projects.Add(newProject);
            context.SaveChanges();
            return newProject.GetViewModel;
        }

        public ProjectViewModel? Update(ProjectBindingModel model)
        {
            using var context = new ConstructionFirmDatabase();
            using var transaction = context.Database.BeginTransaction();
            try
            {
                var Project = context.Projects.FirstOrDefault(x => x.Id == model.Id);
                if (Project == null)
                {
                    return null;
                }
                Project.Update(model);
                context.SaveChanges();
                transaction.Commit();
                return Project.GetViewModel;
            }
            catch
            {
                transaction.Rollback();
                throw;
            }
        }
    }
}