import React, { useRef, useState, ReactNode } from 'react';
import { Box } from '@mui/material';
import { FormikProps, FormikValues } from 'formik';
import ToolOptions, { GetGroupsType } from '@components/options/ToolOptions';
import ToolInputAndResult from '@components/ToolInputAndResult';
import ToolInfo from '@components/ToolInfo';
import Separator from '@components/Separator';
import ToolExamples, {
  CardExampleType
} from '@components/examples/ToolExamples';
import { ToolComponentProps } from '@tools/defineTool';

interface ToolContentPropsBase<T, I> extends ToolComponentProps {
  // Input/Output components
  inputComponent: ReactNode;
  resultComponent: ReactNode;

  // Tool options
  initialValues: T;
  getGroups: GetGroupsType<T> | null;

  // Computation function
  compute: (optionsValues: T, input: I) => void;

  // Tool info (optional)
  toolInfo?: {
    title: string;
    description: string;
  };

  // Input value to pass to the compute function
  input: I;

  // Validation schema (optional)
  validationSchema?: any;
}

interface ToolContentPropsWithExamples<T, I>
  extends ToolContentPropsBase<T, I> {
  exampleCards: CardExampleType<T>[];
  setInput: React.Dispatch<React.SetStateAction<I>>;
}

interface ToolContentPropsWithoutExamples<T, I>
  extends ToolContentPropsBase<T, I> {
  exampleCards?: never;
  setInput?: never;
}

type ToolContentProps<T, I> =
  | ToolContentPropsWithExamples<T, I>
  | ToolContentPropsWithoutExamples<T, I>;

export default function ToolContent<T extends FormikValues, I>({
  title,
  inputComponent,
  resultComponent,
  initialValues,
  getGroups,
  compute,
  toolInfo,
  exampleCards,
  input,
  setInput,
  validationSchema
}: ToolContentProps<T, I>) {
  const formRef = useRef<FormikProps<T>>(null);

  return (
    <Box>
      <ToolInputAndResult input={inputComponent} result={resultComponent} />

      <ToolOptions
        formRef={formRef}
        compute={compute}
        getGroups={getGroups}
        initialValues={initialValues}
        input={input}
        validationSchema={validationSchema}
      />

      {toolInfo && (
        <ToolInfo title={toolInfo.title} description={toolInfo.description} />
      )}

      {exampleCards && exampleCards.length > 0 && (
        <>
          <Separator backgroundColor="#5581b5" margin="50px" />
          <ToolExamples
            title={title}
            exampleCards={exampleCards}
            getGroups={getGroups}
            formRef={formRef}
            setInput={setInput}
          />
        </>
      )}
    </Box>
  );
}