EnvDTE.Client/Impl/AstImpl/CodeElementsImplementation.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using EnvDTE;
using JetBrains.Annotations;
using JetBrains.EnvDTE.Client.Util;
using JetBrains.Rider.Model;

namespace JetBrains.EnvDTE.Client.Impl.AstImpl
{
    public sealed class CodeElementsImplementation : CodeElements
    {
        [NotNull]
        private DteImplementation Implementation { get; }

        [NotNull]
        private EnvDTEElementRegistrar Registrar { get; }

        [NotNull, ItemNotNull]
        private IReadOnlyList<CodeElementModel> CodeElementModels { get; }

        public CodeElementsImplementation(
            [NotNull] EnvDTEElementRegistrar registrar,
            [NotNull, ItemNotNull] IReadOnlyList<CodeElementModel> codeElementModels,
            [CanBeNull] object parent
        )
        {
            Implementation = registrar.Implementation;
            Parent = parent;
            Registrar = registrar;
            CodeElementModels = codeElementModels;
        }

        [NotNull]
        public DTE DTE => Implementation;

        public object Parent { get; }
        public int Count => CodeElementModels.Count;

        public IEnumerator GetEnumerator() => CodeElementModels
            .Select(model => Registrar.Convert(model, Parent))
            .GetEnumerator();

        [CanBeNull]
        public CodeElement Item([NotNull] object index)
        {
            if (!(index is int number)) throw new ArgumentException();
            return Registrar.Convert(CodeElementModels[number], Parent);
        }

        public void Reserved1(object Element) => throw new NotImplementedException();
        public bool CreateUniqueID(string Prefix, string NewName = "0") => throw new NotImplementedException();
    }
}