Class: CookieCryptable::Generators::CookieCryptGenerator

Inherits:
Rails::Generators::NamedBase
  • Object
show all
Defined in:
lib/generators/cookie_crypt/cookie_crypt_generator.rb

Instance Method Summary collapse

Instance Method Details

#generate_1_0_view_files ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/generators/cookie_crypt/cookie_crypt_generator.rb', line 24

def generate_1_0_view_files
  Dir.mkdir("app/views/devise") unless Dir.exists?("app/views/devise")
  unless Dir.exists?("app/views/devise/cookie_crypt")
    puts "Beginning 1.0 views creation..."
    Dir.mkdir("app/views/devise/cookie_crypt") 
    copy_file "max_login_attempts_reached.html.erb", "app/views/devise/cookie_crypt/max_login_attempts_reached.html.erb"
    copy_file "show.html.erb", "app/views/devise/cookie_crypt/show.html.erb"

    $load_all_views = true
  end
end

#generate_1_1_update ⇒ 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
 
# File 'lib/generators/cookie_crypt/cookie_crypt_generator.rb', line 73

def generate_1_1_update
  unless ActiveRecord::Base.class_eval("#{table_name.camelize.singularize}.inspect['security_hash: text'].blank?")
    unless ActiveRecord::Base.class_eval("#{table_name.camelize.singularize}.inspect['security_question_one: string'].blank?")
      puts "Beginning data cleanup, moving 1.0 database data to 1.1 database style..."
      objs = ActiveRecord::Base.class_eval("#{table_name.camelize.singularize}.all")
      objs.each do |obj|
        next if obj.security_question_one.blank?
        h = {}
        h["security_question_1"] = obj.security_question_one
        h["security_answer_1"]   = obj.security_answer_one
        h["security_question_2"] = obj.security_question_two
        h["security_answer_2"] = obj.security_answer_two
        obj.security_hash = h.to_s

        obj.save
      end

      puts "Completed data cleanup, database is now 1.1 ready."
      puts "Generating cleanup migration that will remove now unneeded security_question_one, security_answer_one, security_question_two, security_answer_two fields."

      $generate_1_1_cleanup_migration = true
    end
  end
end

#generate_1_1_view_files ⇒ Object 



61
62
63
64
65
66
67
68
69
70
71
 
# File 'lib/generators/cookie_crypt/cookie_crypt_generator.rb', line 61

def generate_1_1_view_files
  unless File.exist?("app/views/devise/cookie_crypt/show.js.erb")
    puts "Beginning 1.1 views creation..."
    copy_file "show.js.erb", "app/views/devise/cookie_crypt/show.js.erb"
    copy_file "_extra_fields.html.erb", "app/views/devise/cookie_crypt/_extra_fields.html.erb"
    copy_file "_private_login_questions.html.erb", "app/views/devise/cookie_crypt/_private_login_questions.html.erb"
    copy_file "_public_login_questions.html.erb", "app/views/devise/cookie_crypt/_public_login_questions.html.erb"
    File.delete("app/views/devise/cookie_crypt/show.html.erb")
    copy_file "show.html.erb", "app/views/devise/cookie_crypt/show.html.erb"
  end
end

#generate_1_2_view_files ⇒ Object 



98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
 
# File 'lib/generators/cookie_crypt/cookie_crypt_generator.rb', line 98

