bin/burglar in burglar-0.1.3 vs bin/burglar in burglar-0.1.4

@@ -7,25 +7,41 @@
 def add_common_opts(c)
   c.option :config, '-c FILE', '--config FILE', 'Config file to load'
   c.option :begin, '-b DATE', '--begin DATE', 'Beginning of date range'
   c.option :end, '-e DATE', '--end DATE', 'End of date range'
   c.option :pending, '-p', '--pending', 'Include pending transactions'
+  c.option :shell, '-s', '--shell', 'Launch a REPL with the burglar object'
 end
 
 def date_parse_or_default(string, default)
   return default unless string
   Date.parse(string)
 end
 
+def launch_shell(res)
+  res.transactions
+  require 'pry'
+  binding.pry # rubocop:disable Lint/Debugger
+rescue LoadError
+  warn 'Falling back to IRB because Pry failed to load'
+  require 'irb'
+  binding.irb
+end
+
 Mercenary.program(:burglar) do |p|
   p.version Burglar::VERSION
   p.description 'Load data from banks'
   p.syntax 'burglar [options]'
 
   add_common_opts(p)
 
   p.action do |_, options|
     options[:end] = date_parse_or_default(options[:end], Date.today)
     options[:begin] = date_parse_or_default(options[:begin], options[:end] - 30)
-    puts Burglar.new(options).transactions
+    res = Burglar.new(options)
+    if options[:shell]
+      launch_shell res
+    else
+      puts res.transactions
+    end
   end
 end