sumomo 0.10.0 → 0.10.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 346a647def575daedb5dc3e3eba04a73024c441be3ae92740cf9edb668992e44
-  data.tar.gz: 379cbf589f7ace6b8da717054ab71754604ed6b863808aadf3b6996a18df292f
+  metadata.gz: 3f4f67797a004db69b413bc1abd72dd8d26a289847a1ec6943d4b84d3ec2e034
+  data.tar.gz: b28799dfccbebbcb03f3a96af34f383bd94d2ec979c405e354ea36164f11ab88
 SHA512:
-  metadata.gz: a0cd59b9b12bf2ec4e1b968f37962c505e9c9e6b7e47ed8846ad7af3737c162f5947bd4e01bfa1cdddfd4d347c378803db10c25ceadf56c5d7d31e0cecbe6b26
-  data.tar.gz: ed8757d5ca20a22791c1ded35a453ffd00f51260e6f5a96fb38adc9cc55df4617d30dbfa4a83bad353d2758bdc1232f51183e1bdbf09591c6142f6e92d98ff84
+  metadata.gz: e4ca377095ae8386d059a35f32efc6c7f57696dd1457301beeee36e6734d437c1cb1b09f1ff4263f9d3d01d517d6bc0aa42e9377c9c0a653e02325e2c7bf6e7c
+  data.tar.gz: 131bf090f8161b6ebe07ddd0eefb304e34426a7aca4c19eb4215111d9b61641977a73ba27baac471f81c6abc6fa6a908bb3438b5c06ca9230a209a3918ae5bd4

data/exe/sumomo

@@ -1,12 +1,12 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
-require 'trollop'
+require 'optimist'
 require 'sumomo'
 require 'yaml'
 
 SUB_COMMANDS = %w[delete create update outputs show diff testapi].freeze
-global_opts = Trollop.options do
+global_opts = Optimist.options do
   banner <<-USAGE
   Sumomo v#{Sumomo::VERSION}
 
@@ -26,49 +26,67 @@ p ENV['AWS_PROFILE']
 
 cmd = ARGV.shift # get the subcommand
 
