cwa 0.1.1 → 0.2.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e21a07d7fe8b8c5f28bdc354c35a44065e0a755ce05ef34ad75eceeca73d53b8
-  data.tar.gz: 7361cab2144cac541b90609f8321b5722d9cc33ed5ac9d3300382c096f033641
+  metadata.gz: e2829f3d40b0966152d43b204bf2b269d56055ffeda7561ae80fa05b5aae137b
+  data.tar.gz: a1d09ffb821c75d33f710ef5b96034c2a809e698ca4f9e9bd5c7bcf4abd61a62
 SHA512:
-  metadata.gz: e5abc10648836931a6bea3c21221459a6b0fbfac65c1beed03d83adb77de7296095133abee6fcf931697006bca2238f8c6ddcf3772e2bd23b937425b331cdad0
-  data.tar.gz: 3de701a4724fc7afe878057c71c441f248a8c441915b93116621f6ee6db118f4c8da5f35a25414a4d63d0a93721649e473dcfa8c079c5a446ffe992595309ddc
+  metadata.gz: e965333fb74efe2744e5d631733028cc209c95e059ffa5c2bcfca3fe97771a80b0443d57e31e81f56fd7517657e2687747a54f8a8976211d48508eeb94e7055b
+  data.tar.gz: 01571ed454488177e3051d807b0f65e515eb602d6cbfd59f38fea251e3a569078bfc86ea328ec3567236dfbb290d8c9f545224f5daafbfe9cb851129fa0ac387

data/.ruby-version

@@ -0,0 +1 @@
+2.6.5

data/.solargraph.yml

@@ -0,0 +1,16 @@
+---
+include:
+- "**/*.rb"
+exclude:
+- spec/**/*
+- test/**/*
+- vendor/**/*
+- ".bundle/**/*"
+require: []
+domains: []
+reporters:
+- rubocop
+- require_not_found
+require_paths: []
+plugins: []
+max_files: 5000

data/.vim-lsp-settings/settings.json

@@ -0,0 +1,5 @@
+{
+  "solargraph": {
+    "cmd": ["bundle", "exec", "solargraph", "stdio"]
+  }
+}

data/cwa.gemspec

@@ -32,6 +32,7 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
+  spec.add_runtime_dependency"aws-sdk-core",        "~> 3"
   spec.add_runtime_dependency"aws-sdk-cloudwatch",  "~> 1"
   spec.add_runtime_dependency"thor",                "~> 1"
   spec.add_runtime_dependency"terminal-table",      "~> 1.8"
@@ -39,4 +40,5 @@ Gem::Specification.new do |spec|
 
   spec.add_development_dependency "bundler"
   spec.add_development_dependency "rake"
+  spec.add_development_dependency "solargraph"
 end

data/exsample/assume.yml

@@ -0,0 +1,4 @@
+---
+<name>:
+  :arn: <arn>
+  :session_name: <session_name>

data/lib/cwa.rb

@@ -1,6 +1,13 @@
-require "cwa/client"
-require "cwa/version"
+# frozen_string_literal: true
 
+require 'aws-sdk-core'
+require 'cwa/client'
+require 'cwa/version'
+
+#--
+# Copyright (c) 2021 Ito Toshifumi
+# cloudwatch alarm cli
+#++
 module CWA
   class Error < StandardError; end
 
@@ -9,9 +16,21 @@ module CWA
       @aws_opts = opts
     end
 
-    def get
-      @aws_opts = {} unless @aws_opts
+    def get(opts = {})
+      @aws_opts         ||= {}
+      @aws_opts[:profile] = opts[:profile] if opts[:profile]
+      @aws_opts[:region]  = opts[:region]  if opts[:region]
+
       Client.new(@aws_opts)
     end
+
+    def assume_role(opts)
+      role_credentials = Aws::AssumeRoleCredentials.new(
+        client: Aws::STS::Client.new(opts),
+        role_arn: opts[:arn],
+        role_session_name: opts[:session_name]
+      )
+      @aws_opts[:credentials] = role_credentials
+    end
   end
 end

data/lib/cwa/alarms.rb

@@ -1,39 +1,62 @@
+# frozen_string_literal: true
+
 module CWA
   class Alarms
     class Error < StandardError; end
 
+    DEFAULT_OPTS = {
+      alarm_name_prefix: '*',
+      max_records:       100
+    }.freeze
+
     def initialize(client, **opts)
       @opts   = opts
       @client = client
     end
 
     def filter(query)
