196 lines
4.8 KiB
Ruby
196 lines
4.8 KiB
Ruby
|
|
class Pry
|
|
|
|
# Manage the processing of command line options
|
|
class CLI
|
|
|
|
NoOptionsError = Class.new(StandardError)
|
|
|
|
class << self
|
|
|
|
# @return [Proc] The Proc defining the valid command line options.
|
|
attr_accessor :options
|
|
|
|
# @return [Array] The Procs that process the parsed options.
|
|
attr_accessor :option_processors
|
|
|
|
# @return [Array<String>] The input array of strings to process
|
|
# as CLI options.
|
|
attr_accessor :input_args
|
|
|
|
# Add another set of CLI options (a Slop block)
|
|
def add_options(&block)
|
|
if options
|
|
old_options = options
|
|
self.options = proc do
|
|
instance_exec(&old_options)
|
|
instance_exec(&block)
|
|
end
|
|
else
|
|
self.options = block
|
|
end
|
|
|
|
self
|
|
end
|
|
|
|
# Bring in options defined in plugins
|
|
def add_plugin_options
|
|
Pry.plugins.values.each do |plugin|
|
|
plugin.load_cli_options
|
|
end
|
|
|
|
self
|
|
end
|
|
|
|
# Add a block responsible for processing parsed options.
|
|
def process_options(&block)
|
|
self.option_processors ||= []
|
|
option_processors << block
|
|
|
|
self
|
|
end
|
|
|
|
# Clear `options` and `option_processors`
|
|
def reset
|
|
self.options = nil
|
|
self.option_processors = nil
|
|
end
|
|
|
|
def parse_options(args=ARGV.dup)
|
|
unless options
|
|
raise NoOptionsError, "No command line options defined! Use Pry::CLI.add_options to add command line options."
|
|
end
|
|
|
|
self.input_args = args
|
|
|
|
opts = Slop.parse!(args, :help => true, :multiple_switches => false, &options)
|
|
|
|
# Option processors are optional.
|
|
if option_processors
|
|
option_processors.each { |processor| processor.call(opts) }
|
|
end
|
|
|
|
self
|
|
end
|
|
|
|
end
|
|
|
|
reset
|
|
end
|
|
end
|
|
|
|
|
|
# String that is built to be executed on start (created by -e and -exec switches)
|
|
exec_string = ""
|
|
|
|
# Bring in options defined by plugins
|
|
Slop.new do
|
|
on "no-plugins" do
|
|
Pry.config.should_load_plugins = false
|
|
end
|
|
end.parse(ARGV.dup)
|
|
|
|
if Pry.config.should_load_plugins
|
|
Pry::CLI.add_plugin_options
|
|
end
|
|
|
|
# The default Pry command line options (before plugin options are included)
|
|
Pry::CLI.add_options do
|
|
banner %{Usage: pry [OPTIONS]
|
|
Start a Pry session.
|
|
See: `https://github.com/pry` for more information.
|
|
Copyright (c) 2013 John Mair (banisterfiend)
|
|
--
|
|
}
|
|
on :e, :exec=, "A line of code to execute in context before the session starts" do |input|
|
|
exec_string << input + "\n"
|
|
end
|
|
|
|
on "no-pager", "Disable pager for long output" do
|
|
Pry.config.pager = false
|
|
end
|
|
|
|
on "no-history", "Disable history loading" do
|
|
Pry.config.history.should_load = false
|
|
end
|
|
|
|
on "no-color", "Disable syntax highlighting for session" do
|
|
Pry.color = false
|
|
end
|
|
|
|
on :f, "Suppress loading of ~/.pryrc and ./.pryrc" do
|
|
Pry.config.should_load_rc = false
|
|
Pry.config.should_load_local_rc = false
|
|
end
|
|
|
|
on :s, "select-plugin=", "Only load specified plugin (and no others)." do |plugin_name|
|
|
Pry.config.should_load_plugins = false
|
|
Pry.plugins[plugin_name].activate!
|
|
end
|
|
|
|
on :d, "disable-plugin=", "Disable a specific plugin." do |plugin_name|
|
|
Pry.plugins[plugin_name].disable!
|
|
end
|
|
|
|
on "no-plugins", "Suppress loading of plugins." do
|
|
Pry.config.should_load_plugins = false
|
|
end
|
|
|
|
on "installed-plugins", "List installed plugins." do
|
|
puts "Installed Plugins:"
|
|
puts "--"
|
|
Pry.locate_plugins.each do |plugin|
|
|
puts "#{plugin.name}".ljust(18) + plugin.spec.summary
|
|
end
|
|
exit
|
|
end
|
|
|
|
on "simple-prompt", "Enable simple prompt mode" do
|
|
Pry.config.prompt = Pry::SIMPLE_PROMPT
|
|
end
|
|
|
|
on :r, :require=, "`require` a Ruby script at startup" do |file|
|
|
Pry.config.requires << file
|
|
end
|
|
|
|
on :I=, "Add a path to the $LOAD_PATH", :as => Array, :delimiter => ":" do |load_path|
|
|
load_path.map! do |path|
|
|
/\A\.\// =~ path ? path : File.expand_path(path)
|
|
end
|
|
|
|
$LOAD_PATH.unshift(*load_path)
|
|
end
|
|
|
|
on :v, :version, "Display the Pry version" do
|
|
puts "Pry version #{Pry::VERSION} on Ruby #{RUBY_VERSION}"
|
|
exit
|
|
end
|
|
|
|
on(:c, :context=,
|
|
"Start the session in the specified context. Equivalent to `context.pry` in a session.",
|
|
:default => "Pry.toplevel_binding"
|
|
)
|
|
end.process_options do |opts|
|
|
|
|
exit if opts.help?
|
|
|
|
# invoked via cli
|
|
Pry.cli = true
|
|
|
|
# create the actual context
|
|
context = Pry.binding_for(eval(opts[:context]))
|
|
|
|
if Pry::CLI.input_args.any? && Pry::CLI.input_args != ["pry"]
|
|
full_name = File.expand_path(Pry::CLI.input_args.first)
|
|
Pry.load_file_through_repl(full_name)
|
|
exit
|
|
end
|
|
|
|
if Pry.config.should_load_plugins
|
|
parser = Slop.new
|
|
end
|
|
|
|
# Start the session (running any code passed with -e, if there is any)
|
|
Pry.start(context, :input => StringIO.new(exec_string))
|
|
end
|