pwn 0.5.453 → 0.5.454

data/bin/pwn_jenkins_install_plugin

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-sIP', '--ip=IP', '<Required - Jenkins Server IP>') { |s| opts[:ip] = s }
 
   options.on('-dPORT', '--dest-port=PORT', '<Optional - Jenkins Server Port (Defaults to 8080)>') do |port|
@@ -36,11 +32,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 begin
   ip = opts[:ip].to_s.scrub
   port = opts[:port].to_i

data/bin/pwn_jenkins_thinBackup_aws_s3

@@ -5,11 +5,7 @@ require 'aws-sdk'
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-rREGION', '--region=REGION', "<Optional - AWS STS Region (Defaults to 'us-west-1')>") do |sts_region|
     opts[:sts_region] = sts_region
   end
@@ -37,11 +33,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 sts_region = if opts[:sts_region].nil?
                'us-west-1'
              else

data/bin/pwn_jenkins_update_plugins

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-sIP', '--ip=IP', '<Required - Jenkins Server IP>') { |s| opts[:ip] = s }
 
   options.on('-dPORT', '--dest-port=PORT', '<Optional - Jenkins Server Port (Defaults to 8080)>') do |port|
@@ -36,11 +32,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 ip = opts[:ip].to_s.scrub
 port = opts[:port].to_i
 username = opts[:username]

data/bin/pwn_jenkins_useradd

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-sIP', '--ip=IP', '<Required - Jenkins Server IP>') { |s| opts[:ip] = s }
 
   options.on('-dPORT', '--dest-port=PORT', '<Optional - Jenkins Server Port (Defaults to 8080)>') do |port|
@@ -40,11 +36,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 ip = opts[:ip].to_s.scrub
 port = opts[:port].to_i
 

data/bin/pwn_mail_agent

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-aTYPE', '--agent-type=TYPE', '<Required - Mail Agent Type office365|gmail|hotmail_n_live|yahoo|manual>') do |a|
     opts[:agent_type] = a
   end
@@ -22,11 +18,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 private_methods def initiate_action
   puts @list_agent_opts
   if @list_agent_opts

