d2-ui/node_modules-react-15/eslint-plugin-jsx-a11y/src/util/getSuggestion.js

d2-ui

'use strict';

import editDistance from 'damerau-levenshtein';


// Minimum edit distance to be considered a good suggestion.
const THRESHOLD = 2;

/**
 * Returns an array of suggestions given a word and a dictionary and limit of suggestions
 * to return.
 */
export default function getSuggestion(word, dictionary = [], limit = 2) {
  const distances = dictionary.reduce((suggestions, dictionaryWord) => {
    suggestions[dictionaryWord] = editDistance(word.toUpperCase(), dictionaryWord.toUpperCase()).steps;
    return suggestions;
  }, {});

  return Object.keys(distances)
    .filter(suggestion => distances[suggestion] <= THRESHOLD)
    .sort((a, b) => distances[a] - distances[b]) // Sort by distance
    .slice(0, limit);
}