Class: Ahoy::Generators::Messages::MongoidGenerator

Inherits:
Rails::Generators::Base
  • Object
show all
Defined in:
lib/generators/ahoy/messages/mongoid_generator.rb

Instance Method Summary collapse

Instance Method Details

#copy_templates ⇒ Object 



13
14
15
16
17
18
19
20
 
# File 'lib/generators/ahoy/messages/mongoid_generator.rb', line 13

def copy_templates
  case encryption
  when "lockbox"
    template "mongoid_lockbox.rb", "app/models/ahoy/message.rb", lockbox_method: lockbox_method
  else
    template "mongoid.rb", "app/models/ahoy/message.rb"
  end
end

#encryption ⇒ Object 



22
23
24
25
26
27
28
29
 
# File 'lib/generators/ahoy/messages/mongoid_generator.rb', line 22

def encryption
  case options[:encryption]
  when "lockbox", "none"
    options[:encryption]
  else
    abort "Error: encryption must be lockbox or none"
  end
end

#lockbox_method ⇒ Object 



31
32
33
34
35
36
37
 
# File 'lib/generators/ahoy/messages/mongoid_generator.rb', line 31

def lockbox_method
  if defined?(Lockbox::VERSION) && Lockbox::VERSION.to_i < 1
    "encrypts"
  else
    "has_encrypted"
  end
end