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.

getSuggestion.js 1.3 KiB

пре 3 година
123456789101112131415161718192021222324252627282930313233343536373839
  1. 'use strict';
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.default = getSuggestion;
  6. var _damerauLevenshtein = require('damerau-levenshtein');
  7. var _damerauLevenshtein2 = _interopRequireDefault(_damerauLevenshtein);
  8. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  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 = dictionary.reduce(function (suggestions, dictionaryWord) {
  19. var distance = (0, _damerauLevenshtein2.default)(word.toUpperCase(), dictionaryWord.toUpperCase());
  20. var steps = distance.steps;
  21. suggestions[dictionaryWord] = steps; // eslint-disable-line
  22. return suggestions;
  23. }, {});
  24. return Object.keys(distances).filter(function (suggestion) {
  25. return distances[suggestion] <= THRESHOLD;
  26. }).sort(function (a, b) {
  27. return distances[a] - distances[b];
  28. }) // Sort by distance
  29. .slice(0, limit);
  30. }