Show which workers are enabled in ronin-recon workers (closes #175).

Author: postmodern

lib/ronin/recon/cli/commands/workers.rb

@@ -20,6 +20,9 @@
 
 require_relative '../command'
 require_relative '../../registry'
+require_relative '../../config'
+
+require 'command_kit/printing/tables'
 
 module Ronin
   module Recon
@@ -42,6 +45,8 @@ module Commands
         #
         class Workers < Command
 
+          include CommandKit::Printing::Tables
+
           usage '[options] [DIR]'
 
           argument :dir, required: false,
@@ -58,19 +63,23 @@ class Workers < Command
           #   The optional recon worker directory to list.
           #
           def run(dir=nil)
-            files = if dir
-                      dir = "#{dir}/" unless dir.end_with?('/')
+            config = Ronin::Recon::Config.default
 
-                      Ronin::Recon.list_files.select do |file|
-                        file.start_with?(dir)
+            workers = if dir
+                        dir = "#{dir}/" unless dir.end_with?('/')
+
+                        Ronin::Recon.list_files.select do |file|
+                          file.start_with?(dir)
+                        end
+                      else
+                        Ronin::Recon.list_files
                       end
-                    else
-                      Ronin::Recon.list_files
-                    end
 
-            files.each do |file|
-              puts "  #{file}"
+            columns = workers.map do |worker|
+              [worker, ('[enabled]' if config.workers.include?(worker))]
             end
+
+            print_table(columns, padding: 2)
           end
 
         end

spec/cli/commands/workers_spec.rb

@@ -1,15 +1,39 @@
 require 'spec_helper'
 require 'ronin/recon/cli/commands/workers'
+require 'ronin/recon/config'
 
 describe Ronin::Recon::CLI::Commands::Workers do
   describe "#run" do
+    let(:config) { Ronin::Recon::Config.default }
+
     it "must list all worker IDs" do
       expect {
         subject.run
       }.to output(
-        Ronin::Recon.list_files.map { |id|
-          "  #{id}"
-        }.join($/) + $/
+        [
+          "  api/built_with                    ",
+          "  api/crt_sh                        ",
+          "  api/hunter_io                     ",
+          "  api/security_trails               ",
+          "  api/zoom_eye                      ",
+          "  dns/lookup             [enabled]  ",
+          "  dns/mailservers        [enabled]  ",
+          "  dns/nameservers        [enabled]  ",
+          "  dns/reverse_lookup     [enabled]  ",
+          "  dns/srv_enum           [enabled]  ",
+          "  dns/subdomain_enum     [enabled]  ",
+          "  dns/suffix_enum        [enabled]  ",
+          "  net/ip_range_enum      [enabled]  ",
+          "  net/port_scan          [enabled]  ",
+          "  net/service_id         [enabled]  ",
+          "  ssl/cert_enum          [enabled]  ",
+          "  ssl/cert_grab          [enabled]  ",
+          "  web/dir_enum           [enabled]  ",
+          "  web/email_addresses    [enabled]  ",
+          "  web/screenshot                    ",
+          "  web/spider             [enabled]  ",
+          ""
+        ].join($/)
       ).to_stdout
     end
 
@@ -20,11 +44,16 @@
         expect {
           subject.run(dir)
         }.to output(
-          Ronin::Recon.list_files.select { |id|
-            id.start_with?("#{dir}/")
-          }.map { |id|
-            "  #{id}"
-          }.join($/) + $/
+          [
+            "  dns/lookup            [enabled]  ",
+            "  dns/mailservers       [enabled]  ",
+            "  dns/nameservers       [enabled]  ",
+            "  dns/reverse_lookup    [enabled]  ",
+            "  dns/srv_enum          [enabled]  ",
+            "  dns/subdomain_enum    [enabled]  ",
+            "  dns/suffix_enum       [enabled]  ",
+            ""
+          ].join($/)
         ).to_stdout
       end
     end