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

import { Box, Newline, Text } from 'ink';
import { useCallback } from 'react';
import { RadioButtonSelect } from '../components/shared/RadioButtonSelect.js';
import { usePrivacySettings } from '../hooks/usePrivacySettings.js';
import { CloudPaidPrivacyNotice } from './CloudPaidPrivacyNotice.js';

import { Colors } from '../colors.js';
import { useKeypress } from '../hooks/useKeypress.js';
import type { AgentClientSource } from '../cliUiRuntime.js';

interface CloudFreePrivacyNoticeProps {
  config: AgentClientSource;
  onExit: () => void;
}

const PrivacyNoticeHeader: React.FC = () => (
  <>
    <Text bold color={Colors.AccentPurple}>
      Gemini Code Assist for Individuals Privacy Notice
    </Text>
    <Newline />
    <Text color={Colors.Foreground}>
      This notice and our Privacy Policy
      <Text color={Colors.AccentBlue}>[1]</Text> describe how Gemini Code Assist
      handles your data. Please read them carefully.
    </Text>
    <Newline />
  </>
);

const PrivacyNoticeBody: React.FC = () => (
  <>
    <Text color={Colors.Foreground}>
      When you use Gemini Code Assist for individuals with LLxprt Code, Google
      collects your prompts, related code, generated output, code edits, related
      feature usage information, and your feedback to provide, improve, and
      develop Google products and services and machine learning technologies.
    </Text>
    <Newline />
    <Text color={Colors.Foreground}>
      To help with quality and improve our products (such as generative
      machine-learning models), human reviewers may read, annotate, and process
      the data collected above. We take steps to protect your privacy as part of
      this process. This includes disconnecting the data from your Google
      Account before reviewers see or annotate it, and storing those
      disconnected copies for up to 18 months. Please don&apos;t submit
      confidential information or any data you wouldn&apos;t want a reviewer to
      see or Google to use to improve our products, services and
      machine-learning technologies.
    </Text>
    <Newline />
  </>
);

const PrivacyNoticeFooter: React.FC = () => (
  <>
    <Newline />
    <Text color={Colors.Foreground}>
      <Text color={Colors.AccentBlue}>[1]</Text>{' '}
      https://policies.google.com/privacy
    </Text>
    <Newline />
    <Text color={Colors.Gray}>Press Enter to choose an option and exit.</Text>
  </>
);

const OPT_IN_ITEMS = [
  { label: 'Yes', value: true, key: 'true' },
  { label: 'No', value: false, key: 'false' },
];

export const CloudFreePrivacyNotice = ({
  config,
  onExit,
}: CloudFreePrivacyNoticeProps) => {
  const { privacyState, updateDataCollectionOptIn } =
    usePrivacySettings(config);

  const handleSelect = useCallback(
    (value: boolean) => {
      void updateDataCollectionOptIn(value);
      if (!privacyState.error) {
        onExit();
      }
    },
    [updateDataCollectionOptIn, privacyState.error, onExit],
  );

  useKeypress(
    (key) => {
      if (privacyState.error && key.name === 'escape') {
        onExit();
      }
    },
    { isActive: true },
  );

  if (privacyState.isLoading) {
    return <Text color={Colors.Gray}>Loading...</Text>;
  }

  if (privacyState.error) {
    return (
      <Box flexDirection="column" marginY={1}>
        <Text color={Colors.AccentRed}>
          Error loading Opt-in settings: {privacyState.error}
        </Text>
        <Text color={Colors.Gray}>Press Esc to exit.</Text>
      </Box>
    );
  }

  if (privacyState.isFreeTier === false) {
    return <CloudPaidPrivacyNotice onExit={onExit} />;
  }

  return (
    <Box flexDirection="column" marginY={1}>
      <PrivacyNoticeHeader />
      <PrivacyNoticeBody />
      <Box flexDirection="column">
        <Text color={Colors.Foreground}>
          Allow Google to use this data to develop and improve our products?
        </Text>
        <RadioButtonSelect
          items={OPT_IN_ITEMS}
          initialIndex={privacyState.dataCollectionOptIn === true ? 0 : 1}
          onSelect={handleSelect}
        />
      </Box>
      <PrivacyNoticeFooter />
    </Box>
  );
};