class IRB::ExtendCommand::Measure

def execute(type = nil, arg = nil, &block)
  # Please check IRB.init_config in lib/irb/init.rb that sets
  # IRB.conf[:MEASURE_PROC] to register default "measure" methods,
  # "measure :time" (abbreviated as "measure") and "measure :stackprof".
  case type
  when :off
    IRB.conf[:MEASURE] = nil
    IRB.unset_measure_callback(arg)
  when :list
    IRB.conf[:MEASURE_CALLBACKS].each do |type_name, _, arg_val|
      puts "- #{type_name}" + (arg_val ? "(#{arg_val.inspect})" : '')
    end
  when :on
    IRB.conf[:MEASURE] = true
    added = IRB.set_measure_callback(type, arg)
    puts "#{added[0]} is added." if added
  else
    if block_given?
      IRB.conf[:MEASURE] = true
      added = IRB.set_measure_callback(&block)
      puts "#{added[0]} is added." if added
    else
      IRB.conf[:MEASURE] = true
      added = IRB.set_measure_callback(type, arg)
      puts "#{added[0]} is added." if added
    end
  end
  nil
end

def initialize(*args)
  super(*args)
end