require 'rake'
require 'rake/testtask'
require 'rake/rdoctask'
 
require 'rubygems'
 
namespace :db do
  desc 'Create Factory stubs for factory girl plugin for non existing objects.'
 
  task :extract_factories => :environment do
    ActiveRecord::Base.establish_connection
    (ActiveRecord::Base.connection.tables - ["schema_info"]).each do |table_name|
      file_name = "#{RAILS_ROOT}/test/factories/#{table_name.singularize}.rb"
      next if File.exists?( file_name )
      File.open(file_name, 'w') do |file|
 
        columns = ActiveRecord::Schema.columns table_name
        model_name = table_name.singularize.split(/\_/).collect{ |word|
          word.capitalize
        }.to_s
        factory_file_content = %Q!
 
Factory.define :#{table_name.singularize}, :class => #{model_name} do |#{table_name.singularize}|
 
!
        columns.each do |column|
          value = column.default || "Have no default value"
          factory_file_content << "  #{table_name.singularize}.#{column.name} = \"#{value}\"\n"
        end
        factory_file_content << "\nend\n"
        file.write factory_file_content
      end
    end
  end
end