Add a default-off option to use Pry, favoring IRB

Author: davidcelis

Gemfile

@@ -5,3 +5,4 @@ gemspec
 
 gem "rspec", "~> 3.0"
 gem "standardrb"
+gem "pry"

bin/rack-console

@@ -8,20 +8,24 @@ options = {}
 OptionParser.new do |opts|
   opts.banner = "USAGE: rack-console [options]"
 
-  opts.on("-c", "--config [RACKUP_FILE]", "Specify a rackup config file") do |config|
+  opts.on("-c", "--config [RACKUP_FILE]", "Specify a rackup config file (default: ./config.ru)") do |config|
     options[:config] = config
   end
 
-  opts.on("-e", "--environment [ENVIRONMENT]", "Specify the Rack environment") do |environment|
+  opts.on("-e", "--environment [ENVIRONMENT]", "Specify the Rack environment (default: development)") do |environment|
     options[:environment] = environment
   end
 
+  opts.on("-I", "--include [PATHS]", "Add paths (colon-separated) to the $LOAD_PATH") do |paths|
+    options[:include] = paths.split(":")
+  end
+
   opts.on("-r", "--require [LIBRARY]", "Require a file or library before the Rack console loads") do |library|
     options[:require] = library
   end
 
-  opts.on("-I", "--include [PATHS]", "Add paths (colon-separated) to the $LOAD_PATH") do |paths|
-    options[:include] = paths.split(":")
+  opts.on("-P", "--[no-]pry", "Use Pry if available or explicitly disable it") do |pry|
+    options[:pry] = pry
   end
 
   opts.on("-v", "--version", "Print version and exit") do |v|

lib/rack/console.rb

@@ -16,6 +16,7 @@ def initialize(options = {})
       @environment = options[:environment]
       @include = options[:include]
       @require = options[:require]
+      @use_pry = !!options[:pry]
     end
 
     def start
@@ -32,7 +33,7 @@ def start
       main.instance_variable_set(:@app, Rack::Console::Session.new(app))
       main.extend(Rack::Console::Methods)
 
-      if Gem::Specification.find_all_by_name("pry").any?
+      if Gem::Specification.find_all_by_name("pry").any? && @use_pry
         require "pry"
         Pry.start
       else

spec/rack/console_spec.rb

@@ -1,45 +1,63 @@
 require "rack/console"
 require "irb"
+require "pry"
 
 describe Rack::Console do
   before do
     @old_pwd = Dir.pwd
     Dir.chdir File.expand_path("../../support", __FILE__)
 
-    expect(IRB).to receive(:start)
+    allow(IRB).to receive(:start)
+    allow(Pry).to receive(:start)
   end
 
-  it "defaults to a config.ru file in the current working directory" do
-    expect(Rack::Builder).to receive(:parse_file)
-      .with("config.ru")
-      .and_call_original
+  context "when using IRB" do
+    before do
+      expect(IRB).to receive(:start)
+      expect(Pry).not_to receive(:start)
+    end
 
-    Rack::Console.new.start
-  end
+    it "defaults to a config.ru file in the current working directory" do
+      expect(Rack::Builder).to receive(:parse_file)
+        .with("config.ru")
+        .and_call_original
 
-  it "accepts the --config option to override the location of config.ru" do
-    Dir.chdir @old_pwd
-    expect(Rack::Builder).to receive(:parse_file)
-      .with("spec/support/config.ru")
-      .and_call_original
+      Rack::Console.new.start
+    end
 
-    Rack::Console.new(config: "spec/support/config.ru").start
-  end
+    it "accepts the --config option to override the location of config.ru" do
+      Dir.chdir @old_pwd
+      expect(Rack::Builder).to receive(:parse_file)
+        .with("spec/support/config.ru")
+        .and_call_original
 
-  it "accepts the --require option to require a file or library" do
-    Rack::Console.new(require: "json").start
-    expect { JSON }.not_to raise_error
-  end
+      Rack::Console.new(config: "spec/support/config.ru").start
+    end
 
-  it "accepts the --require option to add paths to $LOAD_PATH" do
-    Rack::Console.new(include: ["lib", "spec"]).start
-    expect($LOAD_PATH).to include("lib")
-    expect($LOAD_PATH).to include("spec")
+    it "accepts the --require option to require a file or library" do
+      Rack::Console.new(require: "json").start
+      expect { JSON }.not_to raise_error
+    end
+
+    it "accepts the --require option to add paths to $LOAD_PATH" do
+      Rack::Console.new(include: ["lib", "spec"]).start
+      expect($LOAD_PATH).to include("lib")
+      expect($LOAD_PATH).to include("spec")
+    end
+
+    it "accepts the --environment option to set the environment" do
+      Rack::Console.new(environment: "production").start
+      expect(ENV["RACK_ENV"]).to eq("production")
+    end
   end
 
-  it "accepts an argument to set the environment" do
-    Rack::Console.new(environment: "production").start
-    expect(ENV["RACK_ENV"]).to eq("production")
+  context "when using Pry" do
+    it "accepts the --pry option to enable Pry" do
+      expect(IRB).not_to receive(:start)
+      expect(Pry).to receive(:start)
+
+      Rack::Console.new(pry: true).start
+    end
   end
 
   describe "preamble message" do