purescript 0.1.13 → 0.1.14
raw patch · 7 files changed
+350/−345 lines, 7 files
Files
- purescript.cabal +1/−1
- src/Language/PureScript/CodeGen/JS.hs +58/−62
- src/Language/PureScript/Names.hs +5/−1
- src/Language/PureScript/TypeChecker/Kinds.hs +5/−5
- src/Language/PureScript/TypeChecker/Monad.hs +46/−17
- src/Language/PureScript/TypeChecker/Types.hs +232/−257
- tests/Main.hs +3/−2
purescript.cabal view
@@ -1,5 +1,5 @@ name: purescript-version: 0.1.13+version: 0.1.14 cabal-version: >=1.8 build-type: Simple license: MIT
src/Language/PureScript/CodeGen/JS.hs view
@@ -34,13 +34,14 @@ import Language.PureScript.Pretty.Common import Language.PureScript.CodeGen.Monad import Language.PureScript.CodeGen.JS.AST as AST+import Language.PureScript.TypeChecker.Monad (NameKind(..)) declToJs :: Maybe Ident -> ModulePath -> Declaration -> Environment -> Maybe [JS]-declToJs mod mp (ValueDeclaration ident (Abs args ret)) _ =- Just $ JSFunction (Just ident) args (JSBlock [JSReturn (valueToJs mp ret)]) :+declToJs mod mp (ValueDeclaration ident (Abs args ret)) e =+ Just $ JSFunction (Just ident) args (JSBlock [JSReturn (valueToJs mp e ret)]) : maybe [] (return . setProperty (identToJs ident) (JSVar ident)) mod-declToJs mod mp (ValueDeclaration ident val) _ =- Just $ JSVariableIntroduction ident (Just (valueToJs mp val)) :+declToJs mod mp (ValueDeclaration ident val) e =+ Just $ JSVariableIntroduction ident (Just (valueToJs mp e val)) : maybe [] (return . setProperty (identToJs ident) (JSVar ident)) mod declToJs mod _ (ExternMemberDeclaration member ident _) _ = Just $ JSFunction (Just ident) [Ident "value"] (JSBlock [JSReturn (JSAccessor member (JSVar (Ident "value")))]) :@@ -63,37 +64,32 @@ [JSAssignment (JSAssignVariable (Ident name)) (JSBinary Or (JSVar (Ident name)) (JSObjectLiteral []))]] ++ maybe [] (return . setProperty name (JSVar (Ident name))) mod-declToJs mod omp (ImportDeclaration mp idents) env =- Just $ case idents of- Nothing ->- let idents = map snd . filter ((== mp) . fst) . M.keys $ names env- in map mkLocal idents- Just idents -> map mkLocal idents- where mkLocal ident = JSVariableIntroduction ident (Just (qualifiedToJS identToJs (Qualified mp ident))) declToJs _ _ _ _ = Nothing setProperty :: String -> JS -> Ident -> JS setProperty prop val mod = JSAssignment (JSAssignProperty prop (JSAssignVariable mod)) val -valueToJs :: ModulePath -> Value -> JS-valueToJs _ (NumericLiteral n) = JSNumericLiteral n-valueToJs _ (StringLiteral s) = JSStringLiteral s-valueToJs _ (BooleanLiteral b) = JSBooleanLiteral b-valueToJs m (ArrayLiteral xs) = JSArrayLiteral (map (valueToJs m) xs)-valueToJs m (ObjectLiteral ps) = JSObjectLiteral (map (second (valueToJs m)) ps)-valueToJs m (ObjectUpdate o ps) = JSApp (JSAccessor "extend" (JSVar (Ident "Object"))) [ valueToJs m o, JSObjectLiteral (map (second (valueToJs m)) ps)]-valueToJs m (Constructor name) = qualifiedToJS runProperName name-valueToJs m (Block sts) = JSApp (JSFunction Nothing [] (JSBlock (map (statementToJs m) sts))) []-valueToJs m (Case value binders) = runGen (bindersToJs m binders (valueToJs m value))-valueToJs m (IfThenElse cond th el) = JSConditional (valueToJs m cond) (valueToJs m th) (valueToJs m el)-valueToJs m (Accessor prop val) = JSAccessor prop (valueToJs m val)-valueToJs m (Indexer index val) = JSIndexer (valueToJs m index) (valueToJs m val)-valueToJs m (App val args) = JSApp (valueToJs m val) (map (valueToJs m) args)-valueToJs m (Abs args val) = JSFunction Nothing args (JSBlock [JSReturn (valueToJs m val)])-valueToJs m (Unary op val) = JSUnary op (valueToJs m val)-valueToJs m (Binary op v1 v2) = JSBinary op (valueToJs m v1) (valueToJs m v2)-valueToJs m (Var ident) = qualifiedToJS identToJs ident-valueToJs m (TypedValue val _) = valueToJs m val+valueToJs :: ModulePath -> Environment -> Value -> JS+valueToJs _ _ (NumericLiteral n) = JSNumericLiteral n+valueToJs _ _ (StringLiteral s) = JSStringLiteral s+valueToJs _ _ (BooleanLiteral b) = JSBooleanLiteral b+valueToJs m e (ArrayLiteral xs) = JSArrayLiteral (map (valueToJs m e) xs)+valueToJs m e (ObjectLiteral ps) = JSObjectLiteral (map (second (valueToJs m e)) ps)+valueToJs m e (ObjectUpdate o ps) = JSApp (JSAccessor "extend" (JSVar (Ident "Object"))) [ valueToJs m e o, JSObjectLiteral (map (second (valueToJs m e)) ps)]+valueToJs m e (Constructor name) = qualifiedToJS runProperName name+valueToJs m e (Block sts) = JSApp (JSFunction Nothing [] (JSBlock (map (statementToJs m e) sts))) []+valueToJs m e (Case value binders) = runGen (bindersToJs m e binders (valueToJs m e value))+valueToJs m e (IfThenElse cond th el) = JSConditional (valueToJs m e cond) (valueToJs m e th) (valueToJs m e el)+valueToJs m e (Accessor prop val) = JSAccessor prop (valueToJs m e val)+valueToJs m e (Indexer index val) = JSIndexer (valueToJs m e index) (valueToJs m e val)+valueToJs m e (App val args) = JSApp (valueToJs m e val) (map (valueToJs m e) args)+valueToJs m e (Abs args val) = JSFunction Nothing args (JSBlock [JSReturn (valueToJs m e val)])+valueToJs m e (Unary op val) = JSUnary op (valueToJs m e val)+valueToJs m e (Binary op v1 v2) = JSBinary op (valueToJs m e v1) (valueToJs m e v2)+valueToJs m e (Var ident) = case M.lookup (qualify m ident) (names e) of+ Just (_, Alias aliasModule aliasIdent) -> qualifiedToJS identToJs (Qualified aliasModule aliasIdent)+ _ -> qualifiedToJS identToJs ident+valueToJs m e (TypedValue val _) = valueToJs m e val qualifiedToJS :: (a -> String) -> Qualified a -> JS qualifiedToJS f (Qualified (ModulePath parts) a) =@@ -101,41 +97,41 @@ where delimited [part] = JSVar (Ident (part)) delimited (part:parts) = JSAccessor part (delimited parts) -bindersToJs :: ModulePath -> [(Binder, Value)] -> JS -> Gen JS-bindersToJs m binders val = do+bindersToJs :: ModulePath -> Environment -> [(Binder, Value)] -> JS -> Gen JS+bindersToJs m e binders val = do valName <- fresh- jss <- forM binders $ \(binder, result) -> binderToJs m valName [JSReturn (valueToJs m result)] binder+ jss <- forM binders $ \(binder, result) -> binderToJs m e valName [JSReturn (valueToJs m e result)] binder return $ JSApp (JSFunction Nothing [Ident valName] (JSBlock (concat jss ++ [JSThrow (JSStringLiteral "Failed pattern match")]))) [val] -binderToJs :: ModulePath -> String -> [JS] -> Binder -> Gen [JS]-binderToJs _ varName done NullBinder = return done-binderToJs _ varName done (StringBinder str) =+binderToJs :: ModulePath -> Environment -> String -> [JS] -> Binder -> Gen [JS]+binderToJs _ _ varName done NullBinder = return done+binderToJs _ _ varName done (StringBinder str) = return [JSIfElse (JSBinary EqualTo (JSVar (Ident varName)) (JSStringLiteral str)) (JSBlock done) Nothing]-binderToJs _ varName done (NumberBinder num) =+binderToJs _ _ varName done (NumberBinder num) = return [JSIfElse (JSBinary EqualTo (JSVar (Ident varName)) (JSNumericLiteral num)) (JSBlock done) Nothing]-binderToJs _ varName done (BooleanBinder True) =+binderToJs _ _ varName done (BooleanBinder True) = return [JSIfElse (JSVar (Ident varName)) (JSBlock done) Nothing]-binderToJs _ varName done (BooleanBinder False) =+binderToJs _ _ varName done (BooleanBinder False) = return [JSIfElse (JSUnary Not (JSVar (Ident varName))) (JSBlock done) Nothing]-binderToJs _ varName done (VarBinder ident) =+binderToJs _ e varName done (VarBinder ident) = return (JSVariableIntroduction ident (Just (JSVar (Ident varName))) : done)-binderToJs m varName done (NullaryBinder ctor) =+binderToJs m e varName done (NullaryBinder ctor) = return [JSIfElse (JSBinary EqualTo (JSAccessor "ctor" (JSVar (Ident varName))) (JSStringLiteral (show (uncurry Qualified $ qualify m ctor)))) (JSBlock done) Nothing]-binderToJs m varName done (UnaryBinder ctor b) = do+binderToJs m e varName done (UnaryBinder ctor b) = do value <- fresh- js <- binderToJs m value done b+ js <- binderToJs m e value done b return [JSIfElse (JSBinary EqualTo (JSAccessor "ctor" (JSVar (Ident varName))) (JSStringLiteral (show (uncurry Qualified $ qualify m ctor)))) (JSBlock (JSVariableIntroduction (Ident value) (Just (JSAccessor "value" (JSVar (Ident varName)))) : js)) Nothing]-binderToJs m varName done (ObjectBinder bs) = go done bs+binderToJs m e varName done (ObjectBinder bs) = go done bs where go :: [JS] -> [(String, Binder)] -> Gen [JS] go done [] = return done go done ((prop, binder):bs) = do propVar <- fresh done' <- go done bs- js <- binderToJs m propVar done' binder+ js <- binderToJs m e propVar done' binder return (JSVariableIntroduction (Ident propVar) (Just (JSAccessor prop (JSVar (Ident varName)))) : js)-binderToJs m varName done (ArrayBinder bs rest) = do+binderToJs m e varName done (ArrayBinder bs rest) = do js <- go done rest 0 bs return [JSIfElse (JSBinary cmp (JSAccessor "length" (JSVar (Ident varName))) (JSNumericLiteral (Left (fromIntegral $ length bs)))) (JSBlock js) Nothing] where@@ -145,31 +141,31 @@ go done Nothing _ [] = return done go done (Just binder) index [] = do restVar <- fresh- js <- binderToJs m restVar done binder+ js <- binderToJs m e restVar done binder return (JSVariableIntroduction (Ident restVar) (Just (JSApp (JSAccessor "slice" (JSVar (Ident varName))) [JSNumericLiteral (Left index)])) : js) go done rest index (binder:bs) = do elVar <- fresh done' <- go done rest (index + 1) bs- js <- binderToJs m elVar done' binder+ js <- binderToJs m e elVar done' binder return (JSVariableIntroduction (Ident elVar) (Just (JSIndexer (JSNumericLiteral (Left index)) (JSVar (Ident varName)))) : js)-binderToJs m varName done (NamedBinder ident binder) = do- js <- binderToJs m varName done binder+binderToJs m e varName done (NamedBinder ident binder) = do+ js <- binderToJs m e varName done binder return (JSVariableIntroduction ident (Just (JSVar (Ident varName))) : js)-binderToJs m varName done (GuardedBinder cond binder) = binderToJs m varName done' binder+binderToJs m e varName done (GuardedBinder cond binder) = binderToJs m e varName done' binder where- done' = [JSIfElse (valueToJs m cond) (JSBlock done) Nothing]+ done' = [JSIfElse (valueToJs m e cond) (JSBlock done) Nothing] -statementToJs :: ModulePath -> Statement -> JS-statementToJs m (VariableIntroduction ident value) = JSVariableIntroduction ident (Just (valueToJs m value))-statementToJs m (Assignment target value) = JSAssignment (JSAssignVariable target) (valueToJs m value)-statementToJs m (While cond sts) = JSWhile (valueToJs m cond) (JSBlock (map (statementToJs m) sts))-statementToJs m (For ident start end sts) = JSFor ident (valueToJs m start) (valueToJs m end) (JSBlock (map (statementToJs m) sts))-statementToJs m (ForEach ident arr sts) = JSApp (JSAccessor "forEach" (valueToJs m arr)) [JSFunction Nothing [ident] (JSBlock (map (statementToJs m) sts))]-statementToJs m (If ifst) = ifToJs ifst+statementToJs :: ModulePath -> Environment -> Statement -> JS+statementToJs m e (VariableIntroduction ident value) = JSVariableIntroduction ident (Just (valueToJs m e value))+statementToJs m e (Assignment target value) = JSAssignment (JSAssignVariable target) (valueToJs m e value)+statementToJs m e (While cond sts) = JSWhile (valueToJs m e cond) (JSBlock (map (statementToJs m e) sts))+statementToJs m e (For ident start end sts) = JSFor ident (valueToJs m e start) (valueToJs m e end) (JSBlock (map (statementToJs m e) sts))+statementToJs m e (ForEach ident arr sts) = JSApp (JSAccessor "forEach" (valueToJs m e arr)) [JSFunction Nothing [ident] (JSBlock (map (statementToJs m e) sts))]+statementToJs m e (If ifst) = ifToJs ifst where ifToJs :: IfStatement -> JS- ifToJs (IfStatement cond thens elses) = JSIfElse (valueToJs m cond) (JSBlock (map (statementToJs m) thens)) (fmap elseToJs elses)+ ifToJs (IfStatement cond thens elses) = JSIfElse (valueToJs m e cond) (JSBlock (map (statementToJs m e) thens)) (fmap elseToJs elses) elseToJs :: ElseStatement -> JS- elseToJs (Else sts) = JSBlock (map (statementToJs m) sts)+ elseToJs (Else sts) = JSBlock (map (statementToJs m e) sts) elseToJs (ElseIf ifst) = ifToJs ifst-statementToJs m (Return value) = JSReturn (valueToJs m value)+statementToJs m e (Return value) = JSReturn (valueToJs m e value)
src/Language/PureScript/Names.hs view
@@ -17,7 +17,7 @@ module Language.PureScript.Names where import Data.Data-import Data.List (intercalate)+import Data.List (inits, intercalate) data Ident = Ident String | Op String deriving (Eq, Ord, Data, Typeable) @@ -49,3 +49,7 @@ qualify :: ModulePath -> Qualified a -> (ModulePath, a) qualify mp (Qualified (ModulePath []) a) = (mp, a) qualify _ (Qualified mp a) = (mp, a)++nameResolution :: ModulePath -> Qualified a -> [(ModulePath, a)]+nameResolution (ModulePath mp) (Qualified (ModulePath []) a) = [ (ModulePath mp', a) | mp' <- reverse $ inits mp ]+nameResolution _ (Qualified mp a) = [(mp, a)]
src/Language/PureScript/TypeChecker/Kinds.hs view
@@ -78,10 +78,10 @@ starIfUnknown (FunKind k1 k2) = FunKind (starIfUnknown k1) (starIfUnknown k2) starIfUnknown k = k -inferAll :: Maybe (ProperName, Kind) -> M.Map String Kind -> [Type] -> Subst Check [Kind]+inferAll :: Maybe (ProperName, Kind) -> M.Map String Kind -> [Type] -> Subst [Kind] inferAll name m = mapM (infer name m) -infer :: Maybe (ProperName, Kind) -> M.Map String Kind -> Type -> Subst Check Kind+infer :: Maybe (ProperName, Kind) -> M.Map String Kind -> Type -> Subst Kind infer name m (Array t) = do k <- infer name m t k ~~ Star@@ -102,8 +102,8 @@ Nothing -> throwError $ "Unbound type variable " ++ v infer (Just (name, k)) m c@(TypeConstructor v@(Qualified (ModulePath []) pn)) | name == pn = return k infer name m (TypeConstructor v) = do- env <- lift getEnv- modulePath <- checkModulePath `fmap` lift get+ env <- liftCheck getEnv+ modulePath <- checkModulePath `fmap` get case M.lookup (qualify modulePath v) (types env) of Nothing -> throwError $ "Unknown type constructor '" ++ show v ++ "'" Just (kind, _) -> return kind@@ -118,7 +118,7 @@ infer name (M.insert ident k m) ty infer _ m t = return Star -inferRow :: Maybe (ProperName, Kind) -> M.Map String Kind -> Row -> Subst Check Kind+inferRow :: Maybe (ProperName, Kind) -> M.Map String Kind -> Row -> Subst Kind inferRow _ m (RowVar v) = do case M.lookup v m of Just k -> return k
src/Language/PureScript/TypeChecker/Monad.hs view
@@ -34,7 +34,7 @@ import qualified Data.Map as M -data NameKind = Value | Extern | Alias ModulePath Ident deriving Show+data NameKind = Value | Extern | Alias ModulePath Ident | LocalVariable deriving Show data TypeDeclarationKind = Data | ExternData | TypeSynonym deriving Show @@ -49,6 +49,28 @@ emptyEnvironment :: Environment emptyEnvironment = Environment M.empty M.empty M.empty M.empty M.empty +bindNames :: (MonadState CheckState m) => M.Map (ModulePath, Ident) (Type, NameKind) -> m a -> m a+bindNames newNames action = do+ orig <- get+ modify $ \st -> st { checkEnv = (checkEnv st) { names = newNames `M.union` (names . checkEnv $ st) } }+ a <- action+ modify $ \st -> st { checkEnv = (checkEnv st) { names = names . checkEnv $ orig } }+ return a++bindLocalVariables :: (Functor m, MonadState CheckState m) => [(Ident, Type)] -> m a -> m a+bindLocalVariables bindings action = do+ modulePath <- checkModulePath `fmap` get+ bindNames (M.fromList $ flip map bindings $ \(name, ty) -> ((modulePath, name), (ty, LocalVariable))) action++lookupVariable :: (Functor m, MonadState CheckState m, MonadError String m) => Qualified Ident -> m Type+lookupVariable var = do+ env <- getEnv+ modulePath <- checkModulePath <$> get+ let tries = map (First . flip M.lookup (names env)) (nameResolution modulePath var)+ case getFirst (mconcat tries) of+ Nothing -> throwError $ show var ++ " is undefined"+ Just (ty, _) -> return ty+ data AnyUnifiable where AnyUnifiable :: forall t. (Unifiable t) => t -> AnyUnifiable @@ -60,13 +82,13 @@ newtype Check a = Check { unCheck :: StateT CheckState (Either String) a } deriving (Functor, Monad, Applicative, MonadPlus, MonadState CheckState, MonadError String) -getEnv :: Check Environment-getEnv = fmap checkEnv get+getEnv :: (Functor m, MonadState CheckState m) => m Environment+getEnv = checkEnv <$> get -putEnv :: Environment -> Check ()+putEnv :: (MonadState CheckState m) => Environment -> m () putEnv env = modify (\s -> s { checkEnv = env }) -modifyEnv :: (Environment -> Environment) -> Check ()+modifyEnv :: (MonadState CheckState m) => (Environment -> Environment) -> m () modifyEnv f = modify (\s -> s { checkEnv = f (checkEnv s) }) runCheck :: Check a -> Either String (a, Environment)@@ -98,12 +120,19 @@ data SubstState = SubstState { substSubst :: Substitution , substFutureEscapeChecks :: [AnyUnifiable] } -newtype Subst m a = Subst { unSubst :: StateT SubstState m a }- deriving (Functor, Monad, Applicative, MonadPlus, MonadTrans)+newtype Subst a = Subst { unSubst :: StateT SubstState Check a }+ deriving (Functor, Monad, Applicative, MonadPlus) -deriving instance (MonadError String m) => MonadError String (Subst m)+instance MonadState CheckState Subst where+ get = Subst . lift $ get+ put = Subst . lift . put -runSubst :: (Unifiable a, Monad m) => Subst m a -> m (a, Substitution, [AnyUnifiable])+deriving instance MonadError String Subst++liftCheck :: Check a -> Subst a+liftCheck = Subst . lift++runSubst :: (Unifiable a) => Subst a -> Check (a, Substitution, [AnyUnifiable]) runSubst subst = do (a, s) <- flip runStateT (SubstState mempty []) . unSubst $ subst return (apply (substSubst s) a, substSubst s, substFutureEscapeChecks s)@@ -119,7 +148,7 @@ u2 | u2 == u -> t | otherwise -> unknown u2 -replace :: (Unifiable t) => Unknown t -> t -> Subst Check ()+replace :: (Unifiable t) => Unknown t -> t -> Subst () replace u t' = do sub <- substSubst <$> Subst get let t = apply sub t'@@ -132,25 +161,25 @@ class (Typeable t, Data t, Show t) => Unifiable t where unknown :: Unknown t -> t- (~~) :: t -> t -> Subst Check ()+ (~~) :: t -> t -> Subst () isUnknown :: t -> Maybe (Unknown t) apply :: Substitution -> t -> t unknowns :: t -> [Int] -occursCheck :: (Unifiable t) => Unknown s -> t -> Subst Check ()+occursCheck :: (Unifiable t) => Unknown s -> t -> Subst () occursCheck (Unknown u) t = case isUnknown t of Nothing -> guardWith "Occurs check fails" (u `notElem` unknowns t) _ -> return () -fresh' :: Subst Check Int+fresh' :: Subst Int fresh' = do- n <- lift $ checkNextVar <$> get- lift . modify $ \s -> s { checkNextVar = succ (checkNextVar s) }+ n <- checkNextVar <$> get+ modify $ \s -> s { checkNextVar = succ (checkNextVar s) } return n -fresh :: (Unifiable t) => Subst Check t+fresh :: (Unifiable t) => Subst t fresh = unknown . Unknown <$> fresh' -escapeCheckLater :: (Unifiable t) => t -> Subst Check ()+escapeCheckLater :: (Unifiable t) => t -> Subst () escapeCheckLater t = Subst . modify $ \s -> s { substFutureEscapeChecks = AnyUnifiable t : substFutureEscapeChecks s }
src/Language/PureScript/TypeChecker/Types.hs view
@@ -12,7 +12,7 @@ -- ----------------------------------------------------------------------------- -{-# LANGUAGE DeriveDataTypeable #-}+{-# LANGUAGE DeriveDataTypeable, FlexibleContexts #-} module Language.PureScript.TypeChecker.Types ( typeOf@@ -82,7 +82,7 @@ forM_ int (uncurry (~~)) unifyRows sd1 r1' sd2 r2' where- unifyRows :: [(String, Type)] -> Row -> [(String, Type)] -> Row -> Subst Check ()+ unifyRows :: [(String, Type)] -> Row -> [(String, Type)] -> Row -> Subst () unifyRows [] (RUnknown u) sd r = replace u (rowFromList (sd, r)) unifyRows sd r [] (RUnknown u) = replace u (rowFromList (sd, r)) unifyRows ns@((name, ty):row) r others u@(RUnknown un) = do@@ -102,7 +102,7 @@ unknowns (RCons _ ty r) = unknowns ty ++ unknowns r unknowns _ = [] -unifyTypes :: Type -> Type -> Subst Check ()+unifyTypes :: Type -> Type -> Subst () unifyTypes t1 t2 = rethrow (\e -> "Error unifying type " ++ prettyPrintType t1 ++ " with type " ++ prettyPrintType t2 ++ ":\n" ++ e) $ do unifyTypes' t1 t2 where@@ -134,7 +134,7 @@ ret1 `unifyTypes` ret2 unifyTypes' (TypeVar v1) (TypeVar v2) | v1 == v2 = return () unifyTypes' (TypeConstructor c1) (TypeConstructor c2) = do- modulePath <- checkModulePath `fmap` lift get+ modulePath <- checkModulePath `fmap` get guardWith ("Cannot unify " ++ show c1 ++ " with " ++ show c2 ++ ".") (qualify modulePath c1 == qualify modulePath c2) unifyTypes' (TypeApp t1 t2) (TypeApp t3 t4) = do t1 `unifyTypes` t3@@ -153,17 +153,19 @@ Just ident | isFunction val -> case val of TypedValue val ty -> do- kind <- lift $ kindOf ty+ kind <- liftCheck $ kindOf ty guardWith ("Expected type of kind *, was " ++ prettyPrintKind kind) $ kind == Star- ty' <- lift $ replaceAllTypeSynonyms ty- check (M.singleton ident ty) val ty'+ ty' <- replaceAllTypeSynonyms ty+ modulePath <- checkModulePath <$> get+ bindNames (M.singleton (modulePath, ident) (ty, LocalVariable)) $ check val ty' return ty' _ -> do me <- fresh- ty <- infer (M.singleton ident me) val+ modulePath <- checkModulePath <$> get+ ty <- bindNames (M.singleton (modulePath, ident) (me, LocalVariable)) $ infer val ty ~~ me return ty- _ -> infer M.empty val+ _ -> infer val escapeCheck checks ty sub skolemEscapeCheck ty return $ varIfUnknown $ desaturateAllTypeSynonyms $ setifyAll ty@@ -225,17 +227,17 @@ replace (RowVar v) | v == name = r replace t = t -replaceAllVarsWithUnknowns :: Type -> Subst Check Type+replaceAllVarsWithUnknowns :: Type -> Subst Type replaceAllVarsWithUnknowns (ForAll ident ty) = replaceVarWithUnknown ident ty >>= replaceAllVarsWithUnknowns replaceAllVarsWithUnknowns ty = return ty -replaceVarWithUnknown :: String -> Type -> Subst Check Type+replaceVarWithUnknown :: String -> Type -> Subst Type replaceVarWithUnknown ident ty = do tu <- fresh ru <- fresh return $ replaceRowVars ident ru . replaceTypeVars ident tu $ ty -replaceAllTypeSynonyms :: (D.Data d) => d -> Check d+replaceAllTypeSynonyms :: (Functor m, MonadState CheckState m, MonadError String m) => (D.Data d) => d -> m d replaceAllTypeSynonyms d = do env <- getEnv let syns = map (\((path, name), (args, _)) -> (Qualified path name, length args)) . M.toList $ typeSynonyms env@@ -247,61 +249,62 @@ replace (SaturatedTypeSynonym name args) = foldl TypeApp (TypeConstructor name) args replace t = t -expandAllTypeSynonyms :: Type -> Subst Check Type+expandAllTypeSynonyms :: Type -> Subst Type expandAllTypeSynonyms (SaturatedTypeSynonym name args) = expandTypeSynonym name args >>= expandAllTypeSynonyms expandAllTypeSynonyms ty = return ty -expandTypeSynonym :: Qualified ProperName -> [Type] -> Subst Check Type+expandTypeSynonym :: Qualified ProperName -> [Type] -> Subst Type expandTypeSynonym name args = do- env <- lift getEnv- modulePath <- checkModulePath `fmap` lift get+ env <- getEnv+ modulePath <- checkModulePath `fmap` get case M.lookup (qualify modulePath name) (typeSynonyms env) of Just (synArgs, body) -> return $ replaceAllTypeVars (zip synArgs args) body Nothing -> error "Type synonym was not defined" -ensureNoDuplicateProperties :: [(String, Value)] -> Check ()+ensureNoDuplicateProperties :: (MonadError String m) => [(String, Value)] -> m () ensureNoDuplicateProperties ps = guardWith "Duplicate property names" $ length (nub . map fst $ ps) == length ps -infer :: M.Map Ident Type -> Value -> Subst Check Type-infer m val = rethrow (\e -> "Error inferring type of term " ++ prettyPrintValue val ++ ":\n" ++ e) $ do- ty <- infer' m val+infer :: Value -> Subst Type+infer val = rethrow (\e -> "Error inferring type of term " ++ prettyPrintValue val ++ ":\n" ++ e) $ do+ ty <- infer' val escapeCheckLater ty return ty -infer' _ (NumericLiteral _) = return Number-infer' _ (StringLiteral _) = return String-infer' _ (BooleanLiteral _) = return Boolean-infer' m (ArrayLiteral vals) = do- ts <- mapM (infer m) vals+infer' :: Value -> Subst Type+infer' (NumericLiteral _) = return Number+infer' (StringLiteral _) = return String+infer' (BooleanLiteral _) = return Boolean+infer' (ArrayLiteral vals) = do+ ts <- mapM (infer) vals arr <- fresh forM_ ts $ \t -> arr ~~ Array t return arr-infer' m (Unary op val) = do- t <- infer m val+infer' (Unary op val) = do+ t <- infer val inferUnary op t-infer' m (Binary op left right) = do- t1 <- infer m left- t2 <- infer m right+infer' (Binary op left right) = do+ t1 <- infer left+ t2 <- infer right inferBinary op t1 t2-infer' m (ObjectLiteral ps) = do- lift $ ensureNoDuplicateProperties ps- ts <- mapM (infer m . snd) ps+infer' (ObjectLiteral ps) = do+ ensureNoDuplicateProperties ps+ ts <- mapM (infer . snd) ps let fields = zipWith (\(name, _) t -> (name, t)) ps ts return $ Object $ rowFromList (fields, REmpty)-infer' m (ObjectUpdate o ps) = do- lift $ ensureNoDuplicateProperties ps+infer' (ObjectUpdate o ps) = do+ ensureNoDuplicateProperties ps row <- fresh- newTys <- zipWith (\(name, _) t -> (name, t)) ps <$> mapM (infer m . snd) ps+ newTys <- zipWith (\(name, _) t -> (name, t)) ps <$> mapM (infer . snd) ps oldTys <- zip (map fst ps) <$> replicateM (length ps) fresh- check m o $ Object $ rowFromList (oldTys, row)+ check o $ Object $ rowFromList (oldTys, row) return $ Object $ rowFromList (newTys, row)-infer' m (Indexer index val) = do+infer' (Indexer index val) = do el <- fresh- check m index Number- check m val (Array el)+ check index Number+ check val (Array el) return el-infer' m (Accessor prop val) = do- obj <- infer m val+infer' (Accessor prop val) = do+ obj <- infer val propTy <- inferProperty obj prop case propTy of Nothing -> do@@ -310,61 +313,50 @@ obj `subsumes` Object (RCons prop field rest) return field Just ty -> return ty-infer' m (Abs args ret) = do+infer' (Abs args ret) = do ts <- replicateM (length args) fresh- let m' = m `M.union` M.fromList (zip args ts)- body <- infer m' ret- return $ Function ts body-infer' m app@(App _ _) = do+ bindLocalVariables (zip args ts) $ do+ body <- infer' ret+ return $ Function ts body+infer' app@(App _ _) = do let (f, argss) = unfoldApplication app- ft <- infer m f+ ft <- infer f ret <- fresh- checkFunctionApplications m ft argss ret+ checkFunctionApplications ft argss ret return ret-infer' m (Var var@(Qualified mp name)) = do- case mp of- ModulePath [] ->- case M.lookup name m of- Just ty -> lift $ replaceAllTypeSynonyms ty- Nothing -> lookupGlobal- _ -> lookupGlobal- where- lookupGlobal = do- env <- lift getEnv- modulePath <- checkModulePath `fmap` lift get- case M.lookup (qualify modulePath var) (names env) of- Nothing -> throwError $ show var ++ " is undefined"- Just (ty, _) -> lift $ replaceAllTypeSynonyms ty-infer' m (Block ss) = do+infer' (Var var@(Qualified mp name)) = do+ ty <- lookupVariable var+ replaceAllTypeSynonyms ty+infer' (Block ss) = do ret <- fresh- (allCodePathsReturn, _) <- checkBlock m M.empty ret ss+ (allCodePathsReturn, _) <- checkBlock M.empty ret ss guardWith "Block is missing a return statement" allCodePathsReturn return ret-infer' m (Constructor c) = do- env <- lift getEnv- modulePath <- checkModulePath `fmap` lift get+infer' (Constructor c) = do+ env <- getEnv+ modulePath <- checkModulePath `fmap` get case M.lookup (qualify modulePath c) (dataConstructors env) of Nothing -> throwError $ "Constructor " ++ show c ++ " is undefined"- Just ty -> lift $ replaceAllTypeSynonyms ty-infer' m (Case val binders) = do- t1 <- infer m val+ Just ty -> replaceAllTypeSynonyms ty+infer' (Case val binders) = do+ t1 <- infer val ret <- fresh- checkBinders m t1 ret binders+ checkBinders t1 ret binders return ret-infer' m (IfThenElse cond th el) = do- check m cond Boolean- t2 <- infer m th- t3 <- infer m el+infer' (IfThenElse cond th el) = do+ check cond Boolean+ t2 <- infer th+ t3 <- infer el t2 ~~ t3 return t2-infer' m (TypedValue val ty) = do- kind <- lift $ kindOf ty+infer' (TypedValue val ty) = do+ kind <- liftCheck $ kindOf ty guardWith ("Expected type of kind *, was " ++ prettyPrintKind kind) $ kind == Star- ty' <- lift $ replaceAllTypeSynonyms ty- check m val ty'+ ty' <- replaceAllTypeSynonyms ty+ check val ty' return ty' -inferProperty :: Type -> String -> Subst Check (Maybe Type)+inferProperty :: Type -> String -> Subst (Maybe Type) inferProperty (Object row) prop = do let (props, _) = rowToList row return $ lookup prop props@@ -376,19 +368,19 @@ inferProperty replaced prop inferProperty _ prop = return Nothing -inferUnary :: UnaryOperator -> Type -> Subst Check Type+inferUnary :: UnaryOperator -> Type -> Subst Type inferUnary op val = case fromMaybe (error "Invalid operator") $ lookup op unaryOps of (valTy, resTy) -> do val ~~ valTy return resTy -checkUnary :: M.Map Ident Type -> UnaryOperator -> Value -> Type -> Subst Check ()-checkUnary m op val res =+checkUnary :: UnaryOperator -> Value -> Type -> Subst ()+checkUnary op val res = case fromMaybe (error "Invalid operator") $ lookup op unaryOps of (valTy, resTy) -> do res ~~ resTy- check m val valTy+ check val valTy unaryOps :: [(UnaryOperator, (Type, Type))] unaryOps = [ (Negate, (Number, Number))@@ -396,7 +388,7 @@ , (BitwiseNot, (Number, Number)) ] -inferBinary :: BinaryOperator -> Type -> Type -> Subst Check Type+inferBinary :: BinaryOperator -> Type -> Type -> Subst Type inferBinary op left right | isEqualityTest op = do left ~~ right return Boolean@@ -407,18 +399,18 @@ right ~~ valTy return resTy -checkBinary :: M.Map Ident Type -> BinaryOperator -> Value -> Value -> Type -> Subst Check ()-checkBinary m op left right res | isEqualityTest op = do+checkBinary :: BinaryOperator -> Value -> Value -> Type -> Subst ()+checkBinary op left right res | isEqualityTest op = do res ~~ Boolean- t1 <- infer m left- t2 <- infer m right+ t1 <- infer left+ t2 <- infer right t1 ~~ t2-checkBinary m op left right res =+checkBinary op left right res = case fromMaybe (error "Invalid operator") $ lookup op binaryOps of (valTy, resTy) -> do res ~~ resTy- check m left valTy- check m right valTy+ check left valTy+ check right valTy isEqualityTest :: BinaryOperator -> Bool isEqualityTest EqualTo = True@@ -446,23 +438,23 @@ , (GreaterThanOrEqualTo, (Number, Boolean)) ] -inferBinder :: Type -> Binder -> Subst Check (M.Map Ident Type)+inferBinder :: Type -> Binder -> Subst (M.Map Ident Type) inferBinder _ NullBinder = return M.empty inferBinder val (StringBinder _) = val ~~ String >> return M.empty inferBinder val (NumberBinder _) = val ~~ Number >> return M.empty inferBinder val (BooleanBinder _) = val ~~ Boolean >> return M.empty inferBinder val (VarBinder name) = return $ M.singleton name val inferBinder val (NullaryBinder ctor) = do- env <- lift getEnv- modulePath <- checkModulePath `fmap` lift get+ env <- getEnv+ modulePath <- checkModulePath <$> get case M.lookup (qualify modulePath ctor) (dataConstructors env) of Just ty -> do ty `subsumes` val return M.empty _ -> throwError $ "Constructor " ++ show ctor ++ " is not defined" inferBinder val (UnaryBinder ctor binder) = do- env <- lift getEnv- modulePath <- checkModulePath `fmap` lift get+ env <- getEnv+ modulePath <- checkModulePath <$> get case M.lookup (qualify modulePath ctor) (dataConstructors env) of Just ty -> do Function [obj] ret <- replaceAllVarsWithUnknowns ty@@ -476,7 +468,7 @@ val ~~ Object row return m1 where- inferRowProperties :: Row -> Row -> [(String, Binder)] -> Subst Check (M.Map Ident Type)+ inferRowProperties :: Row -> Row -> [(String, Binder)] -> Subst (M.Map Ident Type) inferRowProperties nrow row [] = nrow ~~ row >> return M.empty inferRowProperties nrow row ((name, binder):binders) = do propTy <- fresh@@ -496,97 +488,95 @@ m <- inferBinder val binder return $ M.insert name val m -inferGuardedBinder :: M.Map Ident Type -> Type -> Binder -> Subst Check (M.Map Ident Type)-inferGuardedBinder m val (GuardedBinder cond binder) = do+inferGuardedBinder :: Type -> Binder -> Subst (M.Map Ident Type)+inferGuardedBinder val (GuardedBinder cond binder) = do m1 <- inferBinder val binder- check (m1 `M.union` m) cond Boolean+ bindLocalVariables (M.toList m1) $ check cond Boolean return m1-inferGuardedBinder m val b = inferBinder val b+inferGuardedBinder val b = inferBinder val b -checkBinders :: M.Map Ident Type -> Type -> Type -> [(Binder, Value)] -> Subst Check ()-checkBinders _ _ _ [] = return ()-checkBinders m nval ret ((binder, val):bs) = do- m1 <- inferGuardedBinder m nval binder- check (m1 `M.union` m) val ret- checkBinders m nval ret bs+checkBinders :: Type -> Type -> [(Binder, Value)] -> Subst ()+checkBinders _ _ [] = return ()+checkBinders nval ret ((binder, val):bs) = do+ m1 <- inferGuardedBinder nval binder+ bindLocalVariables (M.toList m1) $ check val ret+ checkBinders nval ret bs -assignVariable :: Ident -> M.Map Ident Type -> Subst Check ()-assignVariable name m =- case M.lookup name m of- Nothing -> return ()- Just _ -> throwError $ "Variable with name " ++ show name ++ " already exists."+assignVariable :: Ident -> Subst ()+assignVariable name = do+ env <- checkEnv <$> get+ modulePath <- checkModulePath <$> get+ case M.lookup (modulePath, name) (names env) of+ Just (_, LocalVariable) -> throwError $ "Variable with name " ++ show name ++ " already exists."+ _ -> return () -checkStatement :: M.Map Ident Type -> M.Map Ident Type -> Type -> Statement -> Subst Check (Bool, M.Map Ident Type)-checkStatement m mass ret (VariableIntroduction name val) = do- assignVariable name (m `M.union` mass)- t <- infer m val+checkStatement :: M.Map Ident Type -> Type -> Statement -> Subst (Bool, M.Map Ident Type)+checkStatement mass ret (VariableIntroduction name val) = do+ assignVariable name+ t <- infer val return (False, M.insert name t mass)-checkStatement m mass ret (Assignment ident val) = do- t <- infer m val+checkStatement mass ret (Assignment ident val) = do+ t <- infer val case M.lookup ident mass of Nothing -> throwError $ "No local variable with name " ++ show ident Just ty -> do t ~~ ty return (False, mass)-checkStatement m mass ret (While val inner) = do- check m val Boolean- (allCodePathsReturn, _) <- checkBlock m mass ret inner+checkStatement mass ret (While val inner) = do+ check val Boolean+ (allCodePathsReturn, _) <- checkBlock mass ret inner return (allCodePathsReturn, mass)-checkStatement m mass ret (If ifst) = do- allCodePathsReturn <- checkIfStatement m mass ret ifst+checkStatement mass ret (If ifst) = do+ allCodePathsReturn <- checkIfStatement mass ret ifst return (allCodePathsReturn, mass)-checkStatement m mass ret (For ident start end inner) = do- assignVariable ident (m `M.union` mass)- check (m `M.union` mass) start Number- check (m `M.union` mass) end Number- let mass1 = M.insert ident Number mass- (allCodePathsReturn, _) <- checkBlock (m `M.union` mass1) mass1 ret inner+checkStatement mass ret (For ident start end inner) = do+ assignVariable ident+ check start Number+ check end Number+ (allCodePathsReturn, _) <- bindLocalVariables [(ident, Number)] $ checkBlock mass ret inner return (allCodePathsReturn, mass)-checkStatement m mass ret (ForEach ident vals inner) = do- assignVariable ident (m `M.union` mass)+checkStatement mass ret (ForEach ident vals inner) = do+ assignVariable ident val <- fresh- check (m `M.union` mass) vals (Array val)- let mass1 = M.insert ident val mass- (allCodePathsReturn, _) <- checkBlock (m `M.union` mass1) mass1 ret inner+ check vals (Array val)+ (allCodePathsReturn, _) <- bindLocalVariables [(ident, val)] $ checkBlock mass ret inner guardWith "Cannot return from within a foreach block" $ not allCodePathsReturn return (False, mass)-checkStatement m mass ret (Return val) = do- check (m `M.union` mass) val ret+checkStatement mass ret (Return val) = do+ check val ret return (True, mass) -checkIfStatement :: M.Map Ident Type -> M.Map Ident Type -> Type -> IfStatement -> Subst Check Bool-checkIfStatement m mass ret (IfStatement val thens Nothing) = do- check m val Boolean- _ <- checkBlock m mass ret thens+checkIfStatement :: M.Map Ident Type -> Type -> IfStatement -> Subst Bool+checkIfStatement mass ret (IfStatement val thens Nothing) = do+ check val Boolean+ _ <- checkBlock mass ret thens return False-checkIfStatement m mass ret (IfStatement val thens (Just elses)) = do- check m val Boolean- (allCodePathsReturn1, _) <- checkBlock m mass ret thens- allCodePathsReturn2 <- checkElseStatement m mass ret elses+checkIfStatement mass ret (IfStatement val thens (Just elses)) = do+ check val Boolean+ (allCodePathsReturn1, _) <- checkBlock mass ret thens+ allCodePathsReturn2 <- checkElseStatement mass ret elses return $ allCodePathsReturn1 && allCodePathsReturn2 -checkElseStatement :: M.Map Ident Type -> M.Map Ident Type -> Type -> ElseStatement -> Subst Check Bool-checkElseStatement m mass ret (Else elses) = fst <$> checkBlock m mass ret elses-checkElseStatement m mass ret (ElseIf ifst) = checkIfStatement m mass ret ifst+checkElseStatement :: M.Map Ident Type -> Type -> ElseStatement -> Subst Bool+checkElseStatement mass ret (Else elses) = fst <$> checkBlock mass ret elses+checkElseStatement mass ret (ElseIf ifst) = checkIfStatement mass ret ifst -checkBlock :: M.Map Ident Type -> M.Map Ident Type -> Type -> [Statement] -> Subst Check (Bool, M.Map Ident Type)-checkBlock _ mass _ [] = return (False, mass)-checkBlock m mass ret (s:ss) = do- (b1, mass1) <- checkStatement (m `M.union` mass) mass ret s- case (b1, ss) of+checkBlock :: M.Map Ident Type -> Type -> [Statement] -> Subst (Bool, M.Map Ident Type)+checkBlock mass _ [] = return (False, mass)+checkBlock mass ret (s:ss) = do+ (b1, mass1) <- checkStatement mass ret s+ bindLocalVariables (M.toList mass1) $ case (b1, ss) of (True, []) -> return (True, mass1) (True, _) -> throwError "Unreachable code"- (False, ss) -> do- (b2, mass2) <- checkBlock m mass1 ret ss- return (b2, mass2)+ (False, ss) -> checkBlock mass1 ret ss -skolemize :: String -> Type -> Subst Check Type+skolemize :: String -> Type -> Subst Type skolemize ident ty = do tsk <- Skolem <$> fresh' rsk <- RSkolem <$> fresh' return $ replaceRowVars ident rsk $ replaceTypeVars ident tsk ty -check :: M.Map Ident Type -> Value -> Type -> Subst Check ()-check m val ty = rethrow errorMessage $ check' m val ty+check :: Value -> Type -> Subst ()+check val ty = rethrow errorMessage $ check' val ty where errorMessage msg = "Error checking type of term " ++@@ -596,92 +586,77 @@ ":\n" ++ msg -check' :: M.Map Ident Type -> Value -> Type -> Subst Check ()-check' m val (ForAll idents ty) = do+check' :: Value -> Type -> Subst ()+check' val (ForAll idents ty) = do sk <- skolemize idents ty- check m val sk-check' m val u@(TUnknown _) = do- ty <- infer m val+ check val sk+check' val u@(TUnknown _) = do+ ty <- infer val -- Don't unify an unknown with an inferred polytype ty' <- replaceAllVarsWithUnknowns ty ty' ~~ u-check' m (NumericLiteral _) Number = return ()-check' m (StringLiteral _) String = return ()-check' m (BooleanLiteral _) Boolean = return ()-check' m (Unary op val) ty = checkUnary m op val ty-check' m (Binary op left right) ty = checkBinary m op left right ty-check' m (ArrayLiteral vals) (Array ty) = forM_ vals (\val -> check m val ty)-check' m (Indexer index vals) ty = check m index Number >> check m vals (Array ty)-check' m (Abs args ret) (Function argTys retTy) = do+check' (NumericLiteral _) Number = return ()+check' (StringLiteral _) String = return ()+check' (BooleanLiteral _) Boolean = return ()+check' (Unary op val) ty = checkUnary op val ty+check' (Binary op left right) ty = checkBinary op left right ty+check' (ArrayLiteral vals) (Array ty) = forM_ vals (\val -> check val ty)+check' (Indexer index vals) ty = check index Number >> check vals (Array ty)+check' (Abs args ret) (Function argTys retTy) = do guardWith "Incorrect number of function arguments" (length args == length argTys)- let bindings = M.fromList (zip args argTys)- check (bindings `M.union` m) ret retTy-check' m app@(App _ _) ret = do+ bindLocalVariables (zip args argTys) $ check ret retTy+check' app@(App _ _) ret = do let (f, argss) = unfoldApplication app- ft <- infer m f- checkFunctionApplications m ft argss ret-check' m v@(Var var@(Qualified mp name)) ty = do- case mp of- ModulePath [] ->- case M.lookup name m of- Just ty1 -> do- repl <- lift $ replaceAllTypeSynonyms ty1- repl `subsumes` ty- Nothing -> lookupGlobal- _ -> lookupGlobal- where- lookupGlobal = do- env <- lift getEnv- modulePath <- checkModulePath `fmap` lift get- case M.lookup (qualify modulePath var) (names env) of- Nothing -> throwError $ show var ++ " is undefined"- Just (ty1, _) -> do- repl <- lift $ replaceAllTypeSynonyms ty1- repl `subsumes` ty-check' m (TypedValue val ty1) ty2 = do- kind <- lift $ kindOf ty1+ ft <- infer f+ checkFunctionApplications ft argss ret+check' v@(Var var@(Qualified mp name)) ty = do+ ty1 <- lookupVariable var+ repl <- replaceAllTypeSynonyms ty1+ repl `subsumes` ty+check' (TypedValue val ty1) ty2 = do+ kind <- liftCheck $ kindOf ty1 guardWith ("Expected type of kind *, was " ++ prettyPrintKind kind) $ kind == Star ty1 `subsumes` ty2- check m val ty1-check' m (Case val binders) ret = do- t1 <- infer m val- checkBinders m t1 ret binders-check' m (IfThenElse cond th el) ty = do- check m cond Boolean- check m th ty- check m el ty-check' m (ObjectLiteral ps) (Object row) = do- lift $ ensureNoDuplicateProperties ps- checkProperties m ps row False-check' m (ObjectUpdate obj ps) (Object row) = do- lift $ ensureNoDuplicateProperties ps+ check val ty1+check' (Case val binders) ret = do+ t1 <- infer val+ checkBinders t1 ret binders+check' (IfThenElse cond th el) ty = do+ check cond Boolean+ check th ty+ check el ty+check' (ObjectLiteral ps) (Object row) = do+ ensureNoDuplicateProperties ps+ checkProperties ps row False+check' (ObjectUpdate obj ps) (Object row) = do+ ensureNoDuplicateProperties ps us <- zip (map fst ps) <$> replicateM (length ps) fresh let (propsToCheck, rest) = rowToList row propsToRemove = map fst ps remainingProps = filter (\(p, _) -> p `notElem` propsToRemove) propsToCheck- check m obj (Object (rowFromList (us ++ remainingProps, rest)))- checkProperties m ps row True-check' m (Accessor prop val) ty = do+ check obj (Object (rowFromList (us ++ remainingProps, rest)))+ checkProperties ps row True+check' (Accessor prop val) ty = do rest <- fresh- check m val (Object (RCons prop ty rest))-check' m (Block ss) ret = do- (allCodePathsReturn, _) <- checkBlock m M.empty ret ss+ check val (Object (RCons prop ty rest))+check' (Block ss) ret = do+ (allCodePathsReturn, _) <- checkBlock M.empty ret ss guardWith "Block is missing a return statement" allCodePathsReturn-check' m (Constructor c) ty = do- env <- lift getEnv- modulePath <- checkModulePath `fmap` lift get+check' (Constructor c) ty = do+ env <- getEnv+ modulePath <- checkModulePath <$> get case M.lookup (qualify modulePath c) (dataConstructors env) of Nothing -> throwError $ "Constructor " ++ show c ++ " is undefined" Just ty1 -> do- repl <- lift $ replaceAllTypeSynonyms ty1+ repl <- replaceAllTypeSynonyms ty1 repl `subsumes` ty-check' m val (SaturatedTypeSynonym name args) = do+check' val (SaturatedTypeSynonym name args) = do ty <- expandTypeSynonym name args- check m val ty-check' _ val ty = throwError $ prettyPrintValue val ++ " does not have type " ++ prettyPrintType ty+ check val ty+check' val ty = throwError $ prettyPrintValue val ++ " does not have type " ++ prettyPrintType ty -checkProperties :: M.Map Ident Type -> [(String, Value)] -> Row -> Bool -> Subst Check ()-checkProperties m ps row lax = let (ts, r') = rowToList row in go ps ts r' where+checkProperties :: [(String, Value)] -> Row -> Bool -> Subst ()+checkProperties ps row lax = let (ts, r') = rowToList row in go ps ts r' where go [] [] REmpty = return () go [] [] u@(RUnknown _) = u ~~ REmpty go [] [] (RSkolem _) | lax = return ()@@ -689,19 +664,19 @@ | otherwise = throwError $ prettyPrintValue (ObjectLiteral ps) ++ " does not have property " ++ p go ((p,_):_) [] REmpty = throwError $ "Property " ++ p ++ " is not present in closed object type " ++ prettyPrintRow row go ((p,v):ps) [] u@(RUnknown _) = do- ty <- infer m v+ ty <- infer v rest <- fresh u ~~ RCons p ty rest go ps [] rest go ((p,v):ps) ts r = case lookup p ts of Nothing -> do- ty <- infer m v+ ty <- infer v rest <- fresh r ~~ RCons p ty rest go ps ts rest Just ty -> do- check m v ty+ check v ty go ps (delete (p, ty) ts) r go _ _ _ = throwError $ prettyPrintValue (ObjectLiteral ps) ++ " does not have type " ++ prettyPrintType (Object row) @@ -711,16 +686,16 @@ go argss (App f args) = go (args:argss) f go argss f = (f, argss) -checkFunctionApplications :: M.Map Ident Type -> Type -> [[Value]] -> Type -> Subst Check ()-checkFunctionApplications _ _ [] _ = error "Nullary function application"-checkFunctionApplications m fnTy [args] ret = checkFunctionApplication m fnTy args ret-checkFunctionApplications m fnTy (args:argss) ret = do- argTys <- mapM (infer m) args- f <- inferFunctionApplication m fnTy argTys- checkFunctionApplications m f argss ret+checkFunctionApplications :: Type -> [[Value]] -> Type -> Subst ()+checkFunctionApplications _ [] _ = error "Nullary function application"+checkFunctionApplications fnTy [args] ret = checkFunctionApplication fnTy args ret+checkFunctionApplications fnTy (args:argss) ret = do+ argTys <- mapM (infer) args+ f <- inferFunctionApplication fnTy argTys+ checkFunctionApplications f argss ret -checkFunctionApplication :: M.Map Ident Type -> Type -> [Value] -> Type -> Subst Check ()-checkFunctionApplication m fnTy args ret = rethrow errorMessage $ checkFunctionApplication' m fnTy args ret+checkFunctionApplication :: Type -> [Value] -> Type -> Subst ()+checkFunctionApplication fnTy args ret = rethrow errorMessage $ checkFunctionApplication' fnTy args ret where errorMessage msg = "Error applying function of type " ++ prettyPrintType fnTy@@ -728,46 +703,46 @@ ++ ", expecting value of type " ++ prettyPrintType ret ++ ":\n" ++ msg -inferFunctionApplication :: M.Map Ident Type -> Type -> [Type] -> Subst Check Type-inferFunctionApplication m (Function argTys retTy) args = do+inferFunctionApplication :: Type -> [Type] -> Subst Type+inferFunctionApplication (Function argTys retTy) args = do guardWith "Incorrect number of function arguments" (length args == length argTys) zipWithM subsumes args argTys return retTy-inferFunctionApplication m (ForAll ident ty) args = do+inferFunctionApplication (ForAll ident ty) args = do replaced <- replaceVarWithUnknown ident ty- inferFunctionApplication m replaced args-inferFunctionApplication m u@(TUnknown _) args = do+ inferFunctionApplication replaced args+inferFunctionApplication u@(TUnknown _) args = do ret <- fresh args' <- mapM replaceAllVarsWithUnknowns args u ~~ Function args' ret return ret-inferFunctionApplication m (SaturatedTypeSynonym name tyArgs) args = do+inferFunctionApplication (SaturatedTypeSynonym name tyArgs) args = do ty <- expandTypeSynonym name tyArgs- inferFunctionApplication m ty args-inferFunctionApplication _ fnTy args = throwError $ "Cannot apply function of type "+ inferFunctionApplication ty args+inferFunctionApplication fnTy args = throwError $ "Cannot apply function of type " ++ prettyPrintType fnTy ++ " to argument(s) of type(s) " ++ intercalate ", " (map prettyPrintType args) -checkFunctionApplication' :: M.Map Ident Type -> Type -> [Value] -> Type -> Subst Check ()-checkFunctionApplication' m (Function argTys retTy) args ret = do+checkFunctionApplication' :: Type -> [Value] -> Type -> Subst ()+checkFunctionApplication' (Function argTys retTy) args ret = do guardWith "Incorrect number of function arguments" (length args == length argTys)- zipWithM (check m) args argTys+ zipWithM (check) args argTys retTy `subsumes` ret-checkFunctionApplication' m (ForAll ident ty) args ret = do+checkFunctionApplication' (ForAll ident ty) args ret = do replaced <- replaceVarWithUnknown ident ty- checkFunctionApplication m replaced args ret-checkFunctionApplication' m u@(TUnknown _) args ret = do- tyArgs <- mapM (\arg -> infer m arg >>= replaceAllVarsWithUnknowns) args+ checkFunctionApplication replaced args ret+checkFunctionApplication' u@(TUnknown _) args ret = do+ tyArgs <- mapM (\arg -> infer arg >>= replaceAllVarsWithUnknowns) args u ~~ Function tyArgs ret-checkFunctionApplication' m (SaturatedTypeSynonym name tyArgs) args ret = do+checkFunctionApplication' (SaturatedTypeSynonym name tyArgs) args ret = do ty <- expandTypeSynonym name tyArgs- checkFunctionApplication' m ty args ret-checkFunctionApplication' _ fnTy args ret = throwError $ "Applying a function of type "+ checkFunctionApplication' ty args ret+checkFunctionApplication' fnTy args ret = throwError $ "Applying a function of type " ++ prettyPrintType fnTy ++ " to argument(s) " ++ intercalate ", " (map prettyPrintValue args) ++ " does not yield a value of type " ++ prettyPrintType ret ++ "." -subsumes :: Type -> Type -> Subst Check ()+subsumes :: Type -> Type -> Subst () subsumes (ForAll ident ty1) ty2 = do replaced <- replaceVarWithUnknown ident ty1 replaced `subsumes` ty2
tests/Main.hs view
@@ -58,13 +58,14 @@ main :: IO () main = do cd <- getCurrentDirectory+ putStrLn $ cd let examples = cd ++ pathSeparator : "examples" let passing = examples ++ pathSeparator : "passing" passingTestCases <- getDirectoryContents passing- forM_ passingTestCases $ \inputFile -> when (".ps" `isSuffixOf` inputFile) $+ forM_ passingTestCases $ \inputFile -> when (".purs" `isSuffixOf` inputFile) $ assertCompiles (passing ++ pathSeparator : inputFile) let failing = examples ++ pathSeparator : "failing" failingTestCases <- getDirectoryContents failing- forM_ failingTestCases $ \inputFile -> when (".ps" `isSuffixOf` inputFile) $+ forM_ failingTestCases $ \inputFile -> when (".purs" `isSuffixOf` inputFile) $ assertDoesNotCompile (failing ++ pathSeparator : inputFile) exitSuccess