data/bin/pwn_nessus_cloud_scan_crud

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-cYPATH', '--yaml-config=YPATH', '<Required - YAML Config Containing Access & Secret Keys for Authentication, Including Credential Information for Scan Creation (https://developer.tenable.com/docs/determine-settings-for-credential-type)>') do |c|
     opts[:yaml_config] = c
   end
@@ -86,11 +82,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 begin
   # Get Options Passed to pwn_nessus_cloud_create_scan
   # Required Arguments

data/bin/pwn_nessus_cloud_vulnscan

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-cYPATH', '--yaml-config=YPATH', '<Required - YAML Config Containing Access & Secret Keys for Authentication>') do |c|
     opts[:yaml_config] = c
   end
@@ -26,11 +22,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 begin
   yaml_config = opts[:yaml_config]
 

data/bin/pwn_nexpose

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   # Nexpose Flags
   options.on('-CIP', '--console_ip=IP', '<Required - Nexpose Console IP>') { |c| opts[:console_ip] = c }
   options.on('-UCUSER', '--console_username=CUSER', '<Required - Nexpose Console Username>') { |cu| opts[:console_username] = cu }
@@ -17,11 +13,6 @@ OptionParser.new do |options|
   options.on('-RREPORTS', '--report_names=REPORTS', '<Required - Comma Delimited String of Report Names>') { |r| opts[:report_names] = r }
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 # Nexpose
 console_ip = opts[:console_ip].to_s.scrub
 console_username = opts[:console_username].to_s.scrub

data/bin/pwn_nmap_discover_tcp_udp

@@ -7,11 +7,7 @@ require 'pwn'
 require 'time'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-fFILE', '--target-file=FILE', '<Required if --target-range excluded - File containing one supported nmap target / line e.g. foo.bar, 10.1.1.1, 192.168.1.1-20, 192.168.1.0/24, etc>') do |f|
     opts[:target_file] = f
   end
@@ -41,11 +37,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 ftimestr = '%Y-%m-%d %H:%M:%S.%N%z'
 started_at = Time.now.strftime(ftimestr)
 started_at_parse = Time.parse(started_at)

data/bin/pwn_openvas_vulnscan

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-cYPATH', '--yaml-config=YPATH', '<Required - YAML Config Containing Username & Password for Authentication>') do |c|
     opts[:yaml_config] = c
   end
@@ -26,11 +22,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 task_name = opts[:task_name]
 
 report_dir = opts[:report_dir].to_s.scrub

data/bin/pwn_pastebin_sample_filter

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-rPATTERN', '--regex=PATTERN', "<Required - Regex Pattern for Interesting Pastes ('.*' for All)>") do |r|
     opts[:regex] = r
   end
@@ -18,11 +14,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 proxy = opts[:proxy]
 regex = opts[:regex]
 

data/bin/pwn_phone

@@ -8,11 +8,7 @@ require 'ruby-audio'
 require 'waveform'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-tPATH', '--target-file=PATH', '<Required - File Containing List of Targets to Dial>') do |t|
     opts[:target_file] = t
   end
@@ -62,11 +58,6 @@ OptionParser.new do |options|
   # TODO: Append results to pwn_phone_results.json as things are running #save_memory.
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 begin
   pwn_provider = 'ruby-gem'
   # pwn_provider = ENV.fetch('PWN_PROVIDER') if ENV.keys.select { |s| s == 'PWN_PROVIDER' }.any?

data/bin/pwn_rdoc_to_jsonl

@@ -6,19 +6,17 @@ require 'pwn'
 
 # Set the path to your project directory
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+pwn_driver = PWN::Driver::Parser.new do |options|
   options.on('-rPATH', '--rdoc-root-dir=PATH', '<Optional - RDoc root directory (Default: "/opt/pwn/rdoc/PWN")>') do |r|
     opts[:rdoc_root] = r
   end
 
-  options.on('-jPATH', '--jsonl-results=PATH', '<Optional - Path to save JSONL file (Default: "/tmp/RDoc_Parent_Basename-TIMESTAMP.jsonl">') do |j|
+  options.on('-jPATH', '--jsonl-results=PATH', '<Optional - Path to save JSONL file (Default: "/tmp/PWN-[TIMESTAMP].jsonl">') do |j|
     opts[:jsonl_results] = j
   end
-end.parse!
+end
+pwn_driver.auto_opts_help = false
+pwn_driver.parse!
 
 begin
   rdoc_root = opts[:rdoc_root] ||= '/opt/pwn/rdoc/PWN'

data/bin/pwn_sast

@@ -5,20 +5,7 @@ require 'htmlentities'
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
-  # TODO: HOW TO LOAD THE pwn.yaml CONFIGURATION FILE FOR EVERYTHING UNLESS OVERRIDDEN???
-  options.on('-YPATH', '--pwn-env=PATH', '<Optional - PWN YAML File>') do |p|
-    opts[:pwn_env_path] = p
-  end
-
-  options.on('-ZPATH', '--pwn-dec=PATH', '<Optional - Out-of-Band YAML File with :key && :iv>') do |d|
-    opts[:pwn_dec_path] = d
-  end
-
+PWN::Driver::Parser.new do |options|
   options.on('-uGITURI', '--uri-source-root=GITURI', '<Required - HTTP URI of Git Repo Scanned e.g. https://github.com/0dayInc/pwn/tree/master>') do |u|
     opts[:uri_source_root] = u
   end
@@ -48,22 +35,11 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 begin
   timestamp = Time.now.strftime('%Y-%m-%d_%H:%M:%S%z')
 
   pwn_env_path = opts[:pwn_env_path]
   pwn_dec_path = opts[:pwn_dec_path]
-  if pwn_env_path
-    PWN::Config.refresh_env(
-      pwn_env_path: pwn_env_path,
-      pwn_dec_path: pwn_dec_path
-    )
-  end
 
   previous_dir = Dir.pwd
 

data/bin/pwn_serial_check_voicemail

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-vNUM', '--voicemail_num=num', '<Required - Number to Voicemail Service>') do |num|
     opts[:voicemail_num] = num
   end
@@ -17,11 +13,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 voicemail_num = opts[:voicemail_num].to_s.scrub.strip.chomp.delete('-').delete('.').to_i
 
 if opts[:voicemail_pin].nil?

data/bin/pwn_serial_msr206

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+pwn_driver = PWN::Driver::Parser.new do |options|
   options.on('-dDEV', '--block-dev=DEV', '<Optional - MSR206 block device path (defaults to /dev/ttyUSB0)>') do |d|
     opts[:block_dev] = d
   end
@@ -32,7 +28,9 @@ OptionParser.new do |options|
   options.on('-fFLOWCTRL', '--flow-control=FLOWCTRL', '<Optional - none||hard||soft (defaults to soft)>') do |f|
     opts[:flow_control] = f
   end
-end.parse!
+end
+pwn_driver.auto_opts_help = false
+pwn_driver.parse!
 
 begin
   block_dev = opts[:block_dev]

data/bin/pwn_serial_son_micro_sm132_rfid

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+pwn_driver = PWN::Driver::Parser.new do |options|
   options.on('-dDEV', '--block-dev=DEV', '<Optional - SonMicroRFID block device path (defaults to /dev/ttyUSB0)>') do |d|
     opts[:block_dev] = d
   end
@@ -36,7 +32,9 @@ OptionParser.new do |options|
   options.on('-v', '--verbose', '<Optional - Enable verbose output>') do
     opts[:verbose] = true
   end
-end.parse!
+end
+pwn_driver.auto_opts_help = false
+pwn_driver.parse!
 
 begin
   block_dev = opts[:block_dev] if opts[:block_dev] && File.exist?(opts[:block_dev])

data/bin/pwn_shodan_graphql_introspection

@@ -5,17 +5,12 @@ require 'faker'
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
+PWN::Driver::Parser.new do |options|
   options.on('-jFILE', '--json-results=FILE', 'Required - JSON results file from pwn_shodan_search driver') do |j|
     opts[:json_results] = j
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 json_results_path = opts[:json_results]
 raise "ERROR: Shodan JSON Results File #{json_results_path} Does Not Exist." unless File.exist?(json_results_path)
 

data/bin/pwn_shodan_search

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-cYAML', '--config-yaml=YAML', '<Required - YAML config containing api_key from Shodan.io>') do |y|
     opts[:yaml] = y
   end
@@ -26,11 +22,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 begin
   yaml_file = opts[:yaml].to_s
   raise "ERROR: #{yaml_file} does not exist." unless File.exist?(yaml_file)

data/bin/pwn_simple_http_server

@@ -4,10 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
+pwn_driver = PWN::Driver::Parser.new do |options|
   options.on('-iIP', '--bind-ip=IP', '<Optional - HTTP Bind IP to Listen> (Defaults to Localhost)') do |i|
     opts[:bind_ip] = i
   end
@@ -15,7 +12,9 @@ OptionParser.new do |options|
   options.on('-pPORT', '--port=PORT', '<Optional - HTTP Port to Listen (Defaults to 3333)>') do |p|
     opts[:port] = p
   end
-end.parse!
+end
+pwn_driver.auto_opts_help = false
+pwn_driver.parse!
 
 bind_ip = opts[:bind_ip] ||= '127.0.0.1'
 

data/bin/pwn_web_cache_deception

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-tURL', '--target-url=URL', '<Required - URL to Target>') do |t|
     opts[:target_url] = t
   end
@@ -42,11 +38,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 # Colors!
 @dark_red = "\e[1m\e[31m"
 @green = "\e[32m"

data/bin/pwn_www_checkip

@@ -4,10 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
+pwn_driver = PWN::Driver::Parser.new do |options|
   options.on('-tIP_HOST', '--target=IP_HOST', '<Optional - IP or Host to Check (Default - Your Public IP)>') do |t|
     opts[:target] = t
   end
@@ -23,7 +20,12 @@ OptionParser.new do |options|
   options.on('-pPROXY', '--proxy=PROXY', '<Optional - HTTP or Socks Proxy || tor>') do |p|
     opts[:proxy] = p
   end
-end.parse!
+end
+pwn_driver.auto_opts_help = false
+pwn_driver.parse!
+
+pwn_env_path = opts[:pwn_env_path]
+pwn_dec_path = opts[:pwn_dec_path]
 
 target = opts[:target]
 tls_port = opts[:tls_port]

data/bin/pwn_www_uri_buster

@@ -9,11 +9,7 @@ require 'time'
 require 'uri'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-uURL', '--target-url=URL', '<Required - Target URL)>') do |t|
     opts[:target_url] = t
   end
