import { e2e } from '@grafana/e2e';

type WithGrafanaRuntime<T> = T & {
  grafanaRuntime: {
    livePerformance: {
      start: () => void;
      getStats: () => Record<string, unknown>;
    };
  };
};

const hasGrafanaRuntime = <T>(obj: T): obj is WithGrafanaRuntime<T> => {
  return 'grafanaRuntime' in obj;
};

e2e.benchmark({
  name: 'Live performance benchmarking - 4x20hz panels',
  dashboard: {
    folder: '/dashboards/live',
    delayAfterOpening: 1000,
    skipPanelValidation: true,
  },
  repeat: 10,
  duration: 120000,
  appStats: {
    startCollecting: (window) => {
      if (!hasGrafanaRuntime(window)) {
        return;
      }

      return window.grafanaRuntime.livePerformance.start();
    },
    collect: (window) => {
      if (!hasGrafanaRuntime(window)) {
        return {};
      }

      return window.grafanaRuntime.livePerformance.getStats() ?? {};
    },
  },
});