prs/prs.desktop/Panels/Stock Forecast/StockForecastGrid.cs

using Comal.Classes;
using InABox.Clients;
using InABox.Core;
using InABox.DynamicGrid;
using InABox.Wpf;
using InABox.WPF;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Threading;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using PRSDimensionUtils;
using StockMovement = Comal.Classes.StockMovement;

namespace PRSDesktop;

public class StockForecastJobInfo
{
    
    public string JobNumber { get; set; }
    
    public double BOM { get; set; }

    public double Stock { get; set; }

    public double PO { get; set; }

    public double Required => Math.Max(BOM - (Stock + PO), 0.0F);
}
    

public class StockForecastItem : BaseObject
{
    public ProductLink Product => InitializeField(ref _product, nameof(Product));
    private ProductLink? _product;
    
    public ProductStyleLink Style => InitializeField(ref _style, nameof(Style));
    private ProductStyleLink? _style;

    public StockDimensions Dimensions => InitializeField(ref _dimensions, nameof(Dimensions));
    private StockDimensions? _dimensions;

    public bool IsProductInstance => ProductInstances.Count > 0;

    public List<ProductInstance> ProductInstances { get; set; } = new(0);

    public double Usage { get; set; }
    
    public int UsageDays { get; set; }
    
    public double UsageFactor { get; set; }
    
    public double StockFactor { get; set; }
    
    public bool CustomMinStock { get; set; }
    
    public double MinStock { get; set; }
    public double GenStock { get; set; }
    public double GenPO { get; set; }
    public double JobBOM { get; set; }
    public double JobStock { get; set; }
    public double JobPO { get; set; }

    public Dictionary<Guid, StockForecastJobInfo> JobInfo { get; private init; } = [];

    public void AddJobBOM(Guid jobID, double quantity)
    {
        var item = JobInfo.GetValueOrAdd(jobID);
        item.BOM += quantity;
    }
    public void AddJobPO(Guid jobID, double quantity)
    {
        var item = JobInfo.GetValueOrAdd(jobID);
        item.PO += quantity;
    }
    public void AddJobStock(Guid jobID, double quantity)
    {
        var item = JobInfo.GetValueOrAdd(jobID);
        item.Stock += quantity;
    }

    public double CalculatedMinimumStock => Math.Ceiling(CustomMinStock ? MinStock : Usage * StockFactor);
    public double StockRequired => Math.Max(CalculatedMinimumStock - (GenStock + GenPO), 0.0F);
    
    public double Required => Math.Max((CalculatedMinimumStock + JobBOM) - (GenStock + GenPO + JobStock + JobPO), 0.0F);
    
    public double Optimised => Math.Max(Math.Max(CalculatedMinimumStock, JobBOM) - (GenStock + GenPO + JobStock + JobPO), 0.0F);
}

public class StockForecastGrid : DynamicItemsListGrid<StockForecastItem>, IDataModelSource
{

    private enum ColumnTag
    {
        UsageDays,
        Usage,
        MinimumStockRequired,
        GeneralStockHoldings,
        GeneralPurchaseOrders,
        JobStockRequired,
        JobStockHoldings,
        JobPurchaseOrders,
        BalanceRequired
    }
    
    private SupplierProduct[]? _supplierProducts = null;
    
    private static readonly BitmapImage _warning = InABox.Wpf.Resources.warning.AsBitmapImage();
    private static readonly BitmapImage _tick = InABox.Wpf.Resources.tick.AsBitmapImage();
    private static readonly BitmapImage _cart = PRSDesktop.Resources.purchase.AsBitmapImage();
    private static readonly BitmapImage _product = PRSDesktop.Resources.product.AsBitmapImage();

    public Guid[] GroupIDs
    {
        get => _groupIDs;
        set
        {
            SelectedForOrder.RemoveWhere(x => !value.Contains(x.Item1));
            _groupIDs = value;
        }
    }

    public Guid[] JobIDs { get; set; } = [];
    public HashSet<Guid> SupplierIDs { get; set; } = [];

    private readonly Button OrderButton;
    private HashSet<(Guid,Guid,Guid,String)> SelectedForOrder = [];

    private DynamicGridCustomColumnsComponent<StockForecastItem> ColumnsComponent;
    private string ColumnsTag => "StockForecastGrid";

    public StockForecastProperties? Properties { get; set; }
    
    public StockForecastGlobalProperties? GlobalProperties { get; set; }

    public event EventHandler OnPropertiesChanged;
    
