45 lines
1.0 KiB
Ruby
45 lines
1.0 KiB
Ruby
class Pry::Command::ChangePrompt < Pry::ClassCommand
|
|
match 'change-prompt'
|
|
group 'Input and Output'
|
|
description 'Change the current prompt.'
|
|
command_options argument_required: true
|
|
banner <<-BANNER
|
|
Usage: change-prompt [OPTIONS] [NAME]
|
|
|
|
Change the current prompt.
|
|
BANNER
|
|
|
|
def options(opt)
|
|
opt.on(:l, :list, 'List the available prompts')
|
|
end
|
|
|
|
def process(prompt)
|
|
if opts.present?(:l)
|
|
list_prompts
|
|
else
|
|
change_prompt(prompt)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def list_prompts
|
|
prompts = Pry::Prompt.all.map do |name, prompt|
|
|
"#{bold(name)}#{red(' (selected)') if _pry_.prompt == prompt}\n" +
|
|
prompt.description
|
|
end
|
|
output.puts(prompts.join("\n" * 2))
|
|
end
|
|
|
|
def change_prompt(prompt)
|
|
if Pry::Prompt[prompt]
|
|
_pry_.prompt = Pry::Prompt[prompt]
|
|
else
|
|
raise Pry::CommandError, "'#{prompt}' isn't a known prompt. " \
|
|
"Run `change-prompt --list` to see the list of known prompts."
|
|
end
|
|
end
|
|
|
|
Pry::Commands.add_command(self)
|
|
end
|