prs/prs.desktop/Panels/Jobs/DocumentSets/JobDocumentSetTree.xaml.cs

using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.ComponentModel;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Documents;
using System.Windows.Forms;
using System.Windows.Input;
using System.Windows.Media;
using Comal.Classes;
using InABox.Clients;
using InABox.Core;
using InABox.DynamicGrid;
using InABox.WPF;
using Syncfusion.Data.Extensions;
using Syncfusion.UI.Xaml.Grid;
using Syncfusion.UI.Xaml.TreeGrid;
using Syncfusion.UI.Xaml.TreeGrid.Helpers;
using JobDocumentSetFolder = Comal.Classes.JobDocumentSetFolder;
using MessageBox = System.Windows.MessageBox;
using UserControl = System.Windows.Controls.UserControl;

namespace PRSDesktop
{
    
    public class DocumentSetNode : INotifyPropertyChanged
    {

        private DocumentSetNodes _owner = null;
        
        public ObservableCollection<DocumentSetNode> Children => _owner.GetChilden(_id);
        
        private Guid _id;
        public Guid ID 
        {
            get { return _id; }
            set
            {
                _id = value;
                RaisedOnPropertyChanged("ID");
            }
        }
        
        private Guid _parent;
        public Guid Parent 
        {
            get { return _parent; }
            set
            {
                _parent = value;
                RaisedOnPropertyChanged("Parent");
            }
        }
        
        private string _description;
        public string Description
        {
            get { return _description; }
            set
            {
                _description = value;
                RaisedOnPropertyChanged("Description");
            }
        }        
        
        private string _details;
        public string Details
        {
            get { return _details; }
            set
            {
                _details = value;
                RaisedOnPropertyChanged("Details");
            }
        }
        
        public Dictionary<String,String> Blocks { get; private set; }
        
        public event PropertyChangedEventHandler PropertyChanged;

        public void RaisedOnPropertyChanged(string propertyName)
        {
            PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(propertyName));
        }

        public DocumentSetNode(DocumentSetNodes owner)
        {
            _owner = owner;
            Blocks = new Dictionary<String,String>();
            foreach (var column in owner.Columns)
                Blocks[column] = "";
        }
        
