Feat/replace season pack

CHANGELOG.md

@@ -4,6 +4,10 @@
 
 ### Added
 
+- manual search season pack (/~https://github.com/iam4x/bobarr/pull/172)
+
+### Added
+
 - added FlareSolverr for solving CloudFare on certains trackers within jackett (/~https://github.com/iam4x/bobarr/issues/165)
 - update jobs ui
 - update nodejs to v14

packages/api/schema.gql

@@ -293,6 +293,7 @@ type Mutation {
   startFindNewEpisodesJob: GraphQLCommonResponse!
   startDownloadMissingJob: GraphQLCommonResponse!
   downloadMovie(jackettResult: JackettInput!, movieId: Int!): GraphQLCommonResponse!
+  downloadSeason(jackettResult: JackettInput!, seasonNumber: Int!, tvShowTMDBId: Int!): GraphQLCommonResponse!
   downloadTVEpisode(jackettResult: JackettInput!, episodeId: Int!): GraphQLCommonResponse!
   trackMovie(tmdbId: Int!, title: String!): Movie!
   removeMovie(tmdbId: Int!): GraphQLCommonResponse!

packages/api/src/modules/library/library.resolver.ts

@@ -23,10 +23,14 @@ import {
 } from './library.dto';
 
 import { makeCacheInterceptor } from '../redis/cache.interceptor';
+import { TVShowDAO } from 'src/entities/dao/tvshow.dao';
 
 @Resolver()
 export class LibraryResolver {
-  public constructor(private readonly libraryService: LibraryService) {}
+  public constructor(
+    private readonly libraryService: LibraryService,
+    private readonly tvShowDAO: TVShowDAO
+  ) {}
 
   @Query((_returns) => [DownloadingMedia])
   public getDownloadingMedias() {
@@ -97,6 +101,30 @@ export class LibraryResolver {
     return { success: true, message: 'MOVIE_DOWNLOAD_STARTED' };
   }
 
+  @Mutation((_returns) => GraphQLCommonResponse)
+  public async downloadSeason(
+    @Args('tvShowTMDBId', { type: () => Int }) tvShowTMDBId: number,
+    @Args('seasonNumber', { type: () => Int }) seasonNumber: number,
+    @Args('jackettResult', { type: () => JackettInput })
+    jackettResult: JackettInput
+  ) {
+    const { seasons } = await this.tvShowDAO
+      .createQueryBuilder('tvShow')
+      .innerJoinAndSelect(
+        'tvShow.seasons',
+        'season',
+        'season.seasonNumber = :seasonNumber',
+        { seasonNumber }
+      )
+      .where('tvShow.tmdbId = :tvShowTMDBId', { tvShowTMDBId })
+      .getOneOrFail();
+
+    const [{ id: seasonId }] = seasons;
+    await this.libraryService.downloadTVSeason(seasonId, jackettResult, null);
+
+    return { success: true, message: 'TV_EPISODE_DOWNLOAD_STARTED' };
+  }
+
   @Mutation((_returns) => GraphQLCommonResponse)
   public async downloadTVEpisode(
     @Args('episodeId', { type: () => Int }) episodeId: number,

packages/api/src/modules/library/library.service.ts

@@ -348,10 +348,7 @@ export class LibraryService {
     this.logger.info('start download tv season', { seasonId });
     this.logger.info(jackettResult.title);
 
-    await manager!.getCustomRepository(TVSeasonDAO).save({
-      id: seasonId,
-      state: DownloadableMediaState.DOWNLOADING,
-    });
+    await this.replaceSeason(seasonId, manager!);
 
     const torrent = await this.transmissionService.addTorrent(
       {
@@ -580,6 +577,10 @@ export class LibraryService {
       },
     } as const;
 
+    if (mediaType === FileType.SEASON) {
+      await this.replaceSeason(mediaId, manager!);
+    }
+
     if (mediaType === FileType.EPISODE) {
       await this.replaceTVEpisode(mediaId, manager!);
     }
@@ -600,6 +601,65 @@ export class LibraryService {
     });
   }
 
+  private async replaceSeason(seasonId: number, manager: EntityManager) {
+    const tvSeasonDAO = manager!.getCustomRepository(TVSeasonDAO);
+    const torrentDAO = manager!.getCustomRepository(TorrentDAO);
+    const tvEpisodeDAO = manager!.getCustomRepository(TVEpisodeDAO);
+    const fileDAO = manager!.getCustomRepository(FileDAO);
+
+    const tvSeason = await tvSeasonDAO.findOneOrFail({
+      where: { id: seasonId },
+      relations: ['episodes', 'episodes.files'],
+    });
+
+    if (tvSeason.state !== DownloadableMediaState.MISSING) {
+      this.logger.info('tv season already download, removing existing files');
+
+      await forEach(tvSeason.episodes, async (episode) => {
+        const torrent = await torrentDAO.findOne({
+          resourceId: episode.id,
+          resourceType: FileType.EPISODE,
+        });
+
+        if (torrent) {
+          await torrentDAO.remove(torrent);
+          await this.transmissionService.removeTorrentAndFiles(
+            torrent.torrentHash
+          );
+          this.logger.info('episode torrent removed', { torrent: torrent.id });
+        }
+      });
+
+      const tvSeasonFolders = uniq(
+        flatten(
+          tvSeason.episodes.map((episode) =>
+            episode.files.map((file) => path.dirname(file.path))
+          )
+        )
+      );
+
+      await forEachSeries(tvSeasonFolders, (folder) =>
+        childCommand(`rm -rf "${folder}"`)
+      );
+
+      await fileDAO.remove(
+        flatten(tvSeason.episodes.map((episode) => episode.files))
+      );
+    }
+
+    await tvEpisodeDAO.save(
+      tvSeason.episodes.map((v) => ({
+        id: v.id,
+        state: DownloadableMediaState.SEARCHING,
+      }))
+    );
+
+    await tvSeasonDAO.save({
+      id: seasonId,
+      state: DownloadableMediaState.DOWNLOADING,
+    });
+  }
+
   private async replaceTVEpisode(episodeId: number, manager: EntityManager) {
     const tvEpisodeDAO = manager!.getCustomRepository(TVEpisodeDAO);
     const torrentDAO = manager!.getCustomRepository(TorrentDAO);

packages/web/components/manual-search/jackett-results-table.tsx

@@ -16,6 +16,7 @@ import {
   GetMissingDocument,
   useDownloadTvEpisodeMutation,
   GetLibraryTvShowsDocument,
+  useDownloadSeasonMutation,
 } from '../../utils/graphql';
 
 import { Media } from './manual-search.helpers';
@@ -155,6 +156,26 @@ function ManualDownloadMedia({
       }),
   });
 
+  const [downloadTVSeason, { loading: loading3 }] = useDownloadSeasonMutation({
+    awaitRefetchQueries: true,
+    refetchQueries: [
+      { query: GetLibraryTvShowsDocument },
+      { query: GetDownloadingDocument },
+      { query: GetMissingDocument },
+      ...refetchQueries,
+    ],
+    onError: ({ message }) =>
+      notification.error({
+        message: message.replace('GraphQL error: ', ''),
+        placement: 'bottomRight',
+      }),
+    onCompleted: () =>
+      notification.success({
+        message: 'Download episode started',
+        placement: 'bottomRight',
+      }),
+  });
+
   const handleClick = () => {
     if (media.__typename === 'EnrichedMovie') {
       downloadMovie({
@@ -173,9 +194,19 @@ function ManualDownloadMedia({
         },
       });
     }
+
+    if (media.__typename === 'TMDBFormattedTVSeason') {
+      downloadTVSeason({
+        variables: {
+          tvShowTMDBId: media.tvShowTMDBId!,
+          seasonNumber: media.seasonNumber!,
+          jackettResult: jackettInput,
+        },
+      });
+    }
   };
 
-  return loading1 || loading2 ? (
+  return loading1 || loading2 || loading3 ? (
     <LoadingOutlined />
   ) : (
     <DownloadOutlined style={{ cursor: 'pointer' }} onClick={handleClick} />

packages/web/components/manual-search/manual-search.helpers.ts

@@ -7,13 +7,15 @@ import {
   MissingMoviesFragment,
   EnrichedMovie,
   EnrichedTvEpisode,
+  TmdbFormattedTvSeason,
 } from '../../utils/graphql';
 
 export type Media =
   | MissingTvEpisodesFragment
   | MissingMoviesFragment
   | EnrichedMovie
-  | EnrichedTvEpisode;
+  | EnrichedTvEpisode
+  | (TmdbFormattedTvSeason & { tvShowTitle: string; tvShowTMDBId: number });
 
 export function getDefaultSearchQuery(media: Media) {
   if (media.__typename === 'EnrichedTVEpisode') {
@@ -27,5 +29,10 @@ export function getDefaultSearchQuery(media: Media) {
     return `${media.title} ${year}`;
   }
 
+  if (media.__typename === 'TMDBFormattedTVSeason') {
+    const seasonNb = formatNumber(media.seasonNumber!);
+    return `${media.tvShowTitle} S${seasonNb}`;
+  }
+
   return '';
 }

packages/web/components/tvshow-details/tvseason-details.component.tsx

@@ -15,20 +15,21 @@ import {
 
 import { availableIn } from '../../utils/available-in';
 import { ManualSearchComponent } from '../manual-search/manual-search.component';
+import { Media } from '../manual-search/manual-search.helpers';
 
 interface TVSeasonDetailsProps {
   tvShowTMDBId: number;
   season: TmdbFormattedTvSeason;
+  tvShowTitle: string;
 }
 
 export function TVSeasonDetailsComponent({
   tvShowTMDBId,
   season,
+  tvShowTitle,
 }: TVSeasonDetailsProps) {
   const [isOpen, setIsOpen] = useState(false);
-  const [manualSearch, setManualSearch] = useState<EnrichedTvEpisode | null>(
-    null
-  );
+  const [manualSearch, setManualSearch] = useState<Media | null>(null);
 
   const { data, loading } = useGetTvSeasonDetailsQuery({
     pollInterval: 5000,
@@ -90,9 +91,9 @@ export function TVSeasonDetailsComponent({
           <Tag
             icon={<SearchOutlined />}
             onClick={() => setManualSearch(row)}
-            style={{ width: 80, textAlign: 'center', cursor: 'pointer' }}
+            style={{ width: 120, textAlign: 'center', cursor: 'pointer' }}
           >
-            {inLibrary ? 'Replace' : 'Search'}
+            {inLibrary ? 'Replace' : 'Search'} episode
           </Tag>
         );
       },
@@ -121,17 +122,28 @@ export function TVSeasonDetailsComponent({
         className="season"
         style={{ marginBottom: isOpen && season.seasonNumber !== 1 ? 12 : 0 }}
       >
-        <div className="season-title" onClick={toggle}>
-          <div className="season-toggle">
-            {isOpen ? <FaChevronCircleDown /> : <FaChevronCircleRight />}
-          </div>
-          <div className="season-number">Season {season.seasonNumber}</div>
-          {season.airDate && (
-            <div className="season-year">
-              {' '}
-              ({dayjs(season.airDate).format('YYYY')})
+        <div className="season-top">
+          <div className="season-title" onClick={toggle}>
+            <div className="season-toggle">
+              {isOpen ? <FaChevronCircleDown /> : <FaChevronCircleRight />}
             </div>
-          )}
+            <div className="season-number">Season {season.seasonNumber}</div>
+            {season.airDate && (
+              <div className="season-year">
+                {' '}
+                ({dayjs(season.airDate).format('YYYY')})
+              </div>
+            )}
+          </div>
+          <div
+            className="season-replace"
+            onClick={() =>
+              setManualSearch({ ...season, tvShowTitle, tvShowTMDBId })
+            }
+          >
+            {season.inLibrary ? 'Replace' : 'Search'} season
+            <SearchOutlined style={{ marginLeft: 8 }} />
+          </div>
         </div>
         {isOpen && (
           <Table<EnrichedTvEpisode>

packages/web/components/tvshow-details/tvshow-details.component.tsx

@@ -135,6 +135,7 @@ export function TVShowSeasonsModalComponent(
                       key={season.id}
                       season={season}
                       tvShowTMDBId={tvShow.tmdbId}
+                      tvShowTitle={tvShow.title}
                     />
                   ))}
               </div>

packages/web/components/tvshow-details/tvshow-details.styles.tsx

@@ -43,26 +43,45 @@ export const TVShowSeasonsModalComponentStyles = styled(MovieDetailsStyles)`
   .seasons-details {
     padding-top: 12px;
 
-    .season-title {
+    .season-top {
+      margin-bottom: 4px;
+    }
+
+    .season-title,
+    .season-top {
       display: flex;
       align-items: center;
+    }
+
+    .season-title,
+    .season-replace {
       cursor: pointer;
+    }
 
-      .season-number {
-        font-size: 1.25em;
-        font-weight: 600;
-        margin-right: 8px;
-      }
+    .season-replace {
+      font-weight: bold;
+      display: flex;
+      align-items: center;
+      margin-left: 32px;
+      border: 1px solid #fff5;
+      border-radius: 5px;
+      padding: 0 4px;
+    }
 
-      .season-year {
-        font-size: 1em;
-        font-weight: 300;
-      }
+    .season-number {
+      font-size: 1.25em;
+      font-weight: 600;
+      margin-right: 8px;
+    }
 
-      .season-toggle {
-        margin-right: 12px;
-        margin-top: 4px;
-      }
+    .season-year {
+      font-size: 1em;
+      font-weight: 300;
+    }
+
+    .season-toggle {
+      margin-right: 12px;
+      margin-top: 4px;
     }
 
     .ant-table {