    public StockForecastGrid() : base()
    {
        DimensionUtils.ResetDimensionScriptCache();
        
        ColumnsComponent = new DynamicGridCustomColumnsComponent<StockForecastItem>(this, ColumnsTag);
        
        HiddenColumns.Add(x => x.Product.ID);
        HiddenColumns.Add(x => x.Product.Issues);
        HiddenColumns.Add(x => x.Style.ID);
        HiddenColumns.Add(x => x.Dimensions.UnitSize);
        HiddenColumns.Add(x => x.Product.Image.ID);
        HiddenColumns.Add(x => x.Product.Image.FileName);
        HiddenColumns.Add(x => x.Product.Supplier.ID);
        HiddenColumns.Add(x => x.Product.Supplier.SupplierLink.ID);
        HiddenColumns.Add(x => x.Product.OrderStrategy);
        HiddenColumns.Add(x => x.Product.Group.ID);

        HiddenColumns.Add(x => x.Required);
        HiddenColumns.Add(x => x.Optimised);
        
        HiddenColumns.Add(x=>x.CustomMinStock);
        HiddenColumns.Add(x => x.UsageDays);
        HiddenColumns.Add(x => x.MinStock);
        HiddenColumns.Add(x => x.GenStock);
        HiddenColumns.Add(x => x.GenPO);
        HiddenColumns.Add(x => x.JobBOM);
        HiddenColumns.Add(x => x.JobStock);
        HiddenColumns.Add(x => x.JobPO);
        
        ActionColumns.Add(new DynamicImageColumn(ProductInstance_Image, null)
        {
            Position = DynamicActionColumnPosition.Start,
            ToolTip = ProductInstance_ToolTip
        });
        ActionColumns.Add(new DynamicImageColumn(Issues_Image, null)
        {
            ToolTip = Issues_Tooltip,
            Position = DynamicActionColumnPosition.Start
        });
        ActionColumns.Add(new DynamicImagePreviewColumn<StockForecastItem>(x => x.Product.Image)
        {
            Position = DynamicActionColumnPosition.Start
        });
        
        CreateColumn(GetPeriod, ColumnTag.UsageDays, "Days", "")
            .ContextMenu = ProductInstance_Menu;
        
        CreateColumn(GetAverage, ColumnTag.Usage,"Avg.","F2")
            .ContextMenu = ProductInstance_Menu;
        
        CreateColumn(GetMinimumStockLevel, ColumnTag.MinimumStockRequired,"Min.","F2")
            .ContextMenu = ProductInstance_Menu;
        
        CreateColumn(GetGeneralStockLevel, ColumnTag.GeneralStockHoldings,"Hld.","F2");
        CreateColumn(GetGeneralPurchaseOrder, ColumnTag.GeneralPurchaseOrders, "PO.","F2");
        CreateColumn(GetBOMBalance, ColumnTag.JobStockRequired, "BOM.","F2");
        CreateColumn(GetReservedStock, ColumnTag.JobStockHoldings, "Hld.","F2");
        CreateColumn(GetReservedPurchaseOrder, ColumnTag.JobPurchaseOrders, "PO.","F2");
        CreateColumn(GetBalanceRequired, ColumnTag.BalanceRequired,"Req.","");

        ActionColumns.Add(new DynamicImageColumn(SelectForOrder_Image, SelectForOrder_Click)
        {
            Position = DynamicActionColumnPosition.End
        });
        
        OrderButton = AddButton("Order Stock", _cart, OrderStock_Click);
        OrderButton.IsEnabled = false;
    }

    private FrameworkElement? ProductInstance_ToolTip(DynamicActionColumn column, CoreRow? row)
    {
        if(row is null)
        {
            return column.TextToolTip("Does each line match a product instance?");
        }
        else if (LoadItem(row).IsProductInstance)
        {
            return column.TextToolTip("This line matches a product instance.");
        }
        else
        {
            return column.TextToolTip("This line does not match a product instance.");
        }
    }

    private BitmapImage? ProductInstance_Image(CoreRow? row)
    {
        if(row is null || LoadItem(row).IsProductInstance)
        {
            return _product;
        }
        else
        {
            return null;
        }
    }

    #region Columns

    protected override DynamicGridColumns LoadColumns()
    {
        return ColumnsComponent.LoadColumns();
    }

    protected override void LoadColumnsMenu(ContextMenu menu)
    {
        ColumnsComponent.LoadColumnsMenu(menu);
    }

    protected override void SaveColumns(DynamicGridColumns columns)
    {
        ColumnsComponent.SaveColumns(columns);
    }

    #endregion

    private BitmapImage? Issues_Image(CoreRow? row)
    {
        return (row is null)
            ? _warning
            : row.Get<StockForecastItem, string>(x => x.Product.Issues).IsNullOrWhiteSpace() 
                ? null 
                : _warning;
    }

    private FrameworkElement? Issues_Tooltip(DynamicActionColumn column, CoreRow? row)
    {
        return (row is null)
            ? null
            : column.TextToolTip(row.Get<StockForecastItem, string>(x => x.Product.Issues));
    }

    #region UIComponent

    private UIComponent? _uicomponent = null;
    private Guid[] _groupIDs = [];
    
    private class UIComponent : DynamicGridGridUIComponent<StockForecastItem>
    {
        private StockForecastGrid Grid;

        public UIComponent(StockForecastGrid grid)
        {
            Grid = grid;
            Parent = grid;
        }

        public bool CheckSuppliers(CoreRow row)
        {
            if (Grid._supplierProducts == null)
                return false;

            var item = row.ToObject<StockForecastItem>(); //Grid.LoadItem(row));
            return Grid._supplierProducts.Any(r =>
                    Equals(r.Product.ID, item.Product.ID)
                    && Equals(r.Style.ID, item.Style.ID)
                    //&& r.Dimensions.Unit.ID.Equals(item.Dimensions.Unit.ID)
                    && Grid.SupplierIDs.Contains(r.SupplierLink.ID));
        }

        protected override Brush? GetCellForeground(CoreRow row, DynamicColumnBase column)
        {
            if (column is DynamicTextColumn { Tag: ColumnTag _tag } _col)
            {
                if (_tag == ColumnTag.UsageDays || _tag == ColumnTag.Usage || _tag == ColumnTag.MinimumStockRequired)
                {
                    var item = Grid.LoadItem(row);
                    if (!item.IsProductInstance)
                        return Colors.Transparent.ToBrush();
                }
            }

            return base.GetCellForeground(row, column);
        }

