class Pry::Command::ChangePrompt

def change_prompt(prompt)
  if Pry::Prompt[prompt]
    pry_instance.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

def list_prompts
  prompts = Pry::Prompt.all.map do |name, prompt|
    "#{bold(name)}#{red(' (selected)') if pry_instance.prompt == prompt}\n" +
      prompt.description
  end
  output.puts(prompts.join("\n" * 2))
end

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