commandStringExtraction.ts

import type { Middleware } from '../../../middleware';
import type { TextComposerMiddlewareExecutorState } from './TextComposerMiddlewareExecutor';
import { stripTriggerFromText } from './textMiddlewareUtils';
import type { CommandSuggestion } from './types';

export type ApplyCommandSettingsMiddleware = Middleware<
  TextComposerMiddlewareExecutorState<CommandSuggestion>,
  'onChange' | 'onSuggestionItemSelect'
>;

export const createCommandStringExtractionMiddleware =
  (): ApplyCommandSettingsMiddleware => ({
    handlers: {
      onChange: ({ complete, forward, state }) => {
        const { command } = state;

        if (!command) {
          return forward();
        }

        const triggerWithCommand = `/${command.name}`;

        const newText = stripTriggerFromText(state.text, triggerWithCommand);

        return complete({
          ...state,
          selection: {
            end: state.selection.end - triggerWithCommand.length,
            start: state.selection.start - triggerWithCommand.length,
          },
          text: newText,
        });
      },
      onSuggestionItemSelect: ({ next, forward, state }) => {
        const { command } = state;

        if (!command) {
          return forward();
        }

        const triggerWithCommand = `/${command?.name} `;

        const newText = state.text.slice(triggerWithCommand.length);
        return next({
          ...state,
          selection: {
            end: state.selection.end - triggerWithCommand.length,
            start: state.selection.start - triggerWithCommand.length,
          },
          text: newText,
        });
      },
    },
    id: 'stream-io/text-composer/command-string-extraction',
  });