seiton 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f73db8242b517d0858ab9501c36942f9e069d95142c7d11c4b85e42738235096
-  data.tar.gz: d1d0d9a6420570d0d7b685602a1d6bfccdf58d2edd77d6378be47af09ded4248
+  metadata.gz: a1c76283846f2d5849c33ff421c2141e77d6cf2dcff0e4da80b3c443fedb2391
+  data.tar.gz: f54a880b530d61686262d888df107d62021d2907a32d6dcf2d91931d5139fb51
 SHA512:
-  metadata.gz: 96603e9a9eb54aa071d8aaca0fa56a245070cb6cc74aeec2b094db670516252b1be69ee935314cc46ecd325939fb20b1e8e55ccfd6f8324eb2fd3a006abd62e7
-  data.tar.gz: 69bb34f01e5446db1fb113e7295d04805603b6dc6ec80893fe929d7eeb96bc3d1a36d6e2aa7ab7c3709bfdf3c97bb885a4e128bcff907364cd870260f2607eb2
+  metadata.gz: 394b674a35ee00fa9947e64a7782ed2a615fac774239e8809f3a25af0790e82b1e765eb9b732620754ab37d2859af10efc8dc63ecbb0ea151742d699f129ed8e
+  data.tar.gz: d012859cadefc12195b0d8aadd99059b5496c162aee88ffd00df45cd514eccc10b16de411adb92f10c32107699f8c9d0deecbb3eb3d5d247bbdbbd5c33080732

data/.gitignore

@@ -5,7 +5,6 @@
 /coverage/
 /doc/
 /pkg/
-/spec/
 /tmp/
 /vendor/
 .envrc

data/README.md

@@ -29,15 +29,39 @@ $ gem install seiton
 ```sh
 $ bundle exec seiton --help
 Commands:
+  seiton ami             # Delete the EC2 AMI.
   seiton ec2_snapshot    # Delete the EC2 Snapshot.
   seiton eip             # Delete the Elastic IP.
   seiton help [COMMAND]  # Describe available commands or one specific command
+  seiton init            # Initialize seiton.
   seiton instance        # Delete the EC2 Instance.
   seiton rds_snapshot    # Delete the RDS Snapshot.
   seiton sqs_queue       # Delete the SQS Queue.
   seiton version         # Print the version number.
 ```
 
+### Init
+
+#### help
+
+```sh
+$ bundle exec seiton --help init
+Usage:
+  seiton init
+
+Initialize seiton.
+```
+
+#### init
+
+```sh
+$ bundle exec seiton init
+```
+
+* Create a `check` directory in the current directory
+* Create a spec_helper.rb in the `check` directory
+* Create a Rakefile in the current directory
+
 ### Delete EC2 AMI
 
 #### help

data/Rakefile

@@ -1,112 +1,5 @@
 require 'rspec/core/rake_task'
 require 'bundler/gem_tasks'
-require 'aws-sdk'
-require 'highline'
-require 'logger'
 
 RSpec::Core::RakeTask.new('spec')
 task :default => :spec
