feat: Allow setting of operation name for query & mutation

Gemfile

@@ -24,4 +24,5 @@ group :test do
   gem 'sinatra'
   gem 'vcr'
   gem 'webmock'
+  gem 'rspec-instrumentation-matcher'
 end

lib/graphlient/client.rb

@@ -21,8 +21,7 @@ def execute(query, variables = nil)
       query_params = {}
       query_params[:context] = @options if @options
       query_params[:variables] = variables if variables
-      query = client.parse(query) if query.is_a?(String)
-      rc = client.query(query, **query_params)
+      rc = client.query(parse_query(query), **query_params)
       raise Graphlient::Errors::GraphQLError, rc if rc.errors.any?
       # see https://github.com/github/graphql-client/pull/132
       # see https://github.com/exAspArk/graphql-errors/issues/2
@@ -69,5 +68,24 @@ def client
     def errors_in_result?(response)
       response.data && response.data.errors && response.data.errors.all.any?
     end
+
+    def parse_query(query)
+      return query unless query.is_a?(String)
+
+      query = client.parse(query)
+      return query if query.is_a?(GraphQL::Client::OperationDefinition)
+
+      query = query.const_get(query.constants.first)
+      patch_operation_name(query)
+      query
+    end
+
+    def patch_operation_name(query)
+      query.instance_eval do
+        def name
+          super.gsub(/#<Module:(0x[0-9a-f]+)>/, 'Graphlient')
+        end
+      end
+    end
   end
 end

lib/graphlient/query.rb

@@ -57,13 +57,20 @@ def resolve_fragment_constant(value)
     end
 
     def append_node(node, args, arg_processor: nil, &block)
+      hash_arguments = hash_arg(args)
+
       node = "...#{resolve_fragment_constant(node)}".to_sym if node.to_s.start_with?('___')
 
       # add field
-      @query_str << "\n#{indent}#{node}"
+      # when using operation name
+      if ROOT_NODES.include?(node) && hash_arguments && hash_arguments.has_key?(:operation_name)
+        @query_str << "\n#{indent}#{node} #{hash_arguments.delete(:operation_name)} "
+      else
+        @query_str << "\n#{indent}#{node}"
+      end
+
       # add filter
-      hash_arguments = hash_arg(args)
-      @query_str << "(#{args_str(hash_arguments, arg_processor: arg_processor)})" if hash_arguments
+      @query_str << "(#{args_str(hash_arguments, arg_processor: arg_processor)})" if hash_arguments&.any?
 
       if block_given?
         @indents += 1

spec/graphlient/github_query_spec.rb

@@ -18,6 +18,18 @@
     expect(rc.data.viewer.name).to eq 'Daniel Doubrovkine (dB.) @dblockdotorg'
   end
 
+  it 'correctly sets the operation name & type', vcr: { cassette_name: 'github/viewer' } do
+    expect {
+      client.query <<-GRAPHQL
+        query MyViewerAsOperationName {
+          viewer {
+            name
+          }
+        }
+      GRAPHQL
+    }.to instrument("query.graphql").with(hash_including(:operation_name=>"Graphlient__MyViewerAsOperationName", :operation_type=>"query"))
+  end
+
   it 'queries with a parameter', vcr: { cassette_name: 'github/user' } do
     query = <<-GRAPHQL
       query($login: String!) {

spec/graphlient/query_spec.rb

@@ -3,6 +3,18 @@
 describe Graphlient::Query do
   describe '#initialize' do
     context 'query' do
+      it 'returns the operation_name in the query' do
+        query = Graphlient::Query.new do
+          query(operation_name: "MyOperationName") do
+            invoice do
+              line_items
+            end
+          end
+        end
+
+        expect(query.to_s).to eq "query MyOperationName {\n  invoice{\n    line_items\n    }\n  }"
+      end
+
       it 'returns expected query with block' do
         query = Graphlient::Query.new do
           query do
@@ -79,6 +91,19 @@
     end
 
     context 'mutation' do
+
+      it 'returns the operation_name in the query' do
+        mutation = Graphlient::Query.new do
+          mutation(operation_name: "MyUpdateMutationOperationName") do
+            invoice(type: 'test', fee_in_cents: 20_000, total_cents: 50_000, line_items: %w[li1 li2]) do
+              id
+            end
+          end
+        end
+
+        expect(mutation.to_s).to eq "mutation MyUpdateMutationOperationName {\n  invoice(type: \"test\", fee_in_cents: 20000, total_cents: 50000, line_items: [\"li1\", \"li2\"]){\n    id\n    }\n  }"
+      end
+
       it 'returns proper mutation with arguments' do
         mutation = Graphlient::Query.new do
           mutation do

spec/spec_helper.rb

@@ -8,6 +8,7 @@
 require 'webmock/rspec'
 require 'vcr'
 require 'faraday/rack'
+require 'rspec-instrumentation-matcher'
 
 Dir[File.join(File.dirname(__FILE__), 'support', '**/*.rb')].each do |file|
   require file