# Monit capistrano 3.x tasks
#
# Application config and tasks that apply to all services setup
# with monit for the application
#
# Recommendation:
# Let monit monitor any long-running processes to ensure they keep
# within the limits set by you.

require 'capistrano/dsl/base_paths'
require 'capistrano/dsl/monit_paths'
require 'capistrano/helpers/base'
require 'capistrano/helpers/monit'

include Capistrano::DSL::BasePaths
include Capistrano::DSL::MonitPaths
include Capistrano::Helpers::Base
include Capistrano::Helpers::Monit

namespace :load do
  task :defaults do
    set :monit_dir,            proc { shared_path.join('monit') }
    set :monit_available_path, proc { File.join(fetch(:monit_dir), 'available') }
    set :monit_enabled_path,   proc { File.join(fetch(:monit_dir), 'enabled') }
    set :monit_application_group_name,  proc { user_app_env_underscore }

    set :monit_application_conf_template,  File.join(File.expand_path(File.join(File.dirname(__FILE__), '../../../templates', 'monit')), 'app_include.conf.erb')  # rubocop:disable Metrics/LineLength:

    set :monit_application_conf_file, proc { File.join(fetch(:monit_dir), 'monit.conf') }

    set :monit_event_dir, File.join('/var', 'run', 'monit')
  end
end

namespace :monit do
  desc 'Setup monit for the application'
  task :setup do
    on roles(:app) do |host|
      info "MONIT: Setting up initial monit configuration on #{host}"
      if test "[ ! -d #{fetch(:monit_dir)} ]"
        execute :mkdir, "-p #{fetch(:monit_dir)}"
      end
      if test "[ ! -d #{fetch(:monit_available_path)} ]"
        execute :mkdir, "-p #{fetch(:monit_available_path)}"
      end
      if test "[ ! -d #{fetch(:monit_enabled_path)} ]"
        execute :mkdir, "-p #{fetch(:monit_enabled_path)}"
      end

      # Upload application global monit include file
      upload! template_to_s_io(fetch(:monit_application_conf_template)), fetch(:monit_application_conf_file)
    end
  end

  desc 'Enable monit services for application'
  task :enable do
    on roles(:app) do |host|
      if test("[ ! -h #{monit_etc_app_symlink} ]")
        info "MONIT: Enabling for #{fetch(:application)} on #{host}"
        execute :ln, "-sf #{fetch(:monit_application_conf_file)} #{monit_etc_app_symlink}"
      else
        info "MONIT: Already enabled for #{fetch(:application)} on #{host}"
      end
    end
  end

  desc 'Disable monit services for application'
  task :disable do
    on roles(:app) do |host|
      if test("[ -h #{monit_etc_app_symlink} ]")
        info "MONIT: Disabling for #{fetch(:application)} on #{host}"
        execute :rm, "-f #{monit_etc_app_symlink}"
      else
        info "MONIT: Already disabled for #{fetch(:application)} on #{host}"
      end
    end
  end

  desc 'Purge/remove all monit configurations for the application'
  task :purge do
    on roles(:app) do |host|
      info "MONIT: 'Purging config on #{host}"
      execute :rm, "-rf #{fetch(:monit_dir)}" if test("[ -d #{fetch(:monit_dir)} ]")
      execute :rm, "-f #{monit_etc_app_symlink}"
    end
  end

  desc 'Monitor the application'
  task :monitor do
    on roles(:app) do |host|
      info "MONIT: Application: Monitoring on #{host}"
      command_monit_group('monitor')
    end
  end

  desc 'Unmonitor the application'
  task :unmonitor do
    on roles(:app) do |host|
      info "MONIT: Application: Unmonitoring on #{host}"
      command_monit_group('unmonitor')
    end
  end

  desc 'Stop monitoring the application permanent (Monit saves state)'
  task :stop do
    on roles(:app) do |host|
      info "MONIT: Application: Stopping on #{host}"
      command_monit_group('stop')
    end
  end

  desc 'Start monitoring the application permanent (Monit saves state)'
  task :start do
    on roles(:app) do |host|
      info "MONIT: Application: Starting on #{host}"
      command_monit_group('start')
    end
  end

  desc 'Restart monitoring the application'
  task :restart do
    on roles(:app) do |host|
      info "MONIT: Application: Restarting on #{host}"
      command_monit_group('restart')
    end
  end

  desc 'Reload monit config (global)'
  task :reload do
    on roles(:app) do |host|
      info "MONIT: Global: Reloading on #{host}"
      command_monit('reload')
    end
  end

  desc 'Status monit (global)'
  task :status do
    on roles(:app) do |host|
      info "MONIT: Global: Status on #{host}"
      command_monit('status')
    end
  end

  desc 'Summary monit (global)'
  task :summary do
    on roles(:app) do |host|
      info "MONIT: Global: Summary for #{host}"
      command_monit('summary')
    end
  end

  desc 'Validate monit (global)'
  task :validate do
    on roles(:app) do |host|
      info "MONIT: Global: Validating config on #{host}"
      command_monit('validate')
    end
  end
end

after 'monit:enable', 'monit:reload'

before 'monit:disable', 'monit:unmonitor'
after 'monit:disable', 'monit:reload'

before 'monit:purge', 'monit:unmonitor'