def generate_1_2_view_files
  $load_all_views ||= false
  unless File.exist?("app/views/devise/cookie_crypt/_private_login_questions.html.erb")
    unless $load_all_views
      copy_new_files = false
      puts "Enter O  || o  || overwrite to overwrite the view files in app/views/devise/cookie_crypt\n"
      puts "Enter CO || co || copyover  to write new files to app/views/devise/cookie_crypt (you will need to update show.js.erb and show.html.haml yourself)\n"
      puts "Enter N  || n  || no        to not generate anything\n"
      puts "Input:"
      input = STDIN.gets.chomp
    end

    if (input =~ /O|o|overwrite/) == 0
      puts "Beginning 1.2 views creation..."
      File.delete("app/views/devise/cookie_crypt/show.js.erb")
      copy_file "show.js.erb", "app/views/devise/cookie_crypt/show.js.erb"
      File.delete("app/views/devise/cookie_crypt/show.html.erb")
      copy_file "show.html.erb", "app/views/devise/cookie_crypt/show.html.erb"
      File.delete("app/views/devise/cookie_crypt/_extra_fields.html.erb")
      copy_new_files = true
    elsif (input =~ /CO|co|copyover/) == 0
      puts "Beginning 1.2 views creation...(please update the view files yourself)"
      copy_file "show.js.erb", "app/views/devise/cookie_crypt/(copy_into_your)show.js.erb"
      copy_file "show.html.erb", "app/views/devise/cookie_crypt/(copy_into_your)show.html.erb"
      copy_file "_extra_fields.html.erb", "app/views/devise/cookie_crypt/(copy_into_your)_extra_fields.html.erb"
      copy_new_files = true
    elsif $load_all_views
      puts "Beginning 1.2 views creation..."
      copy_new_files = true
    else
      puts "Not updating view files.\nIf you dont want to deal with the cookie_crypt views, you can delete app/views/devise/cookie_crypt and the gem will serve the data."
    end

    if copy_new_files
      copy_file "_private_login_questions.html.erb", "app/views/devise/cookie_crypt/_private_login_questions.html.erb"
      copy_file "_public_login_questions.html.erb", "app/views/devise/cookie_crypt/_public_login_questions.html.erb"
    end
  end
end

#inject_1_0_cookie_crypt_content ⇒ Object

BEGIN 1.0 generator



8
9
10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/generators/cookie_crypt/cookie_crypt_generator.rb', line 8

def inject_1_0_cookie_crypt_content
  if ActiveRecord::Base.class_eval("#{table_name.camelize.singularize}.inspect['security_question_one: string'].blank?") &&
    ActiveRecord::Base.class_eval("#{table_name.camelize.singularize}.inspect['security_hash: text'].blank?")
    puts "Beginning 1.0 content injection..."
    paths = [File.join("app", "models", "#{file_path}.rb"),File.join("config", "initializers", "devise.rb")]
    inject_into_file(paths[0], "cookie_cryptable, :", :after => "devise :") if File.exists?(paths[0])
    if File.exists?(paths[1])
      inject_into_file(paths[1], "\n  # ==> Cookie Crypt Configuration Parameters\n  config.max_cookie_crypt_login_attempts = 3
        \n  # For cookie_deletion_time_frame field, make sure your timeframe parses into an actual date and is a string
        \n  config.cookie_deletion_time_frame = '30.days.from_now'", after: "Devise.setup do |config|")
    end
  end
end

#inject_1_1_cookie_crypt_content ⇒ Object

BEGIN 1.1 generator



39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
 
# File 'lib/generators/cookie_crypt/cookie_crypt_generator.rb', line 39

def inject_1_1_cookie_crypt_content
  if ActiveRecord::Base.class_eval("#{table_name.camelize.singularize}.inspect['security_hash: text'].blank?")
    puts "Beginning 1.1 content injection..."
    paths = [File.join("app", "models", "#{file_path}.rb"),File.join("config", "initializers", "devise.rb")]
    if File.exists?(paths[1])
      inject_into_file(paths[1], "\n  # cookie_crypt_auth_through manages the various styles of authenticating through two factor when the need arises.
        \n  # Valid options are: :one_question_cyclical, :one_question_random, :two_questions_cyclical, :two_questions_random, :all_questions
        \n  config.cookie_crypt_auth_through = :one_question_cyclical
        \n  # cookie_crypt_minimum_questions determines how many questions and answers the user must create the first time they are auth'ing through CC
        \n  # This option must be greater than or equal to 2.
        \n  config.cookie_crypt_minimum_questions = 3
        \n  # cycle_question_on_fail_count determines how many tries the user gets per question(s) before the system changes the questions shown
        \n  # It is recommended to set this to at least 2, but 1 is allowed. This value is ignored if the system is set to :all_questions
        \n  config.cycle_question_on_fail_count = 2
        \n  # enable_custom_question_counts allows users to have *more* than the minimum number of questions. This works via ajax and javascript.
        \n  config.enable_custom_question_counts = false", after: "  # ==> Cookie Crypt Configuration Parameters")
    end
  end
end