stax 0.1.7 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2f2aca89588152ddee959060e8fc3c1dc1bef1c2abb3a57839ab0a35de9aa208
-  data.tar.gz: b9bcf76bba2aa8a472c07291596142b6d4338a566abd42636fb9cf8b05b6b7be
+  metadata.gz: e17c465590b04d1db946eb5cf12e660fb01a63c348aefcfadca2b5d98e568e52
+  data.tar.gz: d585924ea4d53397cc817cda8a617f7a14c430d84befa9f485ee5440a64653af
 SHA512:
-  metadata.gz: b4f6c035f4c10239132e339ef964be95ac86069a47b6bf0f7f1cf7f25a5b1661d3b0d57a518cde8d4ef018d8dd333cc5bfc6b6db6f71e64846b77839bf037958
-  data.tar.gz: 651280004b99781f0d209b9c91179560b75b238dd0faa1431a6f418f94dc1283e9c06e0a6bb9c9636b847898812f9fa8648f052f47aefbe8230d3cc2c3a33074
+  metadata.gz: 4d71a59e526634d2c27d768db9e979569482e53bfeb43ae3907326dd851ba983c3498d5b04ec11727461abcdee5e1dfcf7f634a0c1b6338901718b9092652fb0
+  data.tar.gz: eb8469a0fd373332b3d51171324e9ec9857c77e8a75cdf50c1f6414304af84bcfde1628087ea1866acf6a63b29f4ca5fa9cc13364ec7d8de59e2d330cd20555a

data/bin/stax

@@ -5,6 +5,8 @@ require 'stax'
 Stax.load_staxfile
 begin
   Stax::Cli.start(ARGV)
-rescue Aws::CloudFormation::Errors::ExpiredToken => e
+rescue Aws::CloudFormation::Errors::ExpiredToken,
+       Aws::CloudFormation::Errors::ValidationError,
+       Aws::Errors::MissingCredentialsError => e
   abort(e.message)
 end

data/lib/stax/aws/cfn.rb

@@ -7,6 +7,7 @@ module Stax
         CREATE_IN_PROGRESS CREATE_FAILED CREATE_COMPLETE
         ROLLBACK_IN_PROGRESS ROLLBACK_FAILED ROLLBACK_COMPLETE
         DELETE_IN_PROGRESS DELETE_FAILED
+        IMPORT_IN_PROGRESS IMPORT_COMPLETE IMPORT_ROLLBACK_IN_PROGRESS IMPORT_ROLLBACK_FAILED IMPORT_ROLLBACK_COMPLETE
         UPDATE_IN_PROGRESS UPDATE_COMPLETE_CLEANUP_IN_PROGRESS UPDATE_COMPLETE
         UPDATE_ROLLBACK_IN_PROGRESS UPDATE_ROLLBACK_FAILED UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS UPDATE_ROLLBACK_COMPLETE
         REVIEW_IN_PROGRESS
@@ -17,6 +18,7 @@ module Stax
         CREATE_COMPLETE:      :green,
         DELETE_COMPLETE:      :green,
         UPDATE_COMPLETE:      :green,
+        IMPORT_COMPLETE:      :green,
         CREATE_FAILED:        :red,
         DELETE_FAILED:        :red,
         UPDATE_FAILED:        :red,
@@ -24,6 +26,7 @@ module Stax
         ROLLBACK_COMPLETE:    :red,
         ## resource action
         Add:    :green,
+        Import: :green,
         Modify: :yellow,
         Remove: :red,
       }
@@ -170,4 +173,4 @@ module Stax
 
     end
   end
-end
+end

data/lib/stax/base.rb

@@ -1,6 +1,9 @@
 require 'thor'
 require 'stax/aws/sts'
 
+## clean exit on ctrl-c for all methods
+trap('SIGINT', 'EXIT')
+
 module Stax
   class Base < Thor
 
@@ -68,7 +71,9 @@ module Stax
         end
       end
 
-      def color(string, hash)
+      ## color a string according to current class COLORS hash
+      def color(string, hash = nil)
+        hash ||= self.class::COLORS
         set_color(string, hash.fetch(string.to_sym, :yellow))
       end
 
@@ -118,6 +123,7 @@ module Stax
 
       ## convert a diff in seconds to d h m s
       def human_time_diff(t, n = 5)
