Add workspace to prompt format options

2015-04-06 09:19:49 -05:00 · 2015-04-06 09:19:49 -05:00 · 566c330b83
parent e4d38886de
commit 566c330b83
1 changed files with 3 additions and 0 deletions
--- a/lib/rex/ui/text/shell.rb
+++ b/lib/rex/ui/text/shell.rb
@ -178,6 +178,9 @@ module Shell
          input.prompt.gsub!(/%J/, framework.jobs.length.to_s)
          input.prompt.gsub!(/%L/, Rex::Socket.source_address("50.50.50.50"))
          input.prompt.gsub!(/%D/, ::Dir.getwd)
+          if framework.db && framework.db.workspace
+            input.prompt.gsub!(/%W/, framework.db.workspace.name)
+          end
          self.init_prompt = input.prompt
        end