Add exploits and auxiliary args to the show command

Author: rastating

lib/cli/auto_complete.rb

@@ -41,7 +41,12 @@ def build_cmd_list
       permitted_commands.each { |c| cmds[c] = {} }
       Wpxf::Auxiliary.module_list.each { |m| cmds['use'][m] = {} }
       Wpxf::Exploit.module_list.each { |m| cmds['use'][m] = {} }
-      cmds['show'] = { 'options' => {}, 'advanced' => {} }
+      cmds['show'] = {
+        'options' => {},
+        'advanced' => {},
+        'exploits' => {},
+        'auxiliary' => {}
+      }
       cmds
     end
 

lib/cli/help.rb

@@ -52,11 +52,28 @@ def show_advanced_options
       end
     end
 
+    def show_exploits
+      results = search_modules(['exploit/'])
+      print_good "#{results.length} Exploits"
+      print_module_table results
+    end
+
+    def show_auxiliary
+      results = search_modules(['auxiliary/'])
+      print_good "#{results.length} Auxiliary Modules"
+      print_module_table results
+    end
+
     def show(target)
-      if target.eql?('options')
-        show_options
-      elsif target.eql?('advanced')
-        show_advanced_options
+      handlers = {
+        'options' => 'show_options',
+        'advanced' => 'show_advanced_options',
+        'exploits' => 'show_exploits',
+        'auxiliary' => 'show_auxiliary'
+      }
+
+      if handlers[target]
+        send(handlers[target])
       else
         print_bad("\"#{target}\" is not a valid argument")
       end

lib/cli/modules.rb

@@ -54,17 +54,21 @@ def search_modules(args)
       results
     end
 
+    def print_module_table(modules)
+      modules = modules.sort_by { |k| k[:path] }
+      modules.unshift(path: 'Module', title: 'Title')
+      puts
+      indent_cursor 2 do
+        print_table(modules)
+      end
+    end
+
     def search(*args)
       results = search_modules(args)
 
       if !results.empty?
         print_good "#{results.length} Results for \"#{args.join(' ')}\""
-        results = results.sort_by { |k| k[:path] }
-        results.unshift(path: 'Module', title: 'Title')
-        puts
-        indent_cursor 2 do
-          print_table(results)
-        end
+        print_module_table results
       else
         print_bad "No results for \"#{args.join(' ')}\""
       end