Add additional mutations and queries option

Author: mcelicalderon

lib/graphql_devise/rails/routes.rb

@@ -1,9 +1,11 @@
 module ActionDispatch::Routing
   class Mapper
     def mount_graphql_devise_for(resource, opts = {})
-      custom_operations  = opts[:operations] || {}
-      skipped_operations = opts.fetch(:skip, [])
-      only_operations    = opts.fetch(:only, [])
+      custom_operations    = opts.fetch(:operations, {})
+      skipped_operations   = opts.fetch(:skip, [])
+      only_operations      = opts.fetch(:only, [])
+      additional_mutations = opts.fetch(:additional_mutations, {})
+      additional_queries   = opts.fetch(:additional_queries, {})
 
       if [skipped_operations, only_operations].all?(&:any?)
         raise GraphqlDevise::Error, "Can't specify both `skip` and `only` options when mounting the route."
@@ -62,8 +64,11 @@ def mount_graphql_devise_for(resource, opts = {})
 
         GraphqlDevise::Types::MutationType.field("#{mapping_name}_#{action}", mutation: used_mutation)
       end
+      additional_mutations.each do |action, mutation|
+        GraphqlDevise::Types::MutationType.field(action, mutation: mutation)
+      end
 
-      if used_mutations.present? &&
+      if (used_mutations.present? || additional_mutations.present?) &&
          (Gem::Version.new(GraphQL::VERSION) <= Gem::Version.new('1.10.0') || GraphqlDevise::Schema.mutation.nil?)
         GraphqlDevise::Schema.mutation(GraphqlDevise::Types::MutationType)
       end
@@ -87,8 +92,11 @@ def mount_graphql_devise_for(resource, opts = {})
 
         GraphqlDevise::Types::QueryType.field("#{mapping_name}_#{action}", resolver: used_query)
       end
+      additional_queries.each do |action, resolver|
+        GraphqlDevise::Types::QueryType.field(action, resolver: resolver)
+      end
 
-      if used_queries.blank? && GraphqlDevise::Types::QueryType.fields.blank?
+      if (used_queries.blank? || additional_queries.present?) && GraphqlDevise::Types::QueryType.fields.blank?
         GraphqlDevise::Types::QueryType.field(:dummy, resolver: GraphqlDevise::Resolvers::Dummy)
       end
 

spec/dummy/app/graphql/mutations/register_confirmed_user.rb

@@ -0,0 +1,23 @@
+module Mutations
+  class RegisterConfirmedUser < GraphqlDevise::Mutations::Base
+    argument :email,                 String, required: true
+    argument :name,                  String, required: true
+    argument :password,              String, required: true
+    argument :password_confirmation, String, required: true
+
+    field :user, Types::UserType, null: true
+
+    def resolve(**attrs)
+      user = User.new(attrs.merge(confirmed_at: Time.zone.now))
+
+      if user.save
+        { user: user }
+      else
+        raise_user_error_list(
+          'Custom registration failed',
+          errors: user.errors.full_messages
+        )
+      end
+    end
+  end
+end

spec/dummy/app/graphql/resolvers/public_user.rb

@@ -0,0 +1,11 @@
+module Resolvers
+  class PublicUser < GraphQL::Schema::Resolver
+    type Types::UserType, null: false
+
+    argument :id, Int, required: true
+
+    def resolve(id:)
+      User.find(id)
+    end
+  end
+end

spec/dummy/config/routes.rb

@@ -2,6 +2,10 @@
   mount_graphql_devise_for 'User', at: '/api/v1/graphql_auth', operations: {
     login:   Mutations::Login,
     sign_up: Mutations::SignUp
+  }, additional_mutations: {
+    register_confirmed_user: Mutations::RegisterConfirmedUser
+  }, additional_queries: {
+    public_user: Resolvers::PublicUser
   }
 
   mount_graphql_devise_for(

spec/requests/mutations/additional_mutations_spec.rb

@@ -0,0 +1,65 @@
+require 'rails_helper'
+
+RSpec.describe 'Additional Mutations' do
+  include_context 'with graphql query request'
+
+  let(:name)                  { Faker::Name.name }
+  let(:password)              { Faker::Internet.password }
+  let(:password_confirmation) { password }
+  let(:email)                 { Faker::Internet.email }
+  let(:redirect)              { Faker::Internet.url }
+
+  context 'when using the user model' do
+    let(:query) do
+      <<-GRAPHQL
+        mutation {
+          registerConfirmedUser(
+            email:                "#{email}",
+            name:                 "#{name}",
+            password:             "#{password}",
+            passwordConfirmation: "#{password_confirmation}"
+          ) {
+            user {
+              email
+              name
+            }
+          }
+        }
+      GRAPHQL
+    end
+
+    context 'when params are correct' do
+      it 'creates a new resource that is already confirmed' do
+        expect { post_request }.to(
+          change(User, :count).by(1)
+          .and(not_change(ActionMailer::Base.deliveries, :count))
+        )
+
+        user = User.last
+
+        expect(user).to be_confirmed
+        expect(json_response[:data][:registerConfirmedUser]).to include(
+          user: {
+            email: email,
+            name:  name
+          }
+        )
+      end
+    end
+
+    context 'when params are incorrect' do
+      let(:password_confirmation) { 'not the same' }
+
+      it 'returns descriptive errors' do
+        expect { post_request }.to not_change(User, :count)
+
+        expect(json_response[:errors]).to contain_exactly(
+          hash_including(
+            message:    'Custom registration failed',
+            extensions: { code: 'USER_ERROR', detailed_errors: ["Password confirmation doesn't match Password"] }
+          )
+        )
+      end
+    end
+  end
+end

spec/requests/mutations/additional_queries_spec.rb

@@ -0,0 +1,31 @@
+require 'rails_helper'
+
+RSpec.describe 'Additional Queries' do
+  include_context 'with graphql query request'
+
+  let(:public_user) { create(:user, :confirmed) }
+
+  context 'when using the user model' do
+    let(:query) do
+      <<-GRAPHQL
+        query {
+          publicUser(
+            id: #{public_user.id}
+          ) {
+            email
+            name
+          }
+        }
+      GRAPHQL
+    end
+
+    before { post_request }
+
+    it 'fetches a user by ID' do
+      expect(json_response[:data][:publicUser]).to include(
+        email: public_user.email,
+        name:  public_user.name
+      )
+    end
+  end
+end