Class: Pry::CommandSet

Inherits:
Object show all
Includes:
Enumerable, Helpers::BaseHelpers
Defined in:
lib/pry/command_set.rb

Overview

This class is used to create sets of commands. Commands can be imported from different sets, aliased, removed, etc.

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from Helpers::BaseHelpers

#colorize_code, #heading, #highlight, #not_a_real_file?, #safe_send, #silence_warnings, #stagger_output, #use_ansi_codes?

Constructor Details

#initialize(*imported_sets) { ... } ⇒ CommandSet

Returns a new instance of CommandSet.

Parameters:

  • imported_sets (Array<Commandset>) —

    Sets which will be imported automatically

Yields:

  • Optional block run to define commands



20
21
22
23
24
25
 
# File 'lib/pry/command_set.rb', line 20

def initialize(*imported_sets, &block)
  @commands      = {}
  @helper_module = Module.new
  import(*imported_sets)
  instance_eval(&block) if block
end

Instance Attribute Details

#helper_module ⇒ Object (readonly)

Returns the value of attribute helper_module.



15
16
17
 
# File 'lib/pry/command_set.rb', line 15

def helper_module
  @helper_module
end

Instance Method Details

#[](pattern) ⇒ Pry::Command? Also known as: find_command

Find a command that matches the given line

Parameters:

  • pattern (String) —

    The line that might be a command invocation

Returns:



275
276
277
278
279
280
 
# File 'lib/pry/command_set.rb', line 275

def [](pattern)
  commands = @commands.values.select do |command|
    command.matches?(pattern)
  end
  commands.max_by { |command| command.match_score(pattern) }
end

#[]=(pattern, command) ⇒ Pry::Command

Re-assign the command found at pattern with command.

Examples:

Pry.config.commands["help"] = MyHelpCommand

Parameters:

  • pattern (Regexp, String) —

    The command to add or replace(found at pattern).

  • command (Pry::Command) —

    The command to add.

Returns:

  • (Pry::Command) —

    Returns the new command (matched with “pattern”.)



298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
 
# File 'lib/pry/command_set.rb', line 298

def []=(pattern, command)
  if command.equal?(nil)
    @commands.delete(pattern)
    return
  end

  unless command.is_a?(Class) && command < Pry::Command
    raise TypeError, "command is not a subclass of Pry::Command"
  end

  bind_command_to_pattern = pattern != command.match
  if bind_command_to_pattern
    command_copy = command.dup
    command_copy.match = pattern
    @commands[pattern] = command_copy
  else
    @commands[pattern] = command
  end
end

#add_command(command) ⇒ Object

Add a command to set.

Parameters:

  • command (Command) —

    a subclass of Pry::Command.



324
325
326
 
# File 'lib/pry/command_set.rb', line 324

def add_command(command)
  self[command.match] = command
end

#alias_command(match, action, options = {}) ⇒ Object

Aliases a command

Examples:

Creating an alias for ‘ls -M`

Pry.config.commands.alias_command "lM", "ls -M"

Pass explicit description (overriding default).

Pry.config.commands.alias_command "lM", "ls -M", :desc => "cutiepie"

Parameters:

  • match (String, Regex) —

    The match of the alias (can be a regex).

  • action (String) —

    The action to be performed (typically another command).

  • options (Hash) (defaults to: {}) —

    The optional configuration parameters, accepts the same as the ‘command` method, but also allows the command description to be passed this way too as `:desc`



190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
 
# File 'lib/pry/command_set.rb', line 190

def alias_command(match, action, options = {})
  (cmd = find_command(action)) || raise("command: '#{action}' not found")
  original_options = cmd.options.dup

  options = original_options.merge!(
    desc: "Alias for `#{action}`",
    listing: match.is_a?(String) ? match : match.inspect
  ).merge!(options)

  # ensure default description is used if desc is nil
  desc = options.delete(:desc).to_s

  c = block_command match, desc, options do |*args|
    run action, *args
  end

  # TODO: untested. What's this about?
  c.class_eval do
    define_method(:complete) do |input|
      cmd.new(context).complete(input)
    end
  end

  c.group "Aliases"

  c
end

#block_command(match, description = "No description.", options = {}) { ... } ⇒ Object Also known as: command

Defines a new Pry command.

Examples:

MyCommands = Pry::CommandSet.new do
  command "greet", "Greet somebody" do |name|
    puts "Good afternoon #{name.capitalize}!"
  end
end

# From pry:
# pry(main)> pry_instance.commands = MyCommands
# pry(main)> greet john
# Good afternoon John!
# pry(main)> help greet
# Greet somebody

Regexp command

MyCommands = Pry::CommandSet.new do
  command(
    /number-(\d+)/, "number-N regex command", :listing => "number"
  ) do |num, name|
    puts "hello #{name}, nice number: #{num}"
  end
