Class: DeviseTokenAuth::InstallGenerator

Inherits:
Rails::Generators::Base
  • Object
show all
Includes:
Rails::Generators::Migration
Defined in:
lib/generators/devise_token_auth/install_generator.rb

Instance Method Summary collapse

Instance Method Details

#add_route_mount ⇒ Object 



73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
 
# File 'lib/generators/devise_token_auth/install_generator.rb', line 73

def add_route_mount
  f    = 'config/routes.rb'
  str  = "mount_devise_token_auth_for '#{user_class}', at: '#{mount_path}'"

  if File.exist?(File.join(destination_root, f))
    line = parse_file_for_line(f, 'mount_devise_token_auth_for')

    if line
      existing_user_class = true
    else
      line = 'Rails.application.routes.draw do'
      existing_user_class = false
    end

    if parse_file_for_line(f, str)
      say_status('skipped', "Routes already exist for #{user_class} at #{mount_path}")
    else
      insert_after_line(f, line, str)

      if existing_user_class
        scoped_routes = ''\
          "as :#{user_class.underscore} do\n"\
          "    # Define routes for #{user_class} within this block.\n"\
          "  end\n"
        insert_after_line(f, str, scoped_routes)
      end
    end
  else
    say_status('skipped', "config/routes.rb not found. Add \"mount_devise_token_auth_for '#{user_class}', at: '#{mount_path}'\" to your routes file.")
  end
end

#copy_migrations ⇒ Object 



18
19
20
21
22
23
24
25
26
27
 
# File 'lib/generators/devise_token_auth/install_generator.rb', line 18

def copy_migrations
  if self.class.migration_exists?('db/migrate', "devise_token_auth_create_#{user_class.pluralize.gsub('::','').underscore}")
    say_status('skipped', "Migration 'devise_token_auth_create_#{user_class.pluralize.gsub('::','').underscore}' already exists")
  else
    migration_template(
      'devise_token_auth_create_users.rb.erb',
      "db/migrate/devise_token_auth_create_#{user_class.pluralize.gsub('::','').underscore}.rb"
    )
  end
end

#create_initializer_file ⇒ Object 



14
15
16
 
# File 'lib/generators/devise_token_auth/install_generator.rb', line 14

def create_initializer_file
  copy_file('devise_token_auth.rb', 'config/initializers/devise_token_auth.rb')
end

#create_user_model ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/generators/devise_token_auth/install_generator.rb', line 29

def create_user_model
  fname = "app/models/#{user_class.underscore}.rb"
  if File.exist?(File.join(destination_root, fname))
    inclusion = 'include DeviseTokenAuth::Concerns::User'
    unless parse_file_for_line(fname, inclusion)

      active_record_needle = (Rails::VERSION::MAJOR == 5) ? 'ApplicationRecord' : 'ActiveRecord::Base'
      inject_into_file fname, after: "class #{user_class} < #{active_record_needle}\n" do <<-'RUBY'
        # Include default devise modules.
        devise :database_authenticatable, :registerable,
                :recoverable, :rememberable, :trackable, :validatable,
                :confirmable, :omniauthable
        include DeviseTokenAuth::Concerns::User
        RUBY
      end
    end
  else
    template('user.rb.erb', fname)
  end
end

#include_controller_concerns ⇒ Object 



50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
 
# File 'lib/generators/devise_token_auth/install_generator.rb', line 50

def include_controller_concerns
  fname = 'app/controllers/application_controller.rb'
  line  = 'include DeviseTokenAuth::Concerns::SetUserByToken'

  if File.exist?(File.join(destination_root, fname))
    if parse_file_for_line(fname, line)
      say_status('skipped', 'Concern is already included in the application controller.')
    elsif is_rails_api?
      inject_into_file fname, after: "class ApplicationController < ActionController::API\n" do <<-'RUBY'
  include DeviseTokenAuth::Concerns::SetUserByToken
      RUBY
      end
    else
      inject_into_file fname, after: "class ApplicationController < ActionController::Base\n" do <<-'RUBY'
  include DeviseTokenAuth::Concerns::SetUserByToken
      RUBY
      end
    end
  else
    say_status('skipped', "app/controllers/application_controller.rb not found. Add 'include DeviseTokenAuth::Concerns::SetUserByToken' to any controllers that require authentication.")
  end
end