countable_object.rb

# frozen_string_literal: true

module Types
  class CountableObject < BaseObject
    class << self
      def define_connection(**args, &block)
        super(**args) do
          block&.call

          # this type must be GraphQL::INT_TYPE (not Integer), may be a bug.
          field :totalCount, GraphQL::INT_TYPE, resolve: ->(object, _args, _ctx) {
            relation = object.nodes

            if relation.respond_to?(:total_count)
              relation.total_count # kaminari-zed relation instance
            else
              relation.count
            end
          }
        end
      end
    end
  end
end