Portal/rsconcept/frontend/src/components/select/PickSubstitutions.tsx

'use client';

import { useCallback, useMemo, useState } from 'react';
import { toast } from 'react-toastify';

import BadgeConstituenta from '@/components/info/BadgeConstituenta';
import SelectConstituenta from '@/components/select/SelectConstituenta';
import DataTable, { createColumnHelper, IConditionalStyle } from '@/components/ui/DataTable';
import MiniButton from '@/components/ui/MiniButton';
import { useConceptOptions } from '@/context/ConceptOptionsContext';
import { ILibraryItem } from '@/models/library';
import { ICstSubstitute, IMultiSubstitution } from '@/models/oss';
import { ConstituentaID, IConstituenta, IRSForm } from '@/models/rsform';
import { errors } from '@/utils/labels';

import { IconAccept, IconPageLeft, IconPageRight, IconRemove, IconReplace } from '../Icons';
import NoData from '../ui/NoData';
import SelectLibraryItem from './SelectLibraryItem';

interface PickSubstitutionsProps {
  substitutions: ICstSubstitute[];
  setSubstitutions: React.Dispatch<React.SetStateAction<ICstSubstitute[]>>;
  suggestions?: ICstSubstitute[];

  prefixID: string;
  rows?: number;
  allowSelfSubstitution?: boolean;

  schemas: IRSForm[];
  filter?: (cst: IConstituenta) => boolean;
}

const columnHelper = createColumnHelper<IMultiSubstitution>();

