task 'backup:db:build_number' do
  def get_build_number
    begin
      f = File.open "#{RAILS_ROOT}/.svn/entries"
      # The revision information is on 4 lines in .svn/entries
      3.times {f.gets}
      f.gets.chomp
    rescue
      'x'
    end
  end

  # SVN  Build number
  BUILD_NUMBER = get_build_number
end

desc 'Create YAML fixtures from your DB content'
task 'backup:db:extract_content' => [:environment, 'backup:db:build_number'] do
  sql  = "SELECT * FROM %s"
  ActiveRecord::Base.establish_connection(ActiveRecord::Base.configurations[RAILS_ENV])
  ActiveRecord::Base.connection.tables.each do |table_name|
    i = "000"
    FileUtils.mkdir_p("#{RAILS_ROOT}/backup/#{RAILS_ENV}/build_#{BUILD_NUMBER}/fixtures/") 

    File.open("#{RAILS_ROOT}/backup/#{RAILS_ENV}/build_#{BUILD_NUMBER}/fixtures/#{table_name}.yml", 'w') do |file|
      data = ActiveRecord::Base.connection.select_all(sql % table_name)
      nb_record = data.size
      
      while i.to_i <  nb_record do
        file.write data[i.to_i, 100].inject({}) { |hash, record|
          hash["#{table_name}_#{i.succ!}"] = record
          hash
        }.to_yaml[5..-1]
        # Delete the "--- \n" part in top of all yaml return
      end
    end
  end
end

desc 'Dump the db schema'
task 'backup:db:extract_schema' => [:environment, 'backup:db:build_number'] do
  require 'active_record/schema_dumper'
  ActiveRecord::Base.establish_connection(ActiveRecord::Base.configurations[RAILS_ENV])
  FileUtils.mkdir_p("#{RAILS_ROOT}/backup/#{RAILS_ENV}/build_#{BUILD_NUMBER}/schema/") 
  File.open("#{RAILS_ROOT}/backup/#{RAILS_ENV}/build_#{BUILD_NUMBER}/schema/schema.rb", "w") do |file|
    ActiveRecord::SchemaDumper.dump(ActiveRecord::Base.connection, file)
  end
end

desc 'create a backup folder containing your db schema and content data (see backup/{env}/build_{build number})'
task 'backup:db:dump' => ['backup:db:extract_content', 'backup:db:extract_schema']

desc 'load your backed up data from a previous build. rake backup:db:load BUILD=1182 or rake backup:db:load BUILD=1182 DUMP_ENV=production'
task 'backup:db:load' => [:environment, 'backup:db:build_number'] do
  @build     = ENV['BUILD'] || BUILD_NUMBER
  @env       = ENV['DUMP_ENV'] || RAILS_ENV
  load("#{RAILS_ROOT}/backup/#{@env}/build_#{@build}/schema/schema.rb")

  require 'active_record/fixtures'
  Dir.glob(File.join(RAILS_ROOT, "backup/#{@env}/build_#{@build}/", 'fixtures', '*.yml')).each do |fixture_file|
    Fixtures.create_fixtures("#{RAILS_ROOT}/backup/#{@env}/build_#{@build}/fixtures", File.basename(fixture_file, '.yml'))
  end
end