box-ui-elements-mlh/src/components/form-elements/draft-js-mention-selector/DraftMentionDecorator.js

Box UI Elements

// @flow
import { CompositeDecorator } from 'draft-js';

import DraftMentionItem from './DraftMentionItem';

const mentionStrategy = (contentBlock, callback, contentState) => {
    contentBlock.findEntityRanges(character => {
        const entityKey = character.getEntity();

        const ret = entityKey !== null && contentState.getEntity(entityKey).getType() === 'MENTION';
        return ret;
    }, callback);
};

const DraftMentionDecorator = new CompositeDecorator([
    {
        strategy: mentionStrategy,
        component: DraftMentionItem,
    },
]);

export default DraftMentionDecorator;