gemini-cli/packages/cli/src/ui/components/QuotaDisplay.tsx

/**
 * @license
 * Copyright 2026 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import type React from 'react';
import { Text } from 'ink';
import {
  getStatusColor,
  QUOTA_THRESHOLD_HIGH,
  QUOTA_THRESHOLD_MEDIUM,
} from '../utils/displayUtils.js';
import { formatResetTime } from '../utils/formatters.js';

interface QuotaDisplayProps {
  remaining: number | undefined;
  limit: number | undefined;
  resetTime?: string;
  terse?: boolean;
  forceShow?: boolean;
  lowercase?: boolean;
}

export const QuotaDisplay: React.FC<QuotaDisplayProps> = ({
  remaining,
  limit,
  resetTime,
  terse = false,
  forceShow = false,
  lowercase = false,
}) => {
  if (remaining === undefined || limit === undefined || limit === 0) {
    return null;
  }

  const percentage = (remaining / limit) * 100;

  if (!forceShow && percentage > QUOTA_THRESHOLD_HIGH) {
    return null;
  }

  const color = getStatusColor(percentage, {
    green: QUOTA_THRESHOLD_HIGH,
    yellow: QUOTA_THRESHOLD_MEDIUM,
  });

  const resetInfo =
    !terse && resetTime ? `, ${formatResetTime(resetTime)}` : '';

  if (remaining === 0) {
    let text = terse
      ? 'Limit reached'
      : `/stats Limit reached${resetInfo}${!terse && '. /auth to continue.'}`;
    if (lowercase) text = text.toLowerCase();
    return <Text color={color}>{text}</Text>;
  }

  let text = terse
    ? `${percentage.toFixed(0)}%`
    : `/stats ${percentage.toFixed(0)}% usage remaining${resetInfo}`;
  if (lowercase) text = text.toLowerCase();

  return <Text color={color}>{text}</Text>;
};