namespace :octopus do

desc 'Copy schema version information from master to all shards'
task :copy_schema_versions => :environment do
  abort('Octopus is not enabled for this environment') unless Octopus.enabled?

  connection = ActiveRecord::Base.connection

  current_version  = ActiveRecord::Migrator.current_version
  migrations_paths = ActiveRecord::Migrator.migrations_paths

  connection.send_queries_to_multiple_shards(connection.shard_names) do
    ActiveRecord::Schema.initialize_schema_migrations_table
    ActiveRecord::Schema.assume_migrated_upto_version(current_version, migrations_paths)
  end
end

end