C:/Private/OSS/n2cms/github/n2cms/src/Framework/N2/Collections/BranchHierarchyBuilder.cs

Go to the documentation of this file.

namespace N2.Collections
{
        public class BranchHierarchyBuilder : HierarchyBuilder
        {
                private readonly ContentItem initialItem;
                private readonly ContentItem lastAncestor;
                bool appendAdditionalLevel = false;

                public BranchHierarchyBuilder(ContentItem initialItem, ContentItem lastAncestor, bool appendAdditionalLevel)
                {
                        this.initialItem = initialItem;
                        this.lastAncestor = lastAncestor;
                        this.appendAdditionalLevel = appendAdditionalLevel;
                }

                public BranchHierarchyBuilder(ContentItem initialItem, ContentItem lastAncestor)
                {
                        this.initialItem = initialItem;
                        this.lastAncestor = lastAncestor;
                }

                public override HierarchyNode<ContentItem> Build()
                {
                        if (initialItem == lastAncestor && !appendAdditionalLevel)
                        {
                                return new HierarchyNode<ContentItem>(initialItem);
                        }

                        HierarchyNode<ContentItem> previousNode = null;
                        foreach (ContentItem currentItem in Find.EnumerateParents(initialItem, lastAncestor, appendAdditionalLevel))
                        {
                                HierarchyNode<ContentItem> currentNode = new HierarchyNode<ContentItem>(currentItem);
                                if (previousNode != null)
                                {
                                        previousNode.Parent = currentNode;
                                }

                                foreach (ContentItem childItem in GetChildren(currentItem))
                                {
                                        if (previousNode != null && childItem.Equals(previousNode.Current))
                                        {
                                                currentNode.Children.Add(previousNode);
                                        }
                                        else
                                        {
                                                HierarchyNode<ContentItem> childNode = new HierarchyNode<ContentItem>(childItem);
                                                currentNode.Children.Add(childNode);
                                                childNode.Parent = currentNode;
                                        }
                                }
                                previousNode = currentNode;
                        }
                        return previousNode;
                }
        }
}