+        t = t.round # handle fractional seconds
         mm, ss = t.divmod(60)
         hh, mm = mm.divmod(60)
         dd, hh = hh.divmod(24)

data/lib/stax/cli/ls.rb

@@ -2,37 +2,43 @@ module Stax
   class Cli < Base
 
     no_commands do
+      ## fields to show in output
+      def ls_stack_fields(s)
+        [ s.stack_name, s.creation_time, color(s.stack_status, Aws::Cfn::COLORS) ]
+      end
+
+      ## list stacks from Staxfile in given order
       def ls_staxfile_stacks
-        stacks = Aws::Cfn.stacks.each_with_object({}) { |s, h| h[s.stack_name] = s }
         print_table Stax.stack_list.map { |id|
           name = stack(id).stack_name
-          if (s = stacks[name])
-            [s.stack_name, s.creation_time, color(s.stack_status, Aws::Cfn::COLORS), s.template_description]
+          if (s = Aws::Cfn.describe(name))
+            ls_stack_fields(s)
           else
-            options[:existing] ? nil : [name, '-']
+            [ name, '-' ]
           end
-        }.compact
+        }
       end
 
+      ## list all extant stacks we think match our prefix
       def ls_stacks_with_prefix(prefix)
         print_table Aws::Cfn.stacks.select { |s|
           s.stack_name.start_with?(prefix || stack_prefix)
         }.map { |s|
-          [s.stack_name, s.creation_time, color(s.stack_status, Aws::Cfn::COLORS), s.template_description]
+          ls_stack_fields(s)
         }.sort
       end
 
+      ## list all stacks in account
       def ls_account_stacks
         print_table Aws::Cfn.stacks.map { |s|
-          [s.stack_name, s.creation_time, color(s.stack_status, Aws::Cfn::COLORS), s.template_description]
+          ls_stack_fields(s)
         }.sort
       end
     end
 
     desc 'ls [PREFIX]', 'list stacks'
-    method_option :existing, aliases: '-e', type: :boolean, default: false, desc: 'list just existing stacks'
-    method_option :all,      aliases: '-a', type: :boolean, default: false, desc: 'list all running stacks with our prefix'
-    method_option :account,  aliases: '-A', type: :boolean, default: false, desc: 'list all running stacks in account'
+    method_option :all,     aliases: '-a', type: :boolean, default: false, desc: 'list all running stacks with our prefix'
+    method_option :account, aliases: '-A', type: :boolean, default: false, desc: 'list all running stacks in account'
     def ls(prefix = nil)
       if options[:account]
         ls_account_stacks
@@ -44,4 +50,4 @@ module Stax
     end
 
   end
-end
+end

data/lib/stax/cli.rb

@@ -9,6 +9,7 @@ module Stax
   class Cli < Base
     class_option :branch, type: :string, default: Git.branch, desc: 'git branch to use'
     class_option :app,    type: :string, default: File.basename(Git.toplevel), desc: 'application name'
+
     ## silence deprecation warning
     ## https://github.com/erikhuda/thor/blob/fb625b223465692a9d8a88cc2a483e126f1a8978/CHANGELOG.md#100
     def self.exit_on_failure?

data/lib/stax/mixin/rds.rb

@@ -14,6 +14,7 @@ module Stax
 
       COLORS = {
         available: :green,
+        'in-sync': :green,
         Complete:  :green,
         Active:    :green,
       }
@@ -40,13 +41,36 @@ module Stax
         def stack_db_subnet_groups
           Aws::Cfn.resources_by_type(my.stack_name, 'AWS::RDS::DBSubnetGroup')
         end
+
+        def print_rds_events(opt)
+          Aws::Rds.client.describe_events(opt).map(&:events).flatten.map do |e|
+            [ e.date, e.message ]
+          end.tap(&method(:print_table))
+        end
+
+      end
+
+      desc 'ls', 'list clusters with members'
+      def ls
+        debug("RDS databases for #{my.stack_name}")
+        stack_rds_clusters.map do |c|
+          cluster = [ c.db_cluster_identifier, 'cluster', color(c.status), c.engine ]
+          instances = c.db_cluster_members.map do |m|
+            role = m.is_cluster_writer ? 'writer' : 'reader'
+            i = Aws::Rds.instances(filters: [ { name: 'db-instance-id', values: [ m.db_instance_identifier ] } ]).first
+            [ '- ' + i.db_instance_identifier, role, color(i.db_instance_status), i.engine, i.availability_zone, i.db_instance_class ]
+          end
+          [ cluster ] + instances
+        end.flatten(1).tap do |list|
+          print_table list
+        end
       end
 
       desc 'clusters', 'list db clusters for stack'
       def clusters
         debug("RDS DB clusters for #{my.stack_name}")
         print_table stack_rds_clusters.map { |c|
-          [c.db_cluster_identifier, c.engine, c.engine_version, color(c.status, COLORS), c.cluster_create_time]
+          [c.db_cluster_identifier, c.engine, c.engine_version, color(c.status), c.cluster_create_time]
         }
       end
 