        public DocumentSetNode(DocumentSetNodes owner, Guid id, Guid parent) : this(owner)
        {
            _id = id;
            _parent = parent;
        }
    }

    public class DocumentSetNodes 
    {

        private List<DocumentSetNode> _nodes = null;

        public ObservableCollection<DocumentSetNode> Nodes => new ObservableCollection<DocumentSetNode>(_nodes.Where(x=>x.Parent == Guid.Empty));

        public IEnumerable<String> Columns { get; private set; }
        
        public DocumentSetNodes(IEnumerable<String> columns)
        {
            _nodes = new List<DocumentSetNode>();
            Columns = columns;
        }

        public DocumentSetNode Add(Guid id, Guid parent)
        {
            var node = new DocumentSetNode(this, id, parent);
            _nodes.Add(node);
            return node;
        }

        public ObservableCollection<DocumentSetNode> GetChilden(Guid id)
        {
            return new ObservableCollection<DocumentSetNode>(_nodes.Where(x => x.Parent.Equals(id) && (x.ID != id)));
        }
  
        public DocumentSetNode? GetNode(Guid id)
        {
            return _nodes.FirstOrDefault(x => x.ID == id);
        }
        
    }

    // public class GridColumnSizerExt : TreeGridColumnSizer 
    // { 
    //     public GridColumnSizerExt(SfTreeGrid sfTreeGrid) 
    //         : base() 
    //     { 
    //
    //     } 
    //     public override double SetColumnWidth(TreeGridColumn column, double Width) 
    //     { 
    //         MethodInfo methodInfo = this.TreeGrid.ColumnResizingController.GetType().GetMethod("IsExpanderColumn", BindingFlags.NonPublic | BindingFlags.Instance);             
    //         if ((bool)methodInfo.Invoke(this.TreeGrid.ColumnResizingController, new object[] { column }))       
    //         { 
    //             var columnIndex = this.TreeGrid.Columns.IndexOf(column); 
    //             var scrollColumnIndex = this.TreeGrid.ResolveToScrollColumnIndex(columnIndex); 
    //             var treeGridPanel = this.TreeGrid.GetTreePanel(); 
    //             //You can hide the expander column by setting width as 0 here. 
    //             treeGridPanel.ColumnWidths[scrollColumnIndex] = 100.0; 
    //             return 100.0; 
    //         }                
    //         else 
    //             return base.SetColumnWidth(column, Width); 
    //     }      
    // } 

    public delegate void JobDocumentSetMileStoneSelected(JobDocumentSetMileStoneBlock block);
    
    public partial class JobDocumentSetTree : UserControl
    {

        public event JobDocumentSetMileStoneSelected MileStoneSelected;
    
        private struct MileStone
        {
            public Guid TypeID { get; set; }
            public CoreRow Row { get; set; }
        }
        
        private struct MileStoneType
        {
            public String Code { get; set; }
            public String Description { get; set; }
            public Dictionary<Guid,List<CoreRow>> SetMileStones { get; set; }
            public List<String> Columns { get; set; }
        }
        
        public Guid JobID { get; set; }
        public Guid FolderID { get; set; }

        public bool DisciplineVisible { get; set; }
        public Guid DisciplineID { get; set; }
        
        public bool TypeVisible { get; set; }
        public Guid TypeID { get; set; }
        
        public bool CategoryVisible { get; set; }
        public Guid CategoryID { get; set; }
        
        public bool AreaVisible { get; set; }
        public Guid AreaID { get; set; }

        private Dictionary<Guid,MileStoneType> _types = null;
        private CoreTable _milestones = null;
        public CoreTable Data { get; private set; } = null;

        private CoreTable _files = null;
        
        private bool _hidesuperceded = false;
        private bool _flatlist = false;
        private bool _includeretired = false;

        private DocumentSetNodes _documentsets = null;
        
        public JobDocumentSetTree()
        {
            InitializeComponent();
            AddImage.Source = PRSDesktop.Resources.add.AsBitmapImage();
            EditImage.Source = PRSDesktop.Resources.pencil.AsBitmapImage();
            DeleteImage.Source = PRSDesktop.Resources.delete.AsBitmapImage();
            treeGrid.Loaded += (o, e) =>
            {
                treeGrid.GetTreePanel().RowHeights[1] = 0;
                treeGrid.UpdateDataRow(1);
            };
            
        }
        
        public void Refresh()
        {

            using (new WaitCursor())
            {
                var scrollviewer = WPFUtils.FindVisualChildren<ScrollViewer>(treeGrid).FirstOrDefault();
                var verticalOffset = scrollviewer != null ? scrollviewer.VerticalOffset : 0;
                var horizontalOffset = treeGrid.SelectedItem != null ? scrollviewer.HorizontalOffset : 0;
                treeGrid.ItemsSource = null;

                var setfilter = new Filter<JobDocumentSet>(x => x.Job.ID).IsEqualTo(JobID);
                
                if (FolderID != CoreUtils.FullGuid)
                    setfilter = setfilter.And(x => x.Folder.ID).IsEqualTo(FolderID);  
                
                if (DisciplineID != Guid.Empty)
                    setfilter = setfilter.And(x => x.Discipline.ID).IsEqualTo(DisciplineID);

                if (TypeID != Guid.Empty)
                    setfilter = setfilter.And(x => x.Type.ID).IsEqualTo(TypeID);

                if (CategoryID != Guid.Empty)
                    setfilter = setfilter.And(x => x.Category.ID).IsEqualTo(CategoryID);

                if (AreaID != Guid.Empty)
                    setfilter = setfilter.And(x => x.Area.ID).IsEqualTo(AreaID);
                
                if (!_includeretired)
                    setfilter = setfilter.And(x => x.Retired).IsEqualTo(DateTime.MinValue);

                MultiQuery query = new MultiQuery();
                query.Add(
                    setfilter,
                    new Columns<JobDocumentSet>(x => x.ID)
                        .Add(x => x.Parent.ID)
                        .Add(x => x.Code)
                        .Add(x => x.Description)
                        .Add(x => x.Date)
                        .Add(x => x.Size)
                        .Add(x => x.Scale)
                        .Add(x => x.Employee.Name),
                    new SortOrder<JobDocumentSet>(x => x.Code)
                );


                var milestonefilter = new Filter<JobDocumentSetMileStone>(x => x.DocumentSet.Job.ID).IsEqualTo(JobID);
                if (FolderID != Guid.Empty && FolderID != CoreUtils.FullGuid)
                    milestonefilter = milestonefilter.And(x => x.DocumentSet.Folder.ID).IsEqualTo(FolderID);

                query.Add(
                    milestonefilter,
                    new Columns<JobDocumentSetMileStone>(x => x.ID)
                        .Add(x => x.DocumentSet.ID)
                        .Add(x => x.Type.ID)
                        .Add(x => x.Type.Code)
                        .Add(x => x.Status)
                        .Add(x => x.Notes)
                        .Add(x => x.Revision)
                        .Add(x => x.Due)
                        .Add(x => x.Submitted)
                        .Add(x => x.Closed)
                        .Add(x => x.Attachments)
                        .Add(x => x.Watermark)
                );

                if (_types == null)
                {
                    query.Add<JobDocumentSetMileStoneType>(
                        null,
                        new Columns<JobDocumentSetMileStoneType>(x => x.ID)
                            .Add(x => x.Code)
                            .Add(x => x.Description),
                        new SortOrder<JobDocumentSetMileStoneType>(x => x.Sequence)
                    );
                }

                query.Query();

                Data = query.Get<JobDocumentSet>();
                _milestones = query.Get<JobDocumentSetMileStone>();

                if (_types == null)
                {
                    _types = query.Get<JobDocumentSetMileStoneType>().ToDictionary<JobDocumentSetMileStoneType, Guid, MileStoneType>(
                        x => x.ID,
                        r => new MileStoneType()
                        {
                            Code = r.Get<JobDocumentSetMileStoneType, String>(c => c.Code),
                            Description = r.Get<JobDocumentSetMileStoneType, String>(c => c.Description),
                            SetMileStones = new Dictionary<Guid, List<CoreRow>>(),
                            Columns = new List<string>()
                        }
                    );
                }
                else
                {
                    foreach (var typeid in _types.Keys)
                    {
                        _types[typeid].Columns.Clear();
                        _types[typeid].SetMileStones.Clear();
                    }
                }

                var milestones = _milestones.ToLookup<JobDocumentSetMileStone, Guid, MileStone>(
                    x => x.DocumentSet.ID,
                    r => new MileStone()
                    {
                        TypeID = r.Get<JobDocumentSetMileStone, Guid>(c => c.Type.ID),
                        Row = r
                    }
                );

                foreach (var milestone in milestones)
                {
                    foreach (var entry in milestone)
                    {
                        if (!_types[entry.TypeID].SetMileStones.ContainsKey(milestone.Key))
                            _types[entry.TypeID].SetMileStones[milestone.Key] = new List<CoreRow>();
                        if (_hidesuperceded)
                            _types[entry.TypeID].SetMileStones[milestone.Key].Clear();
                        _types[entry.TypeID].SetMileStones[milestone.Key].Add(entry.Row);
                    }
                }

                List<String> columns = new List<string>();
                foreach (var typeid in _types.Keys)
                {
                    int count = 1;
                    foreach (var setkey in _types[typeid].SetMileStones.Keys)
                        count = Math.Max(count, _types[typeid].SetMileStones[setkey].Count);
                    for (int i = 1; i <= count; i++)
                    {
                        String column = String.Format("{0}_{1}", _types[typeid].Code, i);
                        columns.Add(column);
                        _types[typeid].Columns.Add(String.Format("Blocks[{0}]", column));
                    }
                }

                _documentsets = new DocumentSetNodes(columns);

                foreach (var setrow in Data.Rows)
                {
                    
                    
                    Guid setid = setrow.Get<JobDocumentSet, Guid>(x => x.ID);
                    Guid parentid = _flatlist ? Guid.Empty : setrow.Get<JobDocumentSet, Guid>(x => x.Parent.ID);

                    var node = _documentsets.Add(setid, parentid);
                    
                    
                    JobDocumentSetDescriptionBlock desc = new JobDocumentSetDescriptionBlock()
                    {
                        ID = setid,
                        Code = setrow.Get<JobDocumentSet, String>(c => c.Code),
                        Description = setrow.Get<JobDocumentSet, String>(c => c.Description),
                    };
                    node.Description = Serialization.Serialize(desc);

                    JobDocumentSetDetailsBlock dets = new JobDocumentSetDetailsBlock()
                    {
                        ID = setid,
                        Date = setrow.Get<JobDocumentSet, DateTime>(c => c.Date),
                        Size = setrow.Get<JobDocumentSet, PaperSize>(c => c.Size),
                        Scale = setrow.Get<JobDocumentSet, String>(c => c.Scale),
                        Employee = setrow.Get<JobDocumentSet, String>(c => c.Employee.Name)
                    };
                    node.Details = Serialization.Serialize(dets);

                    foreach (var typeid in _types.Keys)
                    {
                        if (_types[typeid].SetMileStones.TryGetValue(setid, out var rows))
                        {
                            int i = 1;
                            foreach (var row in rows)
                            {
                                JobDocumentSetMileStoneBlock block = new JobDocumentSetMileStoneBlock();

                                block.ID = row.Get<JobDocumentSetMileStone, Guid>(c => c.ID);

                                block.Revision = row.Get<JobDocumentSetMileStone, String>(c => c.Revision);

                                block.Status = row.Get<JobDocumentSetMileStone, JobDocumentSetMileStoneStatus>(c => c.Status);

                                block.Date = (block.Status == JobDocumentSetMileStoneStatus.Approved) ||
                                             (block.Status == JobDocumentSetMileStoneStatus.Cancelled) ||
                                             (block.Status == JobDocumentSetMileStoneStatus.Rejected)
                                    ? row.Get<JobDocumentSetMileStone, DateTime>(c => c.Closed)
                                    : block.Status == JobDocumentSetMileStoneStatus.Submitted
                                        ? block.Date = row.Get<JobDocumentSetMileStone, DateTime>(c => c.Submitted)
                                        : row.Get<JobDocumentSetMileStone, DateTime>(c => c.Due);

                                String[] notes = row.Get<JobDocumentSetMileStone, String[]>(c => c.Notes);
                                block.Notes = notes != null ? String.Join("\n", notes) : "";

                                block.Attachments = row.Get<JobDocumentSetMileStone, int>(c => c.Attachments);
                                block.Watermark = row.Get<JobDocumentSetMileStone, String>(c => c.Watermark);

                                node.Blocks[String.Format("{0}_{1}", _types[typeid].Code, i)] = Serialization.Serialize(block);
                                i++;
                            }
                        }
                    }
                }

                ConfigureColumns(_documentsets);

                ConfigureStackedHeader();

                treeGrid.ItemsSource = _documentsets.Nodes;
                DocumentCount.Content = $"{_documentsets.Nodes.Count} {(_documentsets.Nodes.Count > 1 ? "Records" : "Record")}";
                if (scrollviewer != null)
                {
                    scrollviewer.ScrollToVerticalOffset(verticalOffset);
                    scrollviewer.ScrollToHorizontalOffset(horizontalOffset);
                }
            }

        }

        #region Grid Configuration
        
        private void ConfigureColumns(DocumentSetNodes documentsets)
        {
            treeGrid.Columns.Clear();
            
            treeGrid.Columns.Add(new TreeGridTemplateColumn()
            {
                CellTemplate = FindResource("descriptionTemplate") as DataTemplate,
                MappingName = "Description",
                SetCellBoundValue = true,
                MinimumWidth = 250,
                ColumnSizer = TreeColumnSizer.Star
            });
            
            treeGrid.Columns.Add(new TreeGridTemplateColumn()
            {
                CellTemplate = FindResource("detailsTemplate") as DataTemplate,
                MappingName = "Details",
                SetCellBoundValue = true,
                Width = 120
            });
            
            foreach (var column in documentsets.Columns)
            {
                var col = new TreeGridTemplateColumn()
                {
                    CellTemplate = FindResource("milestoneTemplate") as DataTemplate,
                    MappingName = String.Format("Blocks[{0}]",column),
                    SetCellBoundValue = true,
                    HeaderText = " ",
                    Width = 80,
                    ShowToolTip = true
                };
                treeGrid.Columns.Add(col);
            }
        }

        private void ConfigureStackedHeader()
        {
            stackedHeaderRow.StackedColumns.Clear();
            stackedHeaderRow.StackedColumns.Add(new StackedColumn()
            {
                ChildColumns = "Description,Details",
                HeaderText = "Document Register"
            });
        
            foreach (var typeid in _types.Keys)
            {
                stackedHeaderRow.StackedColumns.Add(new StackedColumn()
                {
                    
                    ChildColumns = String.Join(",", _types[typeid].Columns),
                    HeaderText = _types[typeid].Code
                });
            }
        }
        
        private void TreeGrid_OnItemsSourceChanged(object? sender, TreeGridItemsSourceChangedEventArgs e)
        {
            var panel = treeGrid.GetTreePanel();
            panel.RowHeights[1] = 0;
        }
        
        private void TreeGrid_OnNodeCollapsing(object? sender, NodeCollapsingEventArgs e)
        {
            e.Cancel = true;
        }
        
        public MenuItem CreateCalendar(ContextMenu menu, string text, DateTime startDate, CoreRow[] milestones, Action<CoreRow[], DateTime?>? action)
        {
            var item = new MenuItem();

            var calendarItem = new MenuItem();

            var calendar = new Calendar { DisplayDate = startDate, SelectedDate = null};

            calendar.SelectedDatesChanged += (o, e) =>
            {
                action?.Invoke(milestones, calendar.SelectedDate);
                menu.IsOpen = false;
            };
            calendarItem.Header = calendar;
            calendarItem.Style = DynamicGridUtils.Resources["NonHighlightMenuItem"] as Style;
            item.Header = text;
            item.Items.Add(calendarItem);
            item.IsCheckable = false;
            return item;
        }
        
        
        private void TreeGrid_OnContextMenuOpening(object sender, ContextMenuEventArgs e)
        {
            MileStoneMenu.Items.Clear();

            var tag = (e.OriginalSource as FrameworkElement).Tag;
            Point pos =  Mouse.GetPosition(treeGrid);
            var treeGridPanel = this.treeGrid.GetTreePanel();
            // get the row and column index based on the pointer position 
            var rowColumnIndex = treeGridPanel.PointToCellRowColumnIndex(pos);
            if (rowColumnIndex.IsEmpty)
                return;
            var treeNodeAtRowIndex = treeGrid.GetNodeAtRowIndex(rowColumnIndex.RowIndex);                       
 
            if (rowColumnIndex.ColumnIndex < 2)
            {

                var document = treeGrid.SelectedItem as DocumentSetNode;


                MenuItem edit = new MenuItem();
                edit.Header = "Edit Document Set";
                edit.Click += (o, args) => { EditDocumentSets(new Guid[] { document.ID }); };
                MileStoneMenu.Items.Add(edit);

                MileStoneMenu.Items.Add(new Separator());
                
                MenuItem addchild = new MenuItem();
                addchild.Header = "Add Child";
                addchild.Click += (o,args) => { AddChildDocument(document); };
                MileStoneMenu.Items.Add(addchild);
                
                var documents = treeGrid.SelectedItems.Select(x => (x as DocumentSetNode));
                MenuItem movetofolder = new MenuItem();
                movetofolder.Header = "Move To Folder";
                bool hasfolders = PopulateFolders(movetofolder, documents);
                if (hasfolders)
                {
                    MileStoneMenu.Items.Add(new Separator());
                    MileStoneMenu.Items.Add(movetofolder);
                }

                return;
            }
            
            var mappingname = treeGrid.Columns[rowColumnIndex.ColumnIndex].MappingName;
            var blockkey = mappingname.Replace("Blocks[", "").Replace("]", "");
            var typeid = _types.FirstOrDefault(x => x.Value.Columns.Contains(mappingname)).Key;

            //Guid setid = (treeGrid.SelectedItem as DocumentSetNode).ID;
            Guid[] setids = treeGrid.SelectedItems.Select(x => (x as DocumentSetNode).ID).ToArray();

            //Guid.TryParse(tag.ToString(), out Guid milestoneid);
            var blocks = treeGrid.SelectedItems.Select(x => (x as DocumentSetNode).Blocks[blockkey]).Where(x => !String.IsNullOrWhiteSpace(x))
                .ToArray();
            var milestoneids = blocks.Select(x => Serialization.Deserialize<JobDocumentSetMileStoneBlock>(x).ID).ToArray();

            //var milestone = _milestones.Rows.FirstOrDefault(r => r.Get<JobDocumentSetMileStone, Guid>(c => c.ID) == milestoneid);
            var milestones = _milestones.Rows.Where(r => milestoneids.Contains(r.Get<JobDocumentSetMileStone, Guid>(c => c.ID))).ToArray();
            
            bool canCreateNewMileStones = true;
            foreach (var setid in setids)
            {
                var openmilestones = _milestones.Rows.Any(r =>
                    Guid.Equals(r.Get<JobDocumentSetMileStone, Guid>(c => c.DocumentSet.ID), setid)
                    && Guid.Equals(r.Get<JobDocumentSetMileStone, Guid>(c => c.Type.ID), typeid)
                    && (r.Get<JobDocumentSetMileStone, DateTime>(c => c.Closed).IsEmpty() ||
                        (r.Get<JobDocumentSetMileStone, JobDocumentSetMileStoneStatus>(c => c.Status) == JobDocumentSetMileStoneStatus.Approved))
                );
                if (openmilestones)
                    canCreateNewMileStones = false;
                
            }
            
            if (canCreateNewMileStones)
            {
                MenuItem newmilestone = new MenuItem()
                {
                    Header = "New Milestone",
                    Tag = typeid
                };
                newmilestone.Click += (o, args) => { CreateMileStone(setids, typeid, DateTime.Today); };
                MileStoneMenu.Items.Add(newmilestone);
            }

            if (milestones.Any())
            {
                MenuItem setstatus = new MenuItem() { Header = "Change Status" };
                foreach (JobDocumentSetMileStoneStatus newstatus in Enum.GetValues(typeof(JobDocumentSetMileStoneStatus)))
                {
                    MenuItem setstatus2 = null;
                    switch (newstatus)
                    {
                        
                        case JobDocumentSetMileStoneStatus.Unknown:
                            break;
                        
                        case JobDocumentSetMileStoneStatus.NotStarted:
                        case JobDocumentSetMileStoneStatus.InProgress:
                        case JobDocumentSetMileStoneStatus.OnHold:
                        case JobDocumentSetMileStoneStatus.InfoRequired:
                            setstatus2 = new MenuItem() { Header = newstatus.ToString().SplitCamelCase() };
                            setstatus2.Click += (o, args) => { ChangeMileStoneStatus(milestones, newstatus, DateTime.MinValue, DateTime.MinValue); };
                            break;

                        case JobDocumentSetMileStoneStatus.Submitted:
                            setstatus2 = CreateCalendar(
                                MileStoneMenu,
                                newstatus.ToString().SplitCamelCase(),
                                DateTime.Today,
                                milestones,
                                (r, t) => { ChangeMileStoneStatus(milestones, newstatus, t, DateTime.MinValue); }
                            );
                            break;
                            
                        case JobDocumentSetMileStoneStatus.Approved:
                        case JobDocumentSetMileStoneStatus.Cancelled:
                        case JobDocumentSetMileStoneStatus.Rejected:
                            setstatus2 = CreateCalendar(
                                MileStoneMenu,
                                newstatus.ToString().SplitCamelCase(),
                                DateTime.Today,
                                milestones,
                                (r, t) => { ChangeMileStoneStatus(milestones, newstatus, null, t); }
                            );
                            break;

                    }
                    if (setstatus2 != null)
                        setstatus.Items.Add(setstatus2);
                }

                MileStoneMenu.Items.Add(setstatus);


                //var closed = milestones.Any(r => !r.Get<JobDocumentSetMileStone, DateTime>(c => c.Closed).IsEmpty());
                if ((setids.Length == 1) && (milestones.Length == 1)) // && !closed)
                {
                    
                    MenuItem editmilestone = new MenuItem() { Header = "Edit MileStone" };
                    editmilestone.Click += (o, args) => { EditMileStone(milestones[0]); };
                    MileStoneMenu.Items.Add(editmilestone);

                    var attachments = milestones[0].Get<JobDocumentSetMileStone, int>(x => x.Attachments);
                    if (attachments > 1)
                    {
                        MenuItem splitmilestone = new MenuItem() { Header = "Split MileStone" };
                        splitmilestone.Click += (o, args) => { SplitMileStone(setids[0], milestones[0]); };
                        MileStoneMenu.Items.Add(splitmilestone);
                    }

                    MileStoneMenu.Items.Add(new Separator());
                    
                    MenuItem upload = new MenuItem() { Header = "Upload Files" };
                    upload.Click += (o, args) => { UploadFiles(milestones[0]); };
                    MileStoneMenu.Items.Add(upload);

                }
                
                if (milestones.Any())
                {
                    MenuItem download = new MenuItem() { Header = "Download Files" };
                    download.Items.Add(new MenuItem());
                    download.SubmenuOpened += (o, e) =>
                    {
                        download.Items.Clear();
                        var files = new Client<JobDocumentSetMileStoneFile>().Query(
                            new Filter<JobDocumentSetMileStoneFile>(x => x.EntityLink.ID).InList(milestoneids),
                            new Columns<JobDocumentSetMileStoneFile>(x => x.ID)
                                .Add(x => x.DocumentLink.FileName)
                                .Add(x => x.DocumentLink.ID),
                            new SortOrder<JobDocumentSetMileStoneFile>(x => x.DocumentLink.FileName)
                        );
                        if (files.Rows.Any())
                        {
                            foreach (var row in files.Rows)
                            {
                                MenuItem downloadone = new MenuItem()
                                {
                                    Header = row.Get<JobDocumentSetMileStoneFile, String>(x => x.DocumentLink.FileName),
                                };
                                downloadone.Click += (sender, args) =>
                                {
                                    DownloadFiles(
                                        new CoreRow[] { milestones[0] },
                                        row.Get<JobDocumentSetMileStoneFile, Guid>(x => x.DocumentLink.ID)
                                    );
                                };
                                download.Items.Add(downloadone);
                            }

                            if (download.Items.Count > 1)
                            {
                                download.Items.Add(new Separator());

                                MenuItem downloadall = new MenuItem()
                                {
                                    Header = "Download All",
                                };
                                downloadall.Click += (sender, args) =>
                                {
                                    DownloadFiles(
                                        milestones,
                                        Guid.Empty
                                    );
                                };
                                download.Items.Add(downloadall);
                            }
                        }
                        else
                        {
                            download.Items.Add(
                                new MenuItem()
                                {
                                    Header = "No Files to download",
                                    IsEnabled = false
                                }
                            );
                        }
                    };
                    
                    MileStoneMenu.Items.Add(download);
                }

                // if ((milestoneids.Length == 1)) // && !closed)
                // {
                //     MenuItem managefiles = new MenuItem()
                //     {
                //         Header = "Manage Files"
                //     };
                //     managefiles.Click += (sender, args) => { ManageFiles(milestones[0]); };
                //     MileStoneMenu.Items.Add(managefiles);
                // }
                
                MileStoneMenu.Items.Add(new Separator());
                MenuItem delete = new MenuItem { Header = "Delete MileStone" };
                delete.Click += (o, args) => DeleteMileStone(milestones);
                MileStoneMenu.Items.Add(delete);
            }

            if (MileStoneMenu.Items.Count == 0)
                e.Handled = true;

        }

        private bool PopulateFolders(MenuItem menu, IEnumerable<DocumentSetNode> documents)
        {
            CoreTable data = new Client<JobDocumentSetFolder>().Query(
                new Filter<JobDocumentSetFolder>(x => x.Job.ID).IsEqualTo(JobID),
                new Columns<JobDocumentSetFolder>(x => x.ID)
                    .Add(x => x.Parent.ID)
                    .Add(x => x.Name)
            );
            if (!data.Rows.Any())
                return false;
            DynamicTreeNodes folders = new DynamicTreeNodes();
            folders.Load<JobDocumentSetFolder>(data, x => x.ID, x => x.Parent.ID, x => x.Name);
            foreach (var folder in folders.Nodes)
                DoPopulateFolder(menu, folder, documents);
            return true;
        }

        private void DoPopulateFolder(MenuItem header, DynamicTreeNode folder, IEnumerable<DocumentSetNode> documents)
        {
            MenuItem menu = new MenuItem();
            menu.Header = folder.Description;
            menu.Click += (sender, args) => MoveToFolder(documents, folder);
            header.Items.Add(menu);
            foreach (var childfolder in folder.Children)
                DoPopulateFolder(menu, childfolder, documents);
        }

        private void MoveToFolder(IEnumerable<DocumentSetNode> documents, DynamicTreeNode folder)
        {
            if (FolderID != folder.ID)
            {
                using (new WaitCursor())
                {
                    List<JobDocumentSet> updates = new List<JobDocumentSet>();
                    foreach (var document in documents)
                    {

                        var update = new JobDocumentSet();
                        update.ID = document.ID;
                        update.Folder.ID = folder.ID;
                        update.Parent.ID = Guid.Empty;
                        updates.Add(update);
                    }
                    new Client<JobDocumentSet>().Save(updates, "Moved to Folder: " + folder.Description);
                }

                Refresh();
            }
            else
            {
                MessageBox.Show("Target Folder is the same as Source Folder!");

            }
        }

        private void SplitMileStone(Guid setid, CoreRow milestone)
        {
            if (MessageBox.Show(
                    "Are you sure you wish to split this Document Set?", 
                    "Confirm Delete", 
                    MessageBoxButton.YesNo
                ) != MessageBoxResult.Yes)
                return;
            
            Guid milestoneid = milestone.Get<JobDocumentSetMileStone, Guid>(c => c.ID);
            var dlg = new MultiSelectDialog<JobDocumentSetMileStoneFile>(
                new Filter<JobDocumentSetMileStoneFile>(c => c.EntityLink.ID).IsEqualTo(milestoneid),
                null,
                true
            );
            if (dlg.ShowDialog() == true)
            {
                var files = dlg.Items();
                Progress.ShowModal("Splitting Document Set", (progress) =>
                {

                    JobDocumentSet newset = new Client<JobDocumentSet>().Query(
                        new Filter<JobDocumentSet>(x => x.ID).IsEqualTo(setid)
                    ).Rows.FirstOrDefault()?.ToObject<JobDocumentSet>();
                    if (newset != null)
                    {
                        newset.ID = Guid.Empty;
                        newset.CommitChanges();
                        newset.Parent.ID = setid;
                        newset.Code = String.Format("{0} (COPY)", newset.Code);
                        //newset.Description = "New Child";
                        new Client<JobDocumentSet>().Save(newset, "Created by Splitting MileStone");
                        progress.Report("Creating Milestone");
                        JobDocumentSetMileStone newms = new Client<JobDocumentSetMileStone>().Query(
                            new Filter<JobDocumentSetMileStone>(c=>c.ID).IsEqualTo(milestoneid)
                        ).Rows.FirstOrDefault()?.ToObject<JobDocumentSetMileStone>();
                        if (newms != null)
                        {
                            newms.ID = Guid.Empty;
                            newset.CommitChanges();
                            newms.DocumentSet.ID = newset.ID;
                            new Client<JobDocumentSetMileStone>().Save(newms, "Created By Splitting MileStone");
                            progress.Report("Moving Files");
                            foreach (var file in files)
                                file.EntityLink.ID = newms.ID;
                            new Client<JobDocumentSetMileStoneFile>().Save(files, "Moved when Splitting MileStone");
                        }
                    }
                });
                Refresh();
            }
        }

        private void AddChildDocument(DocumentSetNode node)
        {
            JobDocumentSet newset = new JobDocumentSet();
            newset.Parent.ID = node.ID;
            newset.Job.ID = JobID;
            newset.Folder.ID = FolderID;
            var grid = new DynamicDataGrid<JobDocumentSet>();
            if (grid.EditItems(new[] { newset }))
                Refresh();
            
        }

        // private void ManageFiles(CoreRow milestone)
        // {
        //     var grid = new JobDocumentSetMileStoneFileGrid();
        //     grid.OnGetWaterMark += (row) => milestone.Get<JobDocumentSetMileStone, String>(c => c.Watermark);
        //     grid.ShowSupercededColumn = false;
        //     Window window = new Window();
        //     window.Padding = new Thickness(5);
        //     window.Content = grid;
        //     window.Width = 300;
        //     window.Height = 500;
        //     window.WindowStartupLocation = WindowStartupLocation.CenterScreen;
        //     grid.Load(milestone.ToObject<JobDocumentSetMileStone>(), null);
        //     grid.Margin = new Thickness(5);
        //     grid.Refresh(true, true);
        //     window.ShowDialog();
        //     Refresh();
        // }

        private void DownloadFiles(CoreRow[] rows, Guid id)
        {
            
            FolderBrowserDialog dlg = new FolderBrowserDialog();
            if (dlg.ShowDialog() == DialogResult.OK)
            {
                Progress.ShowModal("Downloading Files", (progress) =>
                {
                    foreach (var row in rows)
                    {
                        var status = row.Get<JobDocumentSetMileStone, JobDocumentSetMileStoneStatus>(c => c.Status);
                        var stage = row.Get<JobDocumentSetMileStone, String>(c => c.Type.Code);
                        var revision = row.Get<JobDocumentSetMileStone, String>(c => c.Revision);

                        String tag = String.Format(" - {0}{1} ({2})", stage, String.IsNullOrWhiteSpace(revision) ? "" : " - Rev " + revision,
                            status.ToString().SplitCamelCase());

                        var filter = id == Guid.Empty
                            ? new Filter<Document>(x => x.ID).InQuery(
                                new Filter<JobDocumentSetMileStoneFile>(x => x.EntityLink.ID).IsEqualTo(
                                    row.Get<JobDocumentSetMileStone, Guid>(c => c.ID)),
                                x => x.DocumentLink.ID
                            )
                            : new Filter<Document>(x => x.ID).IsEqualTo(id);

                        var files = new Client<Document>().Query(filter);

                        foreach (var filerow in files.Rows)
                        {
                            string filename = filerow.Get<Document, String>(c => c.FileName);
                            string extension = Path.GetExtension(filename);
                            string basefilename = Path.GetFileNameWithoutExtension(filename);
                            filename = String.Format("{0}{1}{2}", basefilename, tag, extension);
                            filename = Path.Combine(dlg.SelectedPath, filename);
                            File.WriteAllBytes(filename, filerow.Get<Document, byte[]>(c => c.Data));
                        }
                    }
                });
                Process.Start(new ProcessStartInfo(dlg.SelectedPath) { UseShellExecute = true });
            }
        }
        
        private bool SelectFiles(out String[] files)
        {
            Microsoft.Win32.OpenFileDialog dlg = new Microsoft.Win32.OpenFileDialog();
            dlg.Filter = "PDF Files (*.pdf)|*.pdf";
            dlg.Multiselect = true;
            if ((dlg.ShowDialog() == true) && (dlg.FileNames.Length > 0))
            {
                files = dlg.FileNames.ToArray();
                return true;
            }
            files = null;
            return false;
        }

        private void UploadFiles(CoreRow row)
        {

            Guid id = row.Get<JobDocumentSetMileStone, Guid>(c => c.ID);
            if (SelectFiles(out String[] filenames))
            {
                Progress.ShowModal("Uploading Files", (progress) =>
                {
                    List<Document> documents = new List<Document>();
                    foreach (var file in filenames)
                    {
                        var data = File.ReadAllBytes(file);
                        documents.Add(
                            new Document()
                            {
                                FileName = Path.GetFileName(file).ToLower(),
                                Data = data,
                                CRC = CoreUtils.CalculateCRC(data),
                                TimeStamp = new FileInfo(file).LastWriteTime
                            }
                        );
                    }
                    new Client<Document>().Save(documents.ToArray(), "Uploaded by User");

                    progress.Report("Updating Links");
                    List<JobDocumentSetMileStoneFile> links = new List<JobDocumentSetMileStoneFile>();
                    foreach (var document in documents)
                    {
                        var link = new JobDocumentSetMileStoneFile();
                        link.EntityLink.ID = id;
                        link.DocumentLink.ID = document.ID;
                        links.Add(link);
                    }
                    new Client<JobDocumentSetMileStoneFile>().Save(links, "Uploaded By User");
                });
                MessageBox.Show(String.Format("{0} files uploaded", filenames.Length));
                Refresh();
            }
        }

        private Dictionary<Guid, JobDocumentSetMileStone> GetPreviousMileStones(Guid[] setids, Guid typeid)
        {
            var result = new Dictionary<Guid, JobDocumentSetMileStone>();
            foreach (var setid in setids)
            {
                var typeindex = _types.Keys.IndexOf(typeid);
                JobDocumentSetMileStone? last = null;
                while ((last == null) && (typeindex > 0))
                {
                    last = _milestones.Rows.LastOrDefault(r =>
                            (r.Get<JobDocumentSetMileStone, Guid>(c => c.DocumentSet.ID) == setid) &&
                            (r.Get<JobDocumentSetMileStone, Guid>(c => c.Type.ID) == _types.Keys.ToArray()[typeindex]))
                        ?.ToObject<JobDocumentSetMileStone>();
                    typeindex--;
                }

                if (last != null)
                    result[setid] = last;

            }
            return result;
        }

        private void CreateMileStone(Guid[] setids, Guid typeid, DateTime duedate)
        {
            bool bCopy = false;
            var lastmilestones = GetPreviousMileStones(setids, typeid);
            if (lastmilestones.Any(x => x.Value.Attachments > 0))
            {
                var confirm = MessageBox.Show("Do you wish to copy the files from the previous milestones?", "Copy Files",
                    MessageBoxButton.YesNoCancel);
                if (confirm == MessageBoxResult.Cancel)
                    return;
                bCopy = confirm == MessageBoxResult.Yes;
            }

            Dictionary<JobDocumentSetMileStone,JobDocumentSetMileStoneFile[]> updates = new Dictionary<JobDocumentSetMileStone, JobDocumentSetMileStoneFile[]>();
            foreach (var setid in setids)
            {
                JobDocumentSetMileStone milestone = new JobDocumentSetMileStone();
                milestone.DocumentSet.ID = setid;
                milestone.Type.ID = typeid;
                milestone.Status = JobDocumentSetMileStoneStatus.NotStarted;
                milestone.Due = duedate;

                JobDocumentSetMileStoneFile[] files = new JobDocumentSetMileStoneFile[] { };
                if (bCopy && lastmilestones.TryGetValue(setid, out var lastmilestone))
                {
                    if (lastmilestone.Attachments > 0)
                    {
                        files = new Client<JobDocumentSetMileStoneFile>().Query(
                            new Filter<JobDocumentSetMileStoneFile>(x => x.EntityLink.ID).InList(lastmilestone.ID),
                            new Columns<JobDocumentSetMileStoneFile>(x=>x.EntityLink.DocumentSet.ID)
                                .Add(x => x.DocumentLink.FileName)
                                .Add(x => x.DocumentLink.ID),
                            new SortOrder<JobDocumentSetMileStoneFile>(x => x.DocumentLink.FileName)
                        ).Rows.Select(r=>r.ToObject<JobDocumentSetMileStoneFile>()).ToArray();                        
                    }
                }
                
                updates[milestone] = files;
            }
            var grid = new JobDocumentSetMileStoneGrid();
            grid.OnAfterSave += (editor, items) =>
            {
                if (updates.Keys.Count == 1)
                    return;
                List<JobDocumentSetMileStoneFile> fileupdates = new List<JobDocumentSetMileStoneFile>();
                foreach (var milestone in updates.Keys)
                {
                    foreach (var file in updates[milestone])
                    {
                        file.EntityLink.ID = milestone.ID;
                        fileupdates.Add(file);
                    }
                }

                if (fileupdates.Any())
                    new Client<JobDocumentSetMileStoneFile>().Save(fileupdates,"");
            };

            if (grid.EditItems(updates.Keys.ToArray(), (t) =>
                {
                    if ((t == typeof(JobDocumentSetMileStoneFile)) && (updates.Keys.Count == 1))
                    {
                        CoreTable result = new CoreTable();
                        result.LoadColumns(typeof(JobDocumentSetMileStoneFile));
                        result.LoadRows(updates[updates.Keys.First()]);
                        return result;
                    }
                    return null;
                }, true)
            )
                Refresh();
        }

        private void ChangeMileStoneStatus(CoreRow[] rows, JobDocumentSetMileStoneStatus newstatus, DateTime? issued, DateTime? closed)
        {
            var milestones = rows.Select(r=>r.ToObject<JobDocumentSetMileStone>()).ToArray();
            foreach (var milestone in milestones)
            {
                if (issued.HasValue)
                    milestone.Submitted = issued.Value;
                if (closed.HasValue)
                    milestone.Closed = closed.Value;
                milestone.Status = newstatus;
            }

            using (new WaitCursor())
                new Client<JobDocumentSetMileStone>().Save(milestones, "Changed Status to " + newstatus.ToString().SplitCamelCase());
            Refresh();
        }

        private void EditMileStone(CoreRow row)
        {
            var milestone = new Client<JobDocumentSetMileStone>().Query(
                new Filter<JobDocumentSetMileStone>(x => x.ID).IsEqualTo(row.Get<JobDocumentSetMileStone, Guid>(x => x.ID))
            ).Rows.FirstOrDefault()?.ToObject<JobDocumentSetMileStone>();
            var grid = new JobDocumentSetMileStoneGrid();
            if (grid.EditItems(new[] { milestone }))
                Refresh();
        }

        private void DeleteMileStone(CoreRow[] rows)
        {
            var milestones = rows.Select(r=>r.ToObject<JobDocumentSetMileStone>()).ToArray();
            using (new WaitCursor())
                new Client<JobDocumentSetMileStone>().Delete(milestones,"Deleted by User");
            Refresh();
        }

        private void TreeGrid_OnCellToolTipOpening(object? sender, TreeGridCellToolTipOpeningEventArgs e)
        {
            var column = e.Column.MappingName.Replace("Blocks[","").Replace("]","");
            var data = (e.Record as DocumentSetNode).Blocks[column];
            if (String.IsNullOrWhiteSpace(data))
                return;
            
            var block = Serialization.Deserialize<JobDocumentSetMileStoneBlock>(data.ToString());
            Guid id = block.ID;

            TextBlock text = new TextBlock();
            if (!String.IsNullOrWhiteSpace(block.Notes))
            {
                text.Inlines.Add(new Run("Milestone Notes\n") { FontWeight = FontWeights.Bold, TextDecorations = TextDecorations.Underline });
                text.Inlines.Add(new Run(block.Notes.Replace("=","").Replace("\n\n","\n")) { FontStyle = FontStyles.Italic });
                
            }

            if (block.Attachments > 0)
            {
                if (!String.IsNullOrWhiteSpace(block.Notes))
                    text.Inlines.Add(new Run("\n\n"));
                text.Inlines.Add(new Run("Uploaded Files") { FontWeight = FontWeights.Bold, TextDecorations = TextDecorations.Underline });
                var files = new Client<JobDocumentSetMileStoneFile>().Query(
                    new Filter<JobDocumentSetMileStoneFile>(x => x.EntityLink.ID).IsEqualTo(block.ID),
                    new Columns<JobDocumentSetMileStoneFile>(x => x.DocumentLink.FileName),
                    new SortOrder<JobDocumentSetMileStoneFile>(x => x.DocumentLink.FileName)
                );
                
                foreach (var row in files.Rows)                
                    text.Inlines.Add(new Run("\n"+row.Get<JobDocumentSetMileStoneFile,String>(c=>c.DocumentLink.FileName)) { FontStyle = FontStyles.Italic });
            }

            if (!text.Inlines.Any())
            {
                e.ToolTip.Template = null;
                return;
            }

            e.ToolTip.Template = TemplateGenerator.CreateControlTemplate(
                typeof(System.Windows.Controls.ToolTip),
                () =>
                {
                    var border = new Border
                    {
                        BorderBrush = new SolidColorBrush(Colors.Gray),
                        BorderThickness = new Thickness(0.75),
                        CornerRadius = new CornerRadius(5),
                        Background = new SolidColorBrush(Colors.LightYellow),
                        Padding = new Thickness(5),
                        Child = text
                    };
                    return border;                   
                }
            );

            
        }
        
        #endregion

        #region Button Bar Actions

        private void AddTypes(MenuItem parent, Action<Guid> addfunction)
        {
            if(_types.Count == 0)
            {
                MenuItem item = new MenuItem() { Header = "No Document Milestones", IsEnabled = false };
                parent.Items.Add(item);
            }
            else
            {
                foreach (var type in _types.Keys)
                {
                    MenuItem item = new MenuItem() { Header = _types[type].Description, Tag = type };
                    item.Click += (o, e) => addfunction(type);
                    parent.Items.Add(item);
                }
            }
        }
        
        private void Add_OnClick(object sender, RoutedEventArgs e)
        {
            if (FolderID == Guid.Empty)
            {
                MessageBox.Show("Please choose a Folder first!");
                return;
            }

            ContextMenu menu = new ContextMenu();
            var onetoone = new MenuItem() { Header = "Add Individual Files" };
            AddTypes(onetoone, AddOneToOneFiles);
            menu.Items.Add(onetoone);

            var manytoone = new MenuItem() { Header = "Add Sets of Files" };
            AddTypes(manytoone, AddManyToOneFiles);
            menu.Items.Add(manytoone);

            menu.Items.Add(new Separator());
            
            var manual = new MenuItem() { Header = "Add Document Set Manually" };
            manual.Click += (o, e) => { AddDocumentSet(); };
            menu.Items.Add(manual);

            menu.IsOpen = true;
        }

        private void AddOneToOneFiles(Guid type)
        {
            if (!SelectFiles(out String[] filenames))
                return;

            Progress.ShowModal("Preparing Upload", (progress) =>
            {
                Dictionary<String, Tuple<Document, JobDocumentSet, JobDocumentSetMileStone, JobDocumentSetMileStoneFile>> map =
                    new Dictionary<string, Tuple<Document, JobDocumentSet, JobDocumentSetMileStone, JobDocumentSetMileStoneFile>>();

                foreach (var filename in filenames)
                {
                    var data = File.ReadAllBytes(filename);
                    
                    Document document = new Document()
                    {
                        FileName = Path.GetFileName(filename).ToLower(),
                        Data = data,
                        CRC = CoreUtils.CalculateCRC(data),
                        TimeStamp = new FileInfo(filename).LastWriteTime
                    }; 
                    
                    JobDocumentSet set = new JobDocumentSet();
                    set.Job.ID = JobID;
                    set.Folder.ID = FolderID;
                    set.Code = Path.GetFileNameWithoutExtension(filename).ToUpper();
                    set.Description = Path.GetFileNameWithoutExtension(filename).ToUpper();
                    set.Discipline.ID = DisciplineID;
                    set.Type.ID = TypeID;
                    set.Category.ID = CategoryID;
                    set.Area.ID = AreaID;

                    JobDocumentSetMileStone milestone = new JobDocumentSetMileStone();
                    milestone.Type.ID = type;
                    milestone.Status = JobDocumentSetMileStoneStatus.InProgress;
                    milestone.Due = DateTime.Today;
                
                    JobDocumentSetMileStoneFile file = new JobDocumentSetMileStoneFile();
                    map[filename] = new Tuple<Document, JobDocumentSet, JobDocumentSetMileStone, JobDocumentSetMileStoneFile>(
                        document,
                        set,
                        milestone,
                        file
                    );
                }

                progress.Report("Uploading Files");
                var docs = map.Select(x => x.Value.Item1);
                new Client<Document>().Save(docs, "Uploaded By File Selection");

                progress.Report("Creating Document Sets");
                var sets = map.Select(x => x.Value.Item2);
                new Client<JobDocumentSet>().Save(sets, "Uploaded by File Selection");
                
                progress.Report("Creating MileStones");
                foreach (var key in map.Keys)
                    map[key].Item3.DocumentSet.ID = map[key].Item2.ID; 
                var milestones = map.Select(x => x.Value.Item3);
                new Client<JobDocumentSetMileStone>().Save(milestones, "Uploaded by File Selection");
                
                progress.Report("Linking Documents");
                foreach (var key in map.Keys)
                {
                    map[key].Item4.EntityLink.ID = map[key].Item3.ID;
                    map[key].Item4.DocumentLink.ID = map[key].Item1.ID;
                }
                var files = map.Select(x => x.Value.Item4);
                new Client<JobDocumentSetMileStoneFile>().Save(files, "Uploaded by File Selection");
                
            });
            MessageBox.Show(String.Format("{0} Document Sets Created", filenames.Length));
            Refresh();
        }

        private void AddManyToOneFiles(Guid type)
        {
            if (!SelectFiles(out String[] filenames))
                return;

            JobDocumentSet set = new JobDocumentSet();
            set.Job.ID = JobID;
            set.Folder.ID = FolderID;
            set.Discipline.ID = DisciplineID;
            set.Type.ID = TypeID;
            set.Category.ID = CategoryID;
            set.Area.ID = AreaID;
            
            var grid = new DynamicDataGrid<JobDocumentSet>();
            grid.OnAfterSave += (form, items) =>
            {
                Progress.ShowModal("Creating MileStone", (progress) =>
                {
                    JobDocumentSetMileStone milestone = new JobDocumentSetMileStone();
                    milestone.DocumentSet.ID = set.ID;
                    milestone.Type.ID = type;
                    milestone.Status = JobDocumentSetMileStoneStatus.InProgress;
                    milestone.Due = DateTime.Today;
                    new Client<JobDocumentSetMileStone>().Save(milestone, "Uploaded By File Selection");

                    progress.Report("Uploading Files");
                    List<Document> documents = new List<Document>();
                    foreach (var filename in filenames)
                    {
                        var data = File.ReadAllBytes(filename);
                        Document document = new Document()
                        {
                            FileName = Path.GetFileName(filename).ToLower(),
                            Data = data,
                            CRC = CoreUtils.CalculateCRC(data),
                            TimeStamp = new FileInfo(filename).LastWriteTime
                        };
                        documents.Add(document);
                        new Client<Document>().Save(documents, "Uploaded by File Selection");
                    }

                    progress.Report("Creating File Links");
                    List<JobDocumentSetMileStoneFile> files = new List<JobDocumentSetMileStoneFile>();
                    foreach (var document in documents)
                    {
                        JobDocumentSetMileStoneFile file = new JobDocumentSetMileStoneFile();
                        file.EntityLink.ID = milestone.ID;
                        file.DocumentLink.ID = document.ID;
                        files.Add(file);
                    }

                    new Client<JobDocumentSetMileStoneFile>().Save(files, "Uploaded by File Selection");
                });
            };
            if (grid.EditItems(new[] { set }))
            {
                MessageBox.Show(String.Format("{0} files uploaded", filenames.Length));
                Refresh();
            }
            
        }

        private void AddDocumentSet()
        {
            JobDocumentSet set = new JobDocumentSet();
            set.Job.ID = JobID;
            set.Folder.ID = FolderID;
            set.Discipline.ID = DisciplineID;
            set.Type.ID = TypeID;
            set.Category.ID = CategoryID;
            set.Area.ID = AreaID;
            
            var grid = new DynamicDataGrid<JobDocumentSet>();
            if (grid.EditItems(new[] { set }))
                Refresh();
        }

        private void Edit_OnClick(object sender, RoutedEventArgs e)
        {
            if (treeGrid.SelectedItem == null)
            {
                MessageBox.Show("Please choose a Document Set first");
                return;
            }

            Guid[] setIDs = treeGrid.SelectedItems.Select(x => (x as DocumentSetNode).ID).ToArray();

            EditDocumentSets(setIDs);
        }

        private void EditDocumentSets(Guid[] setIDs)
        {
            var sets = new Client<JobDocumentSet>().Query(
                new Filter<JobDocumentSet>(x => x.ID).InList(setIDs)
            ).Rows.Select(x => x.ToObject<JobDocumentSet>()).ToArray();
            var grid = new DynamicDataGrid<JobDocumentSet>();
            grid.OnCustomiseEditor += (form, items, column, editor) =>
            {
                if (String.Equals(column.ColumnName, "Discipline.ID"))
                    editor.Editable = DisciplineVisible ? Editable.Enabled : Editable.Hidden;
                if (String.Equals(column.ColumnName, "Type.ID"))
                    editor.Editable = TypeVisible ? Editable.Enabled : Editable.Hidden;
                if (String.Equals(column.ColumnName, "Category.ID"))
                    editor.Editable = CategoryVisible ? Editable.Enabled : Editable.Hidden;
                if (String.Equals(column.ColumnName, "Area.ID"))
                    editor.Editable = AreaVisible ? Editable.Enabled : Editable.Hidden;
            };
            if (grid.EditItems(sets))
                UpdateNodes(sets);
        }

        private void UpdateNodes(IEnumerable<JobDocumentSet> sets)
        {
            if (_documentsets == null)
                return;
            foreach (var set in sets)
            {
                var node = _documentsets.GetNode(set.ID);
                if (node != null)
                {
                    JobDocumentSetDescriptionBlock desc = new JobDocumentSetDescriptionBlock()
                    {
                        ID = set.ID,
                        Code = set.Code,
                        Description = set.Description,
                    };
                    node.Description = Serialization.Serialize(desc);

                    JobDocumentSetDetailsBlock dets = new JobDocumentSetDetailsBlock()
                    {
                        ID = set.ID,
                        Date = set.Date,
                        Size = set.Size,
                        Scale = set.Scale,
                        Employee = set.Employee.Name
                    };
                    node.Details = Serialization.Serialize(dets);
                }
            }
            
        }

        private void HideRejected_OnClick(object sender, RoutedEventArgs e)
        {
            _hidesuperceded = !_hidesuperceded;
            HideSupercededLabel.Content = _hidesuperceded ? "Show All" : "Last Only";
            Refresh();
        }

        private void Delete_OnClick(object sender, RoutedEventArgs e)
        {
            if ((treeGrid.SelectedItems == null) || !treeGrid.SelectedItems.Any())
            {
                MessageBox.Show("Please choose a Document Set first");
                return;
            }
            
            if (MessageBox.Show(
                    "Are you sure you wish to delete the selected Document Sets?", 
                    "Confirm Delete", 
                    MessageBoxButton.YesNo
                ) != MessageBoxResult.Yes)
                return;
            
            List<JobDocumentSet> updates = new List<JobDocumentSet>();
            List<DocumentSetNode> orphans = new List<DocumentSetNode>();
            
            var items = treeGrid.SelectedItems.Select(x => (DocumentSetNode)x).ToArray();
            foreach (DocumentSetNode item in items)
            {
                var children = item.Children.Where(x => !items.Contains(x));
                if (children.Any())
                orphans.AddRange(children);
            }

            if (orphans.Any())
            {
                var confirm = MessageBox.Show(
                    "These Document Sets contain children!\nDo you wish to delete these as well?",
                    "Delete Children",
                    MessageBoxButton.YesNoCancel
                );
                if (confirm == MessageBoxResult.Cancel)
                    return;
                if (confirm == MessageBoxResult.No)
                {
                    foreach (var orphan in orphans)
                    {
                        var update = new JobDocumentSet();
                        update.ID = orphan.ID;
                        update.Parent.ID = Guid.Empty;
                        updates.Add(update);
                    }
                    return;
                }
            }
            Progress.ShowModal("Deleting Document Set",(progress) =>
            {
                if (updates.Any())
                    new Client<JobDocumentSet>().Save(updates, "Parent Document Deleted");
                var deletes = items.Select(x=>new JobDocumentSet() { ID = x.ID }).ToArray();
                new Client<JobDocumentSet>().Delete(deletes, "Deleted By User");               
            });
            Refresh();
        }
        
        #endregion

        private void FlatList_OnClick(object sender, RoutedEventArgs e)
        {
            _flatlist = !_flatlist;
            FlatListLabel.Content = _flatlist ? "Tree View" : "Flat List";
            Refresh();
        }

        private void IncludeRetired_OnClick(object sender, RoutedEventArgs e)
        {
            _includeretired = !_includeretired;
            FlatListLabel.Content = _includeretired ? "Active Only" : "Include Retired";
            Refresh();
        }

        private void TreeGrid_OnSelectionChanged(object? sender, GridSelectionChangedEventArgs e)
        {
            //var treeColumn = treeGrid.Columns[e.CurrentRowColumnIndex.ColumnIndex];
            //var column = treeColumn.MappingName.Replace("Blocks[","").Replace("]","");
            // var column = e.Column.MappingName.Replace("Blocks[","").Replace("]","");
            // var data = (e.Record as DocumentSetNode).Blocks[column];
            // if (String.IsNullOrWhiteSpace(data))
            //     return;
            //
            // var block = Serialization.Deserialize<JobDocumentSetMileStoneBlock>(data.ToString());
            // Guid id = block.ID;           
        }

        private void TreeGrid_OnCurrentCellActivated(object? sender, CurrentCellActivatedEventArgs e)
        {

            var node = treeGrid.CurrentItem as DocumentSetNode;
            if (node == null)
                return;
            
            
            var treeColumn = treeGrid.Columns[e.CurrentRowColumnIndex.ColumnIndex];
            var column = treeColumn.MappingName.Replace("Blocks[","").Replace("]","");
            
            if (!node.Blocks.ContainsKey(column))
                MileStoneSelected(null);
            else
            {
                var block = Serialization.Deserialize<JobDocumentSetMileStoneBlock>(node.Blocks[column]);
                MileStoneSelected?.Invoke(block);               
            }

            
        }

        private void TreeGrid_OnCellDoubleTapped(object? sender, TreeGridCellDoubleTappedEventArgs e)
        {
            var set = e.Record as DocumentSetNode;
            if (set != null)
                EditDocumentSets(new Guid[] { set.ID });
        }
    }
}