From e48af381c52533df53484c214d58aa4bddf05d54 Mon Sep 17 00:00:00 2001
From: mrjvs <jellevs@gmail.com>
Date: Mon, 18 Dec 2023 22:53:59 +0100
Subject: [PATCH] its not finished

---
 src/backend/helpers/providerApi.ts | 55 ++++++++++++++++++++++++++++++
 src/hooks/useProviderScrape.tsx    | 29 ++--------------
 2 files changed, 58 insertions(+), 26 deletions(-)
 create mode 100644 src/backend/helpers/providerApi.ts

diff --git a/src/backend/helpers/providerApi.ts b/src/backend/helpers/providerApi.ts
new file mode 100644
index 00000000..4292ee7d
--- /dev/null
+++ b/src/backend/helpers/providerApi.ts
@@ -0,0 +1,55 @@
+import { ScrapeMedia } from "@movie-web/providers";
+
+function scrapeMediaToQueryMedia(media: ScrapeMedia) {
+  let extra: Record<string, string> = {};
+  if (media.type === "show") {
+    extra = {
+      episodeNumber: media.episode.number.toString(),
+      episodeTmdbId: media.episode.tmdbId,
+      seasonNumber: media.season.number.toString(),
+      seasonTmdbId: media.season.tmdbId,
+    };
+  }
+
+  return {
+    type: media.type,
+    releaseYear: media.releaseYear.toString(),
+    imdbId: media.imdbId,
+    title: media.title,
+    ...extra,
+  };
+}
+
+function addQueryDataToUrl(url: URL, data: Record<string, string | undefined>) {
+  Object.entries(data).forEach((entry) => {
+    if (entry[1]) url.searchParams.set(entry[0], entry[1]);
+  });
+}
+
+export function makeProviderUrl(base: string) {
+  const makeUrl = (p: string) => new URL(`${base}${p}`);
+  return {
+    scrapeSource(sourceId: string, media: ScrapeMedia) {
+      const url = makeUrl("/scrape/source");
+      addQueryDataToUrl(url, scrapeMediaToQueryMedia(media));
+      addQueryDataToUrl(url, { id: sourceId });
+      return url.toString();
+    },
+    scrapeAll(media: ScrapeMedia) {
+      const url = makeUrl("/scrape");
+      addQueryDataToUrl(url, scrapeMediaToQueryMedia(media));
+      return url.toString();
+    },
+    scrapeEmbed(embedId: string, embedUrl: string) {
+      const url = makeUrl("/scrape/embed");
+      addQueryDataToUrl(url, { id: embedId, url: embedUrl });
+      return url.toString();
+    },
+  };
+}
+
+export function connectServerSideEvents(url: string, endEvents: string[]) {
+  const;
+
+  return {};
+}
diff --git a/src/hooks/useProviderScrape.tsx b/src/hooks/useProviderScrape.tsx
index 80ab63a4..4c16d6a8 100644
--- a/src/hooks/useProviderScrape.tsx
+++ b/src/hooks/useProviderScrape.tsx
@@ -5,6 +5,7 @@ import {
 } from "@movie-web/providers";
 import { RefObject, useCallback, useEffect, useRef, useState } from "react";
 
+import { makeProviderUrl } from "@/backend/helpers/providerApi";
 import { getLoadbalancedProviderApiUrl, providers } from "@/utils/providers";
 
 export interface ScrapingItems {
@@ -150,32 +151,8 @@ export function useScrape() {
         startScrape();
         const sseOutput = await new Promise<RunOutput | null>(
           (resolve, reject) => {
-            const finalUrl = new URL(`${providerApiUrl}/scrape`);
-            finalUrl.searchParams.append("type", media.type);
-            finalUrl.searchParams.append(
-              "releaseYear",
-              media.releaseYear.toString()
-            );
-            finalUrl.searchParams.append("title", media.title);
-            finalUrl.searchParams.append("tmdbId", media.tmdbId);
-            if (media.imdbId)
-              finalUrl.searchParams.append("imdbId", media.imdbId);
-            if (media.type === "show") {
-              finalUrl.searchParams.append(
-                "episodeNumber",
-                media.episode.number.toString()
-              );
-              finalUrl.searchParams.append(
-                "episodeTmdbId",
-                media.episode.tmdbId
-              );
-              finalUrl.searchParams.append(
-                "seasonNumber",
-                media.season.number.toString()
-              );
-              finalUrl.searchParams.append("seasonTmdbId", media.season.tmdbId);
-            }
-            const scrapeEvents = new EventSource(finalUrl.toString());
+            const baseUrlMaker = makeProviderUrl(providerApiUrl);
+            const scrapeEvents = new EventSource(baseUrlMaker.scrapeAll(media));
             scrapeEvents.addEventListener("init", (e) => {
               initEvent(JSON.parse(e.data));
             });