diff --git a/src/Checker.cc b/src/Checker.cc index 3561ec4f2..24c3b096f 100644 --- a/src/Checker.cc +++ b/src/Checker.cc @@ -336,6 +336,19 @@ std::tuple Checker::inferTypeExpr(TypeEnv& Env, TypeExpres break; } + case NodeKind::ArrowTypeExpression: + { + auto Arrow = static_cast(TE); + auto [ReturnOut, ReturnTy] = inferTypeExpr(Env, Arrow->ReturnType); + Ty = ReturnTy; + for (auto PT: Arrow->ParamTypes) { + auto [ParamOut, ParamTy] = inferTypeExpr(Env, PT); + mergeTo(Out, ParamOut); + Ty = new TFun(ParamTy, Ty); + } + break; + } + default: ZEN_UNREACHABLE