Take custom mutations when mounting gem in routes

.rubocop.yml

@@ -5,14 +5,17 @@ Rails:
   Enabled: true
 
 AllCops:
-  TargetRubyVersion: 2.3
+  TargetRubyVersion: 2.2
   DisplayCopNames: true
   Exclude:
     - bin/**/*
     - db/**/*
     - vendor/**/*
     - tmp/**/*
 
+Rails/HttpPositionalArguments:
+  Enabled: false
+
 Rails/HasAndBelongsToMany:
   Enabled: false
 

Gemfile

@@ -1,5 +1,3 @@
 source 'https://rubygems.org'
 
 gemspec
-
-gem 'listen'

app/controllers/graphql_devise/graphql_controller.rb

@@ -18,6 +18,8 @@ def auth
       render json: result
     end
 
+    attr_accessor :client_id, :token, :resource
+
     private
 
     def execute_params(item)

app/graphql/graphql_devise/mutations/base.rb

@@ -1,18 +1,26 @@
+require 'devise_token_auth/version'
+
 module GraphqlDevise
   module Mutations
     class Base < GraphQL::Schema::Mutation
       private
 
+      def remove_resource
+        controller.resource = nil
+        controller.client_id = nil
+        controller.token = nil
+      end
+
       def single_error_object(error)
         { success: false, errors: [error] }
       end
 
       def request
-        context[:controller].request
+        controller.request
       end
 
       def response
-        context[:controller].response
+        controller.response
       end
 
       def controller
@@ -26,6 +34,14 @@ def resource_class
       def current_resource
         context[:current_resource]
       end
+
+      def client
+        if Gem::Version.new(DeviseTokenAuth::VERSION) <= Gem::Version.new('1.1.0')
+          controller.client_id
+        else
+          controller.token.client if controller.token.present?
+        end
+      end
     end
   end
 end

app/graphql/graphql_devise/mutations/login.rb

@@ -4,9 +4,8 @@ class Login < Base
       argument :email,    String, required: true
       argument :password, String, required: true
 
-      field :authenticable, GraphqlDevise::Types::AuthenticableType, null: true
-      field :success,       Boolean,                                 null: false
-      field :errors,        [String],                                null: false
+      field :success, Boolean,  null: false
+      field :errors,  [String], null: false
 
       def resolve(email:, password:)
         resource = resource_class.find_by(email: email)

app/graphql/graphql_devise/mutations/logout.rb

@@ -0,0 +1,23 @@
+module GraphqlDevise
+  module Mutations
+    class Logout < Base
+      field :success, Boolean,  null: false
+      field :errors,  [String], null: false
+
+      def resolve
+        if current_resource && client && current_resource.tokens[client]
+          current_resource.tokens.delete(client)
+          current_resource.save!
+
+          remove_resource
+
+          yield current_resource if block_given?
+
+          { success: true, errors: [], authenticable: current_resource }
+        else
+          { success: false, errors: [I18n.t('graphql_devise.user_not_found')] }
+        end
+      end
+    end
+  end
+end

app/graphql/graphql_devise/types/mutation_type.rb

@@ -1,7 +1,6 @@
 module GraphqlDevise
   module Types
     class MutationType < GraphQL::Schema::Object
-      field :login, mutation: GraphqlDevise::Mutations::Login
     end
   end
 end

config/locales/en.yml

@@ -1,5 +1,6 @@
 en:
   graphql_devise:
+    user_not_found: "User was not found or was not logged in."
     sessions:
       bad_credentials: "Invalid login credentials. Please try again."
       not_confirmed: "A confirmation email was sent to your account at '%{email}'. You must follow the instructions in the email before your account can be activated"

gemfiles/rails4.2_graphql1.8.gemfile

@@ -2,7 +2,6 @@
 
 source "https://rubygems.org"
 
-gem "listen"
 gem "bundler", "~> 1.17"
 gem "rails", github: "rails/rails", branch: "4-2-stable"
 gem "graphql", "~> 1.8.0"

gemfiles/rails5.0_graphql1.8.gemfile

@@ -2,7 +2,6 @@
 
 source "https://rubygems.org"
 
-gem "listen"
 gem "rails", github: "rails/rails", branch: "5-0-stable"
 gem "graphql", "~> 1.8.0"
 gem "devise_token_auth", "0.1.37"

gemfiles/rails5.0_graphql1.9.gemfile

@@ -2,7 +2,6 @@
 
 source "https://rubygems.org"
 
-gem "listen"
 gem "rails", github: "rails/rails", branch: "5-0-stable"
 gem "graphql", "~> 1.9.0"
 

gemfiles/rails5.1_graphql1.8.gemfile

@@ -2,7 +2,6 @@
 
 source "https://rubygems.org"
 
-gem "listen"
 gem "rails", github: "rails/rails", branch: "5-1-stable"
 gem "graphql", "~> 1.8.0"
 gem "devise_token_auth", "0.1.42"

gemfiles/rails5.1_graphql1.9.gemfile

@@ -2,7 +2,6 @@
 
 source "https://rubygems.org"
 
-gem "listen"
 gem "rails", github: "rails/rails", branch: "5-1-stable"
 gem "graphql", "~> 1.9.0"
 

gemfiles/rails5.2_graphql1.8.gemfile

@@ -2,7 +2,6 @@
 
 source "https://rubygems.org"
 
-gem "listen"
 gem "rails", github: "rails/rails", branch: "5-2-stable"
 gem "graphql", "~> 1.8.0"
 gem "devise_token_auth", "0.1.43"

