tenable-ruby 0.4.3 → 0.5.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ddc71aa2ebeed0c75267eef28edbbe2170a81af29c7ce9cc8be9bcc41d22790e
-  data.tar.gz: 40e8d3c8a4dff5f9b89d88a439753ec1fcd6d92c50a2d0d8c3c77ce1244e6a7d
+  metadata.gz: 4a367b214445184f3805e7d99304cfab5341634a93833e9aa3ee2754fd94212f
+  data.tar.gz: 3cbe4522d7639236cbbdd5a59746cd511ba0c240921390bc595bbcd8165a3022
 SHA512:
-  metadata.gz: b36308298ed4a9a4708699da058c481545c25bc9f95333bbb3ed444777efcecf3108cd3484a4da6892cee8c62181acb4fa10a55e9713f36d7abd7c5dfbadab63
-  data.tar.gz: 82ff9c6ece45a2052532363cc194cdbe8c2af7637e40b5e1d9c2f21f3679ee8461d5b0ae5c7305b6d7917c246a71a1b83b12aa546aa0250d3f1d6243df6fe786
+  metadata.gz: 59c07dcbfa4029a2e24d5b3396a0cfaf17821e723a00beade9f2d4f1fc543578efa66081d79cdfac1804fb67118fe15f7b79131bf3e39fe498f23018e6d0cc62
+  data.tar.gz: 1a9290bc941f16f2c2e5f0dcaf285c2574306271c6459a9aec4e313f09744db17d840f93c87adff1f2ef55068ad8810add28ec4567b2b2dc697ae2e357c7e8c1

data/lib/tenable-ruby.rb

@@ -307,6 +307,29 @@ module TenableRuby
       http_put(options)
     end
 
+    # Unlinks an agent
+    #
+    # Reference
+    # https://developer.tenable.com/reference#agents-delete
+    def agent_unlink(scanner_id, agent_id)
+      http_delete(:uri => "/scanners/#{scanner_id}/agents/#{agent_id}", :fields => header)
+    end
+
+    # Creates a bulk operation task to unlink agents
+    #
+    # Reference
+    # https://developer.tenable.com/reference#bulk-unlink-agents
+    def agent_bulk_unlink(scanner_id, agent_ids)
+      http_post(
+        :uri => "/scanners/#{scanner_id}/agents/_bulk/unlink",
+        :fields => header,
+        :body => {
+          :items => agent_ids
+        }.to_json,
+        :ctype => 'application/json',
+      )
+    end
+
     # Returns details of the specified asset.
     #
     # Reference:
@@ -526,11 +549,11 @@ module TenableRuby
 
     # Performs scan with scan policy provided (uuid of policy or policy name).
     # Name is your scan name and opts is your scan configuration hash
-    # (foldername is optional - folder where to save the scan (if that folder exists))
+    # (scan_folder is optional - folder where to save the scan (if that folder exists))
     # (scanner_id is optional - ID of the scanner/cloud scanner you want to run this scan on)
     #
     # returns: JSON parsed object with scan info
-    def scan_quick_policy(policyname, name, opts = {}, foldername = nil, scanner_id = nil)
+    def scan_quick_policy(policyname, name, opts = {}, scan_folder = nil, scanner_id = nil)
       policies = list_policies['policies']
       if policies.nil?
         raise TenableRuby::Error::TenableError, "Tenable API request 'list_policies' responded with 'nil'"
@@ -548,16 +571,18 @@ module TenableRuby
       settings.merge!(opts)
       settings['name'] = name
       settings['policy_id'] = policy['id']
-      unless foldername.nil?
+      if scan_folder.is_a?(Integer)
+        settings['folder_id'] = scan_folder
+      elsif scan_folder.is_a?(String)
         folders = list_folders['folders']
         if folders.nil?
           raise TenableRuby::Error::TenableError, "Tenable API request 'list_folders' responded with 'nil'"
         end
-        selected_folders = folders.select do |folder|
-          folder['name'] == foldername
-        end
-        unless selected_folders.size == 0
-          settings['folder_id'] = selected_folders.first['id']
+        selected_folder = folders.find { |f| f['name'] == scan_folder }
+        if selected_folder
+          settings['folder_id'] = selected_folder['id']
+        else
+          raise TenableRuby::Error::TenableError, "Could not find folder with name #{scan_folder}"
         end
       end
       unless scanner_id.nil?
@@ -846,7 +871,6 @@ module TenableRuby
       uri = opts[:uri]
       fields = opts[:fields] || {}
       raw_content = opts[:raw_content] || false
-      json = {}
       tries = @httpretry
 
       request = Net::HTTP::Get.new(uri)
@@ -862,11 +886,17 @@ module TenableRuby
           sleep @httpsleep
           retry
         else
-          return json
+          raise TenableRuby::Error::TenableError.new("#{e}: No more retries for http GET '#{opts[:uri]}'")
         end
-      rescue URI::InvalidURIError
-        return json
+      rescue URI::InvalidURIError => e
+        raise TenableRuby::Error::TenableError.new("#{e}: http GET '#{opts[:uri]}'")
+      end
+
+      if response.code.to_s != "200"
+        raise TenableRuby::Error::TenableError.new(response: response),
+          "Tenable API request '#{opts[:uri]}' responded with response code #{response.code}"
       end
+
       if !raw_content
         parse_json(response.body)
       else
@@ -901,7 +931,6 @@ module TenableRuby
       fields = opts[:fields] || {}
       body = opts[:body]
       ctype = opts[:ctype]
-      json = {}
       tries = @httpretry
 
       request = Net::HTTP::Post.new(uri)
@@ -920,10 +949,10 @@ module TenableRuby
           sleep @httpsleep
           retry
         else
-          return json
+          raise TenableRuby::Error::TenableError.new("#{e}: No more retries for http POST '#{opts[:uri]}'")
         end
-      rescue URI::InvalidURIError
-        return json
+      rescue URI::InvalidURIError => e
+        raise TenableRuby::Error::TenableError.new("#{e}: http POST '#{opts[:uri]}'")
       end
 
       if response.code.to_s != "200"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tenable-ruby
 version: !ruby/object:Gem::Version
-  version: 0.4.3
+  version: 0.5.3
 platform: ruby
 authors:
 - Patrick Craston
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-06-15 00:00:00.000000000 Z
+date: 2020-07-15 00:00:00.000000000 Z
 dependencies: []
 description: |-
   Ruby library for communicating with the tenable.io API.