end

# From pry:
# pry(main)> pry_instance.commands = MyCommands
# pry(main)> number-10 john
# hello john, nice number: 10
# pry(main)> help number
# number-N regex command

Parameters:

  • match (String, Regexp) —

    The start of invocations of this command.

  • description (String) (defaults to: "No description.") —

    A description of the command.

  • options (Hash) (defaults to: {}) —

    The optional configuration parameters.

Options Hash (options):

  • :keep_retval (Boolean) —

    Whether or not to use return value of the block for return of ‘command` or just to return `nil` (the default).

  • :interpolate (Boolean) —

    Whether string #{} based interpolation is applied to the command arguments before executing the command. Defaults to true.

  • :listing (String) —

    The listing name of the command. That is the name by which the command is looked up by help and by show-source. Necessary for commands with regex matches.

  • :use_prefix (Boolean) —

    Whether the command uses ‘Pry.config.command_prefix` prefix (if one is defined). Defaults to true.

  • :shellwords (Boolean) —

    Whether the command’s arguments should be split using Shellwords instead of just split on spaces. Defaults to true.

Yields:

  • The action to perform. The parameters in the block determines the parameters the command will receive. All parameters passed into the block will be strings. Successive command parameters are separated by whitespace at the Pry prompt.



78
79
80
81
82
83
84
85
86
87
88
 
# File 'lib/pry/command_set.rb', line 78

def block_command(match, description = "No description.", options = {}, &block)
  if description.is_a?(Hash)
    options = description
    description = "No description."
  end
  options = Pry::Command.default_options(match).merge!(options)

  @commands[match] = Pry::BlockCommand.subclass(
    match, description, options, helper_module, &block
  )
end

#complete(search, context = {}) ⇒ Array<String>

Generate completions for the user’s search.

Parameters:

  • search (String) —

    The line to search for

  • context (Hash) (defaults to: {}) —

    The context to create the command with

Returns:

  • (Array<String>) 


365
366
367
368
369
370
371
372
373
374
 
# File 'lib/pry/command_set.rb', line 365

def complete(search, context = {})
  if (command = find_command(search))
    command.new(context).complete(search)
  else
    keys = @commands.keys.select do |key|
      key.is_a?(String) && key.start_with?(search)
    end
    keys.map { |key| key + " " }
  end
end

#create_command(match, description = "No description.", options = {}) { ... } ⇒ Object

Defines a new Pry command class.

Examples:

Pry::Commands.create_command "echo", "echo's the input", :shellwords => false do
  def options(opt)
    opt.banner "Usage: echo [-u | -d] <string to echo>"
    opt.on :u, :upcase, "ensure the output is all upper-case"
    opt.on :d, :downcase, "ensure the output is all lower-case"
  end

  def process
    if opts.present?(:u) && opts.present?(:d)
      raise Pry::CommandError, "-u and -d makes no sense"
    end
    result = args.join(" ")
    result.downcase! if opts.present?(:downcase)
    result.upcase! if opts.present?(:upcase)
    output.puts result
  end
end

Parameters:

  • match (String, Regexp) —

    The start of invocations of this command.

  • description (String) (defaults to: "No description.") —

    A description of the command.

  • options (Hash) (defaults to: {}) —

    The optional configuration parameters, see #command

Yields:

  • The class body’s definition.



117
118
119
120
121
122
123
124
125
126
127
128
129
 
# File 'lib/pry/command_set.rb', line 117

def create_command(match, description = "No description.", options = {}, &block)
  if description.is_a?(Hash)
    options = description
    description = "No description."
  end
  options = Pry::Command.default_options(match).merge!(options)

  @commands[match] = Pry::ClassCommand.subclass(
    match, description, options, helper_module, &block
  )
  @commands[match].class_eval(&block)
  @commands[match]
end

#delete(*searches) ⇒ Object

Removes some commands from the set

Parameters:

  • searches (Array<String>) —

    the matches or listings of the commands to remove



138
139
140
141
142
143
 
# File 'lib/pry/command_set.rb', line 138

def delete(*searches)
  searches.each do |search|
    cmd = find_command_by_match_or_listing(search)
    @commands.delete cmd.match
  end
end

#desc(search, description = nil) ⇒ Object

Sets or gets the description for a command (replacing the old description). Returns current description if no description parameter provided.

Examples:

Setting

MyCommands = Pry::CommandSet.new do
  desc "help", "help description"
end

Getting

Pry.config.commands.desc "amend-line"

Parameters:

  • search (String, Regexp) —

    The command match.

  • description (String?) (defaults to: nil) —

    (nil) The command description.



253
254
255
256
257
258
 
# File 'lib/pry/command_set.rb', line 253

