Class: Invitational::Generators::MakeInvitableGenerator

Inherits:
Rails::Generators::Base
  • Object
show all
Defined in:
lib/generators/invitational/make_invitable/make_invitable_generator.rb

Instance Method Summary collapse

Instance Method Details

#add_invitational_referenceObject



9
10
11
12
13
14
15
16
17
18
# File 'lib/generators/invitational/make_invitable/make_invitable_generator.rb', line 9

def add_invitational_reference
  @entity_class = entity_class.gsub(/\,/,"").camelize
  @entity_model = @entity_class.underscore
  @path = "app/models/#{@entity_model}.rb"
  @role_list = roles.map{|role| ":" + role.gsub(/\,/,"")}.join(", ")

  content = "  include Invitational::AcceptsInvitationAs\n  accepts_invitation_as #{@role_list}\n"

  inject_into_class @path, @entity_class.constantize, content
end