You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
1.2 KiB
37 lines
1.2 KiB
"use strict"; |
|
|
|
var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault"); |
|
|
|
Object.defineProperty(exports, "__esModule", { |
|
value: true |
|
}); |
|
exports["default"] = getSuggestion; |
|
|
|
var _damerauLevenshtein = _interopRequireDefault(require("damerau-levenshtein")); |
|
|
|
// 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, _damerauLevenshtein["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); |
|
} |
|
|
|
module.exports = exports.default; |