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

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

import React from 'react';
import { Box, Text } from 'ink';
import Gradient from 'ink-gradient';
import { theme } from '../semantic-colors.js';
import { shortAsciiLogo, longAsciiLogo, tinyAsciiLogo } from './AsciiArt.js';
import { getAsciiArtWidth } from '../utils/textUtils.js';
import { useTerminalSize } from '../hooks/useTerminalSize.js';

interface HeaderProps {
  customAsciiArt?: string; // For user-defined ASCII art
  version: string;
  nightly: boolean;
}

const GradientText: React.FC<{ children: React.ReactNode }> = ({
  children,
}) => {
  const textElement = <Text color={theme.text.primary}>{children}</Text>;
  if (theme.ui.gradient && theme.ui.gradient.length > 0) {
    return <Gradient colors={theme.ui.gradient}>{textElement}</Gradient>;
  }
  return textElement;
};

export const Header: React.FC<HeaderProps> = ({
  customAsciiArt,
  version,
  nightly,
}) => {
  const { columns: terminalWidth } = useTerminalSize();
  let displayTitle;
  const widthOfLongLogo = getAsciiArtWidth(longAsciiLogo);
  const widthOfShortLogo = getAsciiArtWidth(shortAsciiLogo);

  if (customAsciiArt) {
    displayTitle = customAsciiArt;
  } else if (terminalWidth >= widthOfLongLogo) {
    displayTitle = longAsciiLogo;
  } else if (terminalWidth >= widthOfShortLogo) {
    displayTitle = shortAsciiLogo;
  } else {
    displayTitle = tinyAsciiLogo;
  }

  const artWidth = getAsciiArtWidth(displayTitle);

  return (
    <Box
      alignItems="flex-start"
      width={artWidth}
      flexShrink={0}
      flexDirection="column"
    >
      <Box>
        <GradientText>{displayTitle}</GradientText>
      </Box>
      {nightly && (
        <Box width="100%" flexDirection="row" justifyContent="flex-end">
          <Text color={theme.text.primary}>v{version}</Text>
        </Box>
      )}
    </Box>
  );
};