-
-namespace :check do
-  targets1 = []
-  targets2 = []
-  Dir.glob(['./spec/*', './check/*']).each do |file|
-    target = File.basename(file)
-    if target.include?('_spec.rb') then
-      targets1 << File.basename(target, '_spec.rb')
-    elsif target.include?('_list.txt') then
-      targets2 << File.basename(target, '_list.txt')
-    end
-  end
-
-  targets1.each do |target|
-    desc "Test for #{target} deleted"
-
-    RSpec::Core::RakeTask.new(target.to_sym) do |t|
-      # t.rspec_opts = ["--format documentation", "--format html", "--out ./result_html/#{target}_result.html"]
-      t.rspec_opts = ["--format documentation", "--format html", "--out ./result_html/#{target}_result.html"]
-      t.pattern = "spec/#{target}_spec.rb"
-      t.verbose = true
-    end if target != 'ec2_snapshot'
-  end
-
-  targets2.each do |target|
-    desc "Test for #{target} deleted"
-    task target.to_sym do
-      log.info('Check that ' + target + ' has been deleted.')
-      resouces = []
-      File.open('./check/' + target + '_list.txt', "r") do |f|
-        f.each_line do |line|
-          resouces << line.chomp
-        end
-      end
-      eval "check_#{target}(resouces)"
-    end
-  end
-end
-
-def ec2_client
-  @ec2_client ||= Aws::EC2::Client.new
-end
-
-def rds_client
-  @rds_client ||= Aws::RDS::Client.new
-end
-
-def hl
-  @hl ||= HighLine.new
-end
-
-def log
-  @log ||= Logger.new(STDOUT)
-end
-
-def ignore_resource?(resource)
-  if resource.include?('ignore|')
-    r = resource.split('|').last
-    log.info('[ ' + hl.color("!", :yellow) + ' ] ' + r + ' is not for deletion.')
-    not_deleted = '[ ' + hl.color("\u2714".encode('utf-8'), :green) + ' ] ' + r + ' exists.'
-    deleted = '[ ' + hl.color("\u2715".encode('utf-8'), :red) + ' ] ' + r + ' has been removed.'
-  else
-    r = resource
-    deleted = '[ ' + hl.color("\u2714".encode('utf-8'), :green) + ' ] ' + r + ' has been removed..'
-    not_deleted = '[ ' + hl.color("\u2715".encode('utf-8'), :red) + ' ] ' + r + ' exists.'
-  end
-  return r, deleted, not_deleted
-end
-
-def check_ec2_snapshots(resouces)
-  resouces.each do |resouce|
-    r, deleted, not_deleted = ignore_resource?(resouce)
-    begin
-      ec2_client.describe_snapshots({ snapshot_ids: [ r ] })
-      log.info(not_deleted)
-    rescue
-      log.info(deleted)
-    end
-  end
-end
-
-def check_ec2_eips(resouces)
-  resouces.each do |resouce|
-    r, deleted, not_deleted = ignore_resource?(resouce)
-    res = ec2_client.describe_addresses({ filters: [{ name: 'public-ip', values: [r] }] })
-    if res.addresses.empty?
-      log.info(deleted)
-    else
-      log.info(not_deleted)
-    end
-  end
-end
-
-def check_db_snapshots(resouces)
-  resouces.each do |resouce|
-    r, deleted, not_deleted = ignore_resource?(resouce)
-    res = rds_client.describe_db_cluster_snapshots(db_cluster_snapshot_identifier: r)
-    if res.db_cluster_snapshots.empty?
-      log.info(deleted)
-    else
-      log.info(not_deleted)
-    end
-  end
-end

data/lib/seiton.rb

@@ -1,14 +1,15 @@
-require "seiton/cli"
-require "seiton/client"
-require "seiton/common"
-require "seiton/ext"
-require "seiton/ec2_check"
-require "seiton/ec2"
-require "seiton/rds_check"
-require "seiton/rds"
-require "seiton/sqs_check"
-require "seiton/sqs"
-require "seiton/version"
+require 'seiton/cli'
+require 'seiton/client'
+require 'seiton/common'
+require 'seiton/ext'
+require 'seiton/ec2_check'
+require 'seiton/ec2'
+require 'seiton/ignores'
+require 'seiton/rds_check'
+require 'seiton/rds'
+require 'seiton/sqs_check'
+require 'seiton/sqs'
+require 'seiton/version'
 
 module Seiton
   # Your code goes here...

data/lib/seiton/cli.rb

@@ -16,72 +16,95 @@ module Seiton
       puts Seiton::VERSION
     end
 
+    desc 'init', 'Initialize seiton.'
+    def init
+      require 'seiton/init'
+      Seiton::Init.welcome
+    end
+
     desc 'ami', 'Delete the EC2 AMI.'
     option :before_datetime, type: :string, aliases: '-b', desc: 'Specify the date and time for deletion (delete resources before the specified date and time.)'
     option :ignores, type: :array, aliases: '-i', desc: 'Specify resources to be undeleted (you can specify multiple resources).'
+    option :ignores_file, type: :string, aliases: nil, desc: 'Specify file name for resources list to be undeleted.'
     option :check, type: :boolean, aliases: '-c', desc: 'Check the resources to be deleted.'
     def ami
       unless options[:before_datetime] then
         puts '--before-datetime must be specified. (--before-datetime=YYYY/MM/DD)'
         exit 1
       end
+
+      ignores = Seiton::Ignores.new(options[:ignores_file], options[:ignores]).generate
       seiton = Seiton::Ec2.new
-      seiton.ec2_image(options[:check], options[:before_datetime], options[:ignores])
+      seiton.ec2_image(options[:check], options[:before_datetime], ignores)
     end
 
     desc 'ec2_snapshot', 'Delete the EC2 Snapshot.'
     option :before_datetime, type: :string, aliases: '-b', desc: 'Specify the date and time for deletion (delete resources before the specified date and time.)'
     option :ignores, type: :array, aliases: '-i', desc: 'Specify resources to be undeleted (you can specify multiple resources).'
