acquia/moonshot

# frozen_string_literal: true

require 'colorize'

module Moonshot
  DoctorCritical = Class.new(RuntimeError)

  #
  # A series of methods for adding "doctor" checks to a mechanism.
  #
  module DoctorHelper
    def doctor_hook
      run_all_checks
    end

    private

    def run_all_checks
      success = true
      puts
      puts self.class.name.split('::').last
      private_methods.each do |meth|
        send(meth) if meth =~ /^doctor_check_/
      rescue DoctorCritical
        # Stop running checks in this Mechanism.
        success = false
        break
      rescue StandardError => e
        success = false
        print '  ✗ '.red
        puts "Exception while running check: #{e.class}: #{e.message.lines.first}"
        break
      end

      success
    end

    def success(str)
      print '  ✓ '.green
      puts str
    end

    def warning(str, additional_info = nil)
      print '  ? '.yellow
      puts str
      additional_info&.lines&.each { |l| puts "      #{l}" }
    end

    def critical(str, additional_info = nil)
      print '  ✗ '.red
      puts str
      additional_info&.lines&.each { |l| puts "      #{l}" }
      raise DoctorCritical
    end
  end
end