        protected override Brush? GetCellBackground(CoreRow row, DynamicColumnBase column)
        {
            if (column is DynamicTextColumn col)
            {
                var item = Grid.LoadItem(row);
                
                var avg = item.UsageDays > 0
                    ? item.CalculatedMinimumStock.IsEffectivelyLessThan(item.Usage * item.StockFactor)
                        ? Colors.LightSalmon.ToBrush(0.5)
                        : Colors.LightGreen.ToBrush(0.5)
                    : Colors.Gainsboro.ToBrush(0.5);
                
                var min = item.CustomMinStock
                    ? Colors.Orchid.ToBrush(0.5)
                    : avg;

                var stock = (item.GenStock + item.GenPO).IsEffectivelyLessThan(item.CalculatedMinimumStock)
                    ? Colors.LightSalmon.ToBrush(0.5)
                    : Colors.LightBlue.ToBrush(0.5);
                
                var job = Math.Max(0.0F, item.JobBOM - (item.JobStock + item.JobPO)).IsEffectivelyEqual(0.0F)
                    ? Colors.LightGreen.ToBrush(0.5)
                    : Colors.LightSalmon.ToBrush(0.5);
                
                var overall = !(Grid.Properties?.Optimise == true ? item.Optimised : item.Required).IsEffectivelyEqual(0.0F)
                    ? Colors.LightSalmon.ToBrush(0.5)
                    : null;
                
                return col.Tag switch
                {
                    ColumnTag.UsageDays => avg,
                    ColumnTag.Usage => avg,
                    ColumnTag.MinimumStockRequired => min,
                    ColumnTag.GeneralStockHoldings => stock,
                    ColumnTag.GeneralPurchaseOrders => stock,
                    ColumnTag.JobStockRequired => job,
                    ColumnTag.JobStockHoldings => job,
                    ColumnTag.JobPurchaseOrders => job,
                    ColumnTag.BalanceRequired => overall,
                    _ => null
                };
            }
            else
            {
                if (Grid.Properties?.AllStock == true && !CheckSuppliers(row))
                    return Colors.Silver.ToBrush(0.5);
            }
            
            return null;
        }


    }
    protected override IDynamicGridUIComponent<StockForecastItem> CreateUIComponent()
    {
        return _uicomponent ??= new UIComponent(this);
    }

    #endregion

    protected override void DoReconfigure(DynamicGridOptions options)
    {
        base.DoReconfigure(options);

        options.Clear();
        options.RecordCount = true;
        options.SelectColumns = true;
        options.FilterRows = true;
        options.ExportData = true;
        options.MultiSelect = true;
        options.HideDatabaseFilters = true;
    }

    protected override void ConfigureColumnGroups()
    {
        base.ConfigureColumnGroups();

        AddColumnGrouping()
            .AddGroup("Usage", GetColumn(ColumnTag.UsageDays), GetColumn(ColumnTag.Usage))
            .AddGroup("General Stock", GetColumn(ColumnTag.MinimumStockRequired), GetColumn(ColumnTag.GeneralPurchaseOrders))
            .AddGroup("Job Stock", GetColumn(ColumnTag.JobStockRequired), GetColumn(ColumnTag.JobPurchaseOrders));
    }

    public override DynamicGridColumns GenerateColumns()
    {
        var columns = new DynamicGridColumns();
        columns.Add<StockForecastItem>(x => x.Product.Code, 120, "Product Code", "", Alignment.MiddleCenter);
        columns.Add<StockForecastItem>(x => x.Product.Name, 0, "Product Name", "", Alignment.MiddleLeft);
        columns.Add<StockForecastItem>(x => x.Style.Code, 120, "Style Code", "", Alignment.MiddleCenter);
        columns.Add<StockForecastItem>(x => x.Dimensions.UnitSize, 120, "Unit Size", "", Alignment.MiddleCenter);
        return columns;
    }

    #region Column Data and Details

    private DynamicTextColumn CreateColumn(DynamicTextColumn.GetTextDelegate calculate, ColumnTag tag, string header, string format)
    {
        var column = new DynamicTextColumn(calculate)
        {
            Width = 65, 
            Format=format, 
            Position = DynamicActionColumnPosition.End, 
            Tag = tag,
            HeaderText = header,
            GetFilter = () => new DynamicColumnFilter<double?>(
                    r => GetColumnCalculatedData(tag, r),
                    filter => GetColumnFilterItems(filter, tag))
        };
        ActionColumns.Add(column);
        return column;
    }

    private DynamicTextColumn GetColumn(ColumnTag tag) => (ActionColumns.First(x => Equals(x.Tag, tag)) as DynamicTextColumn)!;

    private object GetAverage(CoreRow? row)
    {
        if (row is not null)
        {
            var item = LoadItem(row);
            if (item.IsProductInstance)
                return item.Usage;
        }
        return 0.0;
    }

    private object GetPeriod(CoreRow? row)
    {
        if (row is not null)
        {
            var item = LoadItem(row);
            if (item.IsProductInstance)
                return item.UsageDays;
        }
        return 0;
    }

    private object GetMinimumStockLevel(CoreRow? row)
    {
        if (row is not null)
        {
            var item = LoadItem(row);
            if (item.IsProductInstance)
                return item.CalculatedMinimumStock;
        }
        return 0.0;
    }

    private object GetGeneralStockLevel(CoreRow? row)
        => row is not null ? LoadItem(row).GenStock : 0.0;

    private object GetGeneralPurchaseOrder(CoreRow? row)
        => row is not null ? LoadItem(row).GenPO : 0.0;
    
    private object GetBOMBalance(CoreRow? row)
        => row is not null ? LoadItem(row).JobBOM : 0.0;

    private object GetReservedStock(CoreRow? row)
        => row is not null ? LoadItem(row).JobStock : 0.0;

    private object GetReservedPurchaseOrder(CoreRow? row)
        => row is not null ? LoadItem(row).JobPO : 0.0;

    private object GetBalanceRequired(CoreRow? row)
    {
        if(row is not null)
        {
            var item = LoadItem(row);
            var result = Properties?.Optimise == true
                ? (item.Optimised.IsEffectivelyEqual(0.0) ? "" : $"{item.Optimised:F2}")
                : (item.Required.IsEffectivelyEqual(0.0) ? "" : $"{item.Required:F2}");
            return result;

        }
        else
        {
            return "";
        }
    }
    
    private void ShowDetailGrid(String title, params Func<IDynamicDataGrid?>[] gridfuncs)
    {
        var _window = new ThemableWindow { Title = title };
        var _tabcontrol = new DynamicTabControl() { TabStripPlacement = Dock.Bottom, Margin = new Thickness(5) };
        _window.Content = _tabcontrol;
        foreach (var gridfunc in gridfuncs)
        {
            var _grid = gridfunc();
            if (_grid != null)
            {
                _tabcontrol.Items.Add(
                    new DynamicTabItem()
                    {
                        Header = CoreUtils.Neatify(_grid.DataType.Name.Split('.').Last()),
                        Content = _grid
                    }
                );
                _grid.Refresh(true,true);
            }
        }
        _window.ShowDialog();
    }
    
