|
- 'use strict';
-
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.default = getSuggestion;
-
- var _damerauLevenshtein = require('damerau-levenshtein');
-
- var _damerauLevenshtein2 = _interopRequireDefault(_damerauLevenshtein);
-
- function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
-
- // Minimum edit distance to be considered a good suggestion.
- var THRESHOLD = 2;
-
- /**
- * Returns an array of suggestions given a word and a dictionary and limit of suggestions
- * to return.
- */
- function getSuggestion(word) {
- var dictionary = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : [];
- var limit = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : 2;
-
- var distances = dictionary.reduce(function (suggestions, dictionaryWord) {
- var distance = (0, _damerauLevenshtein2.default)(word.toUpperCase(), dictionaryWord.toUpperCase());
- var steps = distance.steps;
-
- suggestions[dictionaryWord] = steps; // eslint-disable-line
- return suggestions;
- }, {});
-
- return Object.keys(distances).filter(function (suggestion) {
- return distances[suggestion] <= THRESHOLD;
- }).sort(function (a, b) {
- return distances[a] - distances[b];
- }) // Sort by distance
- .slice(0, limit);
- }
|