Sha256: f70becae519e36c80b4c31197aed109111323f05401cdf359d16c825ffeeb7c3
Contents?: true
Size: 1.73 KB
Versions: 8
Compression:
Stored size: 1.73 KB
Contents
require 'rails/generators/active_record'
module BootswatchRails
module Generators
class CreatedByGenerator < ActiveRecord::Generators::Base
desc "Add created_by and updated_by to resource"
argument :name, type: :string,
desc: "The resource to be updated"
argument :user, type: :string, default: "user",
banner: "name of the user model"
class_option :init, type: :boolean, default: false,
banner: "create helper for view helpers"
class_option :migration, type: :boolean, default: false,
desc: 'Create a migration for added attributes'
class_option :friendly, type: :boolean, default: false,
desc: 'Add friendly_id finder for user'
source_root File.expand_path('../templates', __FILE__)
def add_helper
return unless options.init?
template "created_by_helper.rb", "app/helpers/created_by_helper.rb"
end
def add_migration
return unless options.migration?
migration_template 'created_by_migration.rb', "db/migrate/#{migration_name}.rb"
end
def update_controller
file = "app/controllers/#{table_name}_controller.rb"
curr = "current_#{user}"
inject_into_file file, after: /def update$/ do
"\n @#{name}.updated_by = #{curr} ? #{curr}.id : nil"
end
inject_into_file file, after: /@#{name} = #{class_name}\.new\(#{name}_params\)$/ do
"\n @#{name}.created_by = #{curr} ? #{curr}.id : nil" +
"\n @#{name}.updated_by = #{curr} ? #{curr}.id : nil"
end
end
protected
def migration_name
"add_created_by_to_#{table_name}"
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems