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

using AsbCloudApp.Data;
using AsbCloudApp.Services;
using ClosedXML.Excel;
using ClosedXML.Excel.Drawings;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;

namespace AsbCloudInfrastructure.Services
{
    public class DrillingProgramService : IDrillingProgramService
    {
        private readonly IFileService fileService;
        private readonly IWellService wellService;
        private const int idFileCategoryDrillingProgramItems = 13;
        private const int idFileCategoryDrillingProgram = 14;

        public DrillingProgramService(IFileService fileService, IWellService wellService)
        {
            this.fileService = fileService;
            this.wellService = wellService;
        }

        public async Task<FileInfoDto> GetAsync(int idWell, CancellationToken token = default)
        {
            var filesInfos = await fileService.GetInfosByCategoryAsync(idWell, idFileCategoryDrillingProgramItems, token)
                .ConfigureAwait(false);

            var well = await wellService.GetAsync(idWell, token)
                .ConfigureAwait(false);

            var matchFiles = await fileService.GetInfosByCategoryAsync(idWell, idFileCategoryDrillingProgram, token)
                .ConfigureAwait(false);

            if (matchFiles is not null && matchFiles.Any())
            {
                matchFiles = matchFiles.OrderByDescending(f => f.UploadDate);
                var matchFilesIterator = matchFiles.GetEnumerator();
                matchFilesIterator.MoveNext();
                var matchFile = matchFilesIterator.Current;
                while (matchFilesIterator.MoveNext())
                    await fileService.DeletedAsync(matchFilesIterator.Current.Id, token)
                        .ConfigureAwait(false);

                if (filesInfos.All(f => f.UploadDate <= matchFile.UploadDate))
                    return matchFile;
                else
                    await fileService.DeletedAsync(matchFile.Id, token)
                        .ConfigureAwait(false);
            }

            var resultFileName = $"Программа бурения {well.Cluster} {well.Caption}.xlsx";

            var fileNames = filesInfos
                .Where(f => f.Name != resultFileName)
                .Select(f => fileService.GetUrl(f));

            var tempResultFilePath = Path.Combine(Path.GetTempPath(), "drillingProgram", resultFileName);

            UniteExcelFiles(fileNames, tempResultFilePath);

            var fileInfo = await fileService.MoveAsync(idWell, null, idFileCategoryDrillingProgram,
                resultFileName, tempResultFilePath, token).ConfigureAwait(false);

            return fileInfo;
        }

        private static void UniteExcelFiles(IEnumerable<string> excelFilesNames, string resultExcelPath)
        {
            var resultExcelFile = new XLWorkbook(XLEventTracking.Disabled);

            const int maxAllowedColumns = 256;

            var filteredFileNames = excelFilesNames.Distinct();

            foreach (var excelFileName in filteredFileNames)
            {
                using var sourceExcelFile = new XLWorkbook(excelFileName, XLEventTracking.Disabled);

                foreach (var sheet in sourceExcelFile.Worksheets)
                {
                    var newSheetName = sheet.Name;
                    int index = 1;

                    if (resultExcelFile.Worksheets.Contains(newSheetName))
                    {
                        var suffix = $"_{index++}";
                        if (newSheetName.Length + suffix.Length >= 31)
                            newSheetName = sheet.Name.Substring(0, (31 - suffix.Length));
                        newSheetName += suffix;
                    }

                    var imagesInfos = sheet.Pictures.Select(p => new ImageInfo
                    {
                        Id = p.Id,
                        Data = p.ImageStream.GetBuffer(),
                        Height = p.Height,
                        Width = p.Width,
                        TopLeftCellAddress = p.TopLeftCell.Address,
                        Left = p.Left,
                        Top = p.Top
                    }).ToList();

                    if (sheet.Columns().Count() > maxAllowedColumns)
                    {
                        var resultSheet = resultExcelFile.Worksheets.Add(newSheetName);

                        var rngData = GetCellsRange(sheet);

                        rngData.CopyTo(resultSheet.Cell(1, 1));

                        var lastRowWithData = rngData.LastRowUsed().RangeAddress
                            .LastAddress.RowNumber;

                        for (int i = 1; i < lastRowWithData; i++)
                        {
                            resultSheet.Row(i).Height = sheet.Row(i).Height;
                            resultSheet.Column(i).Width = sheet.Column(i).Width;
                        }

                        CopyImagesToAnotherSheet(imagesInfos, resultSheet);
                    }
                    else
                    {
                        RemovePicturesFromSheet(sheet);

                        var resultSheet = sheet.CopyTo(resultExcelFile, newSheetName);

                        CopyImagesToAnotherSheet(imagesInfos, resultSheet);
                    }
                }
            }

            resultExcelFile.SaveAs(resultExcelPath, 
                new SaveOptions { EvaluateFormulasBeforeSaving = true});
        }

        private static IXLWorksheet CopyImagesToAnotherSheet(IEnumerable<ImageInfo> imagesInfos,
            IXLWorksheet resultSheet)
        {
            foreach (var image in imagesInfos)
            {
                var stream = new MemoryStream();
                stream.Write(image.Data, 0, image.Data.Length);

                resultSheet.AddPicture(stream)
                           .WithPlacement(XLPicturePlacement.Move)
                           .WithSize(image.Width, image.Height)
                           .MoveTo(resultSheet.Cell(image.TopLeftCellAddress),
                                   image.Left, image.Top);
            }

            return resultSheet;
        }

        private static void RemovePicturesFromSheet(IXLWorksheet sheet)
        {
            var filteredPics = sheet.Pictures.Select(p => p.Name).Distinct().ToList();

            foreach (var n in filteredPics)
                sheet.Pictures.Delete(n);
        }

        private static IXLRange GetCellsRange(IXLWorksheet sheet)
        {
            var firstTableCell = sheet.FirstCellUsed();
            var lastTableCell = sheet.LastCellUsed();
            var rngData = sheet.Range(firstTableCell.Address, lastTableCell.Address);

            return rngData;
        }
    }

    class ImageInfo
    {
        public int Id { get; set; }
        public byte[] Data { get; set; }
        public int Height { get; set; }
        public int Width { get; set; }
        public IXLAddress TopLeftCellAddress { get; set; }
        public int Left { get; set; }
        public int Top { get; set; }
    }
}