@@ -19,10 +19,13 @@
1919module Language.PureScript.Errors where
2020
2121import Data.Either (lefts, rights)
22−import Data.List (intercalate, elemIndex)
22+import Data.List (intercalate)
23+import Data.Function (on)
2324import Data.Monoid
2425import Data.Foldable (fold, foldMap)
2526
27+import qualified Data.Map as M
28+
2629import Control.Monad
2730import Control.Monad.Error.Class (MonadError(..))
2831import Control.Monad.Unify
@@ -252,18 +255,24 @@ errorMessage err = MultipleErrors [err]
252255onErrorMessages :: (ErrorMessage -> ErrorMessage) -> MultipleErrors -> MultipleErrors
253256onErrorMessages f = MultipleErrors . map f . runMultipleErrors
254257
255−replaceUnknowns :: Type -> State [Unknown] Type
256−replaceUnknowns = everywhereOnTypesM replaceUnknowns'
257− where
258− lookupTable :: Unknown -> [Unknown] -> (Unknown, [Unknown])
259− lookupTable x table = case (elemIndex x table) of
260− Nothing -> (length table, table ++ [x])
261− Just i -> (i, table)
258+-- | The various types of things which might need to be relabelled in errors messages.
259+data LabelType = TypeLabel | SkolemLabel String deriving (Show, Eq, Ord)
262260
263− replaceUnknowns' :: Type -> State [Unknown] Type
264− replaceUnknowns' (TUnknown u) = state $ first TUnknown . lookupTable u
265− replaceUnknowns' other = return other
261+-- | A map from rigid type variable name/unknown variable pairs to new variables.
262+type UnknownMap = M.Map (LabelType, Unknown) Unknown
263+
264+replaceUnknowns :: Type -> State UnknownMap Type
265+replaceUnknowns = everywhereOnTypesM replaceTypes
266+ where
267+ lookupTable :: (LabelType, Unknown) -> UnknownMap -> (Unknown, UnknownMap)
268+ lookupTable x m = case M.lookup x m of
269+ Nothing -> let i = length (filter (on (==) fst x) (M.keys m)) in (i, M.insert x i m)
270+ Just i -> (i, m)
266271
272+ replaceTypes :: Type -> State UnknownMap Type
273+ replaceTypes (TUnknown u) = state $ first TUnknown . lookupTable (TypeLabel, u)
274+ replaceTypes (Skolem name s sko) = state $ first (flip (Skolem name) sko) . lookupTable (SkolemLabel name, s)
275+ replaceTypes other = return other
267276
268277onTypesInErrorMessageM :: (Applicative m) => (Type -> m Type) -> ErrorMessage -> m ErrorMessage
269278onTypesInErrorMessageM f = g
@@ -298,7 +307,7 @@ onTypesInErrorMessageM f = g
298307-- |
299308-- Pretty print a single error, simplifying if necessary
300309--
301−prettyPrintSingleError :: Bool -> ErrorMessage -> State [Unknown] Box.Box
310+prettyPrintSingleError :: Bool -> ErrorMessage -> State UnknownMap Box.Box
302311prettyPrintSingleError full e = prettyPrintErrorMessage <$> onTypesInErrorMessageM replaceUnknowns (if full then e else simplifyErrorMessage e)
303312 where
304313 -- |
@@ -569,15 +578,15 @@ prettyPrintSingleError full e = prettyPrintErrorMessage <$> onTypesInErrorMessag
569578-- Pretty print multiple errors
570579--
571580prettyPrintMultipleErrors :: Bool -> MultipleErrors -> String
572−prettyPrintMultipleErrors full = flip evalState [] . prettyPrintMultipleErrorsWith "Error:" "Multiple errors:" full
581+prettyPrintMultipleErrors full = flip evalState M.empty . prettyPrintMultipleErrorsWith "Error:" "Multiple errors:" full
573582
574583-- |
575584-- Pretty print multiple warnings
576585--
577586prettyPrintMultipleWarnings :: Bool -> MultipleErrors -> String
578−prettyPrintMultipleWarnings full = flip evalState [] . prettyPrintMultipleErrorsWith "Warning:" "Multiple warnings:" full
587+prettyPrintMultipleWarnings full = flip evalState M.empty . prettyPrintMultipleErrorsWith "Warning:" "Multiple warnings:" full
579588
580−prettyPrintMultipleErrorsWith :: String -> String -> Bool -> MultipleErrors -> State [Unknown] String
589+prettyPrintMultipleErrorsWith :: String -> String -> Bool -> MultipleErrors -> State UnknownMap String
581590prettyPrintMultipleErrorsWith intro _ full (MultipleErrors [e]) = do
582591 result <- prettyPrintSingleError full e
583592 return $ renderBox $