function PickSubstitutions({
  substitutions,
  setSubstitutions,
  suggestions,
  prefixID,
  rows,
  schemas,
  filter,
  allowSelfSubstitution
}: PickSubstitutionsProps) {
  const { colors } = useConceptOptions();

  const [leftArgument, setLeftArgument] = useState<ILibraryItem | undefined>(
    schemas.length === 1 ? schemas[0] : undefined
  );
  const [rightArgument, setRightArgument] = useState<ILibraryItem | undefined>(
    schemas.length === 1 && allowSelfSubstitution ? schemas[0] : undefined
  );

  const [leftCst, setLeftCst] = useState<IConstituenta | undefined>(undefined);
  const [rightCst, setRightCst] = useState<IConstituenta | undefined>(undefined);

  const [deleteRight, setDeleteRight] = useState(true);
  const toggleDelete = () => setDeleteRight(prev => !prev);

  const [ignores, setIgnores] = useState<ICstSubstitute[]>([]);
  const filteredSuggestions = useMemo(
    () =>
      suggestions?.filter(
        item => !ignores.find(ignore => ignore.original === item.original && ignore.substitution === item.substitution)
      ) ?? [],
    [ignores, suggestions]
  );

  const getSchemaByCst = useCallback(
    (id: ConstituentaID): IRSForm | undefined => {
      for (const schema of schemas) {
        const cst = schema.cstByID.get(id);
        if (cst) {
          return schema;
        }
      }
      return undefined;
    },
    [schemas]
  );

  const getConstituenta = useCallback(
    (id: ConstituentaID): IConstituenta | undefined => {
      for (const schema of schemas) {
        const cst = schema.cstByID.get(id);
        if (cst) {
          return cst;
        }
      }
      return undefined;
    },
    [schemas]
  );

  const substitutionData: IMultiSubstitution[] = useMemo(
    () => [
      ...substitutions.map(item => ({
        original_source: getSchemaByCst(item.original)!,
        original: getConstituenta(item.original)!,
        substitution: getConstituenta(item.substitution)!,
        substitution_source: getSchemaByCst(item.substitution)!,
        is_suggestion: false
      })),
      ...filteredSuggestions.map(item => ({
        original_source: getSchemaByCst(item.original)!,
        original: getConstituenta(item.original)!,
        substitution: getConstituenta(item.substitution)!,
        substitution_source: getSchemaByCst(item.substitution)!,
        is_suggestion: true
      }))
    ],
    [getConstituenta, getSchemaByCst, substitutions, filteredSuggestions]
  );

  function addSubstitution() {
    if (!leftCst || !rightCst) {
      return;
    }
    const newSubstitution: ICstSubstitute = {
      original: deleteRight ? rightCst.id : leftCst.id,
      substitution: deleteRight ? leftCst.id : rightCst.id
    };
    const toDelete = substitutions.map(item => item.original);
    const replacements = substitutions.map(item => item.substitution);
    if (
      toDelete.includes(newSubstitution.original) ||
      toDelete.includes(newSubstitution.substitution) ||
      replacements.includes(newSubstitution.original)
    ) {
      toast.error(errors.reuseOriginal);
      return;
    }
    if (leftArgument === rightArgument) {
      if ((deleteRight && rightCst?.is_inherited) || (!deleteRight && leftCst?.is_inherited)) {
        toast.error(errors.substituteInherited);
        return;
      }
    }
    setSubstitutions(prev => [...prev, newSubstitution]);
    setLeftCst(undefined);
    setRightCst(undefined);
  }

  const handleDeclineSuggestion = useCallback(
    (item: IMultiSubstitution) => {
      setIgnores(prev => [...prev, { original: item.original.id, substitution: item.substitution.id }]);
    },
    [setIgnores]
  );

  const handleAcceptSuggestion = useCallback(
    (item: IMultiSubstitution) => {
      setSubstitutions(prev => [...prev, { original: item.original.id, substitution: item.substitution.id }]);
    },
    [setSubstitutions]
  );

  const handleDeleteSubstitution = useCallback(
    (target: IMultiSubstitution) => {
      handleDeclineSuggestion(target);
      setSubstitutions(prev => {
        const newItems: ICstSubstitute[] = [];
        prev.forEach(item => {
          if (item.original !== target.original.id || item.substitution !== target.substitution.id) {
            newItems.push(item);
          }
        });
        return newItems;
      });
    },
    [setSubstitutions, handleDeclineSuggestion]
  );

  const columns = useMemo(
    () => [
      columnHelper.accessor(item => item.substitution_source.alias, {
        id: 'left_schema',
        size: 100,
        cell: props => <div className='min-w-[10.5rem] text-ellipsis text-left'>{props.getValue()}</div>
      }),
      columnHelper.accessor(item => item.substitution.alias, {
        id: 'left_alias',
        size: 65,
        cell: props => (
          <BadgeConstituenta
            theme={colors}
            value={props.row.original.substitution}
            prefixID={`${prefixID}_${props.row.index}_1_`}
          />
        )
      }),
      columnHelper.display({
        id: 'status',
        size: 0,
        cell: () => <IconPageRight size='1.2rem' />
      }),
      columnHelper.accessor(item => item.original.alias, {
        id: 'right_alias',
        size: 65,
        cell: props => (
          <BadgeConstituenta
            theme={colors}
            value={props.row.original.original}
            prefixID={`${prefixID}_${props.row.index}_2_`}
          />
        )
      }),
      columnHelper.accessor(item => item.original_source.alias, {
        id: 'right_schema',
        size: 100,
        cell: props => <div className='min-w-[8rem] text-ellipsis text-right'>{props.getValue()}</div>
      }),
      columnHelper.display({
        id: 'actions',
        size: 0,
        cell: props =>
          props.row.original.is_suggestion ? (
            <div className='max-w-fit'>
              <MiniButton
                noHover
                title='Принять предложение'
                icon={<IconAccept size='1rem' className='icon-green' />}
                onClick={() => handleAcceptSuggestion(props.row.original)}
              />
              <MiniButton
                noHover
                title='Игнорировать предложение'
                icon={<IconRemove size='1rem' className='icon-red' />}
                onClick={() => handleDeclineSuggestion(props.row.original)}
              />
            </div>
          ) : (
            <div className='max-w-fit'>
              <MiniButton
                noHover
                title='Удалить'
                icon={<IconRemove size='1rem' className='icon-red' />}
                onClick={() => handleDeleteSubstitution(props.row.original)}
              />
            </div>
          )
      })
    ],
    [handleDeleteSubstitution, handleDeclineSuggestion, handleAcceptSuggestion, colors, prefixID]
  );

  const conditionalRowStyles = useMemo(
    (): IConditionalStyle<IMultiSubstitution>[] => [
      {
        when: (item: IMultiSubstitution) => item.is_suggestion,
        style: {
          backgroundColor: colors.bgOrange50
        }
      }
    ],
    [colors]
  );

  return (
    <div className='flex flex-col'>
      <div className='flex items-end gap-3 justify-stretch'>
        <div className='flex-grow flex flex-col basis-1/2 gap-[0.125rem] border-x border-t clr-input'>
          <SelectLibraryItem
            noBorder
            placeholder='Выберите аргумент'
            items={allowSelfSubstitution ? schemas : schemas.filter(item => item.id !== rightArgument?.id)}
            value={leftArgument}
            onSelectValue={setLeftArgument}
          />
          <SelectConstituenta
            noBorder
            items={(leftArgument as IRSForm)?.items.filter(
              cst => !substitutions.find(item => item.original === cst.id) && (!filter || filter(cst))
            )}
            value={leftCst}
            onSelectValue={setLeftCst}
          />
        </div>
        <div className='flex flex-col gap-1'>
          <MiniButton
            title={deleteRight ? 'Заменить правую' : 'Заменить левую'}
            onClick={toggleDelete}
            icon={
              deleteRight ? (
                <IconPageRight size='1.5rem' className='clr-text-primary' />
              ) : (
                <IconPageLeft size='1.5rem' className='clr-text-primary' />
              )
            }
          />
          <MiniButton
            title='Добавить в таблицу отождествлений'
            className='mb-[0.375rem] grow-0'
            icon={<IconReplace size='1.5rem' className='icon-primary' />}
            disabled={!leftCst || !rightCst || leftCst === rightCst}
            onClick={addSubstitution}
          />
        </div>

        <div className='flex-grow basis-1/2 flex flex-col gap-[0.125rem] border-x border-t clr-input'>
          <SelectLibraryItem
            noBorder
            placeholder='Выберите аргумент'
            items={allowSelfSubstitution ? schemas : schemas.filter(item => item.id !== leftArgument?.id)}
            value={rightArgument}
            onSelectValue={setRightArgument}
          />
          <SelectConstituenta
            noBorder
            items={(rightArgument as IRSForm)?.items.filter(
              cst => !substitutions.find(item => item.original === cst.id) && (!filter || filter(cst))
            )}
            value={rightCst}
            onSelectValue={setRightCst}
          />
        </div>
      </div>

      <DataTable
        dense
        noHeader
        noFooter
        className='text-sm border select-none cc-scroll-y'
        rows={rows}
        contentHeight='1.3rem'
        data={substitutionData}
        columns={columns}
        headPosition='0'
        noDataComponent={
          <NoData className='min-h-[2rem]'>
            <p>Список пуст</p>
            <p>Добавьте отождествление</p>
          </NoData>
        }
        conditionalRowStyles={conditionalRowStyles}
      />
    </div>
  );
}

export default PickSubstitutions;