class RubyProf::FlameGraphPrinter

def build_flame_data(call_tree, visited = Set.new)
  node = {
    name: call_tree.target.full_name,
    value: call_tree.total_time,
    self_value: call_tree.self_time,
    called: call_tree.called,
    children: []
  }
  unless visited.include?(call_tree.target)
    visited.add(call_tree.target)
    call_tree.children.sort_by { |c| -c.total_time }.each do |child|
      node[:children] << build_flame_data(child, visited)
    end
    visited.delete(call_tree.target)
  end
  node
end

def flame_data_json
  threads = @result.threads.map do |thread|
    {
      id: thread.id,
      fiber_id: thread.fiber_id,
      total_time: thread.total_time,
      data: build_flame_data(thread.call_tree)
    }
  end
  JSON.generate(threads)
end

def print(output = STDOUT, title: "ruby-prof flame graph",
          min_percent: 0, max_percent: 100, filter_by: :self_time, sort_method: nil, **)
  @min_percent = min_percent
  @max_percent = max_percent
  @filter_by = filter_by
  @sort_method = sort_method
  @title = title
  output << ERB.new(self.template).result(binding)
end

def template
  open_asset('flame_graph_printer.html.erb')
end