tenable-ruby 0.4.2 → 0.5.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a307cd9672be72bc72d8aeee16cf07083250883bd6bb26c38a7e2a6344162b66
-  data.tar.gz: b415f2321cc7d6e14bb1ec621b33093424c05050fcb8da747beb10f554f236bd
+  metadata.gz: 819f8651b369e488bf93369f4ebd57888b78f30d76395252597971ca5d69f06b
+  data.tar.gz: 7b77a9e7f582e4a3de408b702c338b21364ca7c149cf0e7b163918a6038625f1
 SHA512:
-  metadata.gz: 6ef0ab799275410df467d988583b0dcb8d777aaf018806dd46eac988f0e93b0e1151f9fb18e2c313cf3617b0fb4bc656949c26c572a72aab6b36d87444798881
-  data.tar.gz: a5f78981f65993467351de830a3e1c76b85477e3efa049746078e78405c06b40c65dd15d7a804340ca0f1dd6d2a5c8b544ce87ad796cce3c688818703b2fcaca
+  metadata.gz: c2baf2dd3de8b86fab0f6cb88b97957d29fe638a9ef64b247d085d10ad784f389f2a815d64c9b89073fe06620bbe69cf5b50a47931dc71c2904a2d61d9928ab5
+  data.tar.gz: 828494940880467224183077a2b1a535e3173be53b63bb89f7982a415d1961473e8a7ef582a4780dbeda3d6886185c6bd8673f655ff56347abe271ac285dd61a

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
+        },
+        :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?
@@ -701,9 +726,9 @@ module TenableRuby
     #
     # Reference:
     # https://developer.tenable.com/reference#target-groups-create
-    def create_target_group(name, members, acls: [])
+    def create_target_group(name, members, acls: nil)
       http_post(:uri => "/target-groups", :fields => header,
-        :body => {:name => name, :members => members, :acls => alcs})
+        :data => {:name => name, :members => members, :acls => acls})
     end
 
     # Returns the current target groups.
@@ -726,9 +751,9 @@ module TenableRuby
     #
     # Reference:
     # https://developer.tenable.com/reference#target-groups-edit
-    def update_target_group(group_id:, name:, members:, acls: nil)
+    def update_target_group(group_id, name, members, acls: nil)
       http_put(:uri => "/target-groups/#{group_id}", :fields => header,
-        :body => {:name => name, :members => members, :acls => acls})
+        :data => {:name => name, :members => members, :acls => acls})
     end
 
     # Deletes a target group.
@@ -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.2
+  version: 0.5.2
 platform: ruby
 authors:
 - Patrick Craston
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-06-09 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.