pyama86/sac

require 'malsh'
require 'parallel'
module Malsh
  class CLI < Thor
    class_option :slack_webhook, type: :string, aliases: :sw
    class_option :slack_channel, type: :string, aliases: :sc
    class_option :slack_user, type: :string, aliases: :su
    class_option :api_key, type: :string, aliases: :k
    class_option :subject, type: :string, aliases: :s
    class_option :org, type: :string, aliases: :o
    class_option :invert_match, type: :array, aliases: :v
    class_option :invert_role, type: :array, aliases: :r
    class_option :regexp, type: :array, aliases: :e

    desc 'retire', 'check forget retire host'
    def retire
      Malsh.init options

      hosts = Parallel.map(Malsh.hosts) do |h|
        m = Malsh.host_metrics(
          h.id,
          Malsh.host_metric_names(h.id).first,
          Time.now.to_i - 86_400,
          Time.now.to_i
        )
        h if (!m || (m['metrics'] && m['metrics'].size == 0)) &&
             !(h['meta'].has_key?('cloud') && %w[elb rds cloudfront kubernetes].include?(h['meta']['cloud']['provider']))
      end.flatten.compact
      Malsh.notify_host('退役未了ホスト一覧', hosts)
    end

    desc 'maverick', 'check no role'
    def maverick
      Malsh.init options

      hosts = Parallel.map(Malsh.hosts) do |h|
        h if h.roles.keys.size < 1
      end.flatten.compact

      Malsh.notify_host('ロール無所属ホスト一覧', hosts)
    end

    desc 'search', 'search hosts'
    option :past_date, type: :numeric, aliases: :p
    option :cpu_threshold, type: :numeric, aliases: :c
    option :memory_threshold, type: :numeric, aliases: :m
    option :status, type: :string, aliases: :st
    def search
      Malsh.init options
      o = options[:status] ? { status: options[:status] } : {}
      hosts = Malsh.hosts(o)
      Object.const_get('Malsh::HostMetrics').constants.each do |c|
        hosts = Object.const_get("Malsh::HostMetrics::#{c}").check(hosts)
      end
      Malsh.notify_host('ホスト一覧', hosts.compact)
    end

    desc 'alert', 'list alerts'
    def alert
      Malsh.init options
      alerts = Malsh.alerts
      Malsh.notify_alert('アラート一覧', alerts)
    end

    map %w[--version -v] => :__print_version
    desc '--version, -v', 'print the version'
    def __print_version
      puts Malsh::VERSION
    end

    no_commands do
    end
  end
end