2017-12-21 17:36:29 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2010-07-04 15:08:00 +00:00
|
|
|
require 'rails/generators/active_record'
|
|
|
|
require 'generators/devise/orm_helpers'
|
|
|
|
|
|
|
|
module ActiveRecord
|
|
|
|
module Generators
|
|
|
|
class DeviseGenerator < ActiveRecord::Generators::Base
|
2014-02-25 16:42:55 +00:00
|
|
|
argument :attributes, type: :array, default: [], banner: "field:type field:type"
|
2010-07-17 02:20:07 +00:00
|
|
|
|
2010-07-04 15:08:00 +00:00
|
|
|
include Devise::Generators::OrmHelpers
|
|
|
|
source_root File.expand_path("../templates", __FILE__)
|
|
|
|
|
|
|
|
def copy_devise_migration
|
2011-08-30 08:50:12 +00:00
|
|
|
if (behavior == :invoke && model_exists?) || (behavior == :revoke && migration_exists?(table_name))
|
2016-03-07 14:34:19 +00:00
|
|
|
migration_template "migration_existing.rb", "db/migrate/add_devise_to_#{table_name}.rb", migration_version: migration_version
|
2011-08-25 12:05:19 +00:00
|
|
|
else
|
2016-03-07 14:34:19 +00:00
|
|
|
migration_template "migration.rb", "db/migrate/devise_create_#{table_name}.rb", migration_version: migration_version
|
2011-08-25 12:05:19 +00:00
|
|
|
end
|
2010-07-04 15:08:00 +00:00
|
|
|
end
|
|
|
|
|
2011-08-28 21:02:58 +00:00
|
|
|
def generate_model
|
2014-02-25 16:42:55 +00:00
|
|
|
invoke "active_record:model", [name], migration: false unless model_exists? && behavior == :invoke
|
2011-08-28 21:02:58 +00:00
|
|
|
end
|
2011-12-05 10:48:27 +00:00
|
|
|
|
2010-07-04 15:08:00 +00:00
|
|
|
def inject_devise_content
|
2013-03-13 16:15:27 +00:00
|
|
|
content = model_contents
|
2012-04-27 20:57:44 +00:00
|
|
|
|
2012-10-27 21:58:24 +00:00
|
|
|
class_path = if namespaced?
|
|
|
|
class_name.to_s.split("::")
|
|
|
|
else
|
|
|
|
[class_name]
|
|
|
|
end
|
2012-04-27 20:57:44 +00:00
|
|
|
|
2012-05-02 17:49:33 +00:00
|
|
|
indent_depth = class_path.size - 1
|
2012-05-14 13:26:30 +00:00
|
|
|
content = content.split("\n").map { |line| " " * indent_depth + line } .join("\n") << "\n"
|
2012-05-02 17:49:33 +00:00
|
|
|
|
|
|
|
inject_into_class(model_path, class_path.last, content) if model_exists?
|
2010-07-04 15:08:00 +00:00
|
|
|
end
|
2011-12-05 10:28:04 +00:00
|
|
|
|
|
|
|
def migration_data
|
|
|
|
<<RUBY
|
|
|
|
## Database authenticatable
|
2014-02-25 16:42:55 +00:00
|
|
|
t.string :email, null: false, default: ""
|
|
|
|
t.string :encrypted_password, null: false, default: ""
|
2011-12-05 10:28:04 +00:00
|
|
|
|
|
|
|
## Recoverable
|
|
|
|
t.string :reset_password_token
|
|
|
|
t.datetime :reset_password_sent_at
|
|
|
|
|
|
|
|
## Rememberable
|
|
|
|
t.datetime :remember_created_at
|
|
|
|
|
|
|
|
## Trackable
|
2014-02-25 16:42:55 +00:00
|
|
|
t.integer :sign_in_count, default: 0, null: false
|
2011-12-05 10:28:04 +00:00
|
|
|
t.datetime :current_sign_in_at
|
|
|
|
t.datetime :last_sign_in_at
|
2014-05-07 06:50:21 +00:00
|
|
|
t.#{ip_column} :current_sign_in_ip
|
|
|
|
t.#{ip_column} :last_sign_in_ip
|
2011-12-05 10:28:04 +00:00
|
|
|
|
|
|
|
## Confirmable
|
|
|
|
# t.string :confirmation_token
|
|
|
|
# t.datetime :confirmed_at
|
|
|
|
# t.datetime :confirmation_sent_at
|
|
|
|
# t.string :unconfirmed_email # Only if using reconfirmable
|
|
|
|
|
|
|
|
## Lockable
|
2014-02-25 16:42:55 +00:00
|
|
|
# t.integer :failed_attempts, default: 0, null: false # Only if lock strategy is :failed_attempts
|
2011-12-05 10:28:04 +00:00
|
|
|
# t.string :unlock_token # Only if unlock strategy is :email or :both
|
|
|
|
# t.datetime :locked_at
|
|
|
|
RUBY
|
|
|
|
end
|
2014-05-07 06:50:21 +00:00
|
|
|
|
|
|
|
def ip_column
|
|
|
|
# Padded with spaces so it aligns nicely with the rest of the columns.
|
2014-05-07 07:18:10 +00:00
|
|
|
"%-8s" % (inet? ? "inet" : "string")
|
|
|
|
end
|
|
|
|
|
|
|
|
def inet?
|
2015-08-18 23:42:56 +00:00
|
|
|
postgresql?
|
2014-05-07 20:17:40 +00:00
|
|
|
end
|
|
|
|
|
2015-08-18 23:42:56 +00:00
|
|
|
def rails5?
|
|
|
|
Rails.version.start_with? '5'
|
2014-05-07 06:50:21 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def postgresql?
|
2014-09-02 14:24:31 +00:00
|
|
|
config = ActiveRecord::Base.configurations[Rails.env]
|
|
|
|
config && config['adapter'] == 'postgresql'
|
2014-05-07 06:50:21 +00:00
|
|
|
end
|
2016-03-07 14:34:19 +00:00
|
|
|
|
|
|
|
def migration_version
|
|
|
|
if rails5?
|
|
|
|
"[#{Rails::VERSION::MAJOR}.#{Rails::VERSION::MINOR}]"
|
|
|
|
end
|
|
|
|
end
|
2010-07-04 15:08:00 +00:00
|
|
|
end
|
|
|
|
end
|
2010-07-17 02:20:07 +00:00
|
|
|
end
|