Umbraco-CMS/LinqToUmbraco/src/umbraco.Linq/Core/Node/NodeAssociationTree.cs

using System;
using System.Collections.Generic;
using System.Linq;

namespace umbraco.Linq.Core.Node
{
    /// <summary>
    /// Represents a collection of TDocTypeBase retrieved from the umbraco XML cache which are direct children of a node
    /// </summary>
    /// <typeparam name="TDocTypeBase">The type of the doc type base.</typeparam>
    public sealed class NodeAssociationTree<TDocTypeBase> : AssociationTree<TDocTypeBase> where TDocTypeBase : DocTypeBase, new()
    {
        private object lockObject = new object();
        private IEnumerable<TDocTypeBase> _nodes;

        internal NodeAssociationTree(IEnumerable<TDocTypeBase> nodes)
        {
            this._nodes = new List<TDocTypeBase>();
            this._nodes = nodes;
        }

        /// <summary>
        /// Initializes a new instance of the <see cref="NodeAssociationTree&lt;TDocTypeBase&gt;"/> class for a particular tree section
        /// </summary>
        /// <param name="parentNodeId">The parent node id to start from.</param>
        /// <param name="provider">The NodeDataProvider to link the tree with.</param>
        public NodeAssociationTree(int parentNodeId, NodeDataProvider provider)
        {
            this.Provider = provider;
            this.ParentNodeId = parentNodeId;            
        }

        /// <summary>
        /// Gets the enumerator for this Tree collection
        /// </summary>
        /// <returns></returns>
        public override IEnumerator<TDocTypeBase> GetEnumerator()
        {
            if (this._nodes == null) //first access, otherwise it'd be cached
            {
                LoadNodes();
            }
            return this._nodes.GetEnumerator();
        }

        private void LoadNodes()
        {
            var provider = this.Provider as NodeDataProvider;

            provider.CheckDisposed();

            lock (lockObject)
            {
                var parents = provider
                    .Xml
                    .Descendants()
                    .Where(x => x.Attribute("id") != null && (int)x.Attribute("id") == this.ParentNodeId);
                var rawNodes = parents
                    .Single()
                    .Elements()
                    .Where(x => x.Attribute("isDoc") != null)
                    ;
                this._nodes = provider.DynamicNodeCreation(rawNodes).Cast<TDocTypeBase>(); //drop is back to the type which was asked for 
            }
        }

        /// <summary>
        /// Indicates that the NodeAssociationTree is ReadOnly
        /// </summary>
        /// <value>
        /// 	<c>true</c>
        /// </value>
        public override bool IsReadOnly
        {
            get
            {
                return true;
            }
        }

        /// <summary>
        /// Gets or sets the DataProvider associated with this Tree
        /// </summary>
        /// <value>The provider.</value>
        public override UmbracoDataProvider Provider { get; protected set; }

        /// <summary>
        /// Reloads the cache.
        /// </summary>
        public override void ReloadCache()
        {
            this.LoadNodes();
        }

        public override void InsertOnSubmit(TDocTypeBase item)
        {
            throw new NotImplementedException("The NodeAssociationTree does not support Inserting items");
        }

        public override void InsertAllOnSubmit(IEnumerable<TDocTypeBase> items)
        {
            throw new NotImplementedException("The NodeAssociationTree does not support Inserting items");
        }

        public override void DeleteOnSubmit(TDocTypeBase itemm)
        {
            throw new NotImplementedException("The NodeAssociationTree does not support Deleting items");
        }

        public override void DeleteAllOnSubmit(IEnumerable<TDocTypeBase> items)
        {
            throw new NotImplementedException("The NodeAssociationTree does not support Deleting items");
        }
    }
}