summaryrefslogtreecommitdiff
path: root/lib/irb/cmd/measure.rb
blob: adea540e9276d4213d3092459d2ddcb8839de4a4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
require_relative "nop"

# :stopdoc:
module IRB
  module ExtendCommand
    class Measure < Nop
      def initialize(*args)
        super(*args)
      end

      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
    end
  end
end
# :startdoc: