database.cap

namespace :deploy do
  namespace :check do
    task :linked_files => 'config/database.yml'
  end
end

remote_file 'config/database.yml' => '/tmp/database.yml', roles: :app

file '/tmp/database.yml' do |t|
  default_template = <<-EOF
      base: &base
        adapter: sqlite3
        timeout: 5000
      development:
        database: #{shared_path}/db/development.sqlite3
        <<: *base
      test:
        database: #{shared_path}/db/test.sqlite3
        <<: *base
      production:
        database: #{shared_path}/db/production.sqlite3
        <<: *base
  EOF

  location = fetch(:template_dir, 'lib/capistrano/tasks/templates') + '/database.yml.erb'
  template = File.file?(location) ? File.read(location) : default_template

  config = ERB.new(template)
  File.open t.name, 'w' do |f|
    f.puts config.result(binding)
  end
end