pry--pry/lib/pry.rb

# (C) John Mair (banisterfiend) 2010
# MIT License

direc = File.dirname(__FILE__)

require 'rubygems'
require 'readline'
require 'ruby_parser'
require "#{direc}/pry/version"

module Pry
  class << self
    attr_accessor :default_prompt, :wait_prompt,
    :session_start_msg, :session_end_msg
  end
  
  @default_prompt = proc { |v| "pry(#{v})> " }
  @wait_prompt = proc { |v| "pry(#{v})* " }
  @session_start_msg = proc { |v| "Beginning Pry session for #{v}" }
  @session_end_msg = proc { |v| "Ending Pry session for #{v}" }

  # useful for ending all Pry sessions currently active
  @dead = false
  
  # loop
  def self.repl(target=TOPLEVEL_BINDING)
    target = binding_for(target)
    target_self = target.eval('self')
    puts session_start_msg.call(target_self)

    loop do
      if catch(:pop) { rep(target) } == :return || @dead
        break 
      end
    end

    puts session_end_msg.call(target_self)

    target_self
  end

  class << self
    alias_method :into, :repl
    alias_method :start, :repl
  end
  
  # print
  def self.rep(target=TOPLEVEL_BINDING)
    target = binding_for(target)
    value = re(target)
    case value
    when Exception
      puts "#{value.class}: #{value.message}"
    else
      puts "=> #{value.inspect}"
    end
  end

  # eval
  def self.re(target=TOPLEVEL_BINDING)
    target = binding_for(target)
    target.eval r(target)
  rescue StandardError => e
    e
  end

  # read
  def self.r(target=TOPLEVEL_BINDING)
    target = binding_for(target)
    eval_string = ""
    loop do
      val = Readline.readline(prompt(eval_string, target), true)
      eval_string += "#{val}\n"
      process_commands(val, eval_string, target)
      
      break eval_string if valid_expression?(eval_string)
    end
  end
  
  def self.process_commands(val, eval_string, target)
    case val
    when "#exit", "#quit"
      exit
    when "!"
      eval_string.replace("")
      puts "Refreshed REPL."
    when "exit", "quit"
      throw(:pop, :return)
    end
  end

  def self.prompt(eval_string, target)
    target_self = target.eval('self')
    
    if eval_string.empty?
      default_prompt.call(target_self)
    else
      wait_prompt.call(target_self)
    end
  end

  def self.valid_expression?(code)
    RubyParser.new.parse(code)
  rescue Racc::ParseError, SyntaxError
    false
  else
    true
  end

  def self.binding_for(target)
    if target.is_a?(Binding)
      target
    else
      target.instance_eval { binding }
    end
  end

  def self.kill
    @dead = true
  end

  def self.revive
    @dead = false
  end
end