prs/prs.desktop/Panels/Timesheets/TimeSheetStandardLeaveGrid.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Windows.Media.Imaging;
using Comal.Classes;
using InABox.Clients;
using InABox.Core;
using InABox.DynamicGrid;
using InABox.WPF;
using PRS.Shared;

namespace PRSDesktop
{
    public class TimeSheetStandardLeaveGrid : DynamicDataGrid<StandardLeave>, ISpecificGrid
    {
        public DateTime From { get; set; }
        public DateTime To { get; set; }
        
        public TimeSheetStandardLeaveGrid() : base()
        {
            ColumnsTag = "TimeSheetStandardLeave";
            
            ActionColumns.Add(new DynamicImageColumn(SelectedImage, SelectedAction));
            HiddenColumns.Add(x => x.LeaveType.ID);
            HiddenColumns.Add(x => x.From);
            HiddenColumns.Add(x => x.FromTime);
            HiddenColumns.Add(x => x.To);
            HiddenColumns.Add(x => x.ToTime);
            HiddenColumns.Add(x => x.Name);
        }
        protected override void DoReconfigure(DynamicGridOptions options)
        {
            base.DoReconfigure(options);
            options.Clear();
            options.SelectColumns = true;
        }

        public override DynamicGridColumns GenerateColumns()
        {
            var columns = new DynamicGridColumns();
            columns.Add<StandardLeave>(x => x.Name, 0, "Name", "", Alignment.MiddleLeft);
            columns.Add<StandardLeave>(x => x.LeaveType.Description, 200, "Type", "", Alignment.MiddleCenter);
            columns.Add<StandardLeave>(x => x.From, 80, "", "dd MMM yy", Alignment.MiddleCenter);
            columns.Add<StandardLeave>(x => x.To, 80, "", "dd MMM yy", Alignment.MiddleCenter);
            return columns;
        }
        
        private BitmapImage tick = PRSDesktop.Resources.tick.AsBitmapImage();
        public HashSet<Guid> SelectedIDs = new HashSet<Guid>();

        private bool SelectedAction(CoreRow? arg)
        {
            if (arg == null)
            {
                foreach (var row in Data.Rows)
                    ToggleRow(row);
            }
            else
                ToggleRow(arg);
            return false;
        }

        private void ToggleRow(CoreRow row)
        {
            Guid id = row.Get<StandardLeave, Guid>(x => x.ID);
            if (!SelectedIDs.Add(id))
                SelectedIDs.Remove(id);
            InvalidateRow(row);
        }

        private BitmapImage? SelectedImage(CoreRow? arg)
        {
            return arg == null
                ? tick
                : SelectedIDs.Contains(arg.Get<StandardLeave,Guid>(x=>x.ID))
                    ? tick
                    : null;
        }

        protected override void Reload(
        	Filters<StandardLeave> criteria, Columns<StandardLeave> columns, ref SortOrder<StandardLeave>? sort,
        	CancellationToken token, Action<CoreTable?, Exception?> action)
        {
            criteria.Add(new InABox.Core.Filter<StandardLeave>(x => x.From).IsLessThanOrEqualTo(To)
                .And(x => x.To).IsGreaterThanOrEqualTo(From)
                .And(x =>x.Processed).IsEqualTo(DateTime.MinValue));
            base.Reload(criteria, columns, ref sort, token, action);
        }

        public void Process(IProgress<String> progress)
        {
            if (SelectedIDs.Count == 0)
                return;
            
            progress.Report("Loading Standard Leaves");
            var empdates = new Filter<Employee>().All();
            empdates.Ands.Add(new Filter<Employee>(x => x.StartDate).IsEqualTo(DateTime.MinValue)
                .Or(x => x.StartDate).IsLessThanOrEqualTo(To));
            empdates.Ands.Add(new Filter<Employee>(x => x.FinishDate).IsEqualTo(DateTime.MinValue)
                .Or(x => x.FinishDate).IsGreaterThanOrEqualTo(From));

            var employees = new Client<Employee>().Query(
                empdates,
                Columns.None<Employee>().Add(x => x.ID)
                    .Add(x => x.Name)
                    .Add(x => x.RosterStart)
                    .Add(x => x.StartDate)
                    .Add(x => x.FinishDate))
                .ToArray<Employee>();
            var rosters = Client.Query(
                new Filter<EmployeeRosterItem>(x => x.Employee.ID).InList(employees.ToArray(x => x.ID)),
                orderby: new SortOrder<EmployeeRosterItem>(x => x.Day))
                .ToObjects<EmployeeRosterItem>()
                .GroupBy(x => x.Employee.ID)
                .ToDictionary(x => x.Key, x => x.ToArray());

            int i = 0;
            var leaves = new List<StandardLeave>();
            foreach (var row in Data.Rows)
            {
                if (SelectedIDs.Contains(row.Get<StandardLeave,Guid>(x=>x.ID)))
                {
                    var leave = row.ToObject<StandardLeave>();
                    foreach (var employee in employees)
                    {
                        var roster = rosters.GetValueOrDefault(employee.ID);
                        progress.Report($"Processing Standard Leave {((double)i * 100.0F)/((double)SelectedIDs.Count * employees.Length):F2}%");
                        var timesheets = RosterUtils.CreateLeaveTimesheets(
                            employee,
                            roster,
                            leave.From,
                            leave.FromTime,
                            leave.To,
                            leave.ToTime,
                            leave.LeaveType.ID,
                            (timesheet) =>
                            {
                                timesheet.Notes = leave.Name;
                                timesheet.StandardLeaveLink.ID = leave.ID;
                            }
                        );
                        Client.Save(timesheets, "Updated from TimeSheet leave Processor");
                        i++;
                    }
                    leave.Processed = DateTime.Now;
                    leaves.Add(leave);
                }
            }

            if (leaves.Count != 0)
            {
                progress.Report("Saving Standard Leaves");
                Client.Save(leaves, "Updated from TimeSheet leave Processor");
            }
        }
        
    }
}