tenable-ruby 0.4.3 → 0.4.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ddc71aa2ebeed0c75267eef28edbbe2170a81af29c7ce9cc8be9bcc41d22790e
-  data.tar.gz: 40e8d3c8a4dff5f9b89d88a439753ec1fcd6d92c50a2d0d8c3c77ce1244e6a7d
+  metadata.gz: 77917278fce3594ee5fce1156ba4309ef5ed2fdeeaace91d2b3c3626818acd67
+  data.tar.gz: 26b073b7ef3de954d470002eb035e3aa52d2bb30bde90b211680f6f2f0bb82b8
 SHA512:
-  metadata.gz: b36308298ed4a9a4708699da058c481545c25bc9f95333bbb3ed444777efcecf3108cd3484a4da6892cee8c62181acb4fa10a55e9713f36d7abd7c5dfbadab63
-  data.tar.gz: 82ff9c6ece45a2052532363cc194cdbe8c2af7637e40b5e1d9c2f21f3679ee8461d5b0ae5c7305b6d7917c246a71a1b83b12aa546aa0250d3f1d6243df6fe786
+  metadata.gz: fa26c48c98bfea63c85c82402c9850c724de8fe77d81315b23d58c6ce45527f9fd2e6f9302b0ac466abedbffe8ec35caa5598275a5e420381be42d584901e18d
+  data.tar.gz: 933cbeb33c6c6027b18431edfa5347ffd7c00de1483360f8b067ef9d9ceab1b569861bbac4013567a78c9ecf5f7617535ac42cfea374ef3d04caa442140298fa

data/lib/tenable-ruby.rb

@@ -526,11 +526,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 +548,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?

metadata

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