Coroutine action for NPBehave

CoroutineAction.cs

using NPBehave;
using System.Collections;

public class CoroutineAction : Task
{
    System.Func<IEnumerator> starter;
    IEnumerator coro;

    public CoroutineAction(System.Func<IEnumerator> starter) : base("CoroutineAction") {
        this.starter = starter;
        this.coro = null;
    }

    protected override void DoStart() {
        coro = starter();

        if (!coro.MoveNext()) Stopped(true);
        else if (coro.Current == null) RootNode.Clock.AddUpdateObserver(OnUpdate);
        else Stopped(false);
    }

    void OnUpdate() {
        if (!coro.MoveNext()) {
            RootNode.Clock.RemoveUpdateObserver(OnUpdate);
            Stopped(true);
        }
        else if (coro.Current != null) {
            RootNode.Clock.RemoveUpdateObserver(OnUpdate);
            Stopped(false);
        }
    }

    protected override void DoStop() {
        coro = null;
        RootNode.Clock.RemoveUpdateObserver(OnUpdate);
        Stopped(false);
    }
}