From a1f43287a01b695ca50620651176387c2bd2940b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Herv=C3=A9=20BECHER?= <hbdu68@gmail.com>
Date: Thu, 9 May 2024 16:28:05 +0200
Subject: [PATCH] allow creating RandomPicker from iterable and separately
 specified length

---
 src/random.ts | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/random.ts b/src/random.ts
index 99e98b3..f693253 100644
--- a/src/random.ts
+++ b/src/random.ts
@@ -91,20 +91,21 @@ export class RandomPicker<T> {
   readonly #flags: BitArray;
   readonly #options: Required<RandomOptions<T>>;
 
-  public constructor(elements: T[], options?: RandomOptions<T>) {
+  public constructor(elements: Iterable<T>, length: number, options?: RandomOptions<T>) {
     this.#elements = elements;
-    this.#flags = BitArray.create(elements.length);
+    this.#flags = BitArray.create(length);
     this.#options = withDefaultOptions(mergeOptions({ predicate: i => this.#flags.get(i) }, options));
 
     this.reset();
   }
 
   public static from<T>(iterable: Iterable<T>, options?: RandomOptions<T>) {
-    return new this<T>(asArray(iterable), options);
+    const arr = asArray(iterable);
+    return new this<T>(arr, arr.length, options);
   }
 
   public static of<T>(options?: RandomOptions<T>, ...values: T[]) {
-    return new this<T>(values, options);
+    return new this<T>(values, values.length, options);
   }
 
   public get state() {