prs/prs.desktop/Panels/Jobs/DocumentSets/JobDocumentSetFolderTree.cs

using System;
using System.Linq;
using System.Linq.Expressions;
using System.Windows;
using System.Windows.Controls;
using Comal.Classes;
using InABox.Clients;
using InABox.Core;
using InABox.DynamicGrid;
using InABox.WPF;

namespace PRSDesktop
{
    public class JobDocumentSetFolderTree : DynamicTreeView<JobDocumentSetFolder>
    {
        
        public Guid JobID { get; set; }
        
        public JobDocumentSetFolderTree() : base()
        {
            Options.AddRange(DynamicTreeOption.Add, DynamicTreeOption.Edit, DynamicTreeOption.Delete);
            MaxRowHeight = 30D;
            OnContextMenuOpening += JobDocumentSetFolderTree_OnContextMenuOpening;

        }

        private void JobDocumentSetFolderTree_OnContextMenuOpening(DynamicTreeNode node, ContextMenu menu)
        {
            if(node.ID != CoreUtils.FullGuid)
            {
                menu.AddItem("Add Child Folder", null, node, (n) => DoAddItem(n.ID,true));
            }
        }

        protected override Expression<Func<JobDocumentSetFolder, Guid>> ID => x => x.ID;
        protected override Expression<Func<JobDocumentSetFolder, Guid>> ParentID => x => x.Parent.ID;
        protected override Expression<Func<JobDocumentSetFolder, string>> Description => x => x.Name;

        protected override void DoRefresh(Action<CoreTable?, Exception?> action)
        {
            try
            {
                var data = new Client<JobDocumentSetFolder>().Query(
                    new Filter<JobDocumentSetFolder>(x => x.Job.ID).IsEqualTo(JobID),
                    new Columns<JobDocumentSetFolder>(
                        x => x.ID, x => x.Parent.ID,
                        x => x.Name, x => x.Documents)
                );

                var newRow = data.NewRow();
                newRow.Set<JobDocumentSetFolder, Guid>(x => x.ID, CoreUtils.FullGuid);
                newRow.Set<JobDocumentSetFolder, Guid>(x => x.Parent.ID, Guid.Empty);
                newRow.Set<JobDocumentSetFolder, string>(x => x.Name, "All Items");
                newRow.Set<JobDocumentSetFolder, int>(x => x.Documents, int.MaxValue);

                data.Rows.Insert(0, newRow);
                action(data, null);
            }
            catch(Exception e)
            {
                action(null, e);
            }
        }
        
        protected override JobDocumentSetFolder DoCreateItem(Guid parent)
        {
            if ((JobID == Guid.Empty) || (JobID == CoreUtils.FullGuid))
                throw new Exception("Please Select a Job First!");
            
            var folder = base.DoCreateItem(parent);
            folder.Job.ID = JobID;
            folder.Name = "New Folder";
            return folder;
        }

        protected override JobDocumentSetFolder? DoLoadItem(Guid id)
        {
            if (id == CoreUtils.FullGuid)
            {
                MessageBox.Show("This item cannot be edited.");
                return null;
            }

            var folder = new Client<JobDocumentSetFolder>().Query(
                new Filter<JobDocumentSetFolder>(x => x.ID).IsEqualTo(id)
            ).Rows.FirstOrDefault()?.ToObject<JobDocumentSetFolder>();
            if(folder == null)
            {
                Logger.Send(LogType.Error, "", $"Cannot load JobDocumentSetFolder: {id} does not exist!");
            }
            return folder;
        }

        protected override void DoSaveItem(JobDocumentSetFolder folder)
        {
            if (folder.ID != CoreUtils.FullGuid)
            {
                new Client<JobDocumentSetFolder>().Save(folder, "Edited by User");
            }
        }

        protected override bool DoDeleteItem(Guid id)
        {
            if(id == CoreUtils.FullGuid)
            {
                MessageBox.Show("This folder cannot be deleted.");
                return false;
            }

            CoreRow? row = Data.Rows.FirstOrDefault(r=>r.Get<JobDocumentSetFolder, Guid>(x => x.ID) == id);
            if(row == null)
            {
                MessageBox.Show("Error: Row does not exist!");
                Logger.Send(LogType.Error, ClientFactory.UserID, $"Error: Job Document Set Folder {id} does not exist or is not loaded");
                return false;
            }

            if (row.Get<JobDocumentSetFolder, int>(x => x.Documents) > 0)
            {
                MessageBox.Show("This folder is not empty, and cannot be deleted!");
                return false;
            }
            new Client<JobDocumentSetFolder>().Delete(new JobDocumentSetFolder() { ID = id }, "Deleted by User");
            return true;
        }
    }
}