class Selenium::DevTools::V85::HeapProfiler

def add_inspected_heap_object(heap_object_id:)
  @devtools.send_cmd('HeapProfiler.addInspectedHeapObject',
                     heapObjectId: heap_object_id)
end

def collect_garbage
  @devtools.send_cmd('HeapProfiler.collectGarbage')
end

def disable
  @devtools.send_cmd('HeapProfiler.disable')
end

def enable
  @devtools.send_cmd('HeapProfiler.enable')
end

def get_heap_object_id(object_id:)
  @devtools.send_cmd('HeapProfiler.getHeapObjectId',
                     objectId: object_id)
end

def get_object_by_heap_object_id(object_id:, object_group: nil)
  @devtools.send_cmd('HeapProfiler.getObjectByHeapObjectId',
                     objectId: object_id,
                     objectGroup: object_group)
end

def get_sampling_profile
  @devtools.send_cmd('HeapProfiler.getSamplingProfile')
end

def initialize(devtools)
  @devtools = devtools
end

def on(event, &block)
  event = EVENTS[event] if event.is_a?(Symbol)
  @devtools.callbacks["HeapProfiler.#{event}"] << block
end

def start_sampling(sampling_interval: nil)
  @devtools.send_cmd('HeapProfiler.startSampling',
                     samplingInterval: sampling_interval)
end

def start_tracking_heap_objects(track_allocations: nil)
  @devtools.send_cmd('HeapProfiler.startTrackingHeapObjects',
                     trackAllocations: track_allocations)
end

def stop_sampling
  @devtools.send_cmd('HeapProfiler.stopSampling')
end

def stop_tracking_heap_objects(report_progress: nil, treat_global_objects_as_roots: nil)
  @devtools.send_cmd('HeapProfiler.stopTrackingHeapObjects',
                     reportProgress: report_progress,
                     treatGlobalObjectsAsRoots: treat_global_objects_as_roots)
end

def take_heap_snapshot(report_progress: nil, treat_global_objects_as_roots: nil)
  @devtools.send_cmd('HeapProfiler.takeHeapSnapshot',
                     reportProgress: report_progress,
                     treatGlobalObjectsAsRoots: treat_global_objects_as_roots)
end