Module: PryMacro

Defined in:: lib/pry-macro.rb,
lib/pry-macro/version.rb

Defined Under Namespace

Constant Summary collapse

Commands =

Pry::CommandSet.new do
  create_command 'record', 'Starts a recording session' do
    def process
      # Define a few extra ivars on the current pry session so we can persist some state to use
      unless _pry_.instance_variable_defined?(:@record_stack)
        _pry_.instance_variable_set(:@record_stack, [])
        _pry_.instance_variable_set(:@macro_strings, [])
        _pry_.class.class_eval 'attr_accessor :record_stack, :macro_strings'
      end

      # By using a stack we can potentially have multiple sessions going. Use at your own peril though...
      _pry_.record_stack << Pry.history.session_line_count
    end
  end

  create_command 'stop', 'Stops a recording session, and saves a macro' do
    def setup
      if !_pry_.instance_variable_defined?(:@record_stack) && _pry_.record_stack.empty?
        raise 'Cannot stop recording when no recorder is running'
      end

      session_begin = _pry_.record_stack.pop
      session_end   = Pry.history.session_line_count

      # Get the history between the start and end of the recording session
      @history = 
        Pry.history
           .to_a
           .last(session_end - session_begin - 1)
           .reduce(StringIO.new) { |io, item| io.puts(item); io }

      # Have to have a name to execute this later
      @name = ask('Macro Name: ')
    end

    def process
      history_lines = @history.string.lines.map { |s| "      #{s}"}.join.chomp.tap { |h|
        h.sub!(/^ {6}/,'') # First line won't need the spacing
      }

      # Save the command into a string, and make it look decent
      # Tinge of a heredocs hack
      command_string = "        Pry::Commands.block_command '\#{@name}', 'no description' do\n          _pry_.input = StringIO.new(\n            <<-MACRO.gsub(/^ {4,6}/, '')\n              \#{history_lines}\n            MACRO\n          )\n        end\n      COMMAND_STRING\n\n      puts command_string\n\n      # ...so that we can save the contents for saving later (optional)\n      _pry_.macro_strings << MacroString.new(@name, command_string)\n      # ...as well as evaluating it and making it immediately usable to us.\n      eval command_string\n    end\n  end\n\n  create_command 'save_macro', 'Saves a named macro to your .pryrc file on the tail end' do\n    def options(opt)\n      # Options for later:\n      # -p --path : path to save in\n      # -n --name : name of the command (override)\n      # -d --desc : description of the command\n    end\n\n    def process\n      raise 'No Macros are defined!' unless _pry_.instance_variable_defined?(:@macro_strings)\n      @macro = _pry_.macro_strings.find(\n        -> { raise \"Command \#{args.first} not found!\" } # If nothing is found, raise the error\n      ) { |macro| macro.name == args.first }\n\n      # Append the Pryrc with the macro, leaving blank lines\n      File.open(File.join(Dir.home, '.pryrc'), 'a') { |f| f.puts '', @macro.command, '' }\n    end\n  end\nend\n".gsub(/^ {10}/, '')

VERSION =

"0.0.1"