+    option :ignores_file, type: :string, aliases: nil, desc: 'Specify file name for resources list to be undeleted.'
     option :check, type: :boolean, aliases: '-c', desc: 'Check the resources to be deleted.'
     def ec2_snapshot
       unless options[:before_datetime] then
         puts '--before-datetime must be specified. (--before-datetime=YYYY/MM/DD)'
         exit 1
       end
+
+      ignores = Seiton::Ignores.new(options[:ignores_file], options[:ignores]).generate
       seiton = Seiton::Ec2.new
-      seiton.ec2_snapshots(options[:check], options[:before_datetime], options[:ignores])
+      seiton.ec2_snapshots(options[:check],
+                           options[:before_datetime], ignores)
     end
 
     desc 'instance', 'Delete the EC2 Instance.'
     option :before_datetime, type: :string, aliases: '-b', desc: 'Specify the date and time for deletion (delete resources before the specified date and time.)'
     option :ignores, type: :array, aliases: '-i', desc: 'Specify resources to be undeleted (you can specify multiple resources).'
+    option :ignores_file, type: :string, aliases: nil, desc: 'Specify file name for resources list to be undeleted.'
     option :check, type: :boolean, aliases: '-c', desc: 'Check the resources to be deleted.'
     def instance
       unless options[:before_datetime] then
         puts '--before-datetime must be specified. (--before-datetime=YYYY/MM/DD)'
         exit 1
       end
+
+      ignores = Seiton::Ignores.new(options[:ignores_file], options[:ignores]).generate
       seiton = Seiton::Ec2.new
-      seiton.ec2_instance(options[:check], options[:before_datetime], options[:ignores])
+      seiton.ec2_instance(options[:check], options[:before_datetime], ignores)
     end
 
     desc 'rds_snapshot', 'Delete the RDS Snapshot.'
     option :before_datetime, type: :string, aliases: '-b', desc: 'Specify the date and time for deletion (delete resources before the specified date and time.)'
     option :ignores, type: :array, aliases: '-i', desc: 'Specify resources to be undeleted (you can specify multiple resources).'
+    option :ignores_file, type: :string, aliases: nil, desc: 'Specify file name for resources list to be undeleted.'
     option :check, type: :boolean, aliases: '-c', desc: 'Check the resources to be deleted.'
     def rds_snapshot
       unless options[:before_datetime] then
         puts '--before-datetime must be specified. (--before-datetime=YYYY/MM/DD)'
         exit 1
       end
+
+      ignores = Seiton::Ignores.new(options[:ignores_file], options[:ignores]).generate
       seiton = Seiton::Rds.new
-      seiton.rds_snapshot(options[:check], options[:before_datetime], options[:ignores])
+      seiton.rds_snapshot(options[:check], options[:before_datetime], ignores)
     end
 
     desc 'eip', 'Delete the Elastic IP.'
     option :ignores, type: :array, aliases: '-i', desc: 'Specify resources to be undeleted (you can specify multiple resources).'
+    option :ignores_file, type: :string, aliases: nil, desc: 'Specify file name for resources list to be undeleted.'
     option :check, type: :boolean, aliases: '-c', desc: 'Check the resources to be deleted.'
     def eip
+      ignores = Seiton::Ignores.new(options[:ignores_file], options[:ignores]).generate
       seiton = Seiton::Ec2.new
-      seiton.eip(options[:check], options[:ignores])
+      seiton.eip(options[:check], ignores)
     end
 
     desc 'sqs_queue', 'Delete the SQS Queue.'
     option :ignores, type: :array, aliases: '-i', desc: 'Specify resources to be undeleted (you can specify multiple resources).'
+    option :ignores_file, type: :string, aliases: nil, desc: 'Specify file name for resources list to be undeleted.'
     option :check, type: :boolean, aliases: '-c', desc: 'Check the resources to be deleted.'
     def sqs_queue
+      ignores = Seiton::Ignores.new(options[:ignores_file], options[:ignores]).generate
       seiton = Seiton::Sqs.new
-      seiton.sqs_queue(options[:check], options[:ignores])
+      seiton.sqs_queue(options[:check], ignores)
     end
 
   end

data/lib/seiton/common.rb

@@ -38,5 +38,25 @@ module Seiton
       end
       volume_ids
     end