    private IDynamicDataGrid BuildDetailGrid<TEntity>(
        String tag, 
        Expression<Func<TEntity,object?>> productcol, 
        Guid productid, 
        Expression<Func<TEntity,object?>> stylecol, 
        Guid? styleid, 
        Expression<Func<TEntity, IDimensions>> dimcol, 
        IDimensions? dimensions,
        Expression<Func<TEntity,object?>>? jobcol,
        Filter<TEntity>? extrafilter,
        Func<CoreRow,bool>? rowfilter
    )
    {
        var _grid = (Activator.CreateInstance(typeof(DynamicDataGrid<>).MakeGenericType(typeof(TEntity))) as IDynamicDataGrid);
        if (_grid == null)
        {
            MessageWindow.ShowError($"Cannot create Grid for [{typeof(TEntity).Name}]", "", shouldLog: false);
            return null;
        }
        _grid.ColumnsTag = $"{ColumnsTag}.{tag}";
        _grid.Reconfigure(options =>
        {
            options.Clear();
            options.FilterRows = true;
            options.SelectColumns = true;
        });
        _grid.OnDefineFilter += t =>
        {
            var _filter = new Filter<TEntity>(productcol).IsEqualTo(productid);
            if(dimensions is not null)
                _filter = _filter.And(CoreUtils.GetFullPropertyName(dimcol, ".")).DimensionEquals(dimensions);
            
            if (styleid.HasValue)
                _filter = _filter.And(stylecol).IsEqualTo(styleid);

            if (jobcol != null)
                _filter = _filter.And(new Filter<TEntity>(jobcol).InList(JobIDs));
                
            if (extrafilter != null)
                _filter = _filter.And(extrafilter);
            
            return _filter;
        };
        _grid.OnFilterRecord += row => rowfilter?.Invoke(row) ?? true;
        return _grid;
    }

    protected override void DoDoubleClick(object sender, DynamicGridCellClickEventArgs args)
    {
        //base.DoDoubleClick(sender, args);
        if (args.Row is null || args.Column?.Tag is not ColumnTag tag) return;

        var item = LoadItem(args.Row);

        var styleid = HasStyle() ? item.Style.ID : (Guid?)null;

        switch (tag)
        {
            case ColumnTag.Usage :
                
                ShowDetailGrid(
                    "Stock Usage", 
                    () => BuildDetailGrid<StockMovement>(
                        ColumnTag.Usage.ToString(), 
                        x => x.Product.ID, 
                        item.Product.ID, 
                        x => x.Style.ID, 
                        styleid, 
                        x => x.Dimensions,
                        item.Dimensions,
                        null,
                        new Filter<StockMovement>(x=>x.Job.ID).IsEqualTo(Guid.Empty)
                            .And(x=>x.Date).IsGreaterThanOrEqualTo(DateTime.Today.AddDays(0-item.UsageDays * item.UsageFactor))
                            .And(x=>x.Type).InList(
                                StockMovementType.Issue, 
                                StockMovementType.TransferOut, 
                                StockMovementType.TransferIn, 
                                StockMovementType.StockTake),
                        null
                    )
                );               

                break;
            case ColumnTag.GeneralStockHoldings:
                ShowDetailGrid(
                    "Stock Holdings", 
                    () => BuildDetailGrid<StockHolding>(
                        ColumnTag.GeneralStockHoldings.ToString(), 
                        x => x.Product.ID, 
                     item.Product.ID, 
                        x => x.Style.ID, 
                        styleid, 
                        x => x.Dimensions,
                        item.Dimensions,
                        null,
                        new Filter<StockHolding>(x=>x.Job.ID).IsEqualTo(Guid.Empty),
                        null
                    )
                );               
                break;
            case ColumnTag.GeneralPurchaseOrders:
                //ShowDetailGrid(
                //    "Purchase Order Allocations", 
                //    () => BuildDetailGrid<PurchaseOrderItemAllocation>(
                //        ColumnTag.GeneralPurchaseOrders.ToString(), 
                //        x => x.Item.Product.ID, 
                //        item.Product.ID, 
                //        x => x.Item.Style.ID, 
                //        styleid, 
                //        x => x.Item.Dimensions,
                //        item.Dimensions,
                //        null,
                //        new Filter<PurchaseOrderItemAllocation>(x=>x.Job.ID).IsEqualTo(Guid.Empty)
                //            .And(x=>x.Item.ReceivedDate).IsEqualTo(DateTime.MinValue),
                //        null
                //    )
                //);
                break;
            case ColumnTag.JobStockRequired:
                ShowDetailGrid(
                    "Bills Of Materials", 
                    () => BuildDetailGrid<JobBillOfMaterialsItem>(
                        ColumnTag.JobStockRequired.ToString(), 
                        x => x.Product.ID, 
                        item.Product.ID, 
                        x => x.Style.ID, 
                        styleid, 
                        x => x.Dimensions,
                        item.Dimensions,
                        x => x.Job.ID,
                        new Filter<JobBillOfMaterialsItem>(x=>x.BillOfMaterials.Approved).IsNotEqualTo(DateTime.MinValue),
                        null
                    ),
                    () => BuildDetailGrid<StockMovement>(
                        "JobStockIssued", 
                        x => x.Product.ID, 
                        item.Product.ID, 
                        x => x.Style.ID, 
                        styleid, 
                        x => x.Dimensions,
                        item.Dimensions,
                        x => x.Job.ID,
                        new Filter<StockMovement>(x=>x.Type).IsEqualTo(StockMovementType.Issue),
                        null
                    )
                    
                );
                break;
            case ColumnTag.JobStockHoldings:
                ShowDetailGrid(
                    "Stock Holdings", 
                    () => BuildDetailGrid<StockHolding>(
                        ColumnTag.JobStockHoldings.ToString(), 
                        x => x.Product.ID, 
                        item.Product.ID, 
                        x => x.Style.ID, 
                        styleid, 
                        x => x.Dimensions,
                        item.Dimensions,
                        x => x.Job.ID,
                        null,
                        null
                    )
                );
                break;
            case ColumnTag.JobPurchaseOrders:
                //ShowDetailGrid(
                //    "Purchase Orders", 
                //    () => BuildDetailGrid<PurchaseOrderItemAllocation>(
                //        ColumnTag.GeneralPurchaseOrders.ToString(), 
                //        x => x.Item.Product.ID, 
                //        item.Product.ID, 
                //        x => x.Item.Style.ID, 
                //        styleid, 
                //        x => x.Item.Dimensions,
                //        item.Dimensions,
                //        x => x.Job.ID,
                //        new Filter<PurchaseOrderItemAllocation>(x => x.Item.ReceivedDate).IsEqualTo(DateTime.MinValue),
                //        null
                //    )
                //);                  
                break;
        }
    }
    
