Capistrano::Configuration.instance(:must_exist).load do

  # Override default start/stop/restart tasks for Passenger
  namespace :deploy do
    desc <<-DESC
      Overrides the default Capistrano deploy:start.
    DESC
    task :start, :roles => :web do
      run "touch #{current_release}/tmp/restart.txt"
    end
    
    desc <<-DESC
      Overrides the default Capistrano deploy:stop.
    DESC
    task :stop, :roles => :web do
      # Do nothing, 
    end
    
    desc <<-DESC
      Overrides the default Capistrano deploy:restart.
    DESC
    task :restart, :roles => :web do
      run "touch #{current_release}/tmp/restart.txt"
    end
    
    desc <<-DESC
      Run the migrate rake task. By default, it runs this in most recently \
      deployed version of the app. However, you can specify a different release \
      via the migrate_target variable, which must be one of :latest (for the \
      default behavior), or :current (for the release indicated by the `current' \
      symlink). latest release to be deployed with the update_code task). Strings will work \
      for those values instead of symbols, too. You can also specify additional \
      environment variables to pass to rake via the migrate_env variable. Finally, \
      you can specify the full path to the rake executable by setting the rake \
      variable. The defaults are:
    
      set :rake,           \"rake\"
      set :rails_env,      \"production\"
      set :migrate_env,    \"\"
      set :migrate_target, :latest"
    DESC
      task :migrate, :roles => :db, :only => { :primary => true } do
        rake = fetch(:rake, "rake")
        rails_env = fetch(:rails_env, "production")
        migrate_env = fetch(:migrate_env, "")
        migrate_target = fetch(:migrate_target, :latest)
    
        directory = case migrate_target.to_sym
          when :current then current_path
          when :latest  then current_release
          else raise ArgumentError, "unknown migration target #{migrate_target.inspect}"
          end
    
        run "cd #{directory}; rvmsudo bundle exec #{rake} RAILS_ENV=#{rails_env} #{migrate_env} db:migrate"
      end
    
  end
end