Example #1
0
// SuggestCommands can be used to match a given word
// which is very similar to one among the list of available commands.
func SuggestCommands(given string, available []string) []string {
	commandSet := make(map[string]struct{})
	for _, command := range available {
		commandSet[command] = struct{}{}
	}
	if len(commandSet) == 0 || given == "" {
		return nil
	}

	if _, ok := commandSet[given]; ok {
		return nil
	}

	var scores cmdScores

	for command := range commandSet {
		levenshtein := smetrics.WagnerFischer(command, given, 1, 1, 1)
		jaroWinkler := smetrics.JaroWinkler(command, given, 0.7, 4)

		if levenshtein > levenThreshold {
			continue
		}
		if jaroWinkler < jaroThreshold {
			continue
		}
		scores = append(
			scores,
			cmdScore{
				name:        command,
				levenshtein: levenshtein,
				jaroWinkler: jaroWinkler,
			},
		)
	}
	if len(scores) == 0 {
		return nil
	}
	sort.Sort(scores)

	levenshtein := scores[0].levenshtein

	var matches []string
	for _, score := range scores {
		if score.levenshtein != levenshtein {
			break
		}
		matches = append(matches, score.name)
	}

	return matches
}
Example #2
0
// MostSimilarLineTo returns the most similar line in the output to the given
// string, if any of them have a JaroWinkler score >0.1. It returns the string
// (or empty), the index of that line, and a bool indicating if the score was
// greater than 0.1
func (out TestOutput) MostSimilarLineTo(s string) (
	winner string, index int, goodMatch bool) {
	index = -1
	if s == "" {
		return
	}
	max := 0.0
	for i, l := range out.Lines() {
		score := smetrics.JaroWinkler(l, s, 0.7, 4)
		if score > max {
			winner = l
			index = i
			max = score
		}
	}
	return winner, index, max > 0.1
}