    private ContextMenu? ProductInstance_Menu(CoreRow[]? rows)
    {
        if (rows is null || rows.Length == 0 || !Security.CanEdit<ProductInstance>()) 
            return null;
        
        var _items = LoadItems(rows);
        if (_items.Any(x => x.IsProductInstance))
        {
            var _menu = new ContextMenu();
            _menu.AddItem("Edit ProductInstance", null, _items, EditProductInstance_Click);
            return _menu;
        }
        
        return null;
    }

    private DynamicDataGrid<ProductInstance>? _instanceGrid;
    
    private void EditProductInstance_Click(StockForecastItem[] sfItems)
    {
        var items = sfItems.SelectMany(x=>x.ProductInstances).ToArray();
        _instanceGrid ??= new DynamicDataGrid<ProductInstance>();
        if (_instanceGrid.EditItems(items))
        {
            Client.Save(items.Where(x=>x.IsChanged()),"Updated from Stock Forecast Screen");
            Refresh(false,true);
        }
    }
    
    #endregion

    #region Refresh

    private bool HasStyle()
    {
        return DataColumns().ColumnNames().Any(x => x.StartsWith("Style.") && !x.Equals("Style.ID"));
    }

    private CoreRow[] GetRows<TSource>(CoreTable table, Guid productid, Guid? styleid, IDimensions dimensions, Guid[] jobids) where TSource : IJobMaterial
    {
        int productcol = table.GetColumnIndex<TSource>(x => x.Product.ID);
        int stylecol = styleid.HasValue ? table.GetColumnIndex<TSource>(x => x.Style.ID) : -1;
        var dimCols = Dimensions.GetFilterColumnIndices<TSource>(table, x => x.Dimensions);
        int jobcol = table.GetColumnIndex<TSource>(x => x.Job.ID);
        
        var subset = table.Rows
            .Where(r =>
                Guid.Equals(r.Values[productcol], productid)
                && (!styleid.HasValue || Guid.Equals(r.Values[stylecol], styleid))
                && r.ToDimensions<StockDimensions>(dimCols).Equals(dimensions)
                && jobids.Any(x=>Equals(x,r.Values[jobcol]))
            );
        
        return subset.ToArray();
    }

    private double Aggregate<TSource>(CoreTable table, IEnumerable<CoreRow> rows, bool hasstyle, bool hasjob, Expression<Func<TSource, object>> source, CoreRow? target = null, Expression<Func<ProductInstance, object>>? aggregate = null)
    {
        int srcol = table.GetColumnIndex(source);
        
        if (srcol == -1)
            return 0.00;

        var total =  rows.Aggregate(0d, (value, row) => value + (double)(row.Values[srcol] ?? 0.0d));
        
        // int productcol = columns.IndexOf(x => x.Product.ID);
        // int stylecol = hasstyle ? columns.IndexOf(x => x.Style.ID) : -1;
        // int jobcol = hasjob ? columns.IndexOf(x => x.Job.ID) : -1;
        // int unitcol = columns.IndexOf(x => x.Dimensions.UnitSize);
        //
        // var tuples = rows.Select(r => new Tuple<Guid, Guid?, Guid?, String, double>(
        //     (Guid)(r.Values[productcol] ?? Guid.Empty),
        //     (hasstyle ? (Guid)(r.Values[stylecol] ?? Guid.Empty) : null),
        //     (hasjob ? (Guid)(r.Values[jobcol] ?? Guid.Empty) : null),
        //     (String)(r.Values[unitcol] ?? ""),
        //     (double)(r.Values[aggcol] ?? 0.0d))
        // ).ToArray();
        //
        // var total = tuples.Aggregate(0d, (value, tuple) => value + tuple.Item5);

        if(aggregate is not null)
        {
            target?.Set(aggregate, total);
        }
        return total;
    }

    private double? GetColumnCalculatedData(ColumnTag tag, CoreRow row)
    {
        return tag switch
        {
            ColumnTag.Usage => row.Get<StockForecastItem, double>(x => x.Usage),
            ColumnTag.UsageDays => row.Get<StockForecastItem,int>(x => x.UsageDays),
            ColumnTag.MinimumStockRequired => row.Get<StockForecastItem, double>(x => x.CalculatedMinimumStock),
            ColumnTag.GeneralStockHoldings => row.Get<StockForecastItem, double>(x => x.GenStock),
            ColumnTag.GeneralPurchaseOrders => row.Get<StockForecastItem, double>(x => x.GenPO),
            ColumnTag.JobStockRequired => row.Get<StockForecastItem, double>(x => x.JobBOM),
            ColumnTag.JobStockHoldings => row.Get<StockForecastItem, double>(x => x.JobStock),
            ColumnTag.JobPurchaseOrders => row.Get<StockForecastItem, double>(x => x.JobPO),
            ColumnTag.BalanceRequired => (Properties?.Optimise == true ? row.Get<StockForecastItem, double>(x => x.Optimised) : row.Get<StockForecastItem, double>(x => x.Required)),
            _ => null
        };
    }

