Method: IRB.set_measure_callback

Defined in:

lib/irb/init.rb

.set_measure_callback(type = nil, arg = nil, &block) ⇒ Object

# File 'lib/irb/init.rb', line 163

def IRB.set_measure_callback(type = nil, arg = nil, &block)
  added = nil
  if type
    type_sym = type.upcase.to_sym
    if IRB.conf[:MEASURE_PROC][type_sym]
      added = [type_sym, IRB.conf[:MEASURE_PROC][type_sym], arg]
    end
  elsif IRB.conf[:MEASURE_PROC][:CUSTOM]
    added = [:CUSTOM, IRB.conf[:MEASURE_PROC][:CUSTOM], arg]
  elsif block_given?
    added = [:BLOCK, block, arg]
    found = IRB.conf[:MEASURE_CALLBACKS].find{ |m| m[0] == added[0] && m[2] == added[2] }
    if found
      found[1] = block
      return added
    else
      IRB.conf[:MEASURE_CALLBACKS] << added
      return added
    end
  else
    added = [:TIME, IRB.conf[:MEASURE_PROC][:TIME], arg]
  end
  if added
    found = IRB.conf[:MEASURE_CALLBACKS].find{ |m| m[0] == added[0] && m[2] == added[2] }
    if found
      # already added
      nil
    else
      IRB.conf[:MEASURE_CALLBACKS] << added if added
      added
    end
  else
    nil
  end
end