Set resource_name when mounting resource

Author: mcelicalderon

.gitignore

@@ -21,3 +21,5 @@
 # rvm config files
 .ruby-version
 .ruby-gemset
+
+.env

app/controllers/graphql_devise/graphql_controller.rb

@@ -2,8 +2,6 @@
 
 module GraphqlDevise
   class GraphqlController < ApplicationController
-    before_action :set_user_by_token
-
     def auth
       result = if params[:_json]
         GraphqlDevise::Schema.multiplex(
@@ -26,11 +24,7 @@ def execute_params(item)
       {
         operation_name: item[:operationName],
         variables:      ensure_hash(item[:variables]),
-        context:        {
-          current_resource: @resource,
-          controller:       self,
-          resource_class:   resource_class
-        }
+        context:        { controller: self }
       }
     end
 

app/graphql/graphql_devise/mutations/base.rb

@@ -4,6 +4,10 @@ module GraphqlDevise
   module Mutations
     class Base < GraphQL::Schema::Mutation
       include Concerns::ControllerMethods
+
+      class << self
+        attr_accessor :resource_name
+      end
     end
   end
 end

app/graphql/graphql_devise/resolvers/base.rb

@@ -4,6 +4,10 @@ module GraphqlDevise
   module Resolvers
     class Base < GraphQL::Schema::Resolver
       include Concerns::ControllerMethods
+
+      class << self
+        attr_accessor :resource_name
+      end
     end
   end
 end

app/graphql/graphql_devise/resolvers/confirm_account.rb

@@ -30,12 +30,6 @@ def resolve(confirmation_token:, redirect_url:)
           raise_user_error(I18n.t('graphql_devise.confirmations.invalid_token'))
         end
       end
-
-      private
-
-      def resource_name
-        resource_class.to_s.underscore.tr('/', '_')
-      end
     end
   end
 end

lib/graphql_devise/concerns/controller_methods.rb

@@ -31,8 +31,12 @@ def controller
         context[:controller]
       end
 
+      def resource_name
+        self.class.resource_name
+      end
+
       def resource_class
-        context[:resource_class]
+        controller.send(:resource_class, resource_name)
       end
 
       def recoverable_enabled?
@@ -44,7 +48,9 @@ def confirmable_enabled?
       end
 
       def current_resource
-        context[:current_resource]
+        return @current_resource if @current_resource.present?
+
+        @current_resource = controller.send(:set_user_by_token, resource_name)
       end
 
       def client

lib/graphql_devise/rails/routes.rb

@@ -4,7 +4,7 @@ def mount_graphql_devise_for(resource, opts = {})
       custom_operations = opts[:operations] || {}
 
       path         = opts.fetch(:at, '/')
-      mapping_name = resource.underscore.tr('/', '_')
+      mapping_name = resource.underscore.tr('/', '_').to_sym
 
       devise_for(
         resource.pluralize.underscore.tr('/', '_').to_sym,
@@ -34,6 +34,7 @@ def mount_graphql_devise_for(resource, opts = {})
 
           new_mutation
         end
+        used_mutation.resource_name = mapping_name
 
         GraphqlDevise::Types::MutationType.field("#{mapping_name}_#{action}", mutation: used_mutation)
       end
@@ -53,13 +54,14 @@ def mount_graphql_devise_for(resource, opts = {})
 
           new_query
         end
+        used_query.resource_name = mapping_name
 
         GraphqlDevise::Types::QueryType.field("#{mapping_name}_#{action}", resolver: used_query)
       end
 
       Devise.mailer.helper(GraphqlDevise::MailerHelper)
 
-      devise_scope mapping_name.to_sym do
+      devise_scope mapping_name do
         post "#{path}/graphql_auth", to: 'graphql_devise/graphql#auth'
         get  "#{path}/graphql_auth", to: 'graphql_devise/graphql#auth'
       end

spec/requests/mutations/logout_spec.rb

@@ -39,4 +39,27 @@
       )
     end
   end
+
+  context 'when using the admin model' do
+    let(:query) do
+      <<-GRAPHQL
+        mutation {
+          adminLogout {
+            authenticable { email }
+          }
+        }
+      GRAPHQL
+    end
+    let(:admin)   { create(:admin, :confirmed) }
+    let(:headers) { admin.create_new_auth_token }
+
+    it 'logs out the admin' do
+      expect(response).not_to include_auth_headers
+      expect(admin.reload.tokens.keys).to be_empty
+      expect(json_response[:data][:adminLogout]).to match(
+        authenticable: { email: admin.email }
+      )
+      expect(json_response[:errors]).to be_nil
+    end
+  end
 end