DD.WellWorkover.Cloud/AsbCloudInfrastructure/Services/PlannedTrajectoryService/PlannedTrajectoryService.cs

using AsbCloudApp.Data;
using AsbCloudApp.Exceptions;
using AsbCloudApp.Services;
using AsbCloudDb.Model;
using Mapster;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;

namespace AsbCloudInfrastructure.Services.PlannedTrajectory
{
#nullable enable
    public class PlannedTrajectoryService : IPlannedTrajectoryService
    {
        private readonly IAsbCloudDbContext db;
        private readonly IWellService wellService;
        public PlannedTrajectoryService(IAsbCloudDbContext db, IWellService wellService)
        {
            this.db = db;
            this.wellService = wellService;
        }
        public async Task<int> AddRangeAsync(IEnumerable<PlannedTrajectoryDto> plannedTrajectoryRows, CancellationToken token)
        {            
            var entitys = plannedTrajectoryRows
                .Select(e => Convert(e));
            db.PlannedTrajectorys.AddRange(entitys);
            return await db.SaveChangesAsync(token)
                .ConfigureAwait(false);
        }

        public async Task<int> AddAsync(PlannedTrajectoryDto plannedTrajectoryRow, CancellationToken token)
        {            
            var entity = Convert(plannedTrajectoryRow);
            db.PlannedTrajectorys.Add(entity);            
            return await db.SaveChangesAsync(token)
                .ConfigureAwait(false);
        }


        public async Task<int> DeleteRangeAsync(IEnumerable<int> ids, CancellationToken token)
        {
            var query = db.PlannedTrajectorys
                .Where(e => ids.Contains(e.Id));
            db.PlannedTrajectorys.RemoveRange(query);
            return await db.SaveChangesAsync(token)
                .ConfigureAwait(false);
        }

        public async Task<int> DeleteByIdWellAsync(int idWell, CancellationToken token)
        {
            var query = db.PlannedTrajectorys
                .Where(e => e.IdWell == idWell);
            db.PlannedTrajectorys.RemoveRange(query);
            return await db.SaveChangesAsync(token)
                .ConfigureAwait(false);
        }


        public async Task<IEnumerable<PlannedTrajectoryDto>> GetAsync(int idWell, CancellationToken token)
        {
            var well = wellService.GetOrDefault(idWell);
            var timezone = wellService.GetTimezone(idWell);
            if (well is null || well.Timezone is null)
                throw new ArgumentInvalidException("idWell doesn`t exist", nameof(idWell));
            var query = db.PlannedTrajectorys
                .AsNoTracking()
                .Where(x => x.IdWell == idWell);                           
            var entities = await query
                .OrderBy(e => e.UpdateDate)
                .ToListAsync(token);            
            var result = entities
                .Select(r => Convert(r));
            return result;
        }

        public async Task<int> UpdateAsync(int idRow, PlannedTrajectoryDto row, CancellationToken token)
        {            
            var entity = Convert(row, idRow);
            db.PlannedTrajectorys.Update(entity);
            return await db.SaveChangesAsync(token)
                .ConfigureAwait(false);
        }

        private PlannedTrajectoryDto Convert(AsbCloudDb.Model.PlannedTrajectory entity)
        {
            var timezone = wellService.GetTimezone(entity.IdWell);
            var dto = entity.Adapt<PlannedTrajectoryDto>(); 
            dto.UpdateDate = entity.UpdateDate.ToRemoteDateTime(timezone.Hours);
            return dto;
        }

        private AsbCloudDb.Model.PlannedTrajectory Convert(PlannedTrajectoryDto dto)
        {
            var timezone = wellService.GetTimezone(dto.IdWell);
            var entity = dto.Adapt<AsbCloudDb.Model.PlannedTrajectory>();
            entity.IdWell = dto.IdWell;
            entity.Id = 0;
            entity.UpdateDate = DateTime.Now.ToUtcDateTimeOffset(timezone.Hours);
            entity.IdUser = dto.IdUser;
            return entity;
        }
        private AsbCloudDb.Model.PlannedTrajectory Convert(PlannedTrajectoryDto dto, int idRow)
        {
            var timezone = wellService.GetTimezone(dto.IdWell);
            var entity = dto.Adapt<AsbCloudDb.Model.PlannedTrajectory>();
            entity.IdWell = dto.IdWell;
            entity.Id = idRow;
            entity.UpdateDate = DateTime.Now.ToUtcDateTimeOffset(timezone.Hours);
            entity.IdUser = dto.IdUser;
            return entity;
        }
    }
#nullable disable
}