gemini-cli/packages/cli/src/ui/hooks/useBanner.ts

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

import { useState, useEffect, useRef } from 'react';
import { persistentState } from '../../utils/persistentState.js';
import type { Config } from '@google/gemini-cli-core';
import crypto from 'node:crypto';

const DEFAULT_MAX_BANNER_SHOWN_COUNT = 5;

interface BannerData {
  defaultText: string;
  warningText: string;
}

export function useBanner(bannerData: BannerData, config: Config) {
  const { defaultText, warningText } = bannerData;

  const [previewEnabled, setPreviewEnabled] = useState(
    config.getPreviewFeatures(),
  );

  useEffect(() => {
    const isEnabled = config.getPreviewFeatures();
    if (isEnabled !== previewEnabled) {
      setPreviewEnabled(isEnabled);
    }
  }, [config, previewEnabled]);

  const [bannerCounts] = useState(
    () => persistentState.get('defaultBannerShownCount') || {},
  );

  const hashedText = crypto
    .createHash('sha256')
    .update(defaultText)
    .digest('hex');

  const currentBannerCount = bannerCounts[hashedText] || 0;

  const showDefaultBanner =
    warningText === '' &&
    !previewEnabled &&
    currentBannerCount < DEFAULT_MAX_BANNER_SHOWN_COUNT;

  const rawBannerText = showDefaultBanner ? defaultText : warningText;
  const bannerText = rawBannerText.replace(/\\n/g, '\n');

  const lastIncrementedKey = useRef<string | null>(null);

  useEffect(() => {
    if (showDefaultBanner && defaultText) {
      if (lastIncrementedKey.current !== defaultText) {
        lastIncrementedKey.current = defaultText;

        const allCounts = persistentState.get('defaultBannerShownCount') || {};
        const current = allCounts[hashedText] || 0;

        persistentState.set('defaultBannerShownCount', {
          ...allCounts,
          [hashedText]: current + 1,
        });
      }
    }
  }, [showDefaultBanner, defaultText, hashedText]);

  return {
    bannerText,
  };
}