Skip to content

Instantly share code, notes, and snippets.

@i-sync
Forked from kamsar/MultiRootTreeList.cs
Created November 2, 2021 03:51
Show Gist options
  • Save i-sync/51c86757ad95771f5cb21c325f72ed78 to your computer and use it in GitHub Desktop.
Save i-sync/51c86757ad95771f5cb21c325f72ed78 to your computer and use it in GitHub Desktop.
Multi-Root TreeList
using System;
using System.Linq;
using Sitecore.Diagnostics;
using Sitecore.Globalization;
using Sitecore.Shell.Applications.ContentEditor;
using Sitecore.Text;
using Sitecore.Web;
using Sitecore.Web.UI.HtmlControls;
using Sitecore.Web.UI.WebControls;
namespace Kamsar.FieldTypes
{
/// <summary>
/// This field type is like a tree list, but you can specify more than one root item to select from (for example, videos or photos)
/// The data source roots are specified using pipe delimiting just like regular Sitecore Query language
/// </summary>
public class MultiRootTreeList : TreeList
{
protected override void OnLoad(EventArgs args)
{
Assert.ArgumentNotNull(args, "args");
base.OnLoad(args);
if (!Sitecore.Context.ClientPage.IsEvent)
{
// find the existing TreeviewEx that the base OnLoad added, get a ref to its parent, and remove it from controls
var existingTreeView = (TreeviewEx)WebUtil.FindControlOfType(this, typeof(TreeviewEx));
var treeviewParent = existingTreeView.Parent;
existingTreeView.Parent.Controls.Clear(); // remove stock treeviewex, we replace with multiroot
// find the existing DataContext that the base OnLoad added, get a ref to its parent, and remove it from controls
var dataContext = (DataContext)WebUtil.FindControlOfType(this, typeof(DataContext));
var dataContextParent = dataContext.Parent;
dataContextParent.Controls.Remove(dataContext); // remove stock datacontext, we parse our own
// create our MultiRootTreeview to replace the TreeviewEx
var impostor = new Sitecore.Web.UI.WebControls.MultiRootTreeview();
impostor.ID = existingTreeView.ID;
impostor.DblClick = existingTreeView.DblClick;
impostor.Enabled = existingTreeView.Enabled;
impostor.DisplayFieldName = existingTreeView.DisplayFieldName;
// parse the data source and create appropriate data contexts out of it
var dataContexts = ParseDataContexts(dataContext);
impostor.DataContext = string.Join("|", dataContexts.Select(x => x.ID));
foreach(var context in dataContexts) dataContextParent.Controls.Add(context);
// inject our replaced control where the TreeviewEx originally was
treeviewParent.Controls.Add(impostor);
}
}
/// <summary>
/// Parses multiple source roots into discrete data context controls (e.g. 'dataSource=/sitecore/content|/sitecore/media library')
/// </summary>
/// <param name="originalDataContext">The original data context the base control generated. We reuse some of its property values.</param>
/// <returns></returns>
protected virtual DataContext[] ParseDataContexts(DataContext originalDataContext)
{
return new ListString(DataSource).Select(x => CreateDataContext(originalDataContext, x)).ToArray();
}
/// <summary>
/// Creates a DataContext control for a given Sitecore path data source
/// </summary>
protected virtual DataContext CreateDataContext(DataContext baseDataContext, string dataSource)
{
DataContext dataContext = new DataContext();
dataContext.ID = GetUniqueID("D");
dataContext.Filter = baseDataContext.Filter;
dataContext.DataViewName = "Master";
if (!string.IsNullOrEmpty(DatabaseName))
{
dataContext.Parameters = "databasename=" + DatabaseName;
}
dataContext.Root = dataSource;
dataContext.Language = Language.Parse(ItemLanguage);
return dataContext;
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment