From 82d683b146094dec1ac4dc09f56208df2e1dde7e Mon Sep 17 00:00:00 2001 From: Kyrylo Silin Date: Sat, 3 Nov 2018 21:58:16 +0800 Subject: prompt: add basic API for adding prompts Fixes #1836 (Add an API for adding new prompts) --- lib/pry/commands/change_prompt.rb | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) (limited to 'lib/pry/commands/change_prompt.rb') diff --git a/lib/pry/commands/change_prompt.rb b/lib/pry/commands/change_prompt.rb index 13d36cbb..f20b741a 100644 --- a/lib/pry/commands/change_prompt.rb +++ b/lib/pry/commands/change_prompt.rb @@ -11,16 +11,12 @@ class Pry::Command::ChangePrompt < Pry::ClassCommand BANNER def process(prompt) - if prompt_map.key?(prompt) - _pry_.prompt = prompt_map[prompt][:value] + if Pry::Prompt.all.key?(prompt) + _pry_.prompt = Pry::Prompt.all[prompt][:value] else raise Pry::CommandError, "'#{prompt}' isn't a known prompt!" end end -private - def prompt_map - Pry::Prompt::MAP - end Pry::Commands.add_command(self) end -- cgit v1.2.1