wafoo 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 04bbceda52de1c7aa38f36b757a193a844b9802b8191ad22b415db138a67ca16
-  data.tar.gz: df50354e02a1255441f08d332294d1e34c4d53f2b8a50cfe8ce1e6e6a0073ea7
+  metadata.gz: cf968d2be7a9d1632c60d8cb49a0e8751a3d49f8d30685b9c3bc8414a84a5618
+  data.tar.gz: 6249335ba736504e93dd82f8effd9f4d9cd03b6aab9fbc58a4c97589ddc5034c
 SHA512:
-  metadata.gz: 2d2de679881fbd0702c776c5d9f37ce9c8eb4dbe08251a8a6be7643fee70ccaad19817f0c3612f8d5fc893f6eac2ec9385b30d0bd755b4d48cb681473303a843
-  data.tar.gz: e35f3262542c095e87f8b7d4899f96e2ea92d8a7479e2e0dccf8c8bc60b13d87268ce3434402c148a6c0ebaec4366bd4ef5e44d7039106a7a9d5feb0041e1002
+  metadata.gz: a720d8b8b0e4bb647ff85a8c2dff26e92a6699239a8077b312186f8621608b586b7b1496570045216ad1ff93b3c16e68e49ed8b57bbe36cd27d18728f536bd1e
+  data.tar.gz: 596aba582b45439a3820aa0b01ad595102d6dbb568f42f349efe1acd0c4a476bea89881c4ee4a47e6144512e4d6f3e5ac00a837639a94931c44ef290ba449001

data/.gitignore

@@ -9,3 +9,4 @@
 /tmp/
 vendor/bundle/
 .envrc
+ipsets/

data/lib/wafoo/helper.rb

@@ -1,7 +1,7 @@
 module Wafoo
   module Helper
     def output_table(ipsets_list)
-      table = Terminal::Table.new(:headings => ['Type', 'IPSets ID', 'Name'],
+      table = Terminal::Table.new(:headings => ['Type', 'IPSet IDs', 'Name'],
                                   :rows => ipsets_list)
       puts table
     end

data/lib/wafoo/run.rb

@@ -2,12 +2,14 @@ require 'wafoo'
 
 module Wafoo
   class Run
+    IP_SETS_DIR = 'ipsets'
     include Wafoo::Helper
   
     def initialize(options = nil)
       @waf_regional = Aws::WAFRegional::Client.new
       @waf = Aws::WAF::Client.new
       @regional = options[:regional] unless options.nil?
+      FileUtils.mkdir_p(IP_SETS_DIR) unless FileTest.exist?(IP_SETS_DIR)
     end
 
     def read_ipset_from_api(ip_set_id)
@@ -26,7 +28,7 @@ module Wafoo
 
     def read_ipset_from_file(ip_set_id)
       ipsets = []
-      File.open(ip_set_id, 'r') do |file|
+      File.open(IP_SETS_DIR + '/' + ip_set_id, 'r') do |file|
         file.read.split("\n").each do |ipset|
           ipsets << ipset
         end
@@ -64,10 +66,10 @@ module Wafoo
         exit 1
       end
       ipsets.sort.each { |ipset| puts info_print(ipset) }
-      File.open(ip_set_id, 'w') do |f|
+      File.open(IP_SETS_DIR + '/' + ip_set_id, 'w') do |f|
         ipsets.sort.each { |ipset| f.puts(ipset) }
       end
-      puts 'Exported to ' + added_print(ip_set_id)
+      puts 'Exported to ' + added_print(IP_SETS_DIR + '/' + ip_set_id)
     end
 
     def apply_ipset(ipsets, ip_set_id)

data/lib/wafoo/version.rb

@@ -1,3 +1,3 @@
 module Wafoo
-  VERSION = '0.0.4'
+  VERSION = '0.0.5'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: wafoo
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - inokappa