Use prefix filter to match search query

arisgk · Sep 16, 2017 · 1ea98ac · 1ea98ac
1 parent 3422b11
commit 1ea98ac
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 6 deletions.
diff --git a/src/components/Home/AutocompleteEditor/AutocompleteEditor.jsx b/src/components/Home/AutocompleteEditor/AutocompleteEditor.jsx
@@ -9,18 +9,21 @@ import './styles/autocomplete.css';
 const mentionPlugin = createAutocompletePlugin({
   mentionPrefix: '@',
   mentionTrigger: '@',
+  entityMutability: 'IMMUTABLE',
 });
 const MentionSuggestions = mentionPlugin.MentionSuggestions;
 
 const hashtagPlugin = createAutocompletePlugin({
   mentionPrefix: '#',
   mentionTrigger: '#',
+  entityMutability: 'IMMUTABLE',
 });
 const HashtagSuggestions = hashtagPlugin.MentionSuggestions;
 
 const relationPlugin = createAutocompletePlugin({
   mentionPrefix: '<>',
   mentionTrigger: '<>',
+  entityMutability: 'IMMUTABLE',
 });
 const RelationSuggestions = relationPlugin.MentionSuggestions;
 

diff --git a/src/selectors/hashtags/index.js b/src/selectors/hashtags/index.js
@@ -1,10 +1,10 @@
 import { createSelector } from 'reselect';
-import { defaultSuggestionsFilter } from 'draft-js-mention-plugin';
+import prefixFilter from '../../utils/suggestions/prefixFilter';
 
 const getEntities = state => state.home.entities.hashtags;
 const getSearch = state => state.home.ui.hashtags.search;
 
 export const getSuggestions = createSelector(
   [getEntities, getSearch],
-  (entities, search) => defaultSuggestionsFilter(search, entities),
+  (entities, search) => prefixFilter(search, entities, 7),
 );
diff --git a/src/selectors/mentions/index.js b/src/selectors/mentions/index.js
@@ -1,10 +1,10 @@
 import { createSelector } from 'reselect';
-import { defaultSuggestionsFilter } from 'draft-js-mention-plugin';
+import prefixFilter from '../../utils/suggestions/prefixFilter';
 
 const getEntities = state => state.home.entities.mentions;
 const getSearch = state => state.home.ui.mentions.search;
 
 export const getSuggestions = createSelector(
   [getEntities, getSearch],
-  (entities, search) => defaultSuggestionsFilter(search, entities),
+  (entities, search) => prefixFilter(search, entities, 7),
 );
diff --git a/src/selectors/relations/index.js b/src/selectors/relations/index.js
@@ -1,10 +1,10 @@
 import { createSelector } from 'reselect';
-import { defaultSuggestionsFilter } from 'draft-js-mention-plugin';
+import prefixFilter from '../../utils/suggestions/prefixFilter';
 
 const getEntities = state => state.home.entities.relations;
 const getSearch = state => state.home.ui.relations.search;
 
 export const getSuggestions = createSelector(
   [getEntities, getSearch],
-  (entities, search) => defaultSuggestionsFilter(search, entities),
+  (entities, search) => prefixFilter(search, entities, 7),
 );
diff --git a/src/utils/suggestions/prefixFilter.js b/src/utils/suggestions/prefixFilter.js
@@ -0,0 +1,18 @@
+const size = list => (list.constructor.name === 'List'
+  ? list.size
+  : list.length);
+
+const get = (obj, attr) => (obj.get
+  ? obj.get(attr)
+  : obj[attr]);
+
+const filter = (searchValue, suggestions, maxResults) => {
+  const value = searchValue.toLowerCase();
+  const filteredSuggestions = suggestions.filter(suggestion => (
+    !value || get(suggestion, 'name').toLowerCase().startsWith(value)
+  ));
+  const length = size(filteredSuggestions) < maxResults ? size(filteredSuggestions) : maxResults;
+  return filteredSuggestions.slice(0, length);
+};
+
+export default filter;