extra-iterator 0.9.0 → 0.9.2

package/dist/index.d.ts

@@ -106,7 +106,7 @@ export declare class ExtraIterator<T> extends Iterator<T, any, any> {
      *     .toArray()
      *     // returns [1, 2, 3, 4]
      */
-    append(item: T): ExtraIterator<T>;
+    append<U>(item: U): ExtraIterator<T | U>;
     /**
      * Prepends a new value to the beginning of the iterator. The new value will be yielded first, then the rest of this
      * iterator will be yielded.
@@ -118,7 +118,7 @@ export declare class ExtraIterator<T> extends Iterator<T, any, any> {
      *     .toArray()
      *     // returns [0, 1, 2, 3]
      */
-    prepend(item: T): ExtraIterator<T>;
+    prepend<U>(item: U): ExtraIterator<T | U>;
     /**
      * Concatenates multiple values to the end of this iterator.
      *
@@ -129,7 +129,7 @@ export declare class ExtraIterator<T> extends Iterator<T, any, any> {
      *     .toArray()
      *     // returns [1, 2, 3, 4, 5, 6]
      */
-    concat(items: Iterable<T>): ExtraIterator<T>;
+    concat<U>(items: Iterable<U>): ExtraIterator<T | U>;
     /**
      * Concatenates multiple values to the start of this iterator.
      *
@@ -144,7 +144,7 @@ export declare class ExtraIterator<T> extends Iterator<T, any, any> {
      *     .toArray()
      *     // returns [1, 2, 3, 4, 5, 6]
      */
-    prependMany(items: Iterable<T>): ExtraIterator<T>;
+    prependMany<U>(items: Iterable<U>): ExtraIterator<T | U>;
     /**
      * Creates a new iterator that invokes the provided callback function over each element of this iterator and yields
      * the elements for which the callback returns `true`, only for as long as the callback returns `true`.

package/dist/index.js

@@ -137,7 +137,7 @@ export class ExtraIterator extends Iterator {
      * @example ExtraIterator.from([[1, 2], [3, 4]]).flatten().toArray() // returns [1, 2, 3, 4]
      */
     flatten() {
-        return this.flatMap(value => value);
+        return this.flatMap(value => Array.isArray(value) ? new ExtraIterator(value).flatten() : [value]);
     }
     /**
      * Creates a new iterator that yields the values of this iterator, but won't yield any duplicates.

package/dist/index.test.js

@@ -47,8 +47,8 @@ describe('ExtraIterator', () => {
         expect(iterator.toArray()).toEqual([3, 4]);
     });
     it('should flatten nested iterables', () => {
-        const iterator = ExtraIterator.from([[1, 2], [3, 4]]).flatten();
-        expect(iterator.toArray()).toEqual([1, 2, 3, 4]);
+        const iterator = ExtraIterator.from([0, [1, [2, [3, 4]]], [5, [6]], 7]).flatten();
+        expect(iterator.toArray()).toEqual([0, 1, 2, 3, 4, 5, 6, 7]);
     });
     it('should return the first value', () => {
         const iterator = ExtraIterator.from([1, 2, 3]);

package/package.json

@@ -3,7 +3,7 @@
   "description": "An extension of the Iterator class with additional utility helper functions.",
   "author": "Leonardo Raele <leonardoraele@gmail.com>",
   "license": "MIT",
-  "version": "0.9.0",
+  "version": "0.9.2",
   "type": "module",
   "exports": {
     ".": "./dist/index.js",