cwa 0.1.1 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e21a07d7fe8b8c5f28bdc354c35a44065e0a755ce05ef34ad75eceeca73d53b8
-  data.tar.gz: 7361cab2144cac541b90609f8321b5722d9cc33ed5ac9d3300382c096f033641
+  metadata.gz: 1519d445e5671c8a4bde54a85bfc3718352340250ef13c74b76412a10d2a87cc
+  data.tar.gz: 837f764c34d6249820bbf08a59dd72bb9a96e8fed12ac52d811c9d79ce3299a2
 SHA512:
-  metadata.gz: e5abc10648836931a6bea3c21221459a6b0fbfac65c1beed03d83adb77de7296095133abee6fcf931697006bca2238f8c6ddcf3772e2bd23b937425b331cdad0
-  data.tar.gz: 3de701a4724fc7afe878057c71c441f248a8c441915b93116621f6ee6db118f4c8da5f35a25414a4d63d0a93721649e473dcfa8c079c5a446ffe992595309ddc
+  metadata.gz: 13320f6632e46850cfc74c189cdccf3d38ffb4b647c77ae5435df089adcd229b397496898747d74c1febf47ca304ccff0418021b72e529744691db0b678cd7d6
+  data.tar.gz: 432f170c2c6f920315e7aaa84f961a4ca1709fd8e66d60fa8bf3cd52e07d12c28e89b26f56511a4b9c403c898faae2d91047bfa0aad0312d76b50a27b688f3cb

data/.ruby-version

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

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"

data/exsample/assume.yml

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

data/lib/cwa.rb

@@ -1,3 +1,4 @@
+require 'aws-sdk-core'
 require "cwa/client"
 require "cwa/version"
 
@@ -9,9 +10,21 @@ module CWA
       @aws_opts = opts
     end
 
-    def get
-      @aws_opts = {} unless @aws_opts
+    def get(opts = {})
+      @aws_opts         ||= {}
+      @aws_opts[:profile] = opts.delete(:profile) if opts[:profile]
+      @aws_opts[:region]  = opts.delete(:profile) 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/cli.rb

@@ -2,30 +2,45 @@ require 'cwa'
 require 'thor'
 require 'terminal-table'
 require 'colorize'
+require 'yaml'
+require 'fileutils'
 
-OUTPUT_KEYS = [
-  :namespace,
-  :alarm_name,
-  :actions_enabled,
-  :dimensions,
-  :alarm_arn,
-  :alarm_description,
-]
+ASSUME_DIR  = File.join(Dir.home,   '.config', 'cwa')
+ASSUME_FILE = File.join(ASSUME_DIR, 'assume.yml')
+
+OUTPUT_KEYS = %i(
+  namespace
+  alarm_name
+  actions_enabled
+  dimensions
+  alarm_arn
+  alarm_description
+)
+
+
+AWS_OPTIONS = %i(
+  profile
+  region
+)
 
 OPTIONS = "--name ALARMNAME --regexp ALARMNAME --namespae NAMESPACE --dimensions KEY:VALUE"
 
 module CWA
   class Cli < Thor
-    class_option    :verbose, :type => :boolean
+    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"
-    #option ambiguous: :string
     def alarms
       begin
+        _enable_assume if options[:assume_role]
+
         alms  = _output_alms
         raise "not alarms" if alms.empty?
 
@@ -47,7 +62,9 @@ module CWA
     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)
 
@@ -73,7 +90,9 @@ module CWA
     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)
 
@@ -92,9 +111,33 @@ 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
+
     private
     def _output_alms
-      cwa  = CWA.get
+      cwa  = CWA.get(options)
       alms = cwa.alarms(options)
 
       alms.map do |alm|
@@ -106,6 +149,12 @@ module CWA
       end
     end
 
+    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

data/lib/cwa/version.rb

@@ -1,3 +1,3 @@
 module CWA
-  VERSION = "0.1.1"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: cwa
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.1
 platform: ruby
 authors:
 - gen
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-10-03 00:00:00.000000000 Z
+date: 2020-10-05 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
@@ -103,6 +117,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".ruby-version"
 - Gemfile
 - README.md
 - Rakefile
@@ -110,6 +125,7 @@ files:
 - bin/setup
 - cwa.gemspec
 - exe/cwa
+- exsample/assume.yml
 - lib/cwa.rb
 - lib/cwa/alarms.rb
 - lib/cwa/cli.rb
@@ -135,7 +151,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