mirror of https://github.com/vector-im/riot-web
140 lines
4.9 KiB
JavaScript
140 lines
4.9 KiB
JavaScript
/*
|
|
Copyright 2016 Aviral Dasgupta
|
|
Copyright 2017 Vector Creations Ltd
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
import React from 'react';
|
|
import { _t } from '../languageHandler';
|
|
import AutocompleteProvider from './AutocompleteProvider';
|
|
import {emojioneList, shortnameToImage, shortnameToUnicode, asciiRegexp, unicodeRegexp} from 'emojione';
|
|
import FuzzyMatcher from './FuzzyMatcher';
|
|
import sdk from '../index';
|
|
import {PillCompletion} from './Components';
|
|
import type {SelectionRange, Completion} from './Autocompleter';
|
|
import _uniq from 'lodash/uniq';
|
|
import _sortBy from 'lodash/sortBy';
|
|
|
|
import EmojiData from '../stripped-emoji.json';
|
|
|
|
const LIMIT = 20;
|
|
const CATEGORY_ORDER = [
|
|
'people',
|
|
'food',
|
|
'objects',
|
|
'activity',
|
|
'nature',
|
|
'travel',
|
|
'flags',
|
|
'regional',
|
|
'symbols',
|
|
'modifier',
|
|
];
|
|
|
|
// Match for ":wink:" or ascii-style ";-)" provided by emojione
|
|
// (^|\s|(emojiUnicode)) to make sure we're either at the start of the string or there's a
|
|
// whitespace character or an emoji before the emoji. The reason for unicodeRegexp is
|
|
// that we need to support inputting multiple emoji with no space between them.
|
|
const EMOJI_REGEX = new RegExp('(?:^|\\s|' + unicodeRegexp + ')(' + asciiRegexp + '|:\\w*:?)$', 'g');
|
|
|
|
// We also need to match the non-zero-length prefixes to remove them from the final match,
|
|
// and update the range so that we don't replace the whitespace or the previous emoji.
|
|
const MATCH_PREFIX_REGEX = new RegExp('(\\s|' + unicodeRegexp + ')');
|
|
|
|
const EMOJI_SHORTNAMES = Object.keys(EmojiData).map((key) => EmojiData[key]).sort(
|
|
(a, b) => {
|
|
if (a.category === b.category) {
|
|
return a.emoji_order - b.emoji_order;
|
|
}
|
|
return CATEGORY_ORDER.indexOf(a.category) - CATEGORY_ORDER.indexOf(b.category);
|
|
},
|
|
).map((a, index) => {
|
|
return {
|
|
name: a.name,
|
|
shortname: a.shortname,
|
|
aliases_ascii: a.aliases_ascii ? a.aliases_ascii.join(' ') : '',
|
|
// Include the index so that we can preserve the original order
|
|
_orderBy: index,
|
|
};
|
|
});
|
|
|
|
let instance = null;
|
|
|
|
export default class EmojiProvider extends AutocompleteProvider {
|
|
constructor() {
|
|
super(EMOJI_REGEX);
|
|
this.matcher = new FuzzyMatcher(EMOJI_SHORTNAMES, {
|
|
keys: ['aliases_ascii', 'shortname'],
|
|
// For matching against ascii equivalents
|
|
shouldMatchWordsOnly: false,
|
|
});
|
|
this.nameMatcher = new FuzzyMatcher(EMOJI_SHORTNAMES, {
|
|
keys: ['name'],
|
|
// For removing punctuation
|
|
shouldMatchWordsOnly: true,
|
|
});
|
|
}
|
|
|
|
async getCompletions(query: string, selection: SelectionRange) {
|
|
const EmojiText = sdk.getComponent('views.elements.EmojiText');
|
|
|
|
let completions = [];
|
|
const {command, range} = this.getCurrentCommand(query, selection);
|
|
if (command) {
|
|
let matchedString = command[0];
|
|
|
|
// Remove prefix of any length (single whitespace or unicode emoji)
|
|
const prefixMatch = MATCH_PREFIX_REGEX.exec(matchedString);
|
|
if (prefixMatch) {
|
|
matchedString = matchedString.slice(prefixMatch[0].length);
|
|
range.start += prefixMatch[0].length;
|
|
}
|
|
completions = this.matcher.match(matchedString);
|
|
|
|
// Do second match with shouldMatchWordsOnly in order to match against 'name'
|
|
completions = completions.concat(this.nameMatcher.match(matchedString));
|
|
// Reinstate original order
|
|
completions = _sortBy(_uniq(completions), '_orderBy');
|
|
completions = completions.map((result) => {
|
|
const {shortname} = result;
|
|
const unicode = shortnameToUnicode(shortname);
|
|
return {
|
|
completion: unicode,
|
|
component: (
|
|
<PillCompletion title={shortname} initialComponent={<EmojiText style={{maxWidth: '1em'}}>{unicode}</EmojiText>} />
|
|
),
|
|
range,
|
|
};
|
|
}).slice(0, LIMIT);
|
|
}
|
|
return completions;
|
|
}
|
|
|
|
getName() {
|
|
return '😃 ' + _t('Emoji');
|
|
}
|
|
|
|
static getInstance() {
|
|
if (instance == null)
|
|
{instance = new EmojiProvider();}
|
|
return instance;
|
|
}
|
|
|
|
renderCompletions(completions: [React.Component]): ?React.Component {
|
|
return <div className="mx_Autocomplete_Completion_container_pill">
|
|
{completions}
|
|
</div>;
|
|
}
|
|
}
|