@@ -65,7 +89,7 @@ module Stax
       def instances
         debug("RDS DB instances for #{my.stack_name}")
         print_table stack_rds_instances.map { |i|
-          [i.db_instance_identifier, i.engine, i.engine_version, color(i.db_instance_status, COLORS), i.db_instance_class, i.db_subnet_group&.vpc_id, i.availability_zone]
+          [i.db_instance_identifier, i.engine, i.engine_version, color(i.db_instance_status), i.db_instance_class, i.db_subnet_group&.vpc_id, i.availability_zone]
         }
       end
 
@@ -92,11 +116,67 @@ module Stax
         end.flatten.each do |g|
           debug("Subnets for group #{g.db_subnet_group_name}")
           print_table g.subnets.map { |s|
-            [s&.subnet_availability_zone&.name, s&.subnet_identifier, color(s&.subnet_status, COLORS)]
+            [s&.subnet_availability_zone&.name, s&.subnet_identifier, color(s&.subnet_status)]
           }
         end
       end
 
+      desc 'failover', 'failover clusters'
+      method_option :target, type: :string, default: nil, desc: 'id of instance to promote'
+      def failover
+        stack_rds_clusters.each do |c|
+          if yes?("Failover #{c.db_cluster_identifier}?", :yellow)
+            resp = Aws::Rds.client.failover_db_cluster(db_cluster_identifier: c.db_cluster_identifier, target_db_instance_identifier: options[:target])
+            puts "failing over #{resp.db_cluster.db_cluster_identifier}"
+          end
+        end
+      end
+
+      desc 'snapshots', 'list snapshots for stack clusters'
+      def snapshots
+        stack_db_clusters.map(&:physical_resource_id).each do |id|
+          debug("Snapshots for cluster #{id}")
+          Aws::Rds.client.describe_db_cluster_snapshots(db_cluster_identifier: id).map(&:db_cluster_snapshots).flatten.map do |s|
+            [ s.db_cluster_snapshot_identifier, s.snapshot_create_time, "#{s.allocated_storage}G", color(s.status), s.snapshot_type ]
+          end.tap do |list|
+            print_table list
+          end
+        end
+      end
+
+      desc 'events', 'list rds events for this stack'
+      option :duration, aliases: '-d', type: :numeric, default: 60*24, desc: 'duration in mins to show'
+      def events
+        stack_db_clusters.map(&:physical_resource_id).each do |id|
+          debug("Events for cluster #{id}")
+          print_rds_events(duration: options[:duration], source_type: 'db-cluster', source_identifier: id)
+        end
+
+        stack_db_instances.map(&:physical_resource_id).each do |id|
+          debug("Events for instance #{id}")
+          print_rds_events(duration: options[:duration], source_type: 'db-instance', source_identifier: id)
+        end
+      end
+
+      desc 'write-forwarding', 'control write-forwarding'
+      method_option :disable, aliases: '-d', type: :boolean, desc: 'disable write-forwarding'
+      method_option :enable,  aliases: '-e', type: :boolean, desc: 'enable write-forwarding'
+      def write_forwarding
+        stack_db_clusters.map(&:physical_resource_id).each do |cluster|
+          if options[:enable]
+            puts "#{cluster} enabling write-forwarding"
+            Aws::Rds.client.modify_db_cluster(db_cluster_identifier: cluster, enable_global_write_forwarding: true)
+          elsif options[:disable]
+            puts "#{cluster} disabling write-forwarding"
+            Aws::Rds.client.modify_db_cluster(db_cluster_identifier: cluster, enable_global_write_forwarding: false)
+          else
+            print_table Aws::Rds.client.describe_db_clusters(db_cluster_identifier: cluster).db_clusters.map { |c|
+              [ c.db_cluster_identifier, c.global_write_forwarding_status || '-' ]
+            }
+          end
+        end
+      end
+
     end
   end
