From c1c5197d0e8b9fea938e776b0c6b54d29f4e0b30 Mon Sep 17 00:00:00 2001
From: Meier Lukas <meierschlumpf@gmail.com>
Date: Sat, 10 Jun 2023 16:00:52 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=8F=97=EF=B8=8F=20Migrate=20overseerr=20m?=
 =?UTF-8?q?edia=20requests=20to=20tRPC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/overseerr/RequestModal.tsx | 93 +++++++++++++++-----------
 src/server/api/routers/overseerr.ts    | 62 +++++++++++++++++
 2 files changed, 116 insertions(+), 39 deletions(-)

diff --git a/src/modules/overseerr/RequestModal.tsx b/src/modules/overseerr/RequestModal.tsx
index 74fed37be..227ab0e73 100644
--- a/src/modules/overseerr/RequestModal.tsx
+++ b/src/modules/overseerr/RequestModal.tsx
@@ -1,17 +1,16 @@
 import { Alert, Button, Checkbox, createStyles, Group, Modal, Stack, Table } from '@mantine/core';
 import { showNotification, updateNotification } from '@mantine/notifications';
 import { IconAlertCircle, IconCheck, IconDownload } from '@tabler/icons-react';
-import axios from 'axios';
 import Consola from 'consola';
 import { useTranslation } from 'next-i18next';
 
 import { useState } from 'react';
+import { useConfigContext } from '~/config/provider';
+import { api } from '~/utils/api';
 import { useColorTheme } from '../../tools/color';
 import { MovieResult } from './Movie.d';
-import { MediaType, Result } from './SearchResult.d';
+import { Result } from './SearchResult.d';
 import { TvShowResult, TvShowResultSeason } from './TvShow.d';
-import { api } from '~/utils/api';
-import { useConfigContext } from '~/config/provider';
 
 interface RequestModalProps {
   base: Result;
@@ -62,6 +61,7 @@ export function MovieRequestModal({
   setOpened: (opened: boolean) => void;
 }) {
   const { secondaryColor } = useColorTheme();
+  const requestMediaAsync = useMediaRequestMutation();
   const { t } = useTranslation('modules/overseerr');
 
   return (
@@ -97,7 +97,7 @@ export function MovieRequestModal({
           <Button
             variant="outline"
             onClick={() => {
-              askForMedia('movie', result.id, result.title, []);
+              requestMediaAsync('movie', result.id, result.title);
             }}
           >
             {t('popup.item.buttons.request')}
@@ -120,6 +120,7 @@ export function TvRequestModal({
   const [selection, setSelection] = useState<TvShowResultSeason[]>(result.seasons);
   const { classes, cx } = useStyles();
   const { t } = useTranslation('modules/overseerr');
+  const requestMediaAsync = useMediaRequestMutation();
 
   const toggleRow = (container: TvShowResultSeason) =>
     setSelection((current: TvShowResultSeason[]) =>
@@ -200,7 +201,7 @@ export function TvRequestModal({
             variant="outline"
             disabled={selection.length === 0}
             onClick={() => {
-              askForMedia(
+              requestMediaAsync(
                 'tv',
                 result.id,
                 result.name,
@@ -216,37 +217,51 @@ export function TvRequestModal({
   );
 }
 
-function askForMedia(type: MediaType, id: number, name: string, seasons?: number[]) {
-  Consola.info(`Requesting ${type} ${id} ${name}`);
-  showNotification({
-    title: 'Request',
-    id: id.toString(),
-    message: `Requesting media ${name}`,
-    color: 'orange',
-    loading: true,
-    autoClose: false,
-    withCloseButton: false,
-    icon: <IconAlertCircle />,
-  });
-  axios
-    .post(`/api/modules/overseerr/${id}`, { type, seasons })
-    .then(() => {
-      updateNotification({
-        id: id.toString(),
-        title: '',
-        color: 'green',
-        message: ` ${name} requested`,
-        icon: <IconCheck />,
-        autoClose: 2000,
-      });
-    })
-    .catch((err) => {
-      updateNotification({
-        id: id.toString(),
-        color: 'red',
-        title: 'There was an error',
-        message: err.message,
-        autoClose: 2000,
-      });
+const useMediaRequestMutation = () => {
+  const { name: configName } = useConfigContext();
+  const { mutateAsync } = api.overseerr.request.useMutation();
+
+  return async (type: 'tv' | 'movie', id: number, name: string, seasons?: number[]) => {
+    Consola.info(`Requesting ${type} ${id} ${name}`);
+    showNotification({
+      title: 'Request',
+      id: id.toString(),
+      message: `Requesting media ${name}`,
+      color: 'orange',
+      loading: true,
+      autoClose: false,
+      withCloseButton: false,
+      icon: <IconAlertCircle />,
     });
-}
+
+    await mutateAsync(
+      {
+        configName: configName!,
+        id,
+        type,
+        seasons: seasons ?? [],
+      },
+      {
+        onSuccess: () => {
+          updateNotification({
+            id: id.toString(),
+            title: '',
+            color: 'green',
+            message: ` ${name} requested`,
+            icon: <IconCheck />,
+            autoClose: 2000,
+          });
+        },
+        onError: (err) => {
+          updateNotification({
+            id: id.toString(),
+            color: 'red',
+            title: 'There was an error',
+            message: err.message,
+            autoClose: 2000,
+          });
+        },
+      }
+    );
+  };
+};
diff --git a/src/server/api/routers/overseerr.ts b/src/server/api/routers/overseerr.ts
index ba952bb6b..7e4dc10a7 100644
--- a/src/server/api/routers/overseerr.ts
+++ b/src/server/api/routers/overseerr.ts
@@ -104,4 +104,66 @@ export const overseerrRouter = createTRPCRouter({
         });
       return tv;
     }),
+  request: publicProcedure
+    .input(
+      z
+        .object({
+          configName: z.string(),
+          id: z.number(),
+        })
+        .and(
+          z
+            .object({
+              seasons: z.array(z.number()),
+              type: z.literal('tv'),
+            })
+            .or(
+              z.object({
+                type: z.literal('movie'),
+              })
+            )
+        )
+    )
+    .mutation(async ({ input }) => {
+      const config = getConfig(input.configName);
+      const app = config.apps.find(
+        (app) => app.integration?.type === 'overseerr' || app.integration?.type === 'jellyseerr'
+      );
+      const apiKey = app?.integration?.properties.find((x) => x.field === 'apiKey')?.value;
+      if (!apiKey) {
+        throw new TRPCError({
+          code: 'BAD_REQUEST',
+          message: 'No app found',
+        });
+      }
+
+      const appUrl = new URL(app.url);
+      Consola.info('Got an Overseerr request with these arguments', {
+        mediaType: input.type,
+        mediaId: input.id,
+        seasons: input.type === 'tv' ? input.seasons : undefined,
+      });
+      return axios
+        .post(
+          `${appUrl.origin}/api/v1/request`,
+          {
+            mediaType: input.type,
+            mediaId: input.id,
+            seasons: input.type === 'tv' ? input.seasons : undefined,
+          },
+          {
+            headers: {
+              // Set X-Api-Key to the value of the API key
+              'X-Api-Key': apiKey,
+            },
+          }
+        )
+        .then((res) => res.data)
+        .catch((err) => {
+          throw new TRPCError({
+            code: 'INTERNAL_SERVER_ERROR',
+            message: err.message,
+          });
+        });
+    }),
 });