Capistrano::Configuration.instance.load do

  # Check if remote file exists
  #
  def remote_file_exists?(full_path)
    'true' == capture("if [ -e #{full_path} ]; then echo 'true'; fi").strip
  end

  # Check if process is running
  #
  def process_exists?(pid_file)
    capture("ps -p $(cat #{pid_file}) ; true").strip.split("\n").size == 2
  end

  # Set unicorn vars
  #
  _cset (:app_env) { fetch(:rails_env, 'production') }
  _cset (:unicorn_bin) { "unicorn" }
  _cset (:unicorn_pid) { "#{fetch(:current_path)}/tmp/pids/unicorn.pid" }
  _cset (:unicorn_env) { app_env }
  _cset (:unicorn_std_log) { "log/unicorn.stderr.log" }
  _cset (:unicorn_err_log) { "log/unicorn.stderr.log" }
  _cset (:unicorn_worker_processes) { 2 }
  _cset (:unicorn_timeout) { 30 }
  _cset (:unicorn_listen_backlog) { 2048 }
  _cset (:sidekiq_redis_url) { nil }
  _cset (:sidekiq_redis_count) { 1 }
  _cset (:unicorn_hard_restart) { false }

  namespace :unicorn do
    desc 'Start Unicorn'
    task :start, :roles => :app, :except => {:no_release => true} do
      if remote_file_exists?(unicorn_pid)
        if process_exists?(unicorn_pid)
          logger.important("Unicorn is already running!", "Unicorn")
          next
        else
          run "rm #{unicorn_pid}"
        end
      end

      config_path = "#{current_path}/config/unicorn/#{unicorn_env}.rb"
      if remote_file_exists?(config_path)
        logger.important("Starting...", "Unicorn - #{bundle_cmd} ")
        run "cd #{current_path} && #{bundle_cmd} exec #{unicorn_bin} -c #{config_path} -E #{app_env} -D"
      else
        logger.important("Config file for \"#{unicorn_env}\" environment was not found at \"#{config_path}\"", "Unicorn")
      end
    end

    desc 'Stop Unicorn'
    task :stop, :roles => :app, :except => {:no_release => true} do
      if remote_file_exists?(unicorn_pid)
        if process_exists?(unicorn_pid)
          logger.important("Stopping...", "Unicorn")
          run "#{try_sudo} kill `cat #{unicorn_pid}`"
        else
          run "rm #{unicorn_pid}"
          logger.important("Unicorn is not running.", "Unicorn")
        end
      else
        logger.important("No PIDs found. Check if unicorn is running.", "Unicorn")
      end
    end

    desc 'Unicorn graceful shutdown'
    task :graceful_stop, :roles => :app, :except => {:no_release => true} do
      if remote_file_exists?(unicorn_pid)
        if process_exists?(unicorn_pid)
          logger.important("Stopping...", "Unicorn")
          run "#{try_sudo} kill -s QUIT `cat #{unicorn_pid}`"
        else
          run "rm #{unicorn_pid}"
          logger.important("Unicorn is not running.", "Unicorn")
        end
      else
        logger.important("No PIDs found. Check if unicorn is running.", "Unicorn")
      end
    end

    desc 'Reload Unicorn'
    task :reload, :roles => :app, :except => {:no_release => true} do
      if remote_file_exists?(unicorn_pid)
        logger.important("Stopping...", "Unicorn")
        if unicorn_hard_restart
          unicorn.stop
          sleep(2)
          unicorn.start
        else
          run "#{try_sudo} kill -s USR2 `cat #{unicorn_pid}`"
        end
      else
        logger.important("No PIDs found. Starting Unicorn server...", "Unicorn")
        config_path = "#{current_path}/config/unicorn/#{unicorn_env}.rb"
        if remote_file_exists?(config_path)
          run "cd #{current_path} && #{bundle_cmd} exec #{unicorn_bin} -c #{config_path} -E #{app_env} -D"
        else
          logger.important("Config file for \"#{unicorn_env}\" environment was not found at \"#{config_path}\"", "Unicorn")
        end
      end
    end

    desc "Setup unicorn"
    task :setup, :roles => :app, :except => {:no_release => true} do
      run "mkdir -p \"#{shared_path}/config/unicorn\""
      config_path = "#{shared_path}/config/unicorn/#{rails_env}.rb"
      template_path = File.expand_path('../../templates/unicorn/unicorn.rb.erb', __FILE__)
      vars = {
          'application' => application,
          'current_path' => current_path,
          'unicorn_pid' => unicorn_pid,
          'unicorn_std_log' => unicorn_std_log,
          'unicorn_err_log' => unicorn_err_log,
          'stage' => stage,
          'unicorn_listen_backlog' => unicorn_listen_backlog,
          'unicorn_worker_processes' => unicorn_worker_processes,
          'unicorn_timeout' => unicorn_timeout,
          'sidekiq_redis_count' => sidekiq_redis_count,
          'sidekiq_redis_url' => sidekiq_redis_url
      }
      put(render_erb_template(template_path, vars), config_path)
    end
  end

  after :"deploy:restart", :"unicorn:reload"
  after :"deploy:setup", :"unicorn:setup";

  namespace :deploy do
    task :start, :roles => :app do
      unicorn.start
    end

    task :stop, :roles => :app do
      unicorn.stop
    end
  end
end