From 2e8d49f900435873e251f7498587cb3222028de3 Mon Sep 17 00:00:00 2001
From: Vanilagy <1696106+Vanilagy@users.noreply.github.com>
Date: Sat, 4 Jan 2025 01:11:01 +0100
Subject: [PATCH] Add SampleRetrievalOptions to samples iterator

---
 src/media-drain.ts | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/media-drain.ts b/src/media-drain.ts
index a4d4290..282f4dc 100644
--- a/src/media-drain.ts
+++ b/src/media-drain.ts
@@ -41,7 +41,11 @@ export abstract class BaseSampleDrain<Sample extends EncodedVideoSample | Encode
 	abstract getKeySample(timestamp: number, options?: SampleRetrievalOptions): Promise<Sample | null>;
 	abstract getNextKeySample(sample: Sample, options?: SampleRetrievalOptions): Promise<Sample | null>;
 
-	samples(startSample?: Sample, endTimestamp = Infinity): AsyncGenerator<Sample, void, unknown> {
+	samples(
+		startSample?: Sample,
+		endTimestamp = Infinity,
+		options?: SampleRetrievalOptions,
+	): AsyncGenerator<Sample, void, unknown> {
 		const sampleQueue: Sample[] = [];
 
 		let { promise: queueNotEmpty, resolve: onQueueNotEmpty } = promiseWithResolvers();
@@ -60,7 +64,7 @@ export abstract class BaseSampleDrain<Sample extends EncodedVideoSample | Encode
 
 		// The following is the "pump" process that keeps pumping samples into the queue
 		(async () => {
-			let sample = startSample ?? await this.getFirstSample();
+			let sample = startSample ?? await this.getFirstSample(options);
 
 			while (sample && !terminated) {
 				if (sample.timestamp >= endTimestamp) {
@@ -78,7 +82,7 @@ export abstract class BaseSampleDrain<Sample extends EncodedVideoSample | Encode
 				onQueueNotEmpty();
 				({ promise: queueNotEmpty, resolve: onQueueNotEmpty } = promiseWithResolvers());
 
-				sample = await this.getNextSample(sample);
+				sample = await this.getNextSample(sample, options);
 			}
 
 			ended = true;