summaryrefslogtreecommitdiff
path: root/lib/pry/repl_file_loader.rb
blob: 3a06fe7aa6cdf9b7556776498dd057c1f6edc3c0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# frozen_string_literal: true

class Pry
  # A class to manage the loading of files through the REPL loop.
  # This is an interesting trick as it processes your file as if it
  # was user input in an interactive session. As a result, all Pry
  # commands are available, and they are executed non-interactively. Furthermore
  # the session becomes interactive when the repl loop processes a
  # 'make-interactive' command in the file. The session also becomes
  # interactive when an exception is encountered, enabling you to fix
  # the error before returning to non-interactive processing with the
  # 'make-non-interactive' command.

  class REPLFileLoader
    def initialize(file_name)
      full_name = File.expand_path(file_name)
      raise "No such file: #{full_name}" unless File.exist?(full_name)

      define_additional_commands
      @content = File.read(full_name)
    end

    # Switch to interactive mode, i.e take input from the user
    # and use the regular print and exception handlers.
    # @param [Pry] pry_instance the Pry instance to make interactive.
    def interactive_mode(pry_instance)
      pry_instance.config.input = Pry.config.input
      pry_instance.config.print = Pry.config.print
      pry_instance.config.exception_handler = Pry.config.exception_handler
      Pry::REPL.new(pry_instance).start
    end

    # Switch to non-interactive mode. Essentially
    # this means there is no result output
    # and that the session becomes interactive when an exception is encountered.
    # @param [Pry] pry_instance the Pry instance to make non-interactive.
    def non_interactive_mode(pry_instance, content)
      pry_instance.print = proc {}
      pry_instance.exception_handler = proc do |o, _e, p|
        p.run_command "cat --ex"
        o.puts "...exception encountered, going interactive!"
        interactive_mode(pry_instance)
      end

      content.lines.each do |line|
        break unless pry_instance.eval line, generated: true
      end

      return if pry_instance.eval_string.empty?

      pry_instance.output.puts(
        "#{pry_instance.eval_string}...exception encountered, going interactive!"
      )
      interactive_mode(pry_instance)
    end

    # Define a few extra commands useful for flipping back & forth
    # between interactive/non-interactive modes
    def define_additional_commands
      s = self

      Pry::Commands.command "make-interactive", "Make the session interactive" do
        s.interactive_mode(pry_instance)
      end

      Pry::Commands.command(
        "load-file", "Load another file through the repl"
      ) do |file_name|
        s.non_interactive_mode(pry_instance, File.read(File.expand_path(file_name)))
      end
    end

    # Actually load the file through the REPL by setting file content
    # as the REPL input stream.
    def load
      non_interactive_mode(Pry.new, @content)
    end
  end
end