getSuggestion.js 1.2 KB

1234567891011121314151617181920212223242526272829303132
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
  3. Object.defineProperty(exports, "__esModule", {
  4. value: true
  5. });
  6. exports["default"] = getSuggestion;
  7. var _damerauLevenshtein = _interopRequireDefault(require("damerau-levenshtein"));
  8. var _object = _interopRequireDefault(require("object.fromentries"));
  9. // Minimum edit distance to be considered a good suggestion.
  10. var THRESHOLD = 2;
  11. /**
  12. * Returns an array of suggestions given a word and a dictionary and limit of suggestions
  13. * to return.
  14. */
  15. function getSuggestion(word) {
  16. var dictionary = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : [];
  17. var limit = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : 2;
  18. var distances = (0, _object["default"])(dictionary.map(function (dictionaryWord) {
  19. var distance = (0, _damerauLevenshtein["default"])(word.toUpperCase(), dictionaryWord.toUpperCase());
  20. var steps = distance.steps;
  21. return [dictionaryWord, steps];
  22. }));
  23. return Object.keys(distances).filter(function (suggestion) {
  24. return distances[suggestion] <= THRESHOLD;
  25. }).sort(function (a, b) {
  26. return distances[a] - distances[b];
  27. }) // Sort by distance
  28. .slice(0, limit);
  29. }
  30. module.exports = exports.default;