From ca2e20fdbc541f7fc399cd73f469c6a8f86d77f8 Mon Sep 17 00:00:00 2001
From: mrjvs <jellevs@gmail.com>
Date: Tue, 19 Dec 2023 18:09:05 +0100
Subject: [PATCH] Add provider-api to source selection menu

---
 .../player/hooks/useSourceSelection.ts        | 62 +++++++++++++++----
 1 file changed, 49 insertions(+), 13 deletions(-)

diff --git a/src/components/player/hooks/useSourceSelection.ts b/src/components/player/hooks/useSourceSelection.ts
index 094dd458..b3cd5a8e 100644
--- a/src/components/player/hooks/useSourceSelection.ts
+++ b/src/components/player/hooks/useSourceSelection.ts
@@ -5,6 +5,10 @@ import {
 } from "@movie-web/providers";
 import { useAsyncFn } from "react-use";
 
+import {
+  connectServerSideEvents,
+  makeProviderUrl,
+} from "@/backend/helpers/providerApi";
 import {
   scrapeSourceOutputToProviderMetric,
   useReportProviders,
@@ -14,7 +18,7 @@ import { convertRunoutputToSource } from "@/components/player/utils/convertRunou
 import { useOverlayRouter } from "@/hooks/useOverlayRouter";
 import { metaToScrapeMedia } from "@/stores/player/slices/source";
 import { usePlayerStore } from "@/stores/player/store";
-import { providers } from "@/utils/providers";
+import { getLoadbalancedProviderApiUrl, providers } from "@/utils/providers";
 
 export function useEmbedScraping(
   routerId: string,
@@ -31,13 +35,23 @@ export function useEmbedScraping(
   const { report } = useReportProviders();
 
   const [request, run] = useAsyncFn(async () => {
+    const providerApiUrl = getLoadbalancedProviderApiUrl();
     let result: EmbedOutput | undefined;
     if (!meta) return;
     try {
-      result = await providers.runEmbedScraper({
-        id: embedId,
-        url,
-      });
+      if (providerApiUrl) {
+        const baseUrlMaker = makeProviderUrl(providerApiUrl);
+        const conn = connectServerSideEvents<EmbedOutput>(
+          baseUrlMaker.scrapeEmbed(embedId, url),
+          ["completed"]
+        );
+        result = await conn.promise();
+      } else {
+        result = await providers.runEmbedScraper({
+          id: embedId,
+          url,
+        });
+      }
     } catch (err) {
       console.error(`Failed to scrape ${embedId}`, err);
       const notFound = err instanceof NotFoundError;
@@ -85,13 +99,23 @@ export function useSourceScraping(sourceId: string | null, routerId: string) {
   const [request, run] = useAsyncFn(async () => {
     if (!sourceId || !meta) return null;
     const scrapeMedia = metaToScrapeMedia(meta);
+    const providerApiUrl = getLoadbalancedProviderApiUrl();
 
     let result: SourcererOutput | undefined;
     try {
-      result = await providers.runSourceScraper({
-        id: sourceId,
-        media: scrapeMedia,
-      });
+      if (providerApiUrl) {
+        const baseUrlMaker = makeProviderUrl(providerApiUrl);
+        const conn = connectServerSideEvents<SourcererOutput>(
+          baseUrlMaker.scrapeSource(sourceId, scrapeMedia),
+          ["completed"]
+        );
+        result = await conn.promise();
+      } else {
+        result = await providers.runSourceScraper({
+          id: sourceId,
+          media: scrapeMedia,
+        });
+      }
     } catch (err) {
       console.error(`Failed to scrape ${sourceId}`, err);
       const notFound = err instanceof NotFoundError;
@@ -120,10 +144,22 @@ export function useSourceScraping(sourceId: string | null, routerId: string) {
       let embedResult: EmbedOutput | undefined;
       if (!meta) return;
       try {
-        embedResult = await providers.runEmbedScraper({
-          id: result.embeds[0].embedId,
-          url: result.embeds[0].url,
-        });
+        if (providerApiUrl) {
+          const baseUrlMaker = makeProviderUrl(providerApiUrl);
+          const conn = connectServerSideEvents<EmbedOutput>(
+            baseUrlMaker.scrapeEmbed(
+              result.embeds[0].embedId,
+              result.embeds[0].url
+            ),
+            ["completed"]
+          );
+          embedResult = await conn.promise();
+        } else {
+          embedResult = await providers.runEmbedScraper({
+            id: result.embeds[0].embedId,
+            url: result.embeds[0].url,
+          });
+        }
       } catch (err) {
         console.error(`Failed to scrape ${result.embeds[0].embedId}`, err);
         const notFound = err instanceof NotFoundError;