Cpp2IL/Cpp2IL.Core/Extensions/EnumerableExtensions.cs

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

namespace Cpp2IL.Core.Extensions;

internal static class EnumerableExtensions
{
    public static IEnumerable<T> MaybeAppend<T>(this IEnumerable<T> enumerable, T? item) where T : struct
    {
        if (item is not null)
        {
            return enumerable.Append(item.Value);
        }

        return enumerable;
    }

    public static MemoryEnumerable<T> AsEnumerable<T>(this Memory<T> memory) => new(memory);

    public static MemoryEnumerator<T> GetEnumerator<T>(this Memory<T> memory) => new(memory);

    public class MemoryEnumerable<T>(Memory<T> memory) : IEnumerable<T>
    {
        public IEnumerator<T> GetEnumerator() => new MemoryEnumerator<T>(memory);

        IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
    }

    public class MemoryEnumerator<T>(Memory<T> memory) : IEnumerator<T>
    {
        private int _index = -1;

        public bool MoveNext()
        {
            _index++;
            return _index < memory.Length;
        }

        public void Reset()
        {
            _index = -1;
        }

        public T Current => memory.Span[_index];

        object? IEnumerator.Current => Current;

        public void Dispose()
        {
            // Nothing to dispose
        }
    }
}