    private IEnumerable<Tuple<string, double?>> GetColumnFilterItems(DynamicColumnFilter<double?> filter, ColumnTag tag)
    {
        var items = new HashSet<double>();
        foreach(var row in (filter as IDynamicGridColumnFilter).GetRowsToFilter(this))
        {
            var value = GetColumnCalculatedData(tag, row);
            if (value.HasValue && !items.Contains(value.Value))
            {
                items.Add(value.Value);
            }
        }
        var arr = items.ToArray();
        Array.Sort(arr);
        return arr.Select(x => new Tuple<string, double?>(x.ToString("F2"), x));
    }

    private class ItemKey(Guid productID, Guid styleID, StockDimensions dimensions)
    {
        public Guid ProductID { get; set; } = productID;

        public Guid StyleID { get; set; } = styleID;

        public StockDimensions Dimensions { get; set; } = dimensions;

        public override bool Equals(object? obj)
        {
            return obj is ItemKey other
                && ProductID == other.ProductID
                && StyleID == other.StyleID
                && Dimensions.Equals(other.Dimensions);
        }

        public override int GetHashCode()
        {
            return HashCode.Combine(ProductID, StyleID, Dimensions);
        }
    }

    protected override void Reload(Filters<StockForecastItem> criteria, Columns<StockForecastItem> columns, ref SortOrder<StockForecastItem>? sort, CancellationToken token, Action<CoreTable?, Exception?> action)
    {
        // Need to query ProductInstances, StockHoldings, Job BOM and PO.
        KeyedQueryDef<T> GetQuery<T>(Filter<T>? filter = null, Columns<T>? columns = null) where T : Entity, IJobMaterial, IRemotable, IPersistent, new()
        {
            return new KeyedQueryDef<T>(
                Filter<T>.And(
                    new Filter<T>(x => x.Product.Group.ID).InList(GroupIDs)
                        .And(new Filter<T>(x => x.Job.ID).InList(JobIDs)
                            .Or(x => x.Job.ID).IsEqualTo(Guid.Empty)),
                    filter),
                Columns.None<T>()
                    .Add(x => x.Product.ID)
                    .Add(x => x.Product.Group.ID)
                    .Add(x => x.Job.ID)
                    .Add(x => x.Style.ID)
                    .AddDimensionsColumns(x => x.Dimensions, Dimensions.ColumnsType.Data)
                    .Add(x => x.Dimensions.UnitSize)
                    .Add(x => x.Dimensions.Value)
                    .AddRange(columns ?? Enumerable.Empty<Column<T>>()));
        }

        ItemKey GetKeyFromRow(CoreRow row)
        {
            var key = new ItemKey(
                row.Get<IJobMaterial, Guid>(x => x.Product.ID),
                row.Get<IJobMaterial, Guid>(x => x.Style.ID),
                row.ToDimensions<IJobMaterial, StockDimensions>(x => x.Dimensions));
            key.Dimensions.UnitSize = row.Get<IJobMaterial, string>(x => x.Dimensions.UnitSize);
            key.Dimensions.Value = row.Get<IJobMaterial, double>(x => x.Dimensions.Value);
            return key;
        }

        ItemKey GetKey(Guid productid, Guid styleid, StockDimensions dimensions)
        {
            var key = new ItemKey(
                productid,
                styleid,
                dimensions);
            return key;
        }

        var queries = new IKeyedQueryDef[]
        {
            
            new KeyedQueryDef<ProductInstance>(
                new Filter<ProductInstance>(x => x.Product.Group.ID).InList(GroupIDs),
                //new Filter<ProductInstance>(x=>x.MinimumStockLevel).IsNotEqualTo(0.0).And(x => x.Product.Group.ID).InList(GroupIDs),
                Columns.None<ProductInstance>()
                    .Add(x => x.ID)
                    .Add(x => x.Product.ID)
                    .Add(x => x.Style.ID)
                    .AddDimensionsColumns(x => x.Dimensions, Dimensions.ColumnsType.Data)
                    .Add(x => x.Dimensions.UnitSize)
                    .Add(x => x.Dimensions.Value)
                    .Add(x => x.MinimumStock.CustomUsage)
                    .Add(x=>x.MinimumStock.UsageDays)
                    .Add(x=>x.MinimumStock.UsageFactor)
                    .Add(x=>x.MinimumStock.StockFactor)
                    .Add(x=>x.MinimumStock.CustomStockLevel)
                    .Add(x=>x.MinimumStock.StockLevel)
            ),
            
            GetQuery<StockHolding>(
                columns: Columns.None<StockHolding>()
                    .Add(x => x.Units)
            ),
            
            new KeyedQueryDef<PurchaseOrderItem>(
                new Filter<PurchaseOrderItem>(x => x.ReceivedDate).IsEqualTo(DateTime.MinValue)
                    .And(x => x.Product.Group.ID).InList(GroupIDs),
                Columns.None<PurchaseOrderItem>()
                    .Add(x => x.ID)
                    .Add(x => x.Qty)
                    .Add(x => x.Product.ID)
                    .Add(x => x.Style.ID)
                    .AddDimensionsColumns(x => x.Dimensions, Dimensions.ColumnsType.Data)
                    .Add(x => x.Dimensions.UnitSize)
                    .Add(x => x.Dimensions.Value)
            ),
            
            new KeyedQueryDef<PurchaseOrderItemAllocation>(
                new Filter<PurchaseOrderItemAllocation>(x => x.Item.ReceivedDate).IsEqualTo(DateTime.MinValue)
                    .And(x => x.Item.Product.Group.ID).InList(GroupIDs)
                    .And(new Filter<PurchaseOrderItemAllocation>(x => x.Job.ID).InList(JobIDs)
                        .Or(x => x.Job.ID).IsEqualTo(Guid.Empty)),
                Columns.None<PurchaseOrderItemAllocation>()
                    .Add(x => x.Quantity)
                    .Add(x => x.Job.ID)
                    .Add(x => x.Item.ID)
                    .Add(x => x.Item.Product.ID)
                    .Add(x => x.Item.Style.ID)
                    .AddDimensionsColumns(x => x.Item.Dimensions, Dimensions.ColumnsType.Data)
                    .Add(x => x.Item.Dimensions.UnitSize)
                    .Add(x => x.Item.Dimensions.Value)
            ),
            
            GetQuery<JobBillOfMaterialsItem>(
                filter: new Filter<JobBillOfMaterialsItem>(x=>x.BillOfMaterials.Approved).IsNotEqualTo(DateTime.MinValue),
                columns: Columns.None<JobBillOfMaterialsItem>().Add(x => x.Quantity)
            ),
            
            GetQuery<StockMovement>(
                filter: new Filter<StockMovement>(x => x.Type).InList(
                    StockMovementType.Issue, 
                    StockMovementType.TransferOut, 
                    StockMovementType.TransferIn, 
                    StockMovementType.StockTake),
                columns: Columns.None<StockMovement>()
                    .Add(x=>x.Date)
                    .Add(x => x.Units)
            ),
            
            GetQuery<SupplierProduct>(
                columns: Columns.None<SupplierProduct>()
                    .Add(x => x.SupplierLink.ID)
                    .Add(x => x.CostPrice)
                    .Add(x => x.Discount)
                    .Add(x => x.TaxCode.ID)
            )
        };

        Client.QueryMultiple(
            (QueryMultipleResults? results, Exception? error) =>
            {
                if (results is null)
                {
                    action(null, error);
                    return;
                }

                var items = new Dictionary<ItemKey, StockForecastItem>();

                StockForecastItem GetItem(ItemKey key)
                {
                    if(!items.TryGetValue(key, out var item))
                    {
                        item = new StockForecastItem();
                        item.Product.ID = key.ProductID;
                        item.Style.ID = key.StyleID;
                        item.Dimensions.CopyFrom(key.Dimensions);
                        items[key] = item;
                    }
                    return item;
                }
                
                
                var productInstances = results.GetArray<ProductInstance>();
                foreach(var instance in productInstances)
                {
                    var dimensions = new StockDimensions();
                    dimensions.CopyFrom(instance.Dimensions);
                    var minStock = instance.MinimumStock.CustomStockLevel ? instance.MinimumStock.StockLevel : 0.0;
                    var minCost = instance.AverageCost;
                    DimensionUtils.ConvertDimensions(dimensions, ref minStock, ref minCost, Client<ProductDimensionUnit>.Provider);
                    var item = GetItem(new(instance.Product.ID, instance.Style.ID, dimensions));
                    item.UsageDays = Math.Max(item.UsageDays,instance.MinimumStock.CustomUsage ? instance.MinimumStock.UsageDays : (GlobalProperties?.UsageDays ?? 0));
                    item.UsageFactor = Math.Max(item.UsageFactor, instance.MinimumStock.CustomUsage ? instance.MinimumStock.UsageFactor :  (GlobalProperties?.UsageFactor ?? 0));
                    item.StockFactor = Math.Max(item.StockFactor, instance.MinimumStock.CustomUsage ? instance.MinimumStock.StockFactor :  (GlobalProperties?.StockFactor ?? 0));
                    item.CustomMinStock = item.CustomMinStock || instance.MinimumStock.CustomStockLevel;
                    item.MinStock += minStock;
                    item.ProductInstances.Add(instance);
                }

                var holdings = results.Get<StockHolding>();
                foreach(var holdingrow in holdings.Rows)
                {
                    var holding = holdingrow.ToObject<StockHolding>();
                    holding.ConvertDimensions(x => x.Dimensions, x => x.Units, x => x.AverageValue,
                        Client<ProductDimensionUnit>.Provider);

                    var item = GetItem(GetKey(holding.Product.ID, holding.Style.ID, holding.Dimensions));
                    if(holding.Job.ID == Guid.Empty)
                    {
                        item.GenStock += holding.Units;
                    }
                    else
                    {
                        item.JobStock += holding.Units;
                        item.AddJobStock(holding.Job.ID, holding.Units);
                    }
                }

                var purchaseOrderItems = results.GetObjects<PurchaseOrderItem>()
                    .ToDictionary(x => x.ID);
                foreach (var poi in purchaseOrderItems.Values)
                {
                    poi.ConvertDimensions(x => x.Dimensions, x => x.Qty, x => x.Cost,
                        Client<ProductDimensionUnit>.Provider);
                }
                
                var allocations = results.Get<PurchaseOrderItemAllocation>();
                foreach(var allocationrow in allocations.Rows)
                {
                    var allocation  = allocationrow.ToObject<PurchaseOrderItemAllocation>();

                    // POIAs are already converted where necessary, so we don't have to update the quantities again, just update the dimensions
                    DimensionUtils.ConvertDimensions(allocation.Item.Dimensions, allocation.Quantity, allocation.Item.Cost, Client<ProductDimensionUnit>.Provider);
                    
                    var key = new ItemKey(
                        allocation.Item.Product.ID,
                        allocation.Item.Style.ID,
                        allocation.Item.Dimensions);
                    var item = GetItem(key);

                    if(allocation.Job.ID == Guid.Empty)
                    {
                        item.GenPO += allocation.Quantity;
                    }
                    else
                    {
                        item.JobPO += allocation.Quantity;
                        item.AddJobPO(allocation.Job.ID, allocation.Quantity);
                    }

                    if(purchaseOrderItems.TryGetValue(allocation.Item.ID, out var poi))
                        poi.Qty -= allocation.Quantity;
                }

                foreach(var poi in purchaseOrderItems.Values)
                {
                    var key = new ItemKey(
                        poi.Product.ID,
                        poi.Style.ID,
                        poi.Dimensions);
                    var item = GetItem(key);

                    if(poi.Job.ID == Guid.Empty)
                    {
                        item.GenPO += poi.Qty;
                    }
                    else
                    {
                        item.JobPO += poi.Qty;
                        item.AddJobPO(poi.Job.ID, poi.Qty);
                    }
                }

                var jobBOMItems = results.Get<JobBillOfMaterialsItem>();
                foreach(var bomItemRow in jobBOMItems.Rows)
                {
                    var bomItem = bomItemRow.ToObject<JobBillOfMaterialsItem>();
                    bomItem.ConvertDimensions(x => x.Dimensions, x => x.Quantity, x => x.UnitCost, Client<ProductDimensionUnit>.Provider);
                    var key = GetKey(bomItem.Product.ID, bomItem.Style.ID, bomItem.Dimensions);
                    var item = GetItem(key);

                    item.AddJobBOM(bomItem.Job.ID, bomItem.Quantity);
                }

                var movements = results.Get<StockMovement>();
                foreach(var mvtrow in movements.Rows)
                {
                    var movement = mvtrow.ToObject<StockMovement>();
                    movement.ConvertDimensions(x => x.Dimensions, x => x.Units, x => x.Cost,
                        Client<ProductDimensionUnit>.Provider);

                    var item = GetItem(GetKey(movement.Product.ID, movement.Style.ID, movement.Dimensions));
                    if (movement.Job.ID == Guid.Empty)
                    {
                        var _days = item.UsageDays * item.UsageFactor;
                        if (!_days.IsEffectivelyEqual(0.0) && DateTime.Today.Subtract(movement.Date) <= TimeSpan.FromDays(_days))
                            item.Usage += (0.0 - movement.Units) / item.UsageFactor;
                    }
                    else
                        item.AddJobBOM(movement.Job.ID, movement.Units);
                }

                _supplierProducts = results.GetArray<SupplierProduct>();

                Items = items.Values.ToList();

                foreach(var item in Items)
                {
                    foreach(var (job, info) in item.JobInfo)
                    {
                        info.BOM = Math.Max(info.BOM, 0.0);
                    }
                    item.JobBOM = item.JobInfo.Sum(x => x.Value.BOM);
                }

                Items.LoadForeignProperties(columns);
                Items.Sort((a, b) => a.Product.Code.CompareTo(b.Product.Code));

                var result = new CoreTable();
                result.LoadColumns(columns);
                result.LoadRows(Items);
                action(result, null);
            }, queries);
    }

