require 'pry' unless defined? Pry

module PryMoves
  Commands = Pry::CommandSet.new do
    block_command 'step', 'Step execution into the next line or method.' do |param|
      breakout_navigation :step, param
    end

    block_command 'finish', 'Finish - xule tut neponyatnogo' do |param|
      breakout_navigation :finish, param
    end

    block_command 'next', 'Execute the next line stepping into blocks' do |param|
      breakout_navigation :next, param
    end

    block_command 'nn', 'Execute the next line skipping blocks' do |param|
      breakout_navigation :next, 'blockless'
    end

    block_command 'iterate', 'Go to next iteration of current block' do |param|
      breakout_navigation :iterate, param
    end

    block_command 'continue', 'Continue program execution and end the Pry session' do
      check_file_context
      run 'exit-all'
    end

    alias_command 'c', 'continue'
    alias_command 's', 'step'
    alias_command 'n', 'next'
    alias_command 'f', 'finish'

    block_command 'watch', 'Display value of expression on every move' do |param|
      PryMoves::Watch.instance.process_cmd param, target
    end

    block_command 'bt', 'Backtrace' do |param, param2|
      PryMoves::Backtrace.new(target, _pry_).run_command param, param2
    end

    block_command 'debug', '' do
      cmd = arg_string.gsub(/^debug/, '').strip
      breakout_navigation :debug, cmd
    end

    block_command '!', 'exit' do
      PryMoves.unlock
      Pry.config.exit_requested = true
      run '!!!'
    end

    # Hit Enter to repeat last command
    command /^$/, "repeat last command" do
      _pry_.run_command Pry.history.to_a.last
    end

    helpers do
      def breakout_navigation(action, param)
        check_file_context
        _pry_.binding_stack.clear     # Clear the binding stack.
        throw :breakout_nav, {        # Break out of the REPL loop and
          action: action,          #   signal the tracer.
          param:  param,
          binding: target,
          pry: _pry_
        }
      end

      # Ensures that a command is executed in a local file context.
      def check_file_context
        unless PryMoves.check_file_context(target)
          raise Pry::CommandError, 'Cannot find local context. Did you use `binding.pry`?'
        end
      end
    end
  end
end

Pry.commands.import PryMoves::Commands