From e7337cc770e58423205db9cea5408eb77d0a01ef Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Edvin=20Hedenstr=C3=B6m?=
 <54530330+edvinhed@users.noreply.github.com>
Date: Sun, 6 Oct 2024 12:12:01 +0200
Subject: [PATCH] feat: support multiple bitrate values in stream proxy filter
 (#68)

* feat: support multiple bitrate values in stream proxy filter

Updated the stream proxy filter to parse both individual bitrate values and arrays of bitrate values for filtering.

- Modified the input type `br` in the filter function to accept:
  - `number`: Individual bitrate values.
  - `number[]`: Arrays of bitrate values.
  - `"*"`: A wildcard to match any bitrate.

This enhancement allows more flexible filtering based on bitrate specifications.

* chore: ran Prettier & Lint to format code

* Changed method of checking array

Instead of converting it, we now just check if it's an array and then uses .includes()
---
 README.md                      | 17 +++++++++++------
 src/manifests/utils/configs.ts | 17 +++++++++++++----
 2 files changed, 24 insertions(+), 10 deletions(-)

diff --git a/README.md b/README.md
index 91e1309..e0acbd4 100644
--- a/README.md
+++ b/README.md
@@ -121,7 +121,7 @@ Delay Corruption:
     sq?: number | "*", // media sequence number of target segment in playlist. If "*", then target all segments
     rsq?: number,      // relative sequence number from where a livestream is currently at
     ms?: number,       // time to delay in milliseconds
-    br?: number | "*", // apply only to specific bitrate
+    br?: number | number[] | "*", // apply only to specific or an array of specific bitrates
 }
 ```
 
@@ -133,7 +133,7 @@ Status Code Corruption:
     sq?: number | "*", // media sequence number of target segment in playlist. If "*", then target all segments
     rsq?: number,      // relative sequence number from where a livestream is currently at
     code?: number,     // code to return in http response status header instead of media file
-    br?: number | "*", // apply only to specific bitrate
+    br?: number | number[] | "*", // apply only to specific or an array of specific bitrates
 }
 ```
 
@@ -144,7 +144,7 @@ Timeout Corruption:
     i?: number | "*",  // index of target segment in playlist. If "*", then target all segments. (Starts on 0 for HLS / 1 for MPEG-DASH)
     sq?: number | "*", // media sequence number of target segment in playlist. If "*", then target all segments
     rsq?: number,      // relative sequence number from where a livestream is currently at
-    br?: number | "*", // apply only to specific bitrate
+    br?: number | number[] | "*", // apply only to specific or an array of specific bitrates
 }
 ```
 
@@ -155,7 +155,7 @@ Throttle Corruption:
     i?: number | "*",  // index of target segment in playlist. If "*", then target all segments. (Starts on 0 for HLS / 1 for MPEG-DASH)
     sq?: number | "*", // media sequence number of target segment in playlist. If "*", then target all segments
     rsq?: number,      // relative sequence number from where a livestream is currently at
-    br?: number | "*", // apply only to specific bitrate
+    br?: number | number[] | "*", // apply only to specific or an array of specific bitrates
     rate?: number      // rate in bytes per second to limit the segment download speed to
 }
 ```
@@ -204,14 +204,19 @@ https://<chaos-proxy>/api/v2/manifests/hls/proxy-master.m3u8?url=https://maitv-v
 https://<chaos-proxy>/api/v2/manifests/hls/proxy-master.m3u8?url=https://maitv-vod.lab.eyevinn.technology/VINN.mp4/master.m3u8&delay=[{i:5,ms:1500}]&statusCode=[{i:5,code:400,br:2426000}]
 
 ```
+7. VOD: With segment delay of 1500ms on sixth segment, followed by a response code 400 if the bitrate is 1212000 OR 3131000:
 
-7. LIVE: With response of status code 404 on segment with sequence number 105:
+```
+https://<chaos-proxy>/api/v2/manifests/hls/proxy-master.m3u8?url=https://maitv-vod.lab.eyevinn.technology/VINN.mp4/master.m3u8&delay=[{i:5,ms:1500}]&statusCode=[{i:5,code:400,br:[1212000,3131000]}]
+```
+
+8. LIVE: With response of status code 404 on segment with sequence number 105:
 
 ``` 
 https://<chaos-proxy>/api/v2/manifests/hls/proxy-master.m3u8?url=https://demo.vc.eyevinn.technology/channels/demo/master.m3u8&statusCode=[{sq:105,code:400}]
 ```
 
-8. LIVE: Delay response of media manifest ladder 1 and 2 with 500 ms
+9. LIVE: Delay response of media manifest ladder 1 and 2 with 500 ms
 
 ```
 https://<chaos-proxy>/api/v2/manifests/hls/proxy-master.m3u8?url=https://demo.vc.eyevinn.technology/channels/demo/master.m3u8&delay=[{l:1,ms:500},{l:2,ms:500}]
diff --git a/src/manifests/utils/configs.ts b/src/manifests/utils/configs.ts
index 8c0aa60..fd0dd9d 100644
--- a/src/manifests/utils/configs.ts
+++ b/src/manifests/utils/configs.ts
@@ -179,10 +179,19 @@ export const corruptorConfigUtils = function (
           }
 
           // If bitrate is set, filter out segments that doesn't match
-          params = params.filter(
-            (config) =>
-              !config?.br || config?.br === '*' || config?.br === segmentBitrate
-          );
+          params = params.filter((config) => {
+            if (
+              !config?.br ||
+              config?.br === '*' ||
+              config?.br === segmentBitrate
+            ) {
+              return true;
+            } else if (Array.isArray(config?.br)) {
+              return config?.br.includes(segmentBitrate);
+            } else {
+              return false;
+            }
+          });
 
           // Replace relative sequence numbers with absolute ones
           params = params.map((param) => {