♻️ refactor(multicombinations): Extract indices generator.

Author: make-github-pseudonymous-again

src/_multicombinations.js

@@ -0,0 +1,38 @@
+import {list} from '@iterable-iterator/list';
+import {nrepeat} from '@iterable-iterator/repeat';
+
+/**
+ * Yields all k-multicombinations of {0, 1, ..., n-1}.
+ *
+ * @param {number} n
+ * @param {number} r
+ * @returns {IterableIterator<number[]>}
+ */
+export default function* multicombinations(n, r) {
+	if (n === 0 && r > 0) return;
+
+	const indices = list(nrepeat(0, r));
+
+	yield indices;
+
+	while (true) {
+		let i = r - 1;
+
+		// eslint-disable-next-line no-constant-condition
+		while (true) {
+			if (i < 0) return;
+
+			if (indices[i] !== n - 1) {
+				const pivot = ++indices[i];
+
+				for (++i; i < r; ++i) indices[i] = pivot;
+
+				break;
+			}
+
+			--i;
+		}
+
+		yield indices;
+	}
+}

src/multicombinations.js

@@ -1,6 +1,7 @@
 import {list} from '@iterable-iterator/list';
-import {pick} from '@iterable-iterator/map';
-import {nrepeat} from '@iterable-iterator/repeat';
+import {map, pick} from '@iterable-iterator/map';
+
+import _multicombinations from './_multicombinations.js';
 
 /**
  * Yields all combinations, with repetitions, of each possible choice of
@@ -18,40 +19,12 @@ import {nrepeat} from '@iterable-iterator/repeat';
  * @param {number} r - The size of the combinations to generate.
  * @returns {IterableIterator}
  */
-export default function* multicombinations(iterable, r) {
+const multicombinations = (iterable, r) => {
 	const pool = list(iterable);
-	const length = pool.length;
-
-	if (length === 0 && r > 0) {
-		return;
-	}
-
-	const indices = list(nrepeat(0, r));
-
-	yield list(pick(pool, indices));
-
-	while (true) {
-		let i = r - 1;
-
-		// eslint-disable-next-line no-constant-condition
-		while (true) {
-			if (i < 0) {
-				return;
-			}
-
-			if (indices[i] !== length - 1) {
-				const pivot = ++indices[i];
-
-				for (++i; i < r; ++i) {
-					indices[i] = pivot;
-				}
-
-				break;
-			}
-
-			--i;
-		}
+	return map(
+		(indices) => list(pick(pool, indices)),
+		_multicombinations(pool.length, r),
+	);
+};
 
-		yield list(pick(pool, indices));
-	}
-}
+export default multicombinations;