Class: ScaffoldPlus::Generators::CreatedByGenerator

Inherits:
ActiveRecord::Generators::Base
  • Object
show all
Defined in:
lib/generators/scaffold_plus/created_by/created_by_generator.rb

Instance Method Summary collapse

Instance Method Details

#add_migration ⇒ Object 



19
20
21
22
 
# File 'lib/generators/scaffold_plus/created_by/created_by_generator.rb', line 19

def add_migration
  return unless options.migration?
  migration_template 'created_by_migration.rb', "db/migrate/#{migration_name}.rb"
end

#update_controller ⇒ Object 



40
41
42
43
44
45
46
47
48
49
 
# File 'lib/generators/scaffold_plus/created_by/created_by_generator.rb', line 40

def update_controller
  file = "app/controllers/#{table_name}_controller.rb"
  inject_into_file file, after: /def update$/ do
    "\n    @#{name}.updated_by = current_#{user}.id if current_#{user}"
  end
  inject_into_file file, after: /@#{name} = #{class_name}\.new\(#{name}_params\)$/ do
    "\n    @#{name}.created_by = current_#{user}.id if current_#{user}" +
    "\n    @#{name}.updated_by = current_#{user}.id if current_#{user}"
  end
end

#update_model ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/generators/scaffold_plus/created_by/created_by_generator.rb', line 24

def update_model
  lines = options.before? ? [ "" ] : []
  lines << [
    "  def #{user}_data(action, attrib)",
    "    if action.include?('update')",
    "      #{user.camelize}.find(updated_by).try(attrib)",
    "    else",
    "      #{user.camelize}.find(created_by).try(attrib)",
    "    end",
    "  end",
    ""
  ]
  lines << [ "" ] if options.after?
  inject_into_class "app/models/#{name}.rb", class_name, lines.join("\n")
end