feat(graphql-api): Added find related movies

Author: TriPSs

apps/graphql-api/src/app.module.ts

@@ -7,6 +7,7 @@ import { ConfigService } from './shared/config/config.service'
 import { ModelsModule } from './shared/models/models.module'
 import { TorrentModule } from './shared/torrent/torrent.module'
 import { PubSubModule } from './shared/pub-sub/pub-sub.module'
+import { TraktModule } from './shared/trakt/trakt.module'
 
 import { StatusModule } from './status/status.module'
 import { CalendarModule } from './calendar/calendar.module'
@@ -29,6 +30,7 @@ import { WatchModule } from './watch/watch.module'
     ConfigModule,
     TorrentModule,
     PubSubModule,
+    TraktModule,
 
     // GraphQL
     StatusModule,
@@ -60,12 +62,17 @@ import { WatchModule } from './watch/watch.module'
     }),
 
     // Enable Graphql
-    GraphQLModule.forRoot({
-      debug: true,
-      playground: true,
-      installSubscriptionHandlers: true,
-      autoSchemaFile: 'schema.gql',
-      introspection: true
+    GraphQLModule.forRootAsync({
+      imports: [ConfigModule],
+      inject: [ConfigService],
+      useFactory: (configService: ConfigService) => ({
+        debug: configService.isDevelopment,
+        tracing: configService.isDevelopment,
+        playground: true,
+        installSubscriptionHandlers: true,
+        autoSchemaFile: 'schema.gql',
+        introspection: true,
+      })
     })
   ]
 })

apps/graphql-api/src/movies/movies.resolver.ts

@@ -5,11 +5,15 @@ import { Movie } from '@pct-org/mongo-models'
 import { MovieArgs } from './dto/movie.args'
 import { MoviesArgs } from './dto/movies.args'
 import { MoviesService } from './movies.service'
+import { TraktService } from '../shared/trakt/trakt.service'
 
 @Resolver(of => Movie)
 export class MoviesResolver {
 
-  constructor(private readonly moviesService: MoviesService) {}
+  constructor(
+    private readonly moviesService: MoviesService,
+    private readonly traktService: TraktService
+  ) {}
 
   @Query(returns => Movie)
   movie(@Args() movieArgs: MovieArgs): Promise<Movie> {
@@ -21,4 +25,11 @@ export class MoviesResolver {
     return this.moviesService.findAll(moviesArgs)
   }
 
+  @Query(returns => [Movie])
+  relatedMovies(
+    @Args({ name: '_id', description: 'Id of the movie to find related movies for.' }) _id: string
+  ): Promise<Movie[]> {
+    return this.traktService.findRelatedMovies(_id)
+  }
+
 }

apps/graphql-api/src/movies/movies.service.ts

@@ -3,7 +3,6 @@ import { InjectModel } from '@nestjs/mongoose'
 import { Model } from 'mongoose'
 import { Movie } from '@pct-org/mongo-models'
 
-import { MovieArgs } from './dto/movie.args'
 import { MoviesArgs } from './dto/movies.args'
 import { ContentService } from '../shared/content/content.service'
 
@@ -34,6 +33,20 @@ export class MoviesService extends ContentService {
     )
   }
 