+
+    def ignores(ignores_file, ignore_list)
+      ignores = nil
+      begin
+        File.open(ignores_file) do |file|
+          file.read.split("\n").each do |ignore|
+            ignores << ignore
+          end
+        end
+      rescue SystemCallError => e
+        puts e.message
+        exit 1
+      rescue IOError => e
+        puts e.message
+        exit 1
+      end unless ignore_file.nil?
+      ignores.concat(ignore_list) unless ignore_list.nil?
+
+      ignores
+    end
   end
 end

data/lib/seiton/ec2_check.rb

@@ -17,7 +17,8 @@ end
 <%- end -%>
 <% end %>
 EOF
-      File.open("spec/" + "ec2_instance_spec.rb", "w") do |file|
+      FileUtils.mkdir_p("check") unless FileTest.exist?("check")
+      File.open("check/" + "ec2_instance_spec.rb", "w") do |file|
         file.puts "require 'spec_helper'"
         file.puts ERB.new(template, nil, "-").result(binding).gsub(/^\n/, "")
       end
@@ -31,7 +32,8 @@ describe ebs("<%= resource.volume_id %>") do
 end
 <% end %>
 EOF
-      File.open("spec/" + "ec2_volumes_spec.rb", "w") do |file|
+      FileUtils.mkdir_p("check") unless FileTest.exist?("check")
+      File.open("check/" + "ec2_volumes_spec.rb", "w") do |file|
         file.puts "require 'spec_helper'"
         file.puts ERB.new(template, nil, "-").result(binding).gsub(/^\n/, "")
       end
@@ -54,7 +56,8 @@ end
 <%- end -%>
 <% end %>
 EOF
-      File.open("spec/" + "ec2_images_spec.rb", "w") do |file|
+      FileUtils.mkdir_p("check") unless FileTest.exist?("check")
+      File.open("check/" + "ec2_images_spec.rb", "w") do |file|
         file.puts "require 'spec_helper'"
         file.puts ERB.new(template, nil, "-").result(binding).gsub(/^\n/, "")
       end
@@ -89,7 +92,8 @@ end
 <%- end -%>
 <% end %>
 EOF
-      File.open("spec/" + "ec2_eips_spec.rb", "w") do |file|
+      FileUtils.mkdir_p("check") unless FileTest.exist?("check")
+      File.open("check/" + "ec2_eips_spec.rb", "w") do |file|
         file.puts "require 'spec_helper'"
         file.puts ERB.new(template, nil, "-").result(binding).gsub(/^\n/, "")
       end

data/lib/seiton/ignores.rb

@@ -0,0 +1,29 @@
+module Seiton
+  module Ignores
+    def initialize(ignores_file, ignores_list)
+      @file = ignores_file
+      @list = inogres_list
+    end
+
+    def generate
+      ignores = nil
+      begin
+        File.open(@file) do |file|
+          file.read.split("\n").each do |ignore|
+            ignores << ignore
+          end
+        end
+      rescue SystemCallError => e
+        puts e.message
+        exit 1
+      rescue IOError => e
+        puts e.message
+        exit 1
+      end unless @file.nil?
+
+      ignores.concat(@list) unless @list.nil?
+
+      ignores
+    end
+  end
+end

data/lib/seiton/init.rb

