1
0
Fork 0
mirror of https://github.com/pry/pry.git synced 2022-11-09 12:35:05 -05:00
pry--pry/lib/pry/cli.rb
Kyrylo Silin 2e33c98d97 Improve -I option support
Namely, add support for multiple arguments for the `-I` option. For
example:

  % pry -Ilib:../code/src

Where `:` is the delimiter for paths. In the example above `./lib` and
`../code/src` directories will be added to $LOAD_PATH array.

Also, there was a bug in Slop with respect to parsing of "fuzzy options
with arguments"[1]. It is fixed now, so the next version of Slop will
indirectly resolve our issue #674 (-r/--require doesn't behave like
irb's)[2].

And finally, wrap some very long lines in the "cli.rb" (this is just a
cosmetic change).

[1]: https://github.com/injekt/slop/issues/74
[2]: https://github.com/pry/pry/issues/674

Signed-off-by: Kyrylo Silin <kyrylosilin@gmail.com>
2012-08-16 03:05:34 +03:00

185 lines
4.7 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
Pry::CLI.add_plugin_options
# 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) 2011 John Mair (banisterfiend)
--
}
on :e, :exec, "A line of code to execute in context before the session starts", :argument => true 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).", :argument => true do |plugin_name|
Pry.config.should_load_plugins = false
Pry.plugins[plugin_name].activate!
end
on :d, "disable-plugin", "Disable a specific plugin.", :argument => true 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", :argument => true do |file|
Pry.config.requires << file
end
on :I, "Add a path to the $LOAD_PATH", :argument => true, :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.",
:argument => true,
: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
# Start the session (running any code passed with -e, if there is any)
Pry.start(context, :input => StringIO.new(exec_string))
end