prs/prs.desktop/Panels/Invoices/InvoiceAssignmentGrid.cs

using System;
using System.Linq;
using System.Threading;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Media.Imaging;
using Comal.Classes;
using InABox.Clients;
using InABox.Core;
using InABox.DynamicGrid;
using InABox.WPF;

namespace PRSDesktop
{
    public class InvoiceAssignmentGrid : DynamicDataGrid<Assignment>
    {
        private readonly BitmapImage chargeable_excluded = PRSDesktop.Resources.tick.Fade(0.8F).AsGrayScale().AsBitmapImage();
        private readonly BitmapImage chargeable_included = PRSDesktop.Resources.tick.AsBitmapImage();
        
        private readonly BitmapImage unchargeable_excluded = PRSDesktop.Resources.warning.Fade(0.8F).AsGrayScale().AsBitmapImage();
        private readonly BitmapImage unchargeable_included = PRSDesktop.Resources.warning.AsBitmapImage();
        
        private readonly Button IncludeButton;
        
        private readonly Button ShowAllButton;
        private bool _showall = false;

        public InvoiceAssignmentGrid()
        {
            
            ColumnsTag = "InvoiceTimeSheets";

            HiddenColumns.Add(x => x.Invoice.ID);
            HiddenColumns.Add(x => x.Invoice.Deleted);
            HiddenColumns.Add(x => x.Charge.Chargeable);

            ActionColumns.Add(new DynamicImageColumn(IncludeImage, IncludeOne));
            AddButton("Exclude", chargeable_excluded, IncludeSelected, DynamicGridButtonPosition.Right);
            IncludeButton = AddButton("Include", chargeable_included, IncludeSelected, DynamicGridButtonPosition.Right);
            ShowAllButton = AddButton("Show All", null, ToggleShowAll);
        }

        protected override void DoReconfigure(DynamicGridOptions options)
        {
            base.DoReconfigure(options);
            options.Clear();
            options.EditRows = true;
            options.RecordCount = true;
            options.SelectColumns = true;
            options.MultiSelect = true;
            options.FilterRows = true;
        }
        
        public Invoice Invoice { get; set; }

        private bool IncludeSelected(Button sender, CoreRow[] rows)
        {
            if (Invoice.ID != Guid.Empty)
            {
                using (new WaitCursor())
                {
                    var bAdd = sender == IncludeButton;
                    var assignments = rows.Select(r => r.ToObject<Assignment>()).ToArray();
                    foreach (var assignment in assignments)
                    {
                        if (bAdd)
                        {
                            assignment.Invoice.ID = Invoice.ID;
                            assignment.Invoice.Synchronise(Invoice);
                        }
                        else
                        {
                            assignment.Invoice.ID = Guid.Empty;
                            assignment.Invoice.Synchronise(new Invoice());
                        }
                    }
                    new Client<Assignment>().Save(assignments, bAdd ? "Added to Invoice" : "Removed From Invoice", (o, e) => { });
                    foreach (var row in rows)
                    {
                        row.Set<Assignment, Guid>(x => x.Invoice.ID, bAdd ? Invoice.ID : Guid.Empty);
                        InvalidateRow(row);
                    }
                }
            }
            else
                MessageBox.Show("Please Select or Create an Invoice First!");
            return false;
        }

        private bool IncludeOne(CoreRow arg)
        {
            if (arg == null)
                return false;
            
            if (Invoice.ID != Guid.Empty)
            {
                var id = arg.Get<Assignment, Guid>(x => x.ID);
                var assignment = arg.ToObject<Assignment>();
                if (assignment != null)
                {
                    if (assignment.Invoice.IsValid())
                    {
                        assignment.Invoice.ID = Invoice.ID;
                        assignment.Invoice.Synchronise(Invoice);
                    }
                    else
                    {
                        assignment.Invoice.ID = Guid.Empty;
                        assignment.Invoice.Synchronise(new Invoice());
                    }
                    new Client<Assignment>().Save(assignment, "Added to Invoice", (o,e) => { });
                    arg.Set<Assignment, Guid>(x=>x.Invoice.ID, assignment.Invoice.ID);
                    InvalidateRow(arg);
                }
            }
            else
                MessageBox.Show("Please Select or Create an Invoice First!");
            return false;
        }
        
        private bool ToggleShowAll(Button arg1, CoreRow[] rows)
        {
            _showall = !_showall;
            UpdateButton(ShowAllButton, null, _showall ? "Selected Only" : "Show All");
            return true;
        }

        private BitmapImage IncludeImage(CoreRow arg)
        {
            if (arg == null)
                return chargeable_included;
            bool chargeable = arg.Get<Assignment, bool>(x => x.Charge.Chargeable);
            bool included = Entity.IsEntityLinkValid<Assignment, InvoiceLink>(x => x.Invoice, arg);
            return chargeable
                ? included 
                    ? chargeable_included 
                    : chargeable_excluded
                : included
                    ? unchargeable_included
                    : unchargeable_excluded;
        }

        protected override void Reload(
        	Filters<Assignment> criteria, Columns<Assignment> columns, ref SortOrder<Assignment>? sort,
        	CancellationToken token, Action<CoreTable?, Exception?> action)
        {
            
            if (Invoice.ID == Guid.Empty)
                criteria.Add(new Filter<Assignment>().None());
            else
            {
                
                criteria.Add(new Filter<Assignment>(x => x.JobLink.ID).IsEqualTo(Invoice.JobLink.ID));
                
                if (_showall)
                    criteria.Add(new Filter<Assignment>(x => x.Invoice.ID).IsEqualTo(Invoice.ID).Or(x => x.Invoice).NotLinkValid());
                else
                    criteria.Add(new Filter<Assignment>(x => x.Invoice.ID).IsEqualTo(Invoice.ID)); 
                
            }
            base.Reload(criteria, columns, ref sort, token, action);
        }
    }
}