def desc(search, description = nil)
  cmd = find_command_by_match_or_listing(search)
  return cmd.description unless description

  cmd.description = description
end

#each(&block) ⇒ Object 



131
132
133
 
# File 'lib/pry/command_set.rb', line 131

def each(&block)
  @commands.each(&block)
end

#find_command_by_match_or_listing(match_or_listing) ⇒ Command

Returns The command object matched.

Parameters:

  • match_or_listing (String, Regexp) —

    The match or listing of a command. of the command to retrieve.

Returns:

  • (Command) —

    The command object matched.



173
174
175
176
177
 
# File 'lib/pry/command_set.rb', line 173

def find_command_by_match_or_listing(match_or_listing)
  cmd = (@commands[match_or_listing] ||
    Pry::Helpers::BaseHelpers.find_command(match_or_listing, @commands))
  cmd || raise(ArgumentError, "cannot find a command: '#{match_or_listing}'")
end

#find_command_for_help(search) ⇒ Pry::Command?

Find the command that the user might be trying to refer to.

Parameters:

  • search (String) —

    The user’s search.

Returns:



331
332
333
334
335
336
337
338
 
# File 'lib/pry/command_set.rb', line 331

def find_command_for_help(search)
  find_command(search) ||
    (begin
       find_command_by_match_or_listing(search)
     rescue ArgumentError
       nil
     end)
end

#import(*sets) ⇒ Pry::CommandSet

Imports all the commands from one or more sets.

Parameters:

  • sets (Array<CommandSet>) —

    Command sets, all of the commands of which will be imported.

Returns:



149
150
151
152
153
154
155
 
# File 'lib/pry/command_set.rb', line 149

def import(*sets)
  sets.each do |set|
    @commands.merge! set.to_hash
    helper_module.send :include, set.helper_module
  end
  self
end

#import_from(set, *matches) ⇒ Pry::CommandSet

Imports some commands from a set

Parameters:

  • set (CommandSet) —

    Set to import commands from

  • matches (Array<String>) —

    Commands to import

Returns:



161
162
163
164
165
166
167
168
 
# File 'lib/pry/command_set.rb', line 161

def import_from(set, *matches)
  helper_module.send :include, set.helper_module
  matches.each do |match|
    cmd = set.find_command_by_match_or_listing(match)
    @commands[cmd.match] = cmd
  end
  self
end

#list_commands ⇒ Array Also known as: keys

Returns The list of commands provided by the command set.

Returns:

  • (Array) —

    The list of commands provided by the command set.



262
263
264
 
# File 'lib/pry/command_set.rb', line 262

def list_commands
  @commands.keys
end

#process_line(val, context = {}) ⇒ CommandSet::Result

Process the given line to see whether it needs executing as a command.

Parameters:

  • val (String) —

    The line to execute

  • context (Hash) (defaults to: {}) —

    The context to execute the commands with

Returns:

  • (CommandSet::Result) 


351
352
353
354
355
356
357
358
359
 
# File 'lib/pry/command_set.rb', line 351

def process_line(val, context = {})
  if (command = find_command(val))
    context = context.merge(command_set: self)
    retval = command.new(context).process_line(val)
    Result.new(true, retval)
  else
    Result.new(false)
  end
end

#rename_command(new_match, search, options = {}) ⇒ Object

Rename a command. Accepts either match or listing for the search.

Examples:

Renaming the ‘ls` command and changing its description.

Pry.config.commands.rename "dir", "ls", :description => "DOS friendly ls"

Parameters:

  • new_match (String, Regexp) —

    The new match for the command.

  • search (String, Regexp) —

    The command’s current match or listing.

  • options (Hash) (defaults to: {}) —

    The optional configuration parameters, accepts the same as the ‘command` method, but also allows the command description to be passed this way too.



227
228
229
230
231
232
233
234
235
236
237
238
239
240
 
# File 'lib/pry/command_set.rb', line 227

def rename_command(new_match, search, options = {})
  cmd = find_command_by_match_or_listing(search)

  options = {
    listing: new_match,
    description: cmd.description
  }.merge!(options)

  @commands[new_match] = cmd.dup
  @commands[new_match].match = new_match
  @commands[new_match].description = options.delete(:description)
  @commands[new_match].options.merge!(options)
  @commands.delete(cmd.match)
end

#to_hash ⇒ Object Also known as: to_h 



267
268
269
 
# File 'lib/pry/command_set.rb', line 267

def to_hash
  @commands.dup
end

#valid_command?(val) ⇒ Boolean

Is the given line a command invocation?

Parameters:

  • val (String)

Returns:

  • (Boolean) 


343
344
345
 
# File 'lib/pry/command_set.rb', line 343

def valid_command?(val)
  !!find_command(val)
end