    protected override bool FilterRecord(CoreRow row)
    {
        bool result = base.FilterRecord(row);

        if (Properties?.RequiredOnly == true)
        {
            result = result && Properties?.Optimise == true
                ? !row.Get<StockForecastItem, double>(x => x.Optimised).IsEffectivelyEqual(0)
                : !row.Get<StockForecastItem, double>(x => x.Required).IsEffectivelyEqual(0);
        }

        if (Properties?.AllStock != true)
            result = result && _uicomponent?.CheckSuppliers(row) == true;

        return result;
    }

    #endregion

    #region Ordering

    private IEnumerable<CoreRow> FilterRows(IEnumerable<CoreRow> rows)
    {
        var predicates = GetFilterPredicates();
        return rows.Where(r =>
        {
            return predicates.All(x => x.Item2(r));
        });
    }

    private bool SelectForOrder_Click(CoreRow? row)
    {
        if (row is null)
        {
            var menu = new ContextMenu();
            menu.AddItem("Select all", null, () =>
            {
                foreach (var row in FilterRows(Data.Rows))
                {
                    SelectedForOrder.Add(
                        RowToValueTuple(row)
                    );
                    InvalidateRow(row);
                }
                OrderButton.IsEnabled = SelectedForOrder.Count > 0;
            });
            menu.AddItem("Deselect all", null, () =>
            {
                SelectedForOrder.Clear();
                InvalidateGrid();
                OrderButton.IsEnabled = false;
            });
            menu.IsOpen = true;
            return false;
        }
        else
        {
            var vt = RowToValueTuple(row);
            if (!SelectedForOrder.Remove(vt))
            {
                SelectedForOrder.Add(vt);
            }
            OrderButton.IsEnabled = SelectedForOrder.Count > 0;
            InvalidateRow(row);
            return false;
        }
    }

