Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Draft Js Mentions
import { convertFromRaw, convertToRaw, ContentState } from 'draft-js';
const getIndicesOf = (searchStr, str, caseSensitive) => {
let tempStr = str;
let tempSearchStr = searchStr;
const searchStrLen = tempSearchStr.length;
if (searchStrLen === 0) {
return [];
}
let startIndex = 0;
let index;
const indices = [];
if (!caseSensitive) {
tempStr = tempStr.toLowerCase();
tempSearchStr = tempSearchStr.toLowerCase();
}
while ((index = tempStr.indexOf(tempSearchStr, startIndex)) > -1) {
indices.push(index);
startIndex = index + searchStrLen;
}
return indices;
};
const getEntityRanges = (text, mentionName, mentionKey) => {
const indices = getIndicesOf(mentionName, text);
if (indices.length > 0) {
return indices.map(offset => ({
key: mentionKey,
length: mentionName.length,
offset
}));
}
return null;
};
const createMentionEntities = (text, tags) => {
const rawContent = convertToRaw(ContentState.createFromText(text));
const rawState = tags.map(tag => ({
type: 'mention',
mutability: 'IMMUTABLE',
data: { id: tag.id, name: tag.name }
}));
rawContent.entityMap = [...rawState];
rawContent.blocks = rawContent.blocks.map(block => {
const ranges = [];
tags.forEach((tag, index) => {
const entityRanges = getEntityRanges(block.text, tag.name, index);
if (entityRanges) {
ranges.push(...entityRanges);
}
});
return { ...block, entityRanges: ranges };
});
return convertFromRaw(rawContent);
};
export default createMentionEntities;
@ramisalem

This comment has been minimized.

Copy link

commented Oct 19, 2018

Hey there, Do you have any resources on how to create a custom entity I am really confused I could not understand the concept from the draft docs

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.