@@ -0,0 +1,144 @@
+module Seiton
+  class Init
+    def self.welcome
+      puts 'Welcome to seiton (整頓) !!'
+      create_directory
+      write_rakefile
+    end
+
+    def self.create_directory
+      puts 'Create check directory...'
+      FileUtils.mkdir_p('check') unless FileTest.exist?('check')
+      spec_helper = <<~SPEC_HELPER
+require 'awspec'
+Awsecrets.load(secrets_path: File.expand_path('./secrets.yml', File.dirname(__FILE__)))
+      SPEC_HELPER
+
+      File.open('check/spec_helper.rb', 'w') do |f| 
+        f.puts(spec_helper)
+      end
+    end
+
+    def self.write_rakefile
+      puts 'Write Rakefile...'
+      rakefile = <<~RAKE_FILE
+require 'rspec/core/rake_task'
+require 'bundler/gem_tasks'
+require 'aws-sdk'
+require 'highline'
+require 'logger'
+
+RSpec::Core::RakeTask.new('spec')
+task :default => :spec
+
+namespace :check do
+  targets1 = []
+  targets2 = []
+  Dir.glob(['./spec/*', './check/*']).each do |file|
+    target = File.basename(file)
+    if target.include?('_spec.rb') then
+      targets1 << File.basename(target, '_spec.rb')
+    elsif target.include?('_list.txt') then
+      targets2 << File.basename(target, '_list.txt')
+    end
+  end
+
+  targets1.each do |target|
+    desc "Test for \#{target} deleted"
+
+    RSpec::Core::RakeTask.new(target.to_sym) do |t|
+      # t.rspec_opts = ["--format documentation", "--format html", "--out ./result_html/\#{target}_result.html"]
+      t.rspec_opts = ["--format documentation", "--format html", "--out ./result_html/\#{target}_result.html"]
+      t.pattern = "spec/\#{target}_spec.rb"
+      t.verbose = true
+    end if target != 'ec2_snapshot'
+  end
+
+  targets2.each do |target|
+    desc "Test for \#{target} deleted"
+    task target.to_sym do
+      log.info('Check that ' + target + ' has been deleted.')
+      resouces = []
+      File.open('./check/' + target + '_list.txt', "r") do |f|
+        f.each_line do |line|
+          resouces << line.chomp
+        end
+      end
+      eval "check_\#{target}(resouces)"
+    end
+  end
+end
+
+def ec2_client
+  @ec2_client ||= Aws::EC2::Client.new
+end
+
+def rds_client
+  @rds_client ||= Aws::RDS::Client.new
+end
+
+def hl
+  @hl ||= HighLine.new
+end
+
+def log
+  @log ||= Logger.new(STDOUT)
+end
+
+def ignore_resource?(resource)
+  if resource.include?('ignore|')
+    r = resource.split('|').last
+    log.info('[ ' + hl.color("!", :yellow) + ' ] ' + r + ' is not for deletion.')
+    not_deleted = '[ ' + hl.color("\\u2714".encode('utf-8'), :green) + ' ] ' + r + ' exists.'
+    deleted = '[ ' + hl.color("\\u2715".encode('utf-8'), :red) + ' ] ' + r + ' has been removed.'
+  else
+    r = resource
+    deleted = '[ ' + hl.color("\\u2714".encode('utf-8'), :green) + ' ] ' + r + ' has been removed..'
+    not_deleted = '[ ' + hl.color("\\u2715".encode('utf-8'), :red) + ' ] ' + r + ' exists.'
+  end
+  return r, deleted, not_deleted
+end
+
+def check_ec2_snapshots(resouces)
+  resouces.each do |resouce|
+    r, deleted, not_deleted = ignore_resource?(resouce)
+    begin
+      ec2_client.describe_snapshots({ snapshot_ids: [ r ] })
+      log.info(not_deleted)
+    rescue
+      log.info(deleted)
+    end
+  end
+end
+
+def check_ec2_eips(resouces)
+  resouces.each do |resouce|
+    r, deleted, not_deleted = ignore_resource?(resouce)
+    res = ec2_client.describe_addresses({ filters: [{ name: 'public-ip', values: [r] }] })
+    if res.addresses.empty?
+      log.info(deleted)
+    else
+      log.info(not_deleted)
+    end
+  end
+end
+
+def check_db_snapshots(resouces)
+  resouces.each do |resouce|
+    r, deleted, not_deleted = ignore_resource?(resouce)
+    res = rds_client.describe_db_cluster_snapshots(db_cluster_snapshot_identifier: r)
+    if res.db_cluster_snapshots.empty?
+      log.info(deleted)
+    else
+      log.info(not_deleted)
+    end
+  end
+end
+      RAKE_FILE
+
+      File.open('Rakefile', 'w') do |f| 
+        f.puts(rakefile)
+      end unless FileTest.exist?('Rakefile')
+    end
+  end
+end

data/lib/seiton/sqs_check.rb

@@ -17,7 +17,8 @@ end
 <%- end -%>
 <% end %>
 EOF
-      File.open("spec/" + "sqs_queue_spec.rb", "w") do |file|
+      FileUtils.mkdir_p("check") unless FileTest.exist?("check")
+      File.open("check/" + "sqs_queue_spec.rb", "w") do |file|
         file.puts "require 'spec_helper'"
         file.puts ERB.new(template, nil, "-").result(binding).gsub(/^\n/, "")
       end

data/lib/seiton/version.rb

@@ -1,3 +1,3 @@
 module Seiton
-  VERSION = "0.0.2"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: seiton
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - inokappa
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-07-14 00:00:00.000000000 Z
+date: 2020-07-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -130,6 +130,8 @@ files:
 - lib/seiton/ec2.rb
 - lib/seiton/ec2_check.rb
 - lib/seiton/ext.rb
+- lib/seiton/ignores.rb
+- lib/seiton/init.rb
 - lib/seiton/rds.rb
 - lib/seiton/rds_check.rb
 - lib/seiton/sqs.rb