feat: Changed bookmarks to union

apps/api/src/bookmarks/bookmarks.resolver.ts

@@ -3,22 +3,23 @@ import { Content } from '@pct-org/mongo-models'
 
 import { BookmarksArgs } from './dto/bookmarks.args'
 import { BookmarksService } from './bookmarks.service'
+import { BookmarksUnion } from './bookmarks.union'
 import { PubSubService } from '../shared/pub-sub/pub-sub.service'
 
-@Resolver(of => Content)
+@Resolver(of => BookmarksUnion)
 export class BookmarksResolver {
 
   constructor(
     private readonly bookmarksService: BookmarksService,
     private readonly pubSubService: PubSubService
   ) {}
 
-  @Query(returns => [Content])
+  @Query(returns => [BookmarksUnion])
   bookmarks(@Args() bookmarksArgs: BookmarksArgs): Promise<Content[]> {
     return this.bookmarksService.findAll(bookmarksArgs)
   }
 
-  @Mutation(returns => Content)
+  @Mutation(returns => BookmarksUnion)
   async addBookmark(
     @Args('_id') _id: string,
     @Args('type') type: string
@@ -35,7 +36,7 @@ export class BookmarksResolver {
     return updateBookmark
   }
 
-  @Mutation(returns => Content)
+  @Mutation(returns => BookmarksUnion)
   async removeBookmark(
     @Args('_id') _id: string,
     @Args('type') type: string
@@ -52,7 +53,7 @@ export class BookmarksResolver {
     return updateBookmark
   }
 
-  @Subscription(returns => Content)
+  @Subscription(returns => BookmarksUnion)
   bookmarked() {
     return this.pubSubService.asyncIterator('bookmarked')
   }

apps/api/src/bookmarks/bookmarks.union.ts

@@ -0,0 +1,18 @@
+import { createUnionType } from '@nestjs/graphql'
+import { Movie, Show } from '@pct-org/mongo-models'
+
+export const BookmarksUnion = createUnionType({
+  name: 'Bookmark',
+  types: () => [Movie, Show],
+  resolveType(value) {
+    if (value.type === 'movie') {
+      return Movie
+    }
+
+    if (value.type === 'show') {
+      return Show
+    }
+
+    return null
+  }
+})