dronebl.rb 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 379138ce46c0c464447e92e60fa002cdc580db63
-  data.tar.gz: 1213dc5cebb786b9565ee7961b57f36d763589ba
+  metadata.gz: 16aacfdd77793ba2bae1402a73575a2db7912a7e
+  data.tar.gz: 527db5ba7211b2749e9cd2ae3f67f21e84cd1d8f
 SHA512:
-  metadata.gz: ff35d03b19a0cf1bfdcdd1153ffe823ecc62d1549d791be3ef1a437790ba9f29c2f30a3e96f5f5a6318207a0402bbc1e07622d1281a3cdc5ef0d95f2361a2964
-  data.tar.gz: 32bbd489797d86c413099ecae2d739d580cab1b66d1f464e3ef98d0b34bbe741d1c5439b4fb87852b17cd1a4f1b2a56221eb99979adcad2a592bbfa6bb247e22
+  metadata.gz: e2b93b28351acfd7cf7a852e91391c5181bba903bc3e999d53283d0988b44a0a7bd2bd82c9e362d6a46c62b9dc20ef14ac1874f77801162657ec47502fc73fa3
+  data.tar.gz: 7d75a555e29fb494abac771e65cd1c47156289a39d9111eb498592b4d3da4c9a81b7a401100e13b4ac6af6aede924e755809d0ea3c3c56b063defbb95aca02f3

data/bin/dronebl-add

@@ -6,11 +6,12 @@ require 'resolv'
 require 'optparse'
 class Options
   attr_reader :read_from_stdin, :key_str, :key_file, :type, :comment, :dry_run,
-    :use_key_file, :no_check
+    :use_key_file, :no_check, :max_ips_per_chunk
   attr_accessor :ips
   def initialize
     @key_file = File.expand_path "~/.droneblkey"
     @ips = []
+    @max_ips_per_chunk = 100
     @opt_parser = OptionParser.new do |opts|
       opts.banner = "Usage: #{$0} [options]"
       opts.separator ""
@@ -35,6 +36,9 @@ class Options
         puts DroneBL::TYPES.map { |k, v| "#{k} : #{v}"}.join("\n")
         exit
       end
+      opts.on('-M', '--max-ips-per-chunk NUM', Integer, 'The maximum number of IPs to send in a single submission chunk. Default 100.') do |max|
+        @max_ips_per_chunk = max
+      end
       opts.on('-c', '--comment COMMENT', String, 'Attach the given comment to the listings.') do |comment|
         @comment = comment
       end
@@ -93,12 +97,11 @@ opts.ips.uniq!
 prevalid, ipv6 = opts.ips.partition { |ip| (ip.match(Resolv::IPv6::Regex).nil?) }
 valid, invalid = prevalid.partition { |ip| !(ip.match(Resolv::IPv4::Regex).nil?) }
 invalid += ipv6
-if @dry_run
-  puts DroneBL::gen_add_query valid
-  puts "#{valid.count} IPs will be added as type #{opts.type}#{" with comment 'opts.comment if opts.comment}'"}."
+if opts.dry_run
+  valid.each_slice(opts.max_ips_per_chunk) { |submit| puts DroneBL::gen_add_query(submit, opts.type, opts.comment) }
+  puts "#{valid.count} IPs will be added as type #{opts.type}#{" with comment '#{opts.comment if opts.comment}'"}."
 else
-  response = DroneBL::add(valid, opts.type, opts.comment)
-  puts response
+  valid.each_slice(opts.max_ips_per_chunk) { |submit| DroneBL::add(submit, opts.type, opts.comment) }
   puts "#{valid.count} IPs added as type #{opts.type}#{" with comment '#{opts.comment}'" if opts.comment}"
 end
 unless invalid.empty?

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dronebl.rb
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - Rylee Fowler