cfn_manage 0.6.0 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 035ab73bb14da783dcba810528e796a5a5fd21cea291b533b69a54aeeab66930
-  data.tar.gz: c3317a41490aa56e188e5692d1edf9e64be73ca800bd172906c58297137135ec
+  metadata.gz: cb08cd584ef82577a2d39c688d07e5352113d94bc77bb82b260b311aa4928e8d
+  data.tar.gz: 89ea17602c3daddbd0019bc51aa4ae474a975e948470520d9529f0ebe4d17203
 SHA512:
-  metadata.gz: 0e81937b62aa1ae30ae87b18376103d3b27dbf8a942422655524d97818082be1e4da70e455fd6139df6060a49e89aff5bda20af9ea9c422629b20c1a51cad2a5
-  data.tar.gz: 389307f1ac698c5d52e2bc8e612f01d0bb85252c4e71f07cdf14e36a6547577be4f9a46a7474a4b1f8f3f323beddc0465a2d5c9251d82a7359b8d1b13f7e815d
+  metadata.gz: 6e525014378d5c3b3011aa871fce84d0a26175f067c90fd0d2185c6d354eda26340f59ccf2348b6b8eb5c73aa43cb772c689eed0bc16bcfceab8cdfd6202b4cc
+  data.tar.gz: 5b73fc500491e92a9871c1abebd7c9f8d4c480a7a6b1ab06cba002fc46c492648e6c377afdfba4306140fcbb8aab08f68f7643d143737ab4fc6af1ac388651e9

data/bin/cfn_manage

@@ -58,6 +58,10 @@ OptionParser.new do |opts|
     $options['DOCDB_CLUSTER_ID'] = cluster
   end
 
+  opts.on('--transfer-server-id [TRANSFER_SERVER_ID]') do |server|
+    $options['TRANSFER_SERVER_ID'] = server
+  end
+
   opts.on('--ec2-instance-id [EC2_INSTANCE_ID]') do |ec2|
     $options['EC2_INSTANCE_ID'] = ec2
   end
@@ -143,6 +147,12 @@ case command
   when 'start-docdb-cluster'
     CfnManage::CloudFormation::EnvironmentRunStop.new().start_resource($options['DOCDB_CLUSTER_ID'],'AWS::DocDB::DBCluster')
 
+  # docdb cluster commands
+when 'stop-transfer-server'
+  CfnManage::CloudFormation::EnvironmentRunStop.new().stop_resource($options['TRANSFER_SERVER_ID'],'AWS::Transfer::Server')
+when 'start-transfer-server'
+  CfnManage::CloudFormation::EnvironmentRunStop.new().start_resource($options['TRANSFER_SERVER_ID'],'AWS::Transfer::Server')
+      
   # ec2 instance
   when 'stop-ec2'
     CfnManage::CloudFormation::EnvironmentRunStop.new().stop_resource($options['EC2_INSTANCE_ID'],'AWS::EC2::Instance')

data/bin/usage.txt

@@ -26,6 +26,10 @@ cfn_manage stop-docdb-cluster --docdb-cluster-id [DOCDB_CLUSTER_ID]
 
 cfn_manage start-docdb-cluster --docdb-cluster-id [DOCDB_CLUSTER_ID]
 
+cfn_manage stop-transfer-server --transfer-server-id [TRANSFER_SERVER_ID]
+
+cfn_manage start-transfer-server --transfer-server-id [TRANSFER_SERVER_ID]
+
 cfn_manage stop-ec2 --ec2-instance-id [EC2_INSTANCE_ID]
 
 cfn_manage start-ec2 --ec2-instance-id [EC2_INSTANCE_ID]

data/lib/cfn_manage/asg_start_stop_handler.rb

@@ -141,6 +141,7 @@ module CfnManage
               allHealthy += 1
             else
               $log.info("Instance #{instance.instance_id} not currently healthy...")
+              sleep(15)
             end
           end
 

data/lib/cfn_manage/cf_start_stop_environment.rb

@@ -32,6 +32,7 @@ module CfnManage
           'AWS::AutoScaling::AutoScalingGroup' => '200',
           'AWS::EC2::Instance' => '200',
           'AWS::EC2::SpotFleet' => '200',
+          'AWS::Transfer::Server' => '200',
           'AWS::ECS::Cluster' => '250',
           'AWS::CloudWatch::Alarm' => '300'
       }

data/lib/cfn_manage/start_stop_handler_factory.rb

