Set up reference resolvers

Author: NuckChorris

app/graphql/kitsu_schema.rb

@@ -3,6 +3,7 @@
 class KitsuSchema < GraphQL::Schema
   include ApolloFederation::Schema
   federation version: '2.0'
+  import_directives %w[tag key shareable]
 
   default_max_page_size 2000
 

app/graphql/types/account.rb

@@ -1,8 +1,19 @@
+# frozen_string_literal: true
+
 class Types::Account < Types::BaseObject
   implements Types::Interface::WithTimestamps
 
   description 'A user account on Kitsu'
 
+  key fields: %w[id]
+
+  def self.resolve_reference(reference, context)
+    # Only resolve the user if the ID matches the current user
+    return nil if context[:user].nil? || context[:user].id.to_s != reference[:id]
+
+    context[:user]
+  end
+
   field :id, ID, null: false
 
   field :email, [String],

app/graphql/types/anime.rb

@@ -1,8 +1,16 @@
+# frozen_string_literal: true
+
 class Types::Anime < Types::BaseObject
   implements Types::Interface::Media
   implements Types::Interface::Episodic
   implements Types::Interface::WithTimestamps
 
+  key fields: %w[id]
+
+  def self.resolve_reference(reference, _context)
+    Loaders::UnscopedRecordLoader.for(::Anime).load(reference[:id])
+  end
+
   field :subtype, Types::Enum::AnimeSubtype,
     null: false,
     description: 'A secondary type for categorizing Anime.'

app/graphql/types/chapter.rb

@@ -1,9 +1,17 @@
+# frozen_string_literal: true
+
 class Types::Chapter < Types::BaseObject
   implements Types::Interface::Unit
   implements Types::Interface::WithTimestamps
 
   description 'A single chapter of a manga'
 
+  key fields: %w[id]
+
+  def self.resolve_reference(reference, _context)
+    Loaders::UnscopedRecordLoader.for(::Chapter).load(reference[:id])
+  end
+
   field :released_at, GraphQL::Types::ISO8601Date,
     method: :published,
     null: true,

app/graphql/types/character.rb

@@ -5,6 +5,12 @@ class Types::Character < Types::BaseObject
 
   description 'Information about a Character in the Kitsu database'
 
+  key fields: %w[id]
+
+  def self.resolve_reference(reference, _context)
+    Loaders::UnscopedRecordLoader.for(::Character).load(reference[:id])
+  end
+
   field :id, ID, null: false
 
   field :slug, String,

app/graphql/types/episode.rb

@@ -1,9 +1,17 @@
+# frozen_string_literal: true
+
 class Types::Episode < Types::BaseObject
   implements Types::Interface::Unit
   implements Types::Interface::WithTimestamps
 
   description 'An Episode of a Media'
 
+  key fields: %w[id]
+
+  def self.resolve_reference(reference, _context)
+    Loaders::UnscopedRecordLoader.for(::Episode).load(reference[:id])
+  end
+
   field :length, Integer,
     null: true,
     description: 'The length of the episode in seconds'

app/graphql/types/franchise.rb

@@ -1,8 +1,16 @@
+# frozen_string_literal: true
+
 class Types::Franchise < Types::BaseObject
   implements Types::Interface::WithTimestamps
 
   description 'Related media grouped together'
 
+  key fields: %w[id]
+
+  def self.resolve_reference(reference, _context)
+    Loaders::UnscopedRecordLoader.for(::Franchise).load(reference[:id])
+  end
+
   field :id, ID, null: false
 
   field :titles, Types::TitlesList,
@@ -26,7 +34,7 @@ def titles
   def installments(sort: [{ on: :release_order, direction: :asc }])
     Loaders::InstallmentsLoader.connection_for({
       find_by: :franchise_id,
-      sort: sort
+      sort:
     }, object.id)
   end
 end

app/graphql/types/manga.rb

@@ -4,6 +4,12 @@ class Types::Manga < Types::BaseObject
   implements Types::Interface::Media
   implements Types::Interface::WithTimestamps
 
+  key fields: %w[id]
+
+  def self.resolve_reference(reference, _context)
+    Loaders::UnscopedRecordLoader.for(::Manga).load(reference[:id])
+  end
+
   field :subtype, Types::Enum::MangaSubtype,
     null: false,
     description: 'A secondary type for categorizing Manga.'

app/graphql/types/media_reaction.rb

@@ -6,6 +6,12 @@ class Types::MediaReaction < Types::BaseObject
 
   description 'A simple review that is 140 characters long expressing how you felt about a media'
 
+  key fields: %w[id]
+
+  def self.resolve_reference(reference, context)
+    Loaders::RecordLoader.for(::MediaReaction, token: context[:token]).load(reference[:id])
+  end
+
   field :id, ID, null: false
 
   field :author, Types::Profile,

app/graphql/types/person.rb

@@ -8,6 +8,12 @@ class Types::Person < Types::BaseObject
     the creation and localization of media
   DESCRIPTION
 
+  key fields: %w[id]
+
+  def self.resolve_reference(reference, _context)
+    Loaders::UnscopedRecordLoader.for(::Person).load(reference[:id])
+  end
+
   field :id, ID, null: false
 
   field :name, String,