-cmd_opts = case cmd
-           when 'delete'
-             Sumomo.delete_stack(name: ARGV[0], region: global_opts[:region])
-
-           when 'create', 'update', 'show', 'diff'
-             local_opts = Trollop.options do
-               opt :filename, 'File that describes the stack', type: :string, default: 'Sumomofile'
-               opt :changeset, 'Create a changeset instead of directly update', type: :boolean, default: false
-             end
-             Sumomo.manage_stack(name: ARGV[0], cmd: cmd, changeset: !!local_opts[:changeset], region: global_opts[:region]) do
-               proc = proc {}
-               eval File.read(local_opts[:filename]), proc.binding, local_opts[:filename]
-             end
-
-           when 'outputs'
-             puts "Outputs for stack #{ARGV[0]}"
-             puts Sumomo.get_stack_outputs(name: ARGV[0], region: global_opts[:region]).to_yaml
-
-           when 'login'
-             puts "Login to stack #{ARGV[0]} instance at #{ARGV[1]}"
-             `aws s3 cp s3://#{ARGV[0]}/cloudformation/#{ARGV[0]}_master_key.pem x.txt`
-             key = JSON.parse(File.read('x.txt'))['value'].
-              gsub('-----BEGIN RSA PRIVATE KEY----- ', "-----BEGIN RSA PRIVATE KEY-----\n").
-              gsub(' -----END RSA PRIVATE KEY-----', "\n-----END RSA PRIVATE KEY-----").
-              gsub(/(.{64}) /, "\\1\n")
-             File.write('key.pem', key)
-             `chmod 0600 key.pem`
-             exec "ssh -i 'key.pem' ec2-user@#{ARGV[1]} #{ARGV[2]}"
-
-           when 'testapi'
-             local_opts = Trollop.options do
-               opt :filename, 'File that describes the stack', type: :string, default: 'Sumomofile'
-               opt :apiname, 'Name of the API you want to test', type: :string
-               opt :prettyprint, 'Test API outputs JSON with nice indentation', type: :boolean, default: true
-             end
-             puts 'API Test Mode'
-             Sumomo.test_api(local_opts[:apiname], local_opts[:prettyprint]) do
-               proc = proc {}
-               eval File.read(local_opts[:filename]), proc.binding, local_opts[:filename]
-             end
-             exit(0)
-           else
-             Trollop.die "Unknown subcommand #{cmd.inspect}"
+case cmd
+  when 'delete'
+    Sumomo.delete_stack(name: ARGV[0], region: global_opts[:region])
+
+  when 'rollback'
+    Sumomo.rollback_stack(name: ARGV[0], region: global_opts[:region])
+
+  when 'create', 'update', 'show', 'diff'
+    local_opts = Optimist.options do
+      opt :filename, 'File that describes the stack', type: :string, default: 'Sumomofile'
+      opt :changeset, 'Create a changeset instead of directly update', type: :boolean, default: false
+      opt :rollback, 'Specify whether or not to rollback. Allowed values are [enable, disable]', type: :string, default: 'unspecified'
+    end
+
+    changeset = !!local_opts[:changeset]
+    rollback = local_opts[:rollback].to_sym
+
+    if rollback == :disable && local_opts[:changeset] == false
+      Optimist.die "Rollback cannot be set to disable when changeset is set to false, because we will end up in an update-failed state"
+    end
+
+    Sumomo.manage_stack(
+      name: ARGV[0], 
+      cmd: cmd, 
+      changeset: changeset, 
+      region: global_opts[:region],
+      rollback: rollback) do
+
+      proc = proc {}
+      eval File.read(local_opts[:filename]), proc.binding, local_opts[:filename]
+    end
+
+  when 'outputs'
+    puts "Outputs for stack #{ARGV[0]}"
+    puts Sumomo.get_stack_outputs(name: ARGV[0], region: global_opts[:region]).to_yaml
+
+  when 'login'
+    puts "Login to stack #{ARGV[0]} instance at #{ARGV[1]}"
+      `aws s3 cp s3://#{ARGV[0]}/cloudformation/#{ARGV[0]}_master_key.pem x.txt`
+    key = JSON.parse(File.read('x.txt'))['value'].
+      gsub('-----BEGIN RSA PRIVATE KEY----- ', "-----BEGIN RSA PRIVATE KEY-----\n").
+      gsub(' -----END RSA PRIVATE KEY-----', "\n-----END RSA PRIVATE KEY-----").
+      gsub(/(.{64}) /, "\\1\n")
+    File.write('key.pem', key)
+    `chmod 0600 key.pem`
+    exec "ssh -i 'key.pem' ec2-user@#{ARGV[1]} #{ARGV[2]}"
+
+  when 'testapi'
+    local_opts = Optimist.options do
+      opt :filename, 'File that describes the stack', type: :string, default: 'Sumomofile'
+      opt :apiname, 'Name of the API you want to test', type: :string
+      opt :prettyprint, 'Test API outputs JSON with nice indentation', type: :boolean, default: true
+    end
+    puts 'API Test Mode'
+    Sumomo.test_api(local_opts[:apiname], local_opts[:prettyprint]) do
+      proc = proc {}
+      eval File.read(local_opts[:filename]), proc.binding, local_opts[:filename]
+    end
+    exit(0)
+  else
+    Optimist.die "Unknown subcommand #{cmd.inspect}"
 end
 
 unless %w[show diff].include?(cmd)

data/lib/sumomo/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Sumomo
-  VERSION = '0.10.0'
+  VERSION = '0.10.1'
 end

data/lib/sumomo.rb

@@ -26,7 +26,14 @@ module Sumomo
     "cloudformation/#{make_master_key_name(name: name)}.pem"
   end
 
-  def self.manage_stack(name:, region:, cmd:'update', sns_arn: nil, changeset: false, &block)
+  def self.manage_stack(
+    name:, 
+    region:, 
+    cmd:'update', 
+    sns_arn: nil, 
+    rollback: :unspecified, 
+    changeset: false, &block)
+
     cf = Aws::CloudFormation::Client.new(region: region)
     s3 = Aws::S3::Client.new(region: region)
     ec2 = Aws::EC2::Client.new(region: region)
@@ -164,6 +171,7 @@ module Sumomo
       stack_name: name,
       template_url: store.url('cloudformation/template'),
       parameters: hidden_values,
+      disable_rollback: rollback == :disable,
       capabilities: ['CAPABILITY_IAM', 'CAPABILITY_NAMED_IAM']
     }
 
@@ -448,6 +456,12 @@ module Sumomo
     end
   end
 
+  def self.rollback_stack(name:, region:)
+    cf = Aws::CloudFormation::Client.new(region: region)
+
+    cf.rollback_stack(stack_name: name)
+  end
+
   def self.get_stack_outputs(name:, region:)
     cf = Aws::CloudFormation::Client.new(region: region)
 

data/sumomo.gemspec

@@ -38,6 +38,6 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'momo', '0.4.1'
   spec.add_dependency 'rubyzip'
   spec.add_dependency 's3cabinet'
-  spec.add_dependency 'trollop'
+  spec.add_dependency 'optimist'
   spec.add_dependency 'webrick'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sumomo
 version: !ruby/object:Gem::Version
-  version: 0.10.0
+  version: 0.10.1
 platform: ruby
 authors:
 - David Siaw
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-06-08 00:00:00.000000000 Z
+date: 2023-06-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -151,7 +151,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: trollop
+  name: optimist
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="