@@ -67,11 +63,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 def request_path(opts = {})
   target_url = opts[:target_url]
   proxy = opts[:proxy]

data/bin/pwn_xss_dom_vectors

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-tFQDN', '--target-fqdn=FQDN', '<Required - FQDN to Target>') do |t|
     opts[:target_fqdn] = t
   end
@@ -34,11 +30,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 # Required Flag Variables
 target_fqdn = opts[:target_fqdn].to_s.scrub.strip.chomp
 output_dir = opts[:output_dir].to_s.scrub.strip.chomp if Dir.exist?(opts[:output_dir].to_s.scrub.strip.chomp)

data/bin/pwn_zaproxy_active_rest_api_scan

@@ -6,11 +6,7 @@ require 'pwn'
 require 'uri'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-aAPIKEY', '--api_key=APIKEY', '<Required - OWASP Zap API Key (Tools>Options>API)>') do |a|
     opts[:api_key] = a
   end
@@ -60,11 +56,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 begin
   timestamp = Time.now.strftime('%Y-%m-%d_%H-%M-%S%Z')
   logger = PWN::Plugins::PWNLogger.create

data/bin/pwn_zaproxy_active_scan

@@ -4,11 +4,7 @@
 require 'pwn'
 
 opts = PWN::Env[:driver_opts]
-OptionParser.new do |options|
-  options.banner = "USAGE:
-    #{File.basename($PROGRAM_NAME)} [opts]
-  "
-
+PWN::Driver::Parser.new do |options|
   options.on('-aAPIKEY', '--api_key=APIKEY', '<Required - OWASP Zap API Key (Tools>Options>API)>') do |a|
     opts[:api_key] = a
   end
@@ -50,11 +46,6 @@ OptionParser.new do |options|
   end
 end.parse!
 
-if opts.empty?
-  puts `#{File.basename($PROGRAM_NAME)} --help`
-  exit 1
-end
-
 begin
   logger = PWN::Plugins::PWNLogger.create
 

data/lib/pwn/ai/introspection.rb

@@ -45,7 +45,9 @@ module PWN
               system_role_content: system_role_content,
               spinner: false
             )
-            puts response
+            response = response[:choices].last[:content] if response.is_a?(Hash) &&
+                                                            response.key?(:choices) &&
+                                                            response[:choices].last.keys.include?(:content)
           when :openai
             response = PWN::AI::OpenAI.chat(
               request: request.chomp,