From 8db85c545b7bb11812d4778dc1aa227730b8aa1a Mon Sep 17 00:00:00 2001
From: Adrian Castro <22133246+castdrian@users.noreply.github.com>
Date: Mon, 5 Feb 2024 19:49:00 +0100
Subject: [PATCH] feat: set headers in video player

---
 apps/expo/src/app/video-player.tsx   | 33 +++++++++++++++++++++++-----
 packages/provider-utils/src/video.ts | 22 +++++--------------
 2 files changed, 33 insertions(+), 22 deletions(-)

diff --git a/apps/expo/src/app/video-player.tsx b/apps/expo/src/app/video-player.tsx
index 3a9bf9f..60d05a3 100644
--- a/apps/expo/src/app/video-player.tsx
+++ b/apps/expo/src/app/video-player.tsx
@@ -7,6 +7,7 @@ import { useLocalSearchParams, useRouter } from "expo-router";
 import * as ScreenOrientation from "expo-screen-orientation";
 
 import {
+  findHighestQuality,
   getVideoUrl,
   transformSearchResultToScrapeMedia,
 } from "@movie-web/provider-utils";
@@ -28,6 +29,7 @@ interface VideoPlayerProps {
 
 const VideoPlayer: React.FC<VideoPlayerProps> = ({ data }) => {
   const [videoUrl, setVideoUrl] = useState("");
+  const [headers, setHeaders] = useState({} as Record<string, string>);
   const [isLoading, setIsLoading] = useState(true);
   const videoPlayer = useRef<VideoRef>(null);
   const router = useRouter();
@@ -56,19 +58,38 @@ const VideoPlayer: React.FC<VideoPlayerProps> = ({ data }) => {
           episode,
         );
 
-        const videoUrl = await getVideoUrl(scrapeMedia);
-        if (!videoUrl) {
+        const stream = await getVideoUrl(scrapeMedia);
+        if (!stream) {
           await ScreenOrientation.lockAsync(
             ScreenOrientation.OrientationLock.PORTRAIT_UP,
           );
           return router.push("/(tabs)");
         }
-        return videoUrl;
+        return stream;
       };
 
       setIsLoading(true);
-      const url = await fetchVideo();
-      if (url) {
+      const stream = await fetchVideo();
+
+      if (stream) {
+        let highestQuality;
+        let url;
+
+        switch (stream.type) {
+          case "file":
+            highestQuality = findHighestQuality(stream);
+            url = highestQuality ? stream.qualities[highestQuality]?.url : null;
+            return url ?? null;
+          case "hls":
+            url = stream.playlist;
+        }
+
+        const combinedHeaders = {
+          ...stream.headers,
+          ...stream.preferredHeaders,
+        };
+
+        setHeaders(combinedHeaders);
         setVideoUrl(url);
         setIsLoading(false);
       } else {
@@ -123,7 +144,7 @@ const VideoPlayer: React.FC<VideoPlayerProps> = ({ data }) => {
     <SafeAreaView style={styles.container}>
       <Video
         ref={videoPlayer}
-        source={{ uri: videoUrl }}
+        source={{ uri: videoUrl, headers }}
         style={styles.fullScreen}
         fullscreen={true}
         paused={false}
diff --git a/packages/provider-utils/src/video.ts b/packages/provider-utils/src/video.ts
index d483924..8c6bbb0 100644
--- a/packages/provider-utils/src/video.ts
+++ b/packages/provider-utils/src/video.ts
@@ -3,6 +3,7 @@ import type {
   Qualities,
   RunnerOptions,
   ScrapeMedia,
+  Stream,
 } from "@movie-web/providers";
 import {
   makeProviders,
@@ -10,7 +11,7 @@ import {
   targets,
 } from "@movie-web/providers";
 
-export async function getVideoUrl(media: ScrapeMedia): Promise<string | null> {
+export async function getVideoUrl(media: ScrapeMedia): Promise<Stream | null> {
   const providers = makeProviders({
     fetcher: makeStandardFetcher(fetch),
     target: targets.NATIVE,
@@ -23,23 +24,12 @@ export async function getVideoUrl(media: ScrapeMedia): Promise<string | null> {
 
   const results = await providers.runAll(options);
   if (!results) return null;
-
-  let highestQuality;
-  let url;
-
-  switch (results.stream.type) {
-    case "file":
-      highestQuality = findHighestQuality(results.stream);
-      url = highestQuality
-        ? results.stream.qualities[highestQuality]?.url
-        : null;
-      return url ?? null;
-    case "hls":
-      return results.stream.playlist;
-  }
+  return results.stream;
 }
 
-function findHighestQuality(stream: FileBasedStream): Qualities | undefined {
+export function findHighestQuality(
+  stream: FileBasedStream,
+): Qualities | undefined {
   const qualityOrder: Qualities[] = [
     "4k",
     "1080",