mount method receives model constant instead of String

Author: mcelicalderon

README.md

@@ -82,7 +82,7 @@ Will do the following:
   - Add `devise` modules to `Admin` model
   - Other changes that you can find [here](https://devise-token-auth.gitbook.io/devise-token-auth/config)
 - Add the route to `config/routes.rb`
-  - `mount_graphql_devise_for 'Admin', at: 'api/auth'`
+  - `mount_graphql_devise_for Admin, at: 'api/auth'`
 
 `Admin` could be any model name you are going to be using for authentication,
 and `api/auth` could be any mount path you would like to use for auth.
@@ -102,7 +102,7 @@ You can mount this gem's GraphQL auth schema in your routes file like this:
 
 Rails.application.routes.draw do
   mount_graphql_devise_for(
-    'User',
+    User,
     at: 'api/v1',
     authenticatable_type: Types::MyCustomUserType,
     operations: {

app/controllers/graphql_devise/concerns/set_user_by_token.rb

@@ -3,6 +3,8 @@ module Concerns
     SetUserByToken = DeviseTokenAuth::Concerns::SetUserByToken
 
     SetUserByToken.module_eval do
+      attr_accessor :client_id, :token, :resource
+
       def build_redirect_headers(access_token, client, redirect_header_options = {})
         {
           DeviseTokenAuth.headers_names[:"access-token"] => access_token,
@@ -12,6 +14,12 @@ def build_redirect_headers(access_token, client, redirect_header_options = {})
           :token => access_token
         }.merge(redirect_header_options)
       end
+
+      private
+
+      def resource_class(mapping)
+        GraphqlDevise.mapping_by_name(mapping)
+      end
     end
   end
 end

lib/generators/graphql_devise/install_generator.rb

@@ -29,7 +29,7 @@ def execute_dta_installer
 
     def mount_resource_route
       routes_file = 'config/routes.rb'
-      gem_route   = "mount_graphql_devise_for '#{user_class}', at: '#{mount_path}'"
+      gem_route   = "mount_graphql_devise_for #{user_class}, at: '#{mount_path}'"
       dta_route   = "mount_devise_token_auth_for '#{user_class}', at: '#{mount_path}'"
 
       if file_contains_str?(routes_file, gem_route)

lib/graphql_devise.rb

@@ -23,6 +23,10 @@ def self.resource_mounted?(resource)
   end
 
   def self.add_mapping(name, klass)
+    Devise.add_mapping(
+      name.to_s.pluralize.to_sym,
+      module: :devise, class_name: klass.to_s
+    )
     @mappings[name] = klass
   end
 

lib/graphql_devise/rails/routes.rb

@@ -1,18 +1,8 @@
 module ActionDispatch::Routing
   class Mapper
-    DEVISE_OPERATIONS = [
-      :sessions,
-      :registrations,
-      :passwords,
-      :confirmations,
-      :omniauth_callbacks,
-      :unlocks,
-      :invitations
-    ].freeze
-
     def mount_graphql_devise_for(resource, options = {})
       default_operations = GraphqlDevise::DefaultOperations::MUTATIONS.merge(GraphqlDevise::DefaultOperations::QUERIES)
-      mapping_name       = resource.underscore.tr('/', '_').to_sym
+      mapping_name       = resource.to_s.underscore.tr('/', '_').to_sym
 
       # clean_options responds to all keys defined in GraphqlDevise::MountMethod::SUPPORTED_OPTIONS
       clean_options = GraphqlDevise::MountMethod::OptionSanitizer.new(options).call!
@@ -26,13 +16,6 @@ def mount_graphql_devise_for(resource, options = {})
         ]
       ).validate!
 
-      devise_for(
-        resource.pluralize.underscore.tr('/', '_').to_sym,
-        module:     :devise,
-        class_name: resource,
-        skip:       DEVISE_OPERATIONS
-      )
-
       post clean_options.at, to: 'graphql_devise/graphql#auth'
       get  clean_options.at, to: 'graphql_devise/graphql#auth'
 
@@ -71,7 +54,7 @@ def mount_graphql_devise_for(resource, options = {})
         GraphqlDevise::Types::QueryType.field(action, resolver: resolver)
       end
 
-      GraphqlDevise.add_mapping(mapping_name, resource.constantize)
+      GraphqlDevise.add_mapping(mapping_name, resource)
 
       Devise.mailer.helper(GraphqlDevise::MailerHelper)
     end

spec/dummy/config/routes.rb

@@ -1,5 +1,5 @@
 Rails.application.routes.draw do
-  mount_graphql_devise_for 'User', at: '/api/v1/graphql_auth', operations: {
+  mount_graphql_devise_for User, at: '/api/v1/graphql_auth', operations: {
     login:   Mutations::Login,
     sign_up: Mutations::SignUp
   }, additional_mutations: {
@@ -9,20 +9,20 @@
   }
 
   mount_graphql_devise_for(
-    'Admin',
+    Admin,
     authenticatable_type: Types::CustomAdminType,
     skip:                 [:sign_up, :check_password_token],
     at:                   '/api/v1/admin/graphql_auth'
   )
 
   mount_graphql_devise_for(
-    'Guest',
+    Guest,
     only: [:login, :logout, :sign_up],
     at:   '/api/v1/guest/graphql_auth'
   )
 
   mount_graphql_devise_for(
-    'Users::Customer',
+    Users::Customer,
     only: [:login],
     at:   '/api/v1/user_customer/graphql_auth'
   )

spec/generators/graphql_devise/install_generator_spec.rb

@@ -21,7 +21,7 @@
     let(:args) { [] }
 
     it 'creates and updated required files' do
-      assert_file 'config/routes.rb', /^\s{2}mount_graphql_devise_for 'User', at: 'auth'/
+      assert_file 'config/routes.rb', /^\s{2}mount_graphql_devise_for User, at: 'auth'/
       expect(routes_content).not_to match(dta_route)
 
       assert_file 'config/initializers/devise.rb'
@@ -40,7 +40,7 @@
     let(:args) { %w[Admin api] }
 
     it 'creates and updated required files' do
-      assert_file 'config/routes.rb', /^\s{2}mount_graphql_devise_for 'Admin', at: 'api'/
+      assert_file 'config/routes.rb', /^\s{2}mount_graphql_devise_for Admin, at: 'api'/
       expect(routes_content).not_to match(dta_route)
 
       assert_file 'config/initializers/devise.rb'