git_plugin = self

namespace :deploy do
  before :starting, :check_sidekiq_monit_hooks do
    if fetch(:sidekiq_default_hooks) && fetch(:sidekiq_monit_default_hooks)
      invoke 'sidekiq:monit:add_default_hooks'
    end
  end
end

namespace :sidekiq do
  namespace :monit do
    task :add_default_hooks do
      before 'deploy:updating',  'sidekiq:monit:unmonitor'
      after  'deploy:published', 'sidekiq:monit:monitor'
    end

    desc 'Config Sidekiq monit-service'
    task :config do
      on roles(fetch(:sidekiq_roles)) do |role|
        @role = role
        git_plugin.upload_sidekiq_template 'sidekiq_monit', "#{fetch(:tmp_dir)}/monit.conf", @role

        git_plugin.switch_user(role) do
          mv_command = "mv #{fetch(:tmp_dir)}/monit.conf #{fetch(:sidekiq_monit_conf_dir)}/#{fetch(:sidekiq_monit_conf_file)}"

          git_plugin.sudo_if_needed mv_command
          git_plugin.sudo_if_needed "#{fetch(:monit_bin)} reload"
        end
      end
    end

    desc 'Monitor Sidekiq monit-service'
    task :monitor do
      on roles(fetch(:sidekiq_roles)) do |role|
        git_plugin.switch_user(role) do
          begin
            git_plugin.sudo_if_needed "#{fetch(:monit_bin)} monitor #{git_plugin.sidekiq_service_name}"
          rescue
            invoke 'sidekiq:monit:config'
            git_plugin.sudo_if_needed "#{fetch(:monit_bin)} monitor #{git_plugin.sidekiq_service_name}"
          end
        end
      end
    end

    desc 'Unmonitor Sidekiq monit-service'
    task :unmonitor do
      on roles(fetch(:sidekiq_roles)) do |role|
        git_plugin.switch_user(role) do
          begin
            git_plugin.sudo_if_needed "#{fetch(:monit_bin)} unmonitor #{git_plugin.sidekiq_service_name}"
          rescue
            # no worries here
          end
        end
      end
    end
  end

  desc 'Start Sidekiq monit-service'
  task :start do
    on roles(fetch(:sidekiq_roles)) do |role|
      git_plugin.switch_user(role) do
        git_plugin.sudo_if_needed "#{fetch(:monit_bin)} start #{git_plugin.sidekiq_service_name}"
      end
    end
  end

  desc 'Stop Sidekiq monit-service'
  task :stop do
    on roles(fetch(:sidekiq_roles)) do |role|
      git_plugin.switch_user(role) do
        git_plugin.sudo_if_needed "#{fetch(:monit_bin)} stop #{git_plugin.sidekiq_service_name}"
      end
    end
  end

  desc 'Restart Sidekiq monit-service'
  task :restart do
    on roles(fetch(:sidekiq_roles)) do |role|
      git_plugin.sudo_if_needed "#{fetch(:monit_bin)} restart #{git_plugin.sidekiq_service_name}"
    end
  end

  def sidekiq_service_name
    fetch(:sidekiq_service_name, "sidekiq_#{fetch(:application)}_#{fetch(:sidekiq_env)}")
  end

  def sudo_if_needed(command)
    if use_sudo?
      backend.execute :sudo, command
    else
      backend.execute command
    end
  end

  def use_sudo?
    fetch(:sidekiq_monit_use_sudo)
  end

  def upload_sidekiq_template(from, to, role)
    template = sidekiq_template(from, role)
    backend.upload!(StringIO.new(ERB.new(template).result(binding)), to)
  end

  def sidekiq_template(name, role)
    local_template_directory = fetch(:sidekiq_monit_templates_path)

    search_paths = [
      "#{name}-#{role.hostname}-#{fetch(:stage)}.erb",
      "#{name}-#{role.hostname}.erb",
      "#{name}-#{fetch(:stage)}.erb",
      "#{name}.erb"
    ].map { |filename| File.join(local_template_directory, filename) }

    global_search_path = File.expand_path(
      File.join(*%w[.. .. .. generators capistrano sidekiq monit templates], "#{name}.conf.erb"),
      __FILE__
    )

    search_paths << global_search_path

    template_path = search_paths.detect { |path| File.file?(path) }
    File.read(template_path)
  end
end