25 lines
689 B
TypeScript
25 lines
689 B
TypeScript
import type { EnumerateFlags } from '.'
|
|
|
|
export function useIterator<S, F>(
|
|
value: S,
|
|
IterConstructor: { new (value: S): Iterator<F> },
|
|
): EnumerateFlags<F> {
|
|
const enumerate = {
|
|
_value: value,
|
|
|
|
[Symbol.iterator]: function () {
|
|
return new IterConstructor(this._value)
|
|
},
|
|
|
|
forEach: function (callback: (value: F) => void, thisArg?: any) {
|
|
const iter = new IterConstructor(this._value)
|
|
let result = iter.next()
|
|
while (!result.done) {
|
|
callback.call(thisArg, result.value)
|
|
result = iter.next()
|
|
}
|
|
},
|
|
}
|
|
return enumerate as EnumerateFlags<F>
|
|
}
|