Add tab completion to the edit command

Author: wvu

lib/msf/ui/console/command_dispatcher/modules.rb

@@ -26,7 +26,7 @@ class Modules
           def commands
             {
               "back"       => "Move back from the current context",
-              "edit"       => "Edit the current module with the preferred editor",
+              "edit"       => "Edit the current module or a file with the preferred editor",
               "advanced"   => "Displays advanced options for one or more modules",
               "info"       => "Displays information about one or more modules",
               "options"    => "Displays global options or for one or more modules",
@@ -68,14 +68,14 @@ def local_editor
           def cmd_edit_help
             print_line "Usage: edit [file/to/edit.rb]"
             print_line
-            print_line "Edit a local file or the currently active module with #{local_editor}"
-            print_line "If a file path is specified it will automatically be reloaded after editing"
+            print_line "Edit the currently active module or a local file with #{local_editor}."
+            print_line "If a file path is specified, it will automatically be reloaded after editing."
             print_line "Otherwise, you can reload the active module with 'reload' or 'rerun'."
             print_line
           end
 
           #
-          # Edit the currently active module
+          # Edit the currently active module or a local file
           #
           def cmd_edit(*args)
             if args.length > 0
@@ -86,22 +86,32 @@ def cmd_edit(*args)
 
             if path
               editor = local_editor
+
               if editor.nil?
-                editor = 'vim'
+                editor = 'ed'
                 print_warning("LocalEditor or $VISUAL/$EDITOR should be set. Falling back on #{editor}.")
               end
 
               print_status("Launching #{editor} #{path}")
               system(editor, path)
-              
+
+              # XXX: This will try to reload *anything* and break on modules
               if args.length > 0
-                load args[0]
+                print_status("Reloading #{path}")
+                load path
               end
             else
               print_error('Nothing to edit -- try using a module first.')
             end
           end
 
+          #
+          # Tab completion for the edit command
+          #
+          def cmd_edit_tabs(str, words)
+            tab_complete_filenames(str, words)
+          end
+
           def cmd_advanced_help
             print_line 'Usage: advanced [mod1 mod2 ...]'
             print_line