|
1 | 1 | import "../Iterator.prototype.toIterable"; |
2 | 2 |
|
3 | 3 | declare global { |
4 | | - // eslint-disable-next-line @typescript-eslint/no-unused-vars |
5 | | - class Iterator<T> { |
6 | | - static from<TElem>( |
7 | | - object: |
8 | | - | Iterator<TElem> |
9 | | - | Iterable<TElem> |
10 | | - | { next(): IteratorResult<TElem> }, |
11 | | - ): IterableIterator<TElem>; |
| 4 | + interface IteratorConstructor { |
| 5 | + from<T>( |
| 6 | + object: Iterator<T> | Iterable<T> | { next(): IteratorResult<T> }, |
| 7 | + ): IterableIterator<T>; |
12 | 8 | } |
| 9 | + |
| 10 | + const Iterator: IteratorConstructor; |
13 | 11 | } |
14 | 12 |
|
15 | 13 | (globalThis as Record<string, unknown>).Iterator ??= {}; |
16 | 14 |
|
17 | | -( |
18 | | - (globalThis as Record<string, unknown>).Iterator as Record<string, unknown> |
19 | | -).from ??= function <T>( |
20 | | - object: Iterator<T> | Iterable<T> | { next(): IteratorResult<T> }, |
21 | | -): IterableIterator<T> { |
22 | | - if (typeof (object as Iterable<T>)[Symbol.iterator] === "function") { |
23 | | - return (object as Iterable<T>)[Symbol.iterator]() as IterableIterator<T>; |
24 | | - } |
| 15 | +(globalThis as Record<string, unknown>).Iterator = { |
| 16 | + from<T>( |
| 17 | + object: Iterator<T> | Iterable<T> | { next(): IteratorResult<T> }, |
| 18 | + ): IterableIterator<T> { |
| 19 | + if (typeof (object as Iterable<T>)[Symbol.iterator] === "function") { |
| 20 | + const iterator = (object as Iterable<T>)[Symbol.iterator](); |
| 21 | + if (typeof iterator === "object" && Symbol.iterator in iterator) { |
| 22 | + return iterator as IterableIterator<T>; |
| 23 | + } else { |
| 24 | + return { |
| 25 | + ...iterator, |
| 26 | + [Symbol.iterator]() { |
| 27 | + return this; |
| 28 | + }, |
| 29 | + } as IterableIterator<T>; |
| 30 | + } |
| 31 | + } |
25 | 32 |
|
26 | | - if (typeof (object as Iterator<T>).toIterable === "function") { |
27 | | - return (object as Iterator<T>).toIterable(); |
28 | | - } |
| 33 | + if (typeof (object as Iterator<T>).toIterable === "function") { |
| 34 | + return (object as Iterator<T>).toIterable(); |
| 35 | + } |
29 | 36 |
|
30 | | - if (typeof (object as { next(): IteratorResult<T> }).next === "function") { |
31 | | - return (function* () { |
32 | | - let result: IteratorResult<T>; |
33 | | - do { |
34 | | - result = (object as { next(): IteratorResult<T> }).next(); |
35 | | - if (!result.done) { |
| 37 | + if (typeof (object as { next(): IteratorResult<T> }).next === "function") { |
| 38 | + return (function* () { |
| 39 | + let result: IteratorResult<T>; |
| 40 | + while (true) { |
| 41 | + result = (object as { next(): IteratorResult<T> }).next(); |
| 42 | + if (result.done) { |
| 43 | + break; |
| 44 | + } |
36 | 45 | yield result.value; |
37 | 46 | } |
38 | | - } while (!result.done); |
39 | | - })(); |
40 | | - } |
| 47 | + })(); |
| 48 | + } |
41 | 49 |
|
42 | | - throw new TypeError("Object is not an Iterator or Iterable"); |
| 50 | + throw new TypeError( |
| 51 | + "Object is not an iterator, iterable, or an object with a next method", |
| 52 | + ); |
| 53 | + }, |
43 | 54 | }; |
0 commit comments