compiler/src/Reporting/Suggest.hs · elk

on f4401ac help sign in

52 lines 866 B history raw

1{-# OPTIONS_GHC -Wall #-}

2{-# LANGUAGE OverloadedStrings #-}

3module Reporting.Suggest

4 ( distance

5 , sort

6 , rank

7 )

8 where

11import qualified Data.Char as Char

12import qualified Data.List as List

13import qualified Text.EditDistance as Dist

17-- DISTANCE

20distance :: String -> String -> Int

21distance x y =

22 Dist.restrictedDamerauLevenshteinDistance Dist.defaultEditCosts x y

26-- SORT

29sort :: String -> (a -> String) -> [a] -> [a]

30sort target toString values =

31 List.sortOn (distance (toLower target) . toLower . toString) values

34toLower :: String -> String

35toLower string =

36 map Char.toLower string

40-- RANK

43rank :: String -> (a -> String) -> [a] -> [(Int,a)]

44rank target toString values =

45 let

46 toRank v =

47 distance (toLower target) (toLower (toString v))

49 addRank v =

50 (toRank v, v)

51 in

52 List.sortOn fst (map addRank values)