-end
+end

data/lib/stax/stack/changeset.rb

@@ -30,6 +30,23 @@ module Stax
         fail_task(e.message)
       end
 
+      ## create change set to import a resource
+      def change_set_import(resource)
+        Aws::Cfn.changeset(
+          change_set_type: :IMPORT,
+          resources_to_import: [ resource ],
+          stack_name: stack_name,
+          template_body: cfn_template_body,
+          template_url: cfn_template_url,
+          parameters: cfn_parameters_update,
+          capabilities: cfn_capabilities,
+          notification_arns: cfn_notification_arns,
+          change_set_name: change_set_name,
+        ).id
+      rescue ::Aws::CloudFormation::Errors::ValidationError => e
+        fail_task(e.message)
+      end
+
       ## wait and return true if changeset ready for execute
       def change_set_complete?(id)
         begin
@@ -94,5 +111,32 @@ module Stax
       change_set_lock
     end
 
+    desc 'import', 'create and execute changeset to import a resource'
+    option :type,  aliases: '-t', type: :string, default: nil, desc: 'cfn resource (e.g. AWS::DynamoDB::Table)'
+    option :id,    aliases: '-i', type: :string, default: nil, desc: 'logical id (e.g. OrdersTable)'
+    option :key,   aliases: '-k', type: :string, default: nil, desc: 'resource key (e.g. TableName)'
+    option :value, aliases: '-v', type: :string, default: nil, desc: 'resource value (e.g. orders)'
+    def import
+      ## prompt for missing options
+      %i[type id key value].each do |opt|
+        options[opt] ||= ask("Resource #{opt}?", :yellow)
+      end
+
+      ## create import changeset
+      debug("Creating import change set for #{stack_name}")
+      id = change_set_import(
+        resource_type: options[:type],
+        logical_resource_id: options[:id],
+        resource_identifier: {
+          options[:key] => options[:value]
+        }
+      )
+
+      ## wait for changeset, prompt for changes, and execute
+      change_set_complete?(id) || fail_task(change_set_reason(id))
+      change_set_changes(id)
+      change_set_execute(id) && tail && update_warn_imports
+    end
+
   end
 end

data/lib/stax/stack/{imports.rb → exports.rb}

@@ -21,8 +21,8 @@ module Stax
       end
     end
 
-    desc 'imports', 'list imports from this stack'
-    def imports
+    desc 'exports', 'list exports from this stack, and stacks that import them'
+    def exports
       debug("Stacks that import from #{stack_name}")
       print_table Aws::Cfn.exports(stack_name).map { |e|
         imports = (i = Aws::Cfn.imports(e.export_name)).empty? ? '-' : i.join('  ')
@@ -30,5 +30,11 @@ module Stax
       }.sort
     end
 
+    desc 'imports', 'deprecated: use exports'
+    def imports
+      warn("deprecated method: please use 'exports' instead")
+      exports
+    end
+
   end
-end
+end

data/lib/stax/stack/parameters.rb

@@ -21,4 +21,4 @@ module Stax
     end
 
   end
-end
+end

data/lib/stax/version.rb

@@ -1,3 +1,3 @@
 module Stax
-  VERSION = '0.1.7'
+  VERSION = '0.1.10'
 end

data/lib/stax.rb

@@ -16,7 +16,7 @@ require 'stax/stack/crud'
 require 'stax/stack/changeset'
 require 'stax/stack/parameters'
 require 'stax/stack/outputs'
-require 'stax/stack/imports'
+require 'stax/stack/exports'
 require 'stax/stack/resources'
 require 'stax/stack/drift'
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stax
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.10
 platform: ruby
 authors:
 - Richard Lister
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-08-29 00:00:00.000000000 Z
+date: 2022-03-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -582,7 +582,7 @@ files:
 - lib/stax/stack/changeset.rb
 - lib/stax/stack/crud.rb
 - lib/stax/stack/drift.rb
-- lib/stax/stack/imports.rb
+- lib/stax/stack/exports.rb
 - lib/stax/stack/outputs.rb
 - lib/stax/stack/parameters.rb
 - lib/stax/stack/resources.rb
@@ -610,7 +610,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.4
+rubygems_version: 3.1.6
 signing_key:
 specification_version: 4
 summary: Control Cloudformation stack and other stuff.