prs/prs.mobile.new/PRS.Mobile/Data Models/Lists/JobDocumentFolder/JobDocumentFolderModel.cs

using System;
using Comal.Classes;
using InABox.Core;
using InABox.Mobile;

namespace PRS.Mobile
{
    public class JobDocumentFolderModel : ListModel<JobDocumentFolderModel, JobDocumentFolderShell, JobDocumentSetFolder>
    {
        public JobDocumentFolderModel(IModelHost host, Func<Filter<JobDocumentSetFolder>> filter, bool transient = false) : base(host, filter, transient)
        {
        }

        public JobDocumentFolderModel(IModelHost host, Func<Filter<JobDocumentSetFolder>> filter, string filename) : base(host, filter, filename)
        {
        }

        protected override void Initialize()
        {
            base.Initialize();
            DocumentFolders = new CoreTreeNodes();
        }

        public CoreTreeNodes DocumentFolders { get; private set; }

        public override void AfterLoad(MultiQuery query)
        {
            base.AfterLoad(query);
            
            DocumentFolders.Load<JobDocumentSetFolder>(
                query.Get<JobDocumentSetFolder>(), 
                x=>x.ID, 
                x=>x.Parent.ID, 
                x=>x.Name
            );
            var roots = DocumentFolders.GetChilden(Guid.Empty);
            foreach (var root in roots)
                root.Parent = CoreUtils.FullGuid;
            DocumentFolders.Add(CoreUtils.FullGuid, Guid.Empty).Description = "All Folders";
        }
    }
}