require 'capistrano'

Capistrano::Configuration.instance.load do
  set_default(:templates_path, "config/deploy/templates")

  set_default(:nginx_server_name) { Capistrano::CLI.ui.ask "Nginx server name: " }
  set_default(:nginx_use_ssl, false)
  set_default(:nginx_ssl_certificate, "#{nginx_server_name}.crt")
  set_default(:nginx_ssl_certificate_key, "#{nginx_server_name}.key")
  set_default(:nginx_ssl_certificate_local_path) {Capistrano::CLI.ui.ask "Local path to ssl certificate: "}
  set_default(:nginx_ssl_certificate_key_local_path) {Capistrano::CLI.ui.ask "Local path to ssl certificate key: "}

  set_default(:unicorn_pid, "#{current_path}/tmp/pids/unicorn.pid")
  set_default(:unicorn_config, "#{shared_path}/config/unicorn.rb")
  set_default(:unicorn_log, "#{shared_path}/log/unicorn.log")
  set_default(:unicorn_user, user)
  set_default(:unicorn_workers) { Capistrano::CLI.ui.ask "Number of unicorn workers: " }

  namespace :nginx do
    desc "Setup nginx configuration for this application"
    task :setup, roles: :web do
      template("nginx_conf.erb", "/tmp/#{application}")
      run "#{sudo} mv /tmp/#{application} /etc/nginx/sites-available/#{application}"
      run "#{sudo} ln -fs /etc/nginx/sites-available/#{application} /etc/nginx/sites-enabled/#{application}"

      if nginx_use_ssl
        put File.read(nginx_ssl_certificate_local_path), "/tmp/#{nginx_ssl_certificate}"
        put File.read(nginx_ssl_certificate_key_local_path), "/tmp/#{nginx_ssl_certificate_key}"

        run "#{sudo} mv /tmp/#{nginx_ssl_certificate} /etc/ssl/certs/#{nginx_ssl_certificate}"
        run "#{sudo} mv /tmp/#{nginx_ssl_certificate_key} /etc/ssl/private/#{nginx_ssl_certificate_key}"

        run "#{sudo} chown root:root /etc/ssl/certs/#{nginx_ssl_certificate}"
        run "#{sudo} chown root:root /etc/ssl/private/#{nginx_ssl_certificate_key}"
      end
    end

    after "deploy:setup", "nginx:setup"
    after "deploy:setup", "nginx:reload"

    desc "Reload nginx configuration"
    task :reload, roles: :web do
      run "#{sudo} /etc/init.d/nginx reload"
    end
  end

  namespace :unicorn do
    desc "Setup Unicorn initializer and app configuration"
    task :setup, roles: :app do
      run "mkdir -p #{shared_path}/config"
      template "unicorn.rb.erb", unicorn_config
      template "unicorn_init.erb", "/tmp/unicorn_init"
      run "chmod +x /tmp/unicorn_init"
      run "#{sudo} mv /tmp/unicorn_init /etc/init.d/unicorn_#{application}"
      run "#{sudo} update-rc.d -f unicorn_#{application} defaults"
    end

    after "deploy:setup", "unicorn:setup"

    %w[start stop restart].each do |command|
      desc "#{command} unicorn"
      task command, roles: :app do
        run "service unicorn_#{application} #{command}"
      end

      after "deploy:#{command}", "unicorn:#{command}"
    end
  end

  desc "Setup logs rotation for nginx and unicorn"
  task :logrotate, roles: [:web, :app] do
    template("logrotate.erb", "/tmp/#{application}_logrotate")
    run "#{sudo} mv /tmp/#{application}_logrotate /etc/logrotate.d/#{application}"
    run "#{sudo} chown root:root /etc/logrotate.d/#{application}"
  end

  after "deploy:setup", "logrotate"

  def template(template_name, target)
    config_file = "#{templates_path}/#{template_name}"
    # if no customized file, proceed with default
    unless File.exists?(config_file)
      config_file = File.join(File.dirname(__FILE__), "../../generators/capistrano/nginx_unicorn/templates/#{template_name}")
    end
    put ERB.new(File.read(config_file)).result(binding), target
  end
end