56 lines
1.8 KiB
C#
56 lines
1.8 KiB
C#
using System.Collections;
|
|
|
|
namespace PkmnLib.Dynamic.ScriptHandling;
|
|
|
|
/// <summary>
|
|
/// Iterator to figure out the scripts to run.
|
|
/// </summary>
|
|
public class ScriptIterator : IEnumerable<ScriptContainer>
|
|
{
|
|
private readonly IReadOnlyList<IEnumerable<ScriptContainer>> _scripts;
|
|
private int _currentIndex = 0;
|
|
|
|
/// <inheritdoc cref="ScriptIterator"/>
|
|
public ScriptIterator(IReadOnlyList<IEnumerable<ScriptContainer>> scripts)
|
|
{
|
|
_scripts = scripts;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public IEnumerator<ScriptContainer> GetEnumerator()
|
|
{
|
|
while (_currentIndex < _scripts.Count)
|
|
{
|
|
var enumerable = _scripts[_currentIndex++];
|
|
switch (enumerable)
|
|
{
|
|
case IScriptSet set:
|
|
{
|
|
// We can't assume that the set remains the same size, so we need to iterate through it
|
|
// with a for loop.
|
|
for (var j = 0; j < set.Count; j++)
|
|
{
|
|
var script = set.At(j);
|
|
// We can ignore empty scripts.
|
|
if (script.IsEmpty)
|
|
continue;
|
|
yield return script;
|
|
}
|
|
break;
|
|
}
|
|
case ScriptContainer { IsEmpty: true }:
|
|
continue;
|
|
case ScriptContainer container:
|
|
yield return container;
|
|
break;
|
|
default:
|
|
throw new InvalidOperationException(
|
|
$"Unexpected script type: {enumerable.GetType().Name}. Expected IScriptSet or ScriptContainer.");
|
|
}
|
|
}
|
|
_currentIndex = 0;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
|
|
} |