git_plugin = self

namespace :nginx do

task :load_vars do
  set :sites_available,       -> { fetch(:nginx_sites_available_dir) }
  set :sites_enabled,         -> { fetch(:nginx_sites_enabled_dir) }
  set :enabled_application,   -> { File.join(fetch(:sites_enabled),   fetch(:nginx_application_name)) }
  set :available_application, -> { File.join(fetch(:sites_available), fetch(:nginx_application_name)) }
end

# validate_sudo_settings
task :validate_user_settings do
  path_and_dir_keys = [:nginx_log_path, :nginx_sites_enabled_dir, :nginx_sites_available_dir]
  nginx_task_keys   = ['nginx:start', 'nginx:stop', 'nginx:restart', 'nginx:reload', 'nginx:configtest', 'nginx:site:add', 'nginx:site:disable', 'nginx:site:enable', 'nginx:site:remove' ]

  fetch(:nginx_sudo_paths).each do | path |
    abort("Invalid value in :nginx_sudo_paths, Unknown symbol '#{path}'") unless path_and_dir_keys.include? path
  end
  fetch(:nginx_sudo_tasks).each do | task |
    abort("Invalid symbol in :nginx_tasks! Unknown symbol '#{task}'") unless nginx_task_keys.include? task
  end
end

desc "Configtest nginx service"
task :configtest do
  on release_roles fetch(:nginx_roles) do
    nginx_service = fetch(:nginx_service_path)
    execute(
        "sudo #{nginx_service} configtest",
        interaction_handler: PasswdInteractionHandler.new
    )
  end
end

%w[start stop restart reload].each do |command|
  desc "#{command.capitalize} nginx service"
  task command => ['nginx:validate_user_settings'] do
    on release_roles fetch(:nginx_roles) do
      arguments = fetch(:nginx_service_path), command
      git_plugin.add_sudo_if_required arguments, "nginx:#{command}"
      execute *arguments, interaction_handler: PasswdInteractionHandler.new
    end
  end
  before "nginx:#{command}", 'nginx:configtest' unless command == 'stop'
end

task :create_log_paths do
  on release_roles fetch(:nginx_roles) do
    arguments = :mkdir, '-pv', fetch(:nginx_log_path)
    git_plugin.add_sudo_if_required arguments, :nginx_log_path
    execute *arguments, interaction_handler: PasswdInteractionHandler.new
  end
end
after 'deploy:check', 'nginx:create_log_paths' if fetch(:nginx_log_path)

desc 'Compress JS and CSS with gzip'
task :gzip_static => ['nginx:load_vars'] do
  on release_roles fetch(:nginx_roles) do
    within release_path do
      arguments = :find, "'#{fetch(:nginx_static_dir)}' -type f -name '*.js' -o -name '*.css' -exec gzip -v -9 -f -k {} \\;"
      git_plugin.add_sudo_if_required arguments, :gzip_static
      execute *arguments, interaction_handler: PasswdInteractionHandler.new
    end
  end
end

namespace :site do
  desc 'Creates the site configuration and upload it to the available folder'
  task :add => ['nginx:load_vars', 'nginx:validate_user_settings'] do
    on release_roles fetch(:nginx_roles) do
      within fetch(:sites_available) do
        config_file = fetch(:nginx_template)
        if config_file == :default
          config_file = File.expand_path('../../../../templates/nginx.conf.erb', __FILE__)
        end
        config = ERB.new(File.read(config_file)).result(binding)
        upload! StringIO.new(config), '/tmp/nginx.conf'
        arguments = :mv, '/tmp/nginx.conf', fetch(:nginx_application_name)
        git_plugin.add_sudo_if_required arguments, 'nginx:sites:add', :nginx_sites_available_dir
        execute *arguments, interaction_handler: PasswdInteractionHandler.new
      end
    end
  end

  desc 'Enables the site creating a symbolic link into the enabled folder'
  task :enable => ['nginx:load_vars', 'nginx:validate_user_settings'] do
    on release_roles fetch(:nginx_roles) do
      if test "! [ -h #{fetch(:enabled_application)} ]"
        within fetch(:sites_enabled) do
          arguments = :ln, '-nfs', fetch(:available_application), fetch(:enabled_application)
          git_plugin.add_sudo_if_required arguments, 'nginx:sites:enable', :nginx_sites_enabled_dir
          execute *arguments, interaction_handler: PasswdInteractionHandler.new
        end
      end
    end
  end

  desc 'Disables the site removing the symbolic link located in the enabled folder'
  task :disable => ['nginx:load_vars', 'nginx:validate_user_settings'] do
    on release_roles fetch(:nginx_roles) do
      if test "[ -f #{fetch(:enabled_application)} ]"
        within fetch(:sites_enabled) do
          arguments = :rm, '-f', fetch(:nginx_application_name)
          git_plugin.add_sudo_if_required arguments, 'nginx:sites:disable', :nginx_sites_enabled_dir
          execute *arguments, interaction_handler: PasswdInteractionHandler.new
        end
      end
    end
  end

  desc 'Removes the site by removing the configuration file from the available folder'
  task :remove => ['nginx:load_vars', 'nginx:validate_user_settings'] do
    on release_roles fetch(:nginx_roles) do
      if test "[ -f #{fetch(:available_application)} ]"
        within fetch(:sites_available) do
          arguments = :rm, fetch(:nginx_application_name)
          git_plugin.add_sudo_if_required arguments, 'nginx:sites:remove', :nginx_sites_available_dir
          execute *arguments, interaction_handler: PasswdInteractionHandler.new
        end
      end
    end
  end
end

end