Sha256: be461ceca1d010c764566e710a98857809d8740f4cc51400670957ab3a355e95
Contents?: true
Size: 1.5 KB
Versions: 14
Compression:
Stored size: 1.5 KB
Contents
require 'rails/generators'
require 'rails/generators/migration'
STORES = %w(active_record redis)
class SocializationGenerator < Rails::Generators::Base
include Rails::Generators::Migration
source_root File.expand_path(File.join('templates'), File.dirname(__FILE__))
class_option :store, :type => :string, :default => 'active_record', :description => "Type of datastore"
def self.next_migration_number(dirname)
if ActiveRecord::Base.timestamped_migrations
Time.now.utc.strftime("%Y%m%d%H%M%S")
else
"%.3d" % (current_migration_number(dirname) + 1)
end
end
def create_migration_file
options[:store].downcase!
unless STORES.include?(options[:store])
puts "Invalid store '#{options[:store]}'. The following stores are valid: #{STORES.join(', ')}."
exit 1
end
copy_file "#{options[:store]}/model_follow.rb", 'app/models/follow.rb'
copy_file "#{options[:store]}/model_like.rb", 'app/models/like.rb'
copy_file "#{options[:store]}/model_mention.rb", 'app/models/mention.rb'
if options[:store] == 'active_record'
migration_template "#{options[:store]}/migration_follows.rb", 'db/migrate/create_follows.rb'
sleep 1 # wait a second to have a unique migration timestamp
migration_template "#{options[:store]}/migration_likes.rb", 'db/migrate/create_likes.rb'
sleep 1 # wait a second to have a unique migration timestamp
migration_template "#{options[:store]}/migration_mentions.rb", 'db/migrate/create_mentions.rb'
end
end
end
Version data entries
14 entries across 14 versions & 2 rubygems