RhSolutions-AddIn/RhSolutions.AddIn/Services/GuessReader.cs

using System.IO;

namespace RhSolutions.Services;

public class GuessReader : IReader
{
    private readonly Application _application;
    private ProgressBar _progressBar;

    public GuessReader(Application application)
    {
        _application = application;
    }

    public Dictionary<Product, double> ReadProducts(Range range)
    {
        int? productColumnIndex = null;

        for (int column = 1; column < range.Columns.Count + 1; column++)
        {
            if (IsProductColumn(range.Columns[column]))
            {
                productColumnIndex = column;
                break;
            }
        }

        if (productColumnIndex == null)
        {
            throw new ArgumentException("Колонка с артикулом не определена");
        }

        int? amountColumnIndex = null;
        int currentIndex = productColumnIndex.Value + 1;

        while (currentIndex > 0)
        {
            if (currentIndex > range.Columns.Count + 1)
            {
                currentIndex = productColumnIndex.Value - 1;
                continue;
            }
            if (currentIndex > productColumnIndex)
            {
                if (IsAmountColumn(range.Columns[currentIndex++]))
                {
                    amountColumnIndex = currentIndex - 1;
                    break;
                }
            }
            else
            {
                if (IsAmountColumn(range.Columns[currentIndex--]))
                {
                    amountColumnIndex = currentIndex + 1;
                    break;
                }
            }
        }

        if (amountColumnIndex == null)
        {
            throw new ArgumentException("Колонка с количеством не определена");
        }

        else
        {
            return GetDictionaryFromColumns(range.Columns[productColumnIndex],
                range.Columns[amountColumnIndex]);
        }
    }

    private bool IsProductColumn(Range column)
    {
        int successCounter = 0;
        var cells = column.Value2;

        if (cells == null)
        {
            return false;
        }

        for (int row = 1; row < column.Rows.Count + 1; row++)
        {            
            object currentCell = column.Rows.Count == 1 ? cells : cells[row, 1];
            if (currentCell == null)
            {
                continue;
            }

            if (ProductSku.TryParse(currentCell.ToString(), out IEnumerable<ProductSku> skus))
            {
                successCounter++;
            }

            if (successCounter > 2)
            {
                return true;
            }
        }

        return successCounter > 0;
    }

    private bool IsAmountColumn(Range column)
    {
        int successCounter = 0;
        var cells = column.Value2;

        if (cells == null)
        {
            return false;
        }

        for (int row = 1; row < column.Rows.Count + 1; row++)
        {
            object currentCell = column.Rows.Count == 1 ? cells : cells[row, 1];
            if (currentCell == null)
            {
                continue;
            }

            double? value = currentCell as double?; 

            if (value == null || value == 0)
            {
                continue;
            }

            if (value > 30000)
            {
                return false;
            }
            
            if (++successCounter > 3)
            {
                return true;
            }
        }

        return successCounter > 0;
    }

    private Dictionary<Product, double> GetDictionaryFromColumns(Range productColumn, Range amountColumn)
    {
        Dictionary<Product, double> result = new();
        _progressBar = new("Заполняю словарь значений...", productColumn.Rows.Count);

        for (int row = 1; row < productColumn.Rows.Count + 1; row++)
        {
            _progressBar.Update();
            var amountCells = amountColumn.Value2;
            object currentAmountCell = productColumn.Rows.Count == 1 ? amountCells : amountCells[row, 1];

            double? amountValue = currentAmountCell as double?;
            if (amountValue == null)
            {
                continue;
            }

            var productCells = productColumn.Value2;
            object currentProductCell = productColumn.Rows.Count == 1 ? productCells : productCells[row, 1];
            if (currentProductCell == null)
            {
                continue;
            }

            if (ProductSku.TryParse(currentProductCell.ToString(), out IEnumerable<ProductSku> skus))
            {
                Product p = new(skus.First())
                {
                    Name = "Распознанный артикул"
                };
                if (result.ContainsKey(p))
                {
                    result[p] += amountValue.Value;
                }
                else
                {
                    result.Add(p, amountValue.Value);
                }
            }
        }
        return result;
    }

    public List<(string, Dictionary<Product, double>)> ReadProducts(IEnumerable<Worksheet> worksheets)
    {
        List<(string, Dictionary<Product, double>)> result = new();
        foreach (Worksheet worksheet in worksheets)
        {
            string wbName = Path.GetFileNameWithoutExtension(
                worksheet.Parent.Name);
            Range range = worksheet.UsedRange;
            var productsDict = ReadProducts(range);
            result.Add((wbName, productsDict));
        }
        return result;
    }

    public List<(string, Dictionary<Product, double>)> ReadProducts(string[] files)
    {
        _progressBar = new("Открываю исходные файлы...", files.Length);
        List<Worksheet> worksheets = new();

        _application.ScreenUpdating = false;
        foreach (string file in files)
        {
            Workbook wb = _application.Workbooks.Open(file);
            worksheets.Add(wb.ActiveSheet);
            _progressBar.Update();
        }
        _application.ScreenUpdating = true;
        var result = ReadProducts(worksheets);
        foreach (var ws in worksheets)
        {
            ws.Parent.Close();
        }
        return result;
    }

    public void Dispose()
    {
        _progressBar?.Dispose();
    }
}