class Hotch

# File lib/hotch/memory.rb, line 7
def self.memory(name: $PROGRAM_NAME, aggregate: true, &block)
  memory = if aggregate
             $hotch_memory ||= Memory.new(name)
           else
             caller = Kernel.caller_locations(1).first
             name = "#{name}:#{caller.path}:#{caller.lineno}"
             Memory.new(name)
           end

  memory.report_at_exit

  if block
    memory.run(&block)
  else
    memory.start
  end
end

# File lib/hotch.rb, line 11
def initialize(name, viewer: nil, mode: :wall, filter: nil, options: {})
  @name = name
  @viewer = viewer
  @options = options
  @reports = []
  @mode = mode

  @options[:filter] = Regexp.new(filter) if filter
end

# File lib/hotch.rb, line 46
def report
  report = @reports.inject(:+) or return

  dir = Dir.mktmpdir("hotch.#{name}.")

  report_dump(report, dir, "profile.dump")
  dot = report_dot(report, dir, "profile.dot")
  svg = convert_svg(dir, dot, "profile.svg")

  @reports.clear

  return report, svg unless block_given?

  yield report, svg
end

# File lib/hotch.rb, line 62
def report_at_exit
  return if defined? @at_exit_installed

  at_exit do
    stop

    report do |_, svg|
      if viewer
        puts "Profile SVG: #{svg}"
        Kernel.system viewer, svg
      else
        puts "Profile SVG: view #{svg} # no HOTCH_VIEWER set"
      end
    end
  end

  @at_exit_installed = true
end

# File lib/hotch.rb, line 42
def results
  StackProf.results
end

# File lib/hotch.rb, line 35
def run(...)
  start(...)
  yield
ensure
  stop
end

# File lib/hotch.rb, line 21
def start(**options)
  return if StackProf.running?

  stackprof = { mode: @mode }.merge(options)
  StackProf.start(**stackprof)
end

# File lib/hotch.rb, line 28
def stop
  return unless StackProf.running?

  StackProf.stop
  @reports << StackProf::Report.new(results)
end

# File lib/hotch.rb, line 99
def convert_svg(dir, dot, file)
  svg = File.join(dir, file)
  system("dot", "-Tsvg", "-o", svg, dot) or raise "dot: command not found. Please install graphviz"
  svg
end

# File lib/hotch.rb, line 83
def name
  @name.gsub(/\W+/, "_")
end

# File lib/hotch.rb, line 93
def report_dot(report, dir, file)
  write_file(dir, file) do |fh|
    report.print_graphviz(options, fh)
  end
end

# File lib/hotch.rb, line 87
def report_dump(report, dir, file)
  write_file(dir, file) do |fh|
    report.print_dump(fh)
  end
end

# File lib/hotch.rb, line 105
def write_file(dir, file, &block)
  path = File.join(dir, file)
  File.open(path, "wb", &block)
  path
end