sentry/static/app/views/monitors/utils/useMonitorStats.tsx

import {useState} from 'react';

import type {TimeWindowConfig} from 'sentry/components/checkInTimeline/types';
import {useApiQuery} from 'sentry/utils/queryClient';
import {useLocation} from 'sentry/utils/useLocation';
import useOrganization from 'sentry/utils/useOrganization';

import type {MonitorBucket} from '../types';

interface Options {
  /**
   * The list of monitor IDs to fetch stats for
   */
  monitors: string[];
  /**
   * The window configuration object
   */
  timeWindowConfig: TimeWindowConfig;
  /**
   * Do not query stats when set to false
   */
  enabled?: boolean;
}

/**
 * Fetches Monitor stats
 */
export function useMonitorStats({monitors, timeWindowConfig, enabled = true}: Options) {
  const {start, end, rollupConfig} = timeWindowConfig;
  const [now] = useState(() => new Date().getTime() / 1000);

  const until =
    Math.floor(end.getTime() / 1000) +
    rollupConfig.underscanPeriod -
    // XXX(epurkhiser): We are dropping 1 bucket worth of data on the right
    // side to account for the fact that this bucket is actually over-scan
    // becauase the query on the backend is inclusive.
    rollupConfig.interval;

  const selectionQuery = {
    since: Math.floor(start.getTime() / 1000),
    until: Math.min(until, now),
    resolution: `${rollupConfig.interval}s`,
  };

  const organization = useOrganization();
  const location = useLocation();

  const monitorStatsQueryKey = `/organizations/${organization.slug}/monitors-stats/`;

  return useApiQuery<Record<string, MonitorBucket[]>>(
    [
      monitorStatsQueryKey,
      {
        query: {
          monitor: monitors,
          project: location.query.project,
          environment: location.query.environment,
          ...selectionQuery,
        },
      },
    ],
    {
      staleTime: 0,
      enabled: enabled && rollupConfig.totalBuckets > 0,
    }
  );
}