-      alms = _alarms.metric_alarms
+      alms = alarms
 
       # querys
-      name_query       = ->(alm) { alm.alarm_name == query[:name]       }
-      regexp_query  = ->(alm) { alm.alarm_name =~ /#{query[:regexp]}/   }
-      namespace_query  = ->(alm) { alm.namespace  == query[:namespace]  }
-
-      alms = alms.select(&name_query)              if query[:name      ]
-      alms = alms.select(&regexp_query)            if query[:regexp    ]
-      alms = alms.select(&namespace_query)         if query[:namespace ]
-      alms = _dimension?(alms, query[:dimensions]) if query[:dimensions]
+      name_query       = ->(alm) { alm.alarm_name == query[:name]        }
+      regexp_query     = ->(alm) { alm.alarm_name =~ /#{query[:regexp]}/ }
+      namespace_query  = ->(alm) { alm.namespace  == query[:namespace]   }
+
+      alms = alms.select(&name_query)             if query[:name      ]
+      alms = alms.select(&regexp_query)           if query[:regexp    ]
+      alms = alms.select(&namespace_query)        if query[:namespace ]
+      alms = dimension?(alms, query[:dimensions]) if query[:dimensions]
       alms
     end
 
     def refresh
       @alms = nil
-      _alarms
+      alarms
     end
 
     private
-    def _alarms(**opts)
-      opts[:alarm_name_prefix] = "*" unless opts
-      @alms ||= @client.describe_alarms(opts)
+    def alarms(**opts)
+      opts = DEFAULT_OPTS unless opts
+
+      unless @alms
+        @alms      = Array.new
+        alms       = @client.describe_alarms(opts)
+        next_token = alms.next_token
+        @alms     << alms
+
+        while next_token
+          opts[:next_token] = next_token
+          alms              = @client.describe_alarms(opts)
+          next_token        = alms.next_token
+
+          @alms << alms
+        end
+      end
+
+      @alms.map {|e| e.metric_alarms }.flatten
     end
 
-    def _dimension?(alms, dimensions)
+    def dimension?(alms, dimensions)
       alms.select do |alm|
         alm.dimensions.any? do |dims|
           dimensions.keys.any?(dims.name) && dimensions.values.any?(dims.value)

data/lib/cwa/cli.rb

@@ -1,58 +1,75 @@
+# frozen_string_literal: true
+
 require 'cwa'
 require 'thor'
 require 'terminal-table'
 require 'colorize'
+require 'yaml'
+require 'fileutils'
+
+ASSUME_DIR  = File.join(Dir.home,   '.config', 'cwa').freeze
+ASSUME_FILE = File.join(ASSUME_DIR, 'assume.yml').freeze
+
+OUTPUT_KEYS = %i[
+  namespace
+  alarm_name
+  actions_enabled
+  dimensions
+  alarm_arn
+  alarm_description
+].freeze
 
-OUTPUT_KEYS = [
-  :namespace,
-  :alarm_name,
-  :actions_enabled,
-  :dimensions,
-  :alarm_arn,
-  :alarm_description,
-]
 
-OPTIONS = "--name ALARMNAME --regexp ALARMNAME --namespae NAMESPACE --dimensions KEY:VALUE"
+AWS_OPTIONS = %i[
+  profile
+  region
+].freeze
+
+OPTIONS = '--name ALARMNAME --regexp ALARMNAME --namespae NAMESPACE --dimensions KEY:VALUE'.freeze
 
 module CWA
+  # cli class
   class Cli < Thor
-    class_option    :verbose, :type => :boolean
-
-    desc "alarms  #{OPTIONS}", "show cloudwatch alms"
-    option :name,       type: :string, aliases: "n"
-    option :namespace,  type: :string, aliases: "s"
-    option :regexp,     type: :string, aliases: "r"
-    option :dimensions, type: :hash,   aliases: "d"
-    #option ambiguous: :string
+    begin
+    class_option :verbose,      type: :boolean
+    class_option :profile,      type: :string
+    class_option :region,       type: :string
+    class_option :assume_role,  type: :string
+
+    desc "alarms  #{OPTIONS}", 'show cloudwatch alms'
+    option :name,       type: :string, aliases: 'n'
+    option :namespace,  type: :string, aliases: 's'
+    option :regexp,     type: :string, aliases: 'r'
+    option :dimensions, type: :hash,   aliases: 'd'
     def alarms
-      begin
-        alms  = _output_alms
-        raise "not alarms" if alms.empty?
+      enable_assume if options[:assume_role]
 
-        head  = alms.first.keys
-        rows  = alms.map{|alm| alm.values }
-        table = Terminal::Table.new :headings => head, :rows => rows
+      alms  = output_alms
+      raise 'not alarms' if alms.empty?
 
-        puts table
-      rescue => err
-        puts "error => #{err}".colorize(:red)
-        exit 1
-      end
+      head  = alms.first.keys
+      rows  = alms.map{|alm| alm.values }
+      table = Terminal::Table.new :headings => head, :rows => rows
+
+      puts table
     end
 
-    desc "enable  #{OPTIONS}", "enable cloudwatch alms"
-    option :name,       type: :string, aliases: "n"
-    option :namespace,  type: :string, aliases: "s"
-    option :regexp,     type: :string, aliases: "r"
-    option :dimensions, type: :hash,   aliases: "d"
+    desc "enable  #{OPTIONS}", 'enable cloudwatch alms'
+    option :name,       type: :string, aliases: 'n'
+    option :namespace,  type: :string, aliases: 's'
+    option :regexp,     type: :string, aliases: 'r'
+    option :dimensions, type: :hash,   aliases: 'd'
     def enable
       begin
-        cwa  = CWA.get
+        enable_assume if options[:assume_role]
+
+        cwa  = CWA.get(options)
         alms = cwa.alarms(options)
-        alms = _check_alm(alms, :enable)
+        alms = check_alm(alms, :enable)
+
+        raise 'not alarms' if alms.empty?
 
-        raise "not alarms" if alms.empty?
-        _confirm("cloudwatch alarm enable?")
+        confirm('cloudwatch alarm enable?')
 
         alms.each do |alm|
           cwa.enable(alm)
@@ -60,25 +77,28 @@ module CWA
         end
         puts
         alarms
-      rescue => err
-        puts "error => #{err}".colorize(:red)
+      rescue StandardError => e
+        puts "error => #{e}".colorize(:red)
         exit 1
       end
     end
 
-    desc "disable #{OPTIONS}", "disable cloudwatch alms"
-    option :name,       type: :string, aliases: "n"
-    option :namespace,  type: :string, aliases: "s"
-    option :regexp,     type: :string, aliases: "r"
-    option :dimensions, type: :hash,   aliases: "d"
+    desc "disable #{OPTIONS}", 'disable cloudwatch alms'
+    option :name,       type: :string, aliases: 'n'
+    option :namespace,  type: :string, aliases: 's'
+    option :regexp,     type: :string, aliases: 'r'
+    option :dimensions, type: :hash,   aliases: 'd'
     def disable
       begin
-        cwa  = CWA.get
+        _enable_assume if options[:assume_role]
+
+        cwa  = CWA.get(options)
         alms = cwa.alarms(options)
-        alms = _check_alm(alms, :disable)
+        alms = check_alm(alms, :disable)
 
-        raise "not alarms" if alms.empty?
-        _confirm("cloudwatch alarm disable?")
+        raise 'not alarms' if alms.empty?
+
+        confirm('cloudwatch alarm disable?')
 
         alms.each do |alm|
           cwa.disable(alm)
@@ -92,9 +112,37 @@ module CWA
       end
     end
 
+    desc 'configure', 'create config files'
+    def configure
+      configs     = %w[assume_role]
+
+      puts configs
+      print "create type? : "
+      type = $stdin.gets.strip
+      case type
+      when 'assume_role'
+        print "name?         : "
+        name    = $stdin.gets.strip
+        print "arn?          : "
+        arn     = $stdin.gets.strip
+        print "session_name? : "
+        session = $stdin.gets.strip
+
+        assume = {name => { arn: arn, session_name: session}}
+        FileUtils.mkdir_p(ASSUME_DIR) unless Dir.exist?(ASSUME_DIR)
+        file = open(ASSUME_FILE, "w")
+        YAML.dump(assume, file)
+        puts "create => #{ASSUME_FILE.colorize(:yellow)}"
+      end
+    end
+      rescue StandardError => e
+        puts "error => #{e}".colorize(:red)
+        exit 1
+      end
+
     private
-    def _output_alms
-      cwa  = CWA.get
+    def output_alms
+      cwa  = CWA.get(options)
       alms = cwa.alarms(options)
 
       alms.map do |alm|
@@ -106,33 +154,40 @@ module CWA
       end
     end
 
-    def _check_alm(alms, mode)
+    def enable_assume
+      raise 'not config file, pls "cwa configure"' unless File.exist?(ASSUME_FILE)
+
+      assume = YAML.load_file(ASSUME_FILE)[options[:assume_role]]
+      CWA.assume_role(assume)
+    end
+
+    def check_alm(alms, mode)
       check = false if mode == :enable
       check = true  if mode == :disable
       alms.map do |alm|
-        puts "-" * 50
-        puts "namespace       : #{alm[:namespace      ]}"
-        puts "alarm_name      : #{alm[:alarm_name     ]}"
-        puts "dimensions      : #{alm[:dimensions     ]}"
+        puts '-' * 50
+        puts "namespace       : #{alm[:namespace]}"
+        puts "alarm_name      : #{alm[:alarm_name]}"
+        puts "dimensions      : #{alm[:dimensions]}"
         puts "actions_enabled : #{alm[:actions_enabled]}"
         unless alm[:actions_enabled] == check
           puts "=> #{'skip'.colorize(:yellow)}"
-          puts "-" * 50
+          puts '-' * 50
           next
         end
-        puts "-" * 50
+        puts '-' * 50
         alm
       end.compact
     end
 
-    def _confirm(check_word, **opt)
-      true_word  = ( opt[:true] || /yes|y/ )
-      false_word = ( opt[:false] || /no/ )
+    def confirm(check_word, **_opt)
+      true_word  = /yes|y/
+      false_word = /no/
 
       while true
-        print "#{check_word} (#{true_word.inspect.delete("/")}/#{false_word.inspect.delete("/")}) : "
-        case anser = $stdin.gets.strip
-        when true_word then return true
+        print "#{check_word} (#{true_word.inspect.delete('/')}/#{false_word.inspect.delete('/')}) : "
+        case $stdin.gets.strip
+        when true_word  then return true
         when false_word then return false
         end
       end

data/lib/cwa/client.rb

@@ -1,7 +1,10 @@
+# frozen_string_literal: true
+
 require 'aws-sdk-cloudwatch'
-require "cwa/alarms"
+require 'cwa/alarms'
 
 module CWA
+  # AWS client class
   class Client
     class Error < StandardError; end
 
@@ -12,9 +15,7 @@ module CWA
     def alarms(query)
       @alarms ||= Alarms.new(@client)
       alms      = @alarms.filter(query)
-      if block_given?
-        alms.each{|alm| yield alm }
-      end
+      alms.each { |alm| yield alm } if block_given?
       alms
     end
 
@@ -24,12 +25,12 @@ module CWA
 
     def enable(alm)
       alm = alm[:alarm_name]
-      @client.enable_alarm_actions({alarm_names: [alm] })
+      @client.enable_alarm_actions({ alarm_names: [alm] })
     end
 
     def disable(alm)
       alm = alm[:alarm_name]
-      @client.disable_alarm_actions({alarm_names: [alm] })
+      @client.disable_alarm_actions({ alarm_names: [alm] })
     end
   end
 end

data/lib/cwa/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module CWA
-  VERSION = "0.1.1"
+  VERSION = "0.2.5"
 end

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: cwa
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.5
 platform: ruby
 authors:
 - gen
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-10-03 00:00:00.000000000 Z
+date: 2020-10-10 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-core
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
 - !ruby/object:Gem::Dependency
   name: aws-sdk-cloudwatch
   requirement: !ruby/object:Gem::Requirement
@@ -94,6 +108,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: solargraph
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: cloudwatch alarm client
 email:
 - ymdtshm@gmail.com
@@ -103,6 +131,9 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".ruby-version"
+- ".solargraph.yml"
+- ".vim-lsp-settings/settings.json"
 - Gemfile
 - README.md
 - Rakefile
@@ -110,6 +141,7 @@ files:
 - bin/setup
 - cwa.gemspec
 - exe/cwa
+- exsample/assume.yml
 - lib/cwa.rb
 - lib/cwa/alarms.rb
 - lib/cwa/cli.rb
@@ -135,7 +167,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.2
+rubygems_version: 3.0.3
 signing_key:
 specification_version: 4
 summary: cloudwatch alarm client