clearance 的 generator

clearance
Class Rails::Generator::Commands::Create


<br />#這邊是需要的library <br />require File.expand_path(File.dirname(__FILE__) + "/lib/insert_commands.rb") <br />require File.expand_path(File.dirname(__FILE__) + "/lib/rake_commands.rb") <br />require 'factory_girl' <br /> <br />class ClearanceGenerator < Rails::Generator::Base <br /> <br /> def manifest <br /> record do |m| <br /> # 原來可以這樣insert GOOD <br /> m.insert_into "app/controllers/application_controller.rb", <br /> "include Clearance::Authentication" <br /> <br /> user_model = "app/models/user.rb" <br /> if File.exists?(user_model) <br /> m.insert_into user_model, "include Clearance::User" <br /> else <br /> #原來也可以生目錄 <br /> m.directory File.join("app", "models") <br /> m.file "user.rb", user_model <br /> end <br /> <br /> m.directory File.join("test", "factories") <br /> m.file "factories.rb", "test/factories/clearance.rb" <br /> <br /> #還可以選擇用哪個file migrate, 叫啥名字 <br /> m.migration_template "migrations/#{migration_name}.rb", <br /> 'db/migrate', <br /> :migration_file_name => "clearance_#{migration_name}" <br /> <br /> m.readme "README" <br /> end <br /> private <br /> <br /> def migration_name <br /> #可以check table 在不在 <br /> if ActiveRecord::Base.connection.table_exists?(:users) <br /> 'update_users' <br /> else <br /> 'create_users' <br /> end <br /> end <br /> <br />end <br /> <br />#-------------------------------------------------- <br />Rails::Generator::Commands::Create.class_eval do <br /> def insert_into(file, line) <br /> logger.insert "#{line} into #{file}" <br /> unless options[:pretend] || file_contains?(file, line) <br /> gsub_file file, /^(class|module) .+$/ do |match| <br /> "#{match}\n #{line}" <br /> end <br /> end <br /> end <br />end <br /><br />#/clearance/templates/migrations/create_users.rb <br />class ClearanceCreateUsers < ActiveRecord::Migration <br /> def self.up <br /> create_table(:users) do |t| <br /> t.string :email <br /> t.string :encrypted_password, :limit => 128 <br /> t.string :salt, :limit => 128 <br /> t.string :confirmation_token, :limit => 128 <br /> t.string :remember_token, :limit => 128 <br /> t.boolean :email_confirmed, :default => false, :null => false <br /> t.timestamps <br /> end <br /> <br /> add_index :users, [:id, :confirmation_token] <br /> add_index :users, :email <br /> add_index :users, :remember_token <br /> end <br /> <br /> def self.down <br /> drop_table :users <br /> end <br />end <br /> <br /> <br />#------------------------------------- <br />#/clearance/templates/migrations/update_users.rb <br />class ClearanceUpdateUsers < ActiveRecord::Migration <br /> def self.up <br /><% <br /> # ActiveRecord::Base.connection.columns(:users).collect { |each| each.name } = User.columns.map { |each| each.name } <br /> existing_columns = ActiveRecord::Base.connection.columns(:users).collect { |each| each.name } <br /> columns = [ <br /> [:email, 't.string :email'], <br /> [:encrypted_password, 't.string :encrypted_password, :limit => 128'], <br /> [:salt, 't.string :salt, :limit => 128'], <br /> [:confirmation_token, 't.string :confirmation_token, :limit => 128'], <br /> [:remember_token, 't.string :remember_token, :limit => 128'], <br /> [:email_confirmed, 't.boolean :email_confirmed, :default => false, :null => false'] <br /> ].delete_if {|c| existing_columns.include?(c.first.to_s)} <br />-%> <br /> change_table(:users) do |t| <br /><% columns.each do |c| -%> <br /> <%= c.last %> <br /><% end -%> <br /> end <br /> <br /><% <br /> #連有哪些index 也可以看 <br /> existing_indexes = ActiveRecord::Base.connection.indexes(:users) <br /> index_names = existing_indexes.collect { |each| each.name } <br /> new_indexes = [ <br /> [:index_users_on_id_and_confirmation_token, 'add_index :users, [:id, :confirmation_token]'], <br /> [:index_users_on_email, 'add_index :users, :email'], <br /> [:index_users_on_remember_token, 'add_index :users, :remember_token'] <br /> ].delete_if { |each| index_names.include?(each.first.to_s) } <br />-%> <br /><% new_indexes.each do |each| -%> <br /> <%= each.last %> <br /><% end -%> <br /> end <br /> <br /> def self.down <br /> change_table(:users) do |t| <br /><% unless columns.empty? -%> <br /> t.remove <%= columns.collect { |each| ":#{each.first}" }.join(',') %> <br /><% end -%> <br /> end <br /> end <br />end <br />