namespace Azusa.Shared.Extensions;

/// <summary>
/// 使用Range作为参数的迭代器方法
/// <br/>
/// 扩展foreach关键字来实现类似<c>foreach (var i in 1..5)</c>的效果
/// </summary>
public static class ForeachExtensions
{
    /// <summary>
    /// 拓展Range结构实现GetEnumerator方法供foreach读取，实现foreach(var i in x..y)
    /// </summary>
    /// <returns></returns>
    public static CustomIntEnumerator GetEnumerator(this Range range)
    {
        return new CustomIntEnumerator(range);
    }

    /// <summary>
    /// 拓展int类实现GetEnumerator方法供foreach读取，实现foreach(var i in x)
    /// </summary>
    /// <param name="end"></param>
    /// <returns></returns>
    public static CustomIntEnumerator GetEnumerator(this int end)
    {
        return new CustomIntEnumerator(end);
        return new CustomIntEnumerator(new Range(0, end));//在执行空函数时性能比上一句低10倍，Why
    }
}

//使用引用结构体增强性能
public ref struct CustomIntEnumerator
{
    private int _current;
    private readonly int _end;

    public CustomIntEnumerator(Range range)
    {
        //避免某些时候从结尾开始编制
        // x.. 时会产生Range(x,^0)
        if (range.End.IsFromEnd)
        {
            throw new NotSupportedException("不支持从结尾编制索引");
        }
        _current = range.Start.Value - 1;
        _end = range.End.Value - 1;//迭代器不包含范围的尾部
    }

    public CustomIntEnumerator(int end)
    {
        _current = -1;
        _end = end;
    }

    /* 注意，供foeach使用的迭代器不需要实现IEnumerator接口，只需要提供Current属性以及MoveNext方法即可，*/
    public int Current => _current;

    public bool MoveNext()
    {
        _current++;
        return _current <= _end;
    }
}