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

import type React from 'react';
import { useCallback } from 'react';
import { Box } from 'ink';
import { Colors } from '../../colors.js';
import { useKeypress } from '../../hooks/useKeypress.js';
import { WelcomeStep, type WelcomeChoice } from './WelcomeStep.js';
import { ProviderSelectStep } from './ProviderSelectStep.js';
import { ModelSelectStep } from './ModelSelectStep.js';
import { AuthMethodStep } from './AuthMethodStep.js';
import { AuthenticationStep } from './AuthenticationStep.js';
import { CompletionStep } from './CompletionStep.js';
import { SkipExitStep } from './SkipExitStep.js';
import type {
  WelcomeState,
  WelcomeActions,
  ModelInfo,
} from '../../hooks/useWelcomeOnboarding.js';
import { getBorderStyle } from '../../contexts/UnicodeRenderingContext.js';

interface WelcomeDialogProps {
  state: WelcomeState;
  actions: WelcomeActions;
  availableProviders: string[];
  availableModels: ModelInfo[];
  triggerAuth: (
    provider: string,
    method: 'oauth' | 'api_key',
    apiKey?: string,
  ) => Promise<void>;
}

interface WelcomeStepRendererProps {
  state: WelcomeState;
  actions: WelcomeActions;
  availableProviders: string[];
  availableModels: ModelInfo[];
  triggerAuth: (
    provider: string,
    method: 'oauth' | 'api_key',
    apiKey?: string,
  ) => Promise<void>;
  handleWelcomeSelect: (choice: WelcomeChoice) => void;
}

const WelcomeStepRenderer: React.FC<WelcomeStepRendererProps> = ({
  state,
  actions,
  availableProviders,
  availableModels,
  triggerAuth,
  handleWelcomeSelect,
}) => {
  switch (state.step) {
    case 'welcome':
      return <WelcomeStep onSelect={handleWelcomeSelect} />;
    case 'provider':
      return (
        <ProviderSelectStep
          providers={availableProviders}
          onSelect={actions.selectProvider}
          onSkip={actions.skipSetup}
        />
      );
    case 'model':
      if (!state.selectedProvider) return null;
      return (
        <ModelSelectStep
          provider={state.selectedProvider}
          models={availableModels}
          modelsLoadStatus={state.modelsLoadStatus}
          onSelect={actions.selectModel}
          onBack={actions.goBack}
        />
      );
    case 'auth_method':
      if (!state.selectedProvider) return null;
      return (
        <AuthMethodStep
          provider={state.selectedProvider}
          onSelect={actions.selectAuthMethod}
          onBack={actions.goBack}
          error={state.error}
        />
      );
    case 'authenticating':
      if (!state.selectedProvider || !state.selectedAuthMethod) return null;
      return (
        <AuthenticationStep
          provider={state.selectedProvider}
          method={state.selectedAuthMethod}
          onComplete={actions.onAuthComplete}
          onError={actions.onAuthError}
          onBack={actions.goBack}
          triggerAuth={triggerAuth}
        />
      );
    case 'completion':
      if (!state.selectedProvider || !state.selectedAuthMethod) return null;
      return (
        <CompletionStep
          provider={state.selectedProvider}
          model={state.selectedModel}
          authMethod={state.selectedAuthMethod}
          onSaveProfile={actions.saveProfile}
          onDismiss={actions.dismiss}
        />
      );
    case 'skipped':
      return <SkipExitStep onDismiss={actions.dismiss} />;
    default:
      return null;
  }
};

export const WelcomeDialog: React.FC<WelcomeDialogProps> = ({
  state,
  actions,
  availableProviders,
  availableModels,
  triggerAuth,
}) => {
  useKeypress(
    (key) => {
      if (key.name === 'escape' && !state.authInProgress) {
        actions.skipSetup();
      }
    },
    { isActive: state.step !== 'completion' && state.step !== 'skipped' },
  );

  const handleWelcomeSelect = useCallback(
    (choice: WelcomeChoice) => {
      if (choice === 'setup') {
        actions.startSetup();
      } else {
        actions.skipSetup();
      }
    },
    [actions],
  );

  return (
    <Box
      flexDirection="column"
      borderStyle={getBorderStyle('round')}
      borderColor={Colors.AccentCyan}
      padding={1}
      width="100%"
      marginLeft={1}
    >
      <WelcomeStepRenderer
        state={state}
        actions={actions}
        availableProviders={availableProviders}
        availableModels={availableModels}
        triggerAuth={triggerAuth}
        handleWelcomeSelect={handleWelcomeSelect}
      />
    </Box>
  );
};