+  findAllWithIDS(ids: string[], lean = true): Promise<Movie[]> {
+    return this.movieModel.find(
+      {
+        _id: {
+          $in: ids
+        }
+      },
+      {},
+      {
+        lean
+      }
+    )
+  }
+
   updateOne(movie: Movie): Promise<Movie> {
     return this.movieModel.findOneAndUpdate({
         _id: movie._id

apps/graphql-api/src/shared/trakt/trakt.module.ts

@@ -0,0 +1,19 @@
+import { Module, Global } from '@nestjs/common'
+
+import { ShowsService } from '../../shows/shows.service'
+import { MoviesService } from '../../movies/movies.service'
+import { TraktService } from './trakt.service'
+
+@Global()
+@Module({
+  providers: [
+    TraktService,
+    ShowsService,
+    MoviesService
+  ],
+  exports: [
+    TraktService
+  ]
+})
+export class TraktModule {
+}

apps/graphql-api/src/shared/trakt/trakt.service.ts

@@ -0,0 +1,93 @@
+import { Injectable } from '@nestjs/common'
+import * as Trakt from 'trakt.tv'
+import { Show, Movie } from '@pct-org/mongo-models'
+
+import { ConfigService } from '../config/config.service'
+import { ShowsService } from '../../shows/shows.service'
+import { MoviesService } from '../../movies/movies.service'
+
+@Injectable()
+export class TraktService {
+
+  private readonly trakt
+
+  constructor(
+    private readonly configService: ConfigService,
+    private readonly showsService: ShowsService,
+    private readonly moviesService: MoviesService
+  ) {
+    const clientId = this.configService.get(ConfigService.TRAKT_KEY)
+
+    if (clientId) {
+      try {
+        this.trakt = new Trakt({
+          client_id: this.configService.get(ConfigService.TRAKT_KEY)
+        })
+      } catch (err) {
+        // Do nothing
+      }
+    }
+  }
+
+  async mostWatchedWeeklyShows(): Promise<Show[]> {
+    // Trakt is not setup
+    if (!this.trakt) {
+      return []
+    }
+
+    let showIds = []
+
+    try {
+      const traktMostWatched = await this.trakt.shows.watched({
+        period: 'weekly',
+        limit: 20
+      })
+
+      showIds = traktMostWatched.map((item) => item.show.ids.imdb)
+    } catch (e) {
+    }
+
+    if (showIds.length === 0) {
+      return []
+    }
+
+    const shows = await this.showsService.findAllWithIDS(showIds)
+
+    return shows.sort((showA, showB) =>
+      showIds.indexOf(showA._id) - showIds.indexOf(showB._id)
+    )
+  }
+
+  async findRelatedMovies(movieImdbId): Promise<Movie[]> {
+    // Trakt is not setup
+    if (!this.trakt) {
+      return []
+    }
+
+    let movieIds = []
+
+    try {
+      const traktRelatedMovies = await this.trakt.movies.related({
+        id: movieImdbId,
+        limit: 30
+      })
+
+      movieIds = traktRelatedMovies.map((item) => item.ids.imdb)
+    } catch (e) {
+    }
+
+    if (movieIds.length === 0) {
+      return []
+    }
+
+    const movies = await this.moviesService.findAllWithIDS(movieIds)
+
+    return this.sortItemsOnIds(movies, movieIds)
+  }
+
+  private sortItemsOnIds(items, ids) {
+    return items.sort((itemA, itemB) =>
+      ids.indexOf(itemA._id) - ids.indexOf(itemB._id)
+    )
+  }
+}

apps/graphql-api/src/shows/shows.resolver.ts

@@ -1,30 +1,23 @@
 import { Args, Parent, Query, ResolveField, Resolver } from '@nestjs/graphql'
 import { Show, Season } from '@pct-org/mongo-models'
-import * as Trakt from 'trakt.tv'
 
 import { ShowArgs } from './dto/show.args'
 import { ShowsArgs } from './dto/shows.args'
 import { ShowsService } from './shows.service'
 
-import { ConfigService } from '../shared/config/config.service'
 import { SeasonsService } from '../seasons/seasons.service'
 import { DownloadsService } from '../downloads/downloads.service'
+import { TraktService } from '../shared/trakt/trakt.service'
 
 @Resolver(of => Show)
 export class ShowsResolver {
 
-  private readonly trakt
-
   constructor(
     private readonly showsService: ShowsService,
     private readonly seasonsService: SeasonsService,
     private readonly downloadsService: DownloadsService,
-    private readonly configService: ConfigService
+    private readonly traktService: TraktService
   ) {
-    this.trakt = new Trakt({
-      // eslint-disable-next-line @typescript-eslint/camelcase
-      client_id: this.configService.get(ConfigService.TRAKT_KEY)
-    })
   }
 
   /**
@@ -60,26 +53,7 @@ export class ShowsResolver {
    */
   @Query(returns => [Show], { description: 'Get most watched shows.' })
   async mostWatchedShows(): Promise<Show[]> {
-    let showIds = []
-
-    try {
-      const traktMostWatched = await this.trakt.shows.watched({
-        period: 'weekly'
-      })
-
-      showIds = traktMostWatched.map((item) => item.show.ids.imdb)
-    } catch (e) {
-    }
-
-    if (showIds.length === 0) {
-      return []
-    }
-
-    const shows = await this.showsService.findAllWithIDS(showIds)
-
-    return shows.sort((showA, showB) =>
-      showIds.indexOf(showA._id) - showIds.indexOf(showB._id),
-    )
+    return this.traktService.mostWatchedWeeklyShows()
   }
 
   /**