# load libraries
%w{ rubygems yaml what_methods wirble hpricot }.each { |lib| require lib }
# start wirble
Wirble.init
# Wirble.colorize
# Customize the prompt a little
# Taken from => http://dotfiles.org/~sd/.irbrc
unless self.class.const_defined? "IRB_RC_HAS_LOADED"
HISTFILE = "~/.irb.hist"
MAXHISTSIZE = 100
begin # ANSI codes
ANSI_BLACK = "\033[0;30m"
ANSI_GRAY = "\033[1;30m"
ANSI_LGRAY = "\033[0;37m"
ANSI_WHITE = "\033[1;37m"
ANSI_RED ="\033[0;31m"
ANSI_LRED = "\033[1;31m"
ANSI_GREEN = "\033[0;32m"
ANSI_LGREEN = "\033[1;32m"
ANSI_BROWN = "\033[0;33m"
ANSI_YELLOW = "\033[1;33m"
ANSI_BLUE = "\033[0;34m"
ANSI_LBLUE = "\033[1;34m"
ANSI_PURPLE = "\033[0;35m"
ANSI_LPURPLE = "\033[1;35m"
ANSI_CYAN = "\033[0;36m"
ANSI_LCYAN = "\033[1;36m"
ANSI_BACKBLACK = "\033[40m"
ANSI_BACKRED = "\033[41m"
ANSI_BACKGREEN = "\033[42m"
ANSI_BACKYELLOW = "\033[43m"
ANSI_BACKBLUE = "\033[44m"
ANSI_BACKPURPLE = "\033[45m"
ANSI_BACKCYAN = "\033[46m"
ANSI_BACKGRAY = "\033[47m"
ANSI_RESET = "\033[0m"
ANSI_BOLD = "\033[1m"
ANSI_UNDERSCORE = "\033[4m"
ANSI_BLINK = "\033[5m"
ANSI_REVERSE = "\033[7m"
ANSI_CONCEALED = "\033[8m"
XTERM_SET_TITLE = "\033]2;"
XTERM_END = "\007"
ITERM_SET_TAB = "\033]1;"
ITERM_END = "\007"
SCREEN_SET_STATUS = "\033]0;"
SCREEN_END = "\007"
end
begin # Custom Prompt
if ENV['RAILS_ENV']
name = "rails #{ENV['RAILS_ENV']}"
colors = ANSI_BACKBLUE + ANSI_YELLOW
else
name = "ruby"
colors = ANSI_BACKPURPLE + ANSI_YELLOW
end
if IRB and IRB.conf[:PROMPT]
IRB.conf[:PROMPT][:SD] = {
:PROMPT_I => ">> ", # normal prompt
:PROMPT_S => "%l> ", # string continuation
:PROMPT_C => " > ", # code continuation
:PROMPT_N => " > ", # code continuation too?
:RETURN => "#{ANSI_BOLD}# => %s #{ANSI_RESET}\n", # return value
}
IRB.conf[:PROMPT_MODE] = :SD
end
end
IRB_RC_HAS_LOADED = true
end
# A cool way to index in a hash
# h = { :alpha => 'bet', :beta => 'blocker' }
# h/:beta #=> 'blocker'
class Hash
def /(key)
self[key]
end
end
# A real array diff, not a set diff
# a1 = [1,1,2]
# a2 = [1,2]
# a1 - a2 #=> []
# a1.diff(a2) #=> [1]
class Array
def diff(other)
list = self.dup
other.each { |elem| list.delete_at( list.index(elem) ) }
list
end
end
# Where quasi global methods belong (Thanks to aperios for teaching me!)
module Kernel
# Awesome benchmarking function
# Example: bench(100) do ... end
# Source: http://ozmm.org/posts/time_in_irb.html
def time(times=1)
require "benchmark"
ret = nil
Benchmark.bm { |x| x.report { times.to_i.times { ret = yield } } }
ret
end
alias :bench :time
# Copy the last IRB result into the clipboard
# Thanks to aperios (apeiros@gmx.ne) in #ruby-lang
def pbcopy
IO.popen('pbcopy', 'w') { |io| io.write(IRB.CurrentContext.last_value.to_s) }
end
# Open in TextMate
def mate *args
flattened_args = args.map {|arg| "\"#{arg.to_s}\""}.join ' '
`mate #{flattened_args}`
nil
end
# Clear (My alias c=clear inside IRB)
def c
system('clear')
end
# Do something forever, break will exit the loop
def forever
while true
yield
end
end
# Simple regular expression helper
# show_regexp - stolen from the pickaxe
def show_regexp(a, re)
if a =~ re
"#{$`}<<#{$&}>>#{$'}"
else
"no match"
end
end
# Why's aorta method to edit an object in YAML, awesome!
# Source: http://rubyforge.org/snippet/detail.php?type=snippet&id=22
def aorta( obj )
tempfile = File.join('/tmp',"yobj_#{ Time.now.to_i }")
File.open( tempfile, 'w' ) { |f| f << obj.to_yaml }
system( "#{ ENV['EDITOR'] || 'vi' } #{ tempfile }" )
return obj unless File.exists?( tempfile )
content = YAML::load( File.open( tempfile ) )
File.delete( tempfile )
content
end
def aorta!(obj)
obj = aorta(obj)
end
end
class Object
# Print Documentation
# Example: String.ri :sub
# Source: http://github.com/ryanb/dotfiles/blob/145906d11810c691dbb1a47481d790e3ad186dcb/irbrc
def ri(method = nil)
unless method && method =~ /^[A-Z]/ # if class isn't specified
klass = self.kind_of?(Class) ? name : self.class.name
method = [klass, method].compact.join('#')
end
puts `ri '#{method}'`
end
# Methods that aren't in Object.methods
# Example: String.m
def m
self.methods - Object.methods
end
alias :new_methods :m
end
# Convenience method on Regexp so you can do
# /an/.show_match("banana") # => "b<<an>>ana"
class Regexp
def show_match(a)
show_regexp(a, self)
end
end
# Load all my non-test libraries in '~/.util/irb'
UTIL_DIR = File.expand_path('~') + '/.util/irb/*'
Dir[UTIL_DIR].each do |f|
require f unless File.basename(f) =~ /\Atest/
end
