40 lines
1.4 KiB
Ruby
40 lines
1.4 KiB
Ruby
direc = File.dirname(__FILE__)
|
|
|
|
require 'rubygems'
|
|
require "#{direc}/../lib/pry"
|
|
|
|
class MathCommands < Pry::CommandBase
|
|
command "greet", "Greet a person, e.g: greet john" do |name|
|
|
output.puts "Good afternoon #{name.capitalize}! Do you like Math?"
|
|
end
|
|
|
|
command "add", "Add a list of numbers together, e.g: add 1 2 3 4" do |*args|
|
|
output.puts "Total: #{args.map(&:to_f).inject(&:+)}"
|
|
end
|
|
|
|
command "multiply", "Multiply a list of numbers together, e.g: multiply 1 2 3 4" do |*args|
|
|
output.puts "Total: #{args.map(&:to_f).inject(&:*)}"
|
|
end
|
|
|
|
# Explicitly giving a description of "" to prevent command being
|
|
# displayed in 'help'
|
|
command "exit", "" do
|
|
throw :breakout, 0
|
|
end
|
|
|
|
# Bring in the "!" method from Pry::Commands
|
|
import_from Pry::Commands, "!"
|
|
end
|
|
|
|
# Since we provide math commands, let's have mathematical
|
|
# before_session and after_session hooks, and a mathematical prompt
|
|
math_prompt = [proc { "math> " }, proc { "math* " }]
|
|
math_hooks = {
|
|
:before_session => proc { |output, *| output.puts "Welcome! Let's do some math! Type 'help' for a list of commands." },
|
|
:after_session => proc { |output, *| output.puts "Goodbye!" }
|
|
}
|
|
|
|
# Start a Pry session using the commands defined in MyCommands
|
|
# Type 'help' in Pry to get a list of the commands and their descriptions
|
|
Pry.start(TOPLEVEL_BINDING, :commands => MathCommands, :prompt => math_prompt, :hooks => math_hooks)
|