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 Job Job { get; set; }

    public bool HasAllItems { get; set; } = true;
    
    public JobDocumentSetFolderTree() : base()
    {
        Options.AddRange(DynamicTreeOption.Add, DynamicTreeOption.Edit, DynamicTreeOption.Delete);
        MaxRowHeight = 30D;
        OnContextMenuOpening += JobDocumentSetFolderTree_OnContextMenuOpening;
    }

    private void JobDocumentSetFolderTree_OnContextMenuOpening(CoreTreeNode 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(Job.ID),
                new Columns<JobDocumentSetFolder>(
                    x => x.ID, x => x.Parent.ID,
                    x => x.Name, x => x.Documents)
            );

            if (HasAllItems)
            {
                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 ((Job.ID == Guid.Empty) || (Job.ID == CoreUtils.FullGuid))
            throw new Exception("Please Select a Job First!");
        
        var folder = base.DoCreateItem(parent);
        folder.Job.ID = Job.ID;
        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;
    }
}