add tmdb to series update actions (#1004)

ShokoAnime · Aug 21, 2024 · 5e07929 · 5e07929
1 parent f82333c
commit 5e07929
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 0 deletions.
diff --git a/src/components/Collection/Series/EditSeriesTabs/UpdateActionsTab.tsx b/src/components/Collection/Series/EditSeriesTabs/UpdateActionsTab.tsx
@@ -4,7 +4,9 @@ import Action from '@/components/Collection/Series/EditSeriesTabs/Action';
 import toast from '@/components/Toast';
 import {
   useRefreshSeriesAniDBInfoMutation,
+  useRefreshSeriesTMDBInfoMutation,
   useRefreshSeriesTvdbInfoMutatation,
+  useUpdateSeriesTMDBImagesMutation,
 } from '@/core/react-query/series/mutations';
 
 type Props = {
@@ -14,6 +16,8 @@ type Props = {
 const UpdateActionsTab = ({ seriesId }: Props) => {
   const { mutate: refreshAnidb } = useRefreshSeriesAniDBInfoMutation();
   const { mutate: refreshTvdb } = useRefreshSeriesTvdbInfoMutatation();
+  const { mutate: refreshTmdb } = useRefreshSeriesTMDBInfoMutation();
+  const { mutate: updateTmdbImages } = useUpdateSeriesTMDBImagesMutation();
 
   const triggerAnidbRefresh = (force: boolean, cacheOnly: boolean) => {
     refreshAnidb({ seriesId, force, cacheOnly }, {
@@ -31,6 +35,22 @@ const UpdateActionsTab = ({ seriesId }: Props) => {
             onSuccess: () => toast.success('TvDB refresh queued!'),
           })}
       />
+      <Action
+        name="Update TMDB Info"
+        description="Gets the latest series information from TMDB."
+        onClick={() =>
+          refreshTmdb(seriesId, {
+            onSuccess: () => toast.success('TMDB refresh queued!'),
+          })}
+      />
+      <Action
+        name="Update TMDB Images - Force"
+        description="Forces a complete redownload of images from TMDB."
+        onClick={() =>
+          updateTmdbImages({ seriesId, force: true }, {
+            onSuccess: () => toast.success('TMDB image download queued!'),
+          })}
+      />
       <Action
         name="Update AniDB Info"
         description="Gets the latest series information from the AniDB database."

diff --git a/src/core/react-query/series/mutations.ts b/src/core/react-query/series/mutations.ts
@@ -120,3 +120,21 @@ export const useWatchSeriesEpisodesMutation = () =>
       invalidateQueries(['series', seriesId, 'episodes']);
     },
   });
+
+export const useRefreshSeriesTMDBInfoMutation = () =>
+  useMutation({
+    mutationFn: (seriesId: number) =>
+      Promise.all([
+        axios.post(`Series/${seriesId}/TMDB/Show/Action/Refresh`),
+        axios.post(`Series/${seriesId}/TMDB/Movie/Action/Refresh`),
+      ]),
+  });
+
+export const useUpdateSeriesTMDBImagesMutation = () =>
+  useMutation({
+    mutationFn: ({ force = false, seriesId }: { seriesId: number, force: boolean }) =>
+      Promise.all([
+        axios.post(`Series/${seriesId}/TMDB/Show/Action/DownloadImages`, { force }),
+        axios.post(`Series/${seriesId}/TMDB/Movie/Action/DownloadImages`, { force }),
+      ]),
+  });