gemfiles/rails5.2_graphql1.9.gemfile

@@ -2,7 +2,6 @@
 
 source "https://rubygems.org"
 
-gem "listen"
 gem "rails", github: "rails/rails", branch: "5-2-stable"
 gem "graphql", "~> 1.9.0"
 

gemfiles/rails_edge_graphql_edge.gemfile

@@ -2,7 +2,6 @@
 
 source "https://rubygems.org"
 
-gem "listen"
 gem "sqlite3", "~> 1.4"
 gem "devise_token_auth", github: "lynndylanhurley/devise_token_auth", branch: "master"
 gem "devise", ">= 4.7"

graphql_devise.gemspec

@@ -35,7 +35,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'pry'
   spec.add_development_dependency 'rake', '~> 10.0'
   spec.add_development_dependency 'rspec-rails'
-  spec.add_development_dependency 'rubocop'
+  spec.add_development_dependency 'rubocop', '0.68.1'
   spec.add_development_dependency 'rubocop-performance'
   spec.add_development_dependency 'rubocop-rspec'
   spec.add_development_dependency 'sqlite3', '~> 1.3.6'

lib/graphql_devise/rails/routes.rb

@@ -3,29 +3,40 @@
 module ActionDispatch::Routing
   class Mapper
     def mount_graphql_devise_for(resource, opts = {})
-      options = {
-        skip: [
-          :sessions,
-          :registrations,
-          :passwords,
-          :confirmations,
-          :token_validations,
-          :omniauth_callbacks,
-          :unlocks
-        ]
-      }.merge(opts)
+      mutation_options = opts[:mutations] || {}
 
       path         = opts.fetch(:at, '/')
       mapping_name = resource.underscore.tr('/', '_')
 
       devise_for(
         resource.pluralize.underscore.tr('/', '_').to_sym,
-        class_name: resource,
-        module:     :devise,
-        path:       path,
-        skip:       options[:skip] + [:omniauth_callbacks]
+        module: :devise,
+        skip:   [:sessions, :registrations, :passwords, :confirmations, :omniauth_callbacks, :unlocks]
       )
 
+      authenticable_type = opts[:authenticable_type] ||
+                           "Types::#{resource}Type".safe_constantize ||
+                           GraphqlDevise::Types::AuthenticableType
+
+      default_mutations = {
+        login:  GraphqlDevise::Mutations::Login,
+        logout: GraphqlDevise::Mutations::Logout
+      }.freeze
+
+      default_mutations.each do |action, mutation|
+        used_mutation = if mutation_options[action].present?
+          mutation_options[action]
+        else
+          new_mutation = Class.new(mutation)
+          new_mutation.graphql_name("#{resource}#{action.to_s.titleize}")
+          new_mutation.field(:authenticable, authenticable_type, null: true)
+
+          new_mutation
+        end
+
+        GraphqlDevise::Types::MutationType.field("#{mapping_name}_#{action}", mutation: used_mutation)
+      end
+
       devise_scope mapping_name.to_sym do
         post "#{path}/graphql_auth", to: 'graphql_devise/graphql#auth'
       end

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

@@ -0,0 +1,14 @@
+module Mutations
+  class Login < GraphqlDevise::Mutations::Login
+    field :user, Types::UserType, null: true
+
+    def resolve(email:, password:)
+      original_payload = super do |user|
+        user.do_something
+        user.reload
+      end
+
+      original_payload.merge(user: original_payload[:authenticable])
+    end
+  end
+end

spec/dummy/app/graphql/types/user_type.rb

@@ -0,0 +1,7 @@
+module Types
+  class UserType < GraphQL::Schema::Object
+    field :email,         String, null: false
+    field :name,          String, null: false
+    field :sign_in_count, Int,    null: true
+  end
+end

spec/dummy/app/models/user.rb

@@ -10,7 +10,13 @@ class User < ApplicationRecord
 
   include DeviseTokenAuth::Concerns::User
 
+  validates :name, presence: true
+
   def valid_for_authentication?
     auth_available && super
   end
+
+  def do_something
+    'Nothing to see here!'
+  end
 end

spec/dummy/bin/rails

@@ -1,4 +1,3 @@
 #!/usr/bin/env ruby
 APP_PATH = File.expand_path('../config/application', __dir__)
-require_relative '../config/boot'
 require 'rails/commands'

spec/dummy/bin/rake

@@ -1,4 +1,3 @@
 #!/usr/bin/env ruby
-require_relative '../config/boot'
 require 'rake'
 Rake.application.run

spec/dummy/config/environments/development.rb

@@ -42,8 +42,4 @@
 
   # Print deprecation notices to the stderr.
   config.active_support.deprecation = :stderr
-
-  # Use an evented file watcher to asynchronously detect changes in source code,
-  # routes, locales, etc. This feature depends on the listen gem.
-  config.file_watcher = ActiveSupport::EventedFileUpdateChecker
 end

spec/dummy/config/routes.rb

@@ -1,3 +1,5 @@
 Rails.application.routes.draw do
-  mount_graphql_devise_for 'User', at: 'api/v1'
+  mount_graphql_devise_for 'User', at: 'api/v1', mutations: {
+    login: Mutations::Login
+  }
 end

spec/factories/users.rb

@@ -1,5 +1,6 @@
 FactoryBot.define do
   factory :user do
+    name      { Faker::FunnyName.two_word_name }
     email     { Faker::Internet.unique.email }
     password  { Faker::Internet.password }