import {ChatAdapterExtras} from '@shared/types/adapters/chat/chatAdapterExtras';
import {ChatSegment} from '@shared/types/chatSegment/chatSegment';
import {chatSegmentsToChatItems} from '@shared/utils/chat/chatSegmentsToChatItems';
import {useMemo} from 'react';
import {HistoryPayloadSize} from '../../../../../js/core/src';
import {reactPropsToCorePropsInEvents} from '../../utils/reactPropsToCorePropsInEvents';
import {AiChatProps} from '../props';

export const useAdapterExtras = <AiMsg>(
    aiChatProps: AiChatProps<AiMsg>,
    chatSegments: ChatSegment<AiMsg>[],
    historyPayloadSize?: HistoryPayloadSize,
): ChatAdapterExtras<AiMsg> => {
    return useMemo(() => {
        const allHistory = chatSegmentsToChatItems(chatSegments);
        const conversationHistory = (historyPayloadSize === 'max' || historyPayloadSize === undefined)
            ? allHistory
            : (historyPayloadSize > 0)
                ? allHistory.slice(-historyPayloadSize)
                : undefined;

        return {
            aiChatProps: reactPropsToCorePropsInEvents<AiMsg>(aiChatProps),
            conversationHistory,
        };

    }, [aiChatProps, chatSegments, historyPayloadSize]);
};