    private static (Guid,Guid, Guid, string) RowToValueTuple(CoreRow row)
    {
        return (
            row.Get<StockForecastItem,Guid>(x=>x.Product.Group.ID),
            row.Get<StockForecastItem,Guid>(x=>x.Product.ID),
            row.Get<StockForecastItem,Guid>(x=>x.Style.ID),
            row.Get<StockForecastItem,String>(x=>x.Dimensions.UnitSize)
        );
    }

    private BitmapImage? SelectForOrder_Image(CoreRow? row)
    {
        if(row is null)
        {
            return _cart;
        }
        else if(SelectedForOrder.Contains(RowToValueTuple(row)))
        {
            return _cart;
        }
        else
        {
            return null;
        }
    }
    
    private bool OrderStock_Click(Button button, CoreRow[] rows)
    {
        rows = FilterRows(Data.Rows.Where(x => SelectedForOrder.Contains(RowToValueTuple(x)))).ToArray();
        if(rows.Length == 0)
        {
            return false;
        }

        var items = new List<StockForecastOrderData>();
        foreach(var forecastItem in LoadItems(rows))
        {
            var item = new StockForecastOrderData(forecastItem.Product, forecastItem.Style, forecastItem.Dimensions);
            item.RequiredQuantity = Properties?.Optimise == true ? forecastItem.Optimised : forecastItem.Required;
            if(forecastItem.StockRequired > 0)
            {
                item.SetRequiredQuantity(Guid.Empty, Guid.Empty, "", forecastItem.StockRequired);
            }
            
            foreach(var (id, jobInfo) in forecastItem.JobInfo)
            {
                if (jobInfo.Required > 0)
                    item.SetRequiredQuantity(id, Guid.Empty, jobInfo.JobNumber, jobInfo.Required);
            }

            items.Add(item);
        }

        var window = new StockForecastOrderScreen(items) { Owner = App.Current.MainWindow };
        if(window.ShowDialog() != true)
        {
            return false;
        }
        window.CreateOrders("Stock Forecast");
        
        SelectedForOrder.Clear();
        OrderButton.IsEnabled = false;
        return true;
    }

    #endregion

    #region IDataModelSource

    public event DataModelUpdateEvent? OnUpdateDataModel;

    public string SectionName => "Stock Forecast";
    
    public DataModel DataModel(Selection selection)
    {
        return new AutoDataModel<ProductInstance>(null);
    }
    
    #endregion
}