@@ -6,6 +6,7 @@ require 'cfn_manage/alarm_start_stop_handler'
 require 'cfn_manage/spot_fleet_start_stop_handler'
 require 'cfn_manage/ecs_cluster_start_stop_handler'
 require 'cfn_manage/documentdb_cluster_start_stop_handler'
+require 'cfn_manage/transfer_start_stop_handler'
 
 module CfnManage
 
@@ -40,6 +41,9 @@ module CfnManage
         when 'AWS::ECS::Cluster'
           return CfnManage::EcsClusterStartStopHandler.new(resource_id, skip_wait)
 
+        when 'AWS::Transfer::Server'
+          return CfnManage::TransferStartStopHandler.new(resource_id, skip_wait)
+
         else
           return nil
       end

data/lib/cfn_manage/transfer_start_stop_handler.rb

@@ -0,0 +1,97 @@
+require 'cfn_manage/aws_credentials'
+require 'aws-sdk-transfer'
+
+module CfnManage
+
+  class TransferStartStopHandler
+
+    def initialize(server_id, skip_wait)
+      sftpId = server_id.split("/")
+      @server_id = sftpId.last
+      @skip_wait = skip_wait
+      credentials = CfnManage::AWSCredentials.get_session_credentials("startstoptransfer_#{@server_id}")
+      @client = Aws::Transfer::Client.new(retry_limit: 20)
+      if credentials != nil
+        @client = Aws::Transfer::Client.new(credentials: credentials, retry_limit: 20)
+      end
+    end
+
+    def start(configuration)
+
+      state = get_state()
+
+      if state != "OFFLINE"
+        $log.warn("SFTP Server #{@server_id} is in a state of #{state} and can not be started.")
+        return
+      end
+
+      $log.info("Starting SFTP Server #{@server_id}")
+      @client.start_server({
+        server_id: @server_id,
+      })
+      unless @skip_wait
+        $log.info("Waiting for SFTP to start #{@cluster_id}")
+        wait('ONLINE')
+      end
+
+      return configuration
+    end
+
+    def stop()
+
+      state = get_state()
+
+      if state != "ONLINE"
+        $log.warn("SFTP Server #{@server_id} is in a state of #{state} and can not be stopped.")
+        return {}
+      end
+
+      @client.stop_server({
+        server_id: @server_id,
+      })
+      unless @skip_wait
+        $log.info("Waiting for SFTP to stop #{@cluster_id}")
+        wait('OFFLINE')
+      end
+      return {}
+    end
+
+    def get_state()
+      resp = @client.describe_server({
+        server_id: @server_id,
+      })
+      return resp.server.state
+    end
+
+    def wait(completed_state)
+
+      state_count = 0
+      steady_count = 2
+      attempts = 0
+      until attempts == (max_attempts = 60*6) do
+        state = get_state()
+        $log.info("SFTP Server #{@server_id} state: #{state}, waiting for #{completed_state}")
+
+        if state == "#{completed_state}"
+          state_count = state_count + 1
+          $log.info("#{state_count}/#{steady_count}")
+        elsif ["START_FAILED", "STOP_FAILED"].include?(state)
+          $log.error("SFTP Server #{@server_id} failed to reach state #{completed_state} with current state #{state}!")
+          break
+        else
+          state_count = 0
+        end
+        break if state_count == steady_count
+        attempts = attempts + 1
+        sleep(15)
+      end
+
+      if attempts == max_attempts
+        $log.error("SFTP Server #{@server_id} did not enter #{completed_state} state, however continuing operations...")
+      end
+
+    end
+
+  end
+
+end

data/lib/cfn_manage/version.rb

@@ -1,3 +1,3 @@
 module CfnManage
-  VERSION="0.6.0".freeze
+  VERSION="0.7.0".freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cfn_manage
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.7.0
 platform: ruby
 authors:
 - Base2Services
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-07-11 00:00:00.000000000 Z
+date: 2019-08-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core
@@ -212,6 +212,26 @@ dependencies:
     - - "<"
       - !ruby/object:Gem::Version
         version: '2'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-transfer
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -333,6 +353,7 @@ files:
 - lib/cfn_manage/rds_start_stop_handler.rb
 - lib/cfn_manage/spot_fleet_start_stop_handler.rb
 - lib/cfn_manage/start_stop_handler_factory.rb
+- lib/cfn_manage/transfer_start_stop_handler.rb
 - lib/cfn_manage/version.rb
 homepage: https://github.com/base2Services/cfn-start-stop-stack/blob/master/README.md
 licenses: