pry/pry

# frozen_string_literal: true

class Pry
  class Command
    class Play < Pry::ClassCommand
      match 'play'
      group 'Editing'
      description 'Playback a string variable, method, line, or file as input.'

      banner <<-'BANNER'
        Usage: play [OPTIONS] [--help]

        The play command enables you to replay code from files and methods as if they
        were entered directly in the Pry REPL.

        play --lines 149..153   # assumes current context
        play -i 20 --lines 1..3 # assumes lines of the input expression at 20
        play -o 4               # the output of an expression at 4
        play Pry#repl -l 1..-1  # play the contents of Pry#repl method
        play -e 2               # play from specified line until end of valid expression
        play hello.rb           # play a file
        play Rakefile -l 5      # play line 5 of a file
        play -d hi              # play documentation of hi method
        play hi --open          # play hi method and leave it open

        https://github.com/pry/pry/wiki/User-Input#wiki-Play
      BANNER

      def options(opt)
        CodeCollector.inject_options(opt)

        opt.on :open, 'Plays the selected content except the last line. Useful' \
                      ' for replaying methods and leaving the method definition' \
                      ' "open". `amend-line` can then be used to' \
                      ' modify the method.'

        opt.on :e, :expression=, 'Executes until end of valid expression', as: Integer
        opt.on :p, :print, 'Prints executed code'
      end

      def process
        @cc = CodeCollector.new(args, opts, pry_instance)

        perform_play
        show_input
      end

      def perform_play
        eval_string << content_after_options
        run "fix-indent"
      end

      def show_input
        return unless opts.present?(:print)
        return unless Pry::Code.complete_expression?(eval_string)

        run 'show-input'
      end

      def content_after_options
        if opts.present?(:open)
          restrict_to_lines(content, (0..-2))
        elsif opts.present?(:expression)
          content_at_expression
        else
          content
        end
      end

      def content_at_expression
        code_object.expression_at(opts[:expression])
      end

      def code_object
        Pry::Code.new(content)
      end

      def should_use_default_file?
        !args.first && !opts.present?(:in) && !opts.present?(:out)
      end

      def content
        if should_use_default_file?
          file_content
        else
          @cc.content
        end
      end

      # The file to play from when no code object is specified.
      # e.g `play --lines 4..10`
      def default_file
        file =
          if target.respond_to?(:source_location)
            target.source_location.first
          else
            target.eval("__FILE__")
          end
        file && File.expand_path(file)
      end

      def file_content
        if !default_file || !File.exist?(default_file)
          raise CommandError, "File does not exist! File was: #{default_file.inspect}"
        end

        @cc.restrict_to_lines(File.read(default_file), @cc.line_range)
      end
    end

    Pry::Commands.add_command(Pry::Command::Play)
  end
end