Readline sub-command

readline.rb

require 'readline'

MAIN = ['help', 'show', 'set' , 'exit'].sort
SHOW = ['exploits', 'payloads', 'wiki'].sort
USE  = ['exploit' , 'encoder'].sort
SET  = ['payload', 'RHOST'].sort


comp = proc do |s|
  if Readline.line_buffer =~ /help.* /
    MAIN.grep( /^#{Regexp.escape(s)}/ )
  elsif Readline.line_buffer =~ /set.*/
    SET.grep( /^#{Regexp.escape(s)}/ )
  elsif Readline.line_buffer =~ /show.*/
    SHOW.grep( /^#{Regexp.escape(s)}/ )
  else
    MAIN.grep( /^#{Regexp.escape(s)}/ )
  end
end


Readline.completion_proc = comp
Readline.completion_append_character = " "

while line = Readline.readline('-> ', true)
  print("-> ", line, "\n")
end