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

import {
  type ConfirmationRequest,
  type HistoryItemWithoutId,
  type PermissionConfirmationRequest,
} from '../types.js';
import { type ReactNode } from 'react';
import { type RunEventNotificationEvent } from '../../utils/terminalNotifications.js';
import { getConfirmingToolState } from './confirmingTool.js';

export interface PendingAttentionNotification {
  key: string;
  event: RunEventNotificationEvent;
}

function keyFromReactNode(node: ReactNode): string {
  if (typeof node === 'string' || typeof node === 'number') {
    return String(node);
  }
  if (Array.isArray(node)) {
    return node.map((item) => keyFromReactNode(item)).join('|');
  }
  return 'react-node';
}

export function getPendingAttentionNotification(
  pendingHistoryItems: HistoryItemWithoutId[],
  commandConfirmationRequest: ConfirmationRequest | null,
  authConsentRequest: ConfirmationRequest | null,
  permissionConfirmationRequest: PermissionConfirmationRequest | null,
  hasConfirmUpdateExtensionRequests: boolean,
  hasLoopDetectionConfirmationRequest: boolean,
): PendingAttentionNotification | null {
  const confirmingToolState = getConfirmingToolState(pendingHistoryItems);
  if (confirmingToolState) {
    const details = confirmingToolState.tool.confirmationDetails;
    if (details?.type === 'ask_user') {
      const firstQuestion = details.questions.at(0)?.header;
      return {
        key: `ask_user:${confirmingToolState.tool.callId}`,
        event: {
          type: 'attention',
          heading: 'Answer requested by agent',
          detail: firstQuestion || 'The agent needs your response to continue.',
        },
      };
    }

    const toolTitle = details?.title || confirmingToolState.tool.description;
    return {
      key: `tool_confirmation:${confirmingToolState.tool.callId}`,
      event: {
        type: 'attention',
        heading: 'Approval required',
        detail: toolTitle
          ? `Approve tool action: ${toolTitle}`
          : 'Approve a pending tool action to continue.',
      },
    };
  }

  if (commandConfirmationRequest) {
    const promptKey = keyFromReactNode(commandConfirmationRequest.prompt);
    return {
      key: `command_confirmation:${promptKey}`,
      event: {
        type: 'attention',
        heading: 'Confirmation required',
        detail: 'A command is waiting for your confirmation.',
      },
    };
  }

  if (authConsentRequest) {
    const promptKey = keyFromReactNode(authConsentRequest.prompt);
    return {
      key: `auth_consent:${promptKey}`,
      event: {
        type: 'attention',
        heading: 'Authentication confirmation required',
        detail: 'Authentication is waiting for your confirmation.',
      },
    };
  }

  if (permissionConfirmationRequest) {
    const filesKey = permissionConfirmationRequest.files.join('|');
    return {
      key: `filesystem_permission_confirmation:${filesKey}`,
      event: {
        type: 'attention',
        heading: 'Filesystem permission required',
        detail: 'Read-only path access is waiting for your confirmation.',
      },
    };
  }

  if (hasConfirmUpdateExtensionRequests) {
    return {
      key: 'extension_update_confirmation',
      event: {
        type: 'attention',
        heading: 'Extension update confirmation required',
        detail: 'An extension update is waiting for your confirmation.',
      },
    };
  }

  if (hasLoopDetectionConfirmationRequest) {
    return {
      key: 'loop_detection_confirmation',
      event: {
        type: 'attention',
        heading: 'Loop detection confirmation required',
        detail: 'A loop detection prompt is waiting for your response.',
      },
    };
  }

  return null;
}