require 'ripl/rc/u'

# from https://github.com/janlelis/ripl-multi_line
module Ripl::Rc::Multiline
  include Ripl::Rc::U

  # test those:
  # ruby -e '"'
  # ruby -e '{'
  # ruby -e '['
  # ruby -e '('
  # ruby -e '/'
  # ruby -e 'class C'
  # ruby -e 'def f'
  # ruby -e 'begin'
  ERROR_REGEXP = Regexp.new(
    [ # string or regexp
      "unterminated \\w+ meets end of file",
      # mri and rubinius
      "syntax error, unexpected \\$end",
      # rubinius
      "expecting '.+'( or '.+')*",
      # jruby
      "syntax error, unexpected end-of-file",
    ].join('|'))

  def before_loop
    return super if Multiline.disabled?
    @rc_multiline_buffer = []
    super
  end

  def prompt
    return super if Multiline.disabled?
    if @rc_multiline_buffer.empty?
      super
    else
      "#{' '*(@prompt.size-2)}| "
    end
  end

  def loop_once
    return super if Multiline.disabled?
    catch(:rc_multiline_cont) do
      super
      @rc_multiline_buffer.clear
    end
  end

  def print_eval_error(e)
    return super if Multiline.disabled?
    if e.is_a?(SyntaxError) && e.message =~ ERROR_REGEXP
      @rc_multiline_buffer << @input if @rc_multiline_buffer.empty?
      throw :rc_multiline_cont
    else
      super
    end
  end

  def loop_eval(input)
    return super if Multiline.disabled?
    if @rc_multiline_buffer.empty?
      super
    else
      @rc_multiline_buffer << input
      super @rc_multiline_buffer.join("\n")
    end
  end

  def handle_interrupt
    return super if Multiline.disabled?
    if @rc_multiline_buffer.empty?
      super
    else
      line = @rc_multiline_buffer.pop
      print "[removed this line: #{line}]"
      super
      throw :rc_multiline_cont
    end
  end
end

Ripl::Shell.include(Ripl::Rc::Multiline)