wallet screen

app/graphql/x_stake_schema.rb

@@ -4,15 +4,22 @@ class XStakeSchema < GraphQL::Schema
   query(Types::QueryType)
 
   def self.resolve_type(abstract_type, obj, ctx)
-    raise(GraphQL::RequiredImplementationMissingError)
+    case obj
+    when Currency
+      Types::CurrencyType
+    when Balance
+      Types::BalanceType
+    else
+      raise(GraphQL::RequiredImplementationMissingError, "Unexpected object: #{obj}")
+    end
   end
 
-  def self.id_from_object(object, type_definition, query_ctx)
+  def self.id_from_object(object, type_definition, ctx)
     GraphQL::Schema::UniqueWithinType.encode(type_definition.name, object.id)
   end
 
-  def self.object_from_id(id, query_ctx)
+  def self.object_from_id(id, ctx)
     type_name, item_id = GraphQL::Schema::UniqueWithinType.decode(id)
-    type_name.constantize.find(item_id)
+    Pundit.policy_scope(ctx[:current_user], type_name.constantize).find(item_id)
   end
 end