void parseBlockExp (Lexer& lex, ExpList& list)
{
switch (lex.current().tok)
{
case tSemicolon:
lex.advance();
break;
case tLet:
list.push_back(parseLet(lex));
lex.eat(tSemicolon);
break;
case tLCurl:
list.push_back(parseBlock(lex));
break;
case tIf:
list.push_back(parseCond(lex));
break;
case tLoop:
list.push_back(parseLoop(lex));
break;
case tFor:
list.push_back(parseFor(lex));
break;
// everthing else does
default:
list.push_back(parseAssign(lex, parseExp(lex)));
if (lex.current() != tSemicolon)
lex.expect(tRCurl);
else
lex.eat(tSemicolon);
break;
}
}
ExpPtr parseFor (Lexer& lex)
{
Span spStart, spEnd;
ExpPtr val1, val2, body;
spStart = lex.eat(tFor).span;
auto var = lex.eat(tIdent).str;
lex.eat(tColon);
val1 = parseExp(lex);
if (lex.current() == tArrow)
{
lex.advance();
val2 = parseExp(lex);
}
else
val2 = nullptr;
body = parseBlock(lex);
spEnd = body->span;
if (val2 == nullptr)
return Exp::make(eForEach, var,
{ val1, body }, spStart + spEnd);
else
return Exp::make(eForRange, var,
{ val1, val2, body }, spStart + spEnd);
}
ExpPtr parseLambda (Lexer& lex)
{
Span spStart, spEnd;
SigPtr sig;
ExpPtr body;
spStart = lex.current().span;
if (lex.current() == tFunc)
{
lex.advance();
sig = parseSigParens(lex);
body = parseBlock(lex);
}
else
{
lex.eat(tLambda);
sig = parseSig(lex, true);
lex.eat(tArrow);
body = parseExp(lex);
}
spEnd = body->span;
return Exp::make(eLambda, sig->toSigType(), "", { body }, spStart + spEnd);
}
ImportDecl parseImportDecl (Lexer& lex)
{
Span spStart, spEnd;
spStart = lex.eat(tImport).span;
auto nameTok = lex.eat(tIdent, tString);
auto name = nameTok.str;
spEnd = nameTok.span;
return ImportDecl
{
.isPublic = false,
.name = name,
.span = spStart + spEnd
};
}
FuncDecl parseFuncDecl (Lexer& lex)
{
Span spStart, spEnd;
spStart = lex.eat(tFunc).span;
auto name = lex.eat(tIdent).str;
auto sig = parseSigParens(lex);
ExpPtr body;
/* if (lex.current() == tEqual)
{
lex.advance();
body = parseExp(lex);
}
else */
body = parseBlock(lex);
spEnd = sig->span;
return FuncDecl
{
.isPublic = false,
.name = name,
.signature = sig,
.body = body,
.span = spStart + spEnd
};
}
static FuncDecl parseConstructor (Lexer& lex)
{
Span spStart, spEnd;
FuncDecl result;
result.name = lex.current().str;
spStart = lex.eat(tIdent).span;
result.signature = parseSigParens(lex);
spEnd = result.signature->span;
result.span = spStart + spEnd;
result.body = nullptr;
return result;
}
ExpPtr parseMem (Lexer& lex, ExpPtr in)
{
lex.eat(tDot);
auto mem = lex.eat(tIdent);
return Exp::make(eMem, mem.str, { in },
in->span + mem.span);
}
TyPtr parseTypePoly (Lexer& lex, Span& sp)
{
Span spStart, spEnd;
spStart = lex.eat(tLambda).span;
spEnd = lex.current().span;
auto name = lex.eat(tIdent).str;
sp = spStart + spEnd;
return Ty::makePoly(name);
}
TyPtr parseTypeList (Lexer& lex, Span& sp)
{
Span spStart, spEnd;
spStart = lex.eat(tLBrack).span;
auto inner = parseType(lex, spEnd);
spEnd = lex.eat(tRBrack).span;
sp = spStart + spEnd;
return Ty::makeConcrete("List", { inner });
}
SigPtr parseSigParens (Lexer& lex)
{
Span spStart, spEnd;
spStart = lex.eat(tLParen).span;
auto res = parseSig(lex, false);
spEnd = lex.eat(tRParen).span;
res->span = spStart + spEnd;
return res;
}
ExpPtr parseBlock (Lexer& lex)
{
Span spStart, spEnd;
ExpList exps;
spStart = lex.eat(tLCurl).span;
while (lex.current() != tRCurl)
parseBlockExp(lex, exps);
spEnd = lex.eat(tRCurl).span;
return Exp::make(eBlock, exps, spStart + spEnd);
}
ExpPtr parseiGet (Lexer& lex)
{
// ^get <ty> <idx> <term>
Span spStart, spEnd, spType;
spStart = lex.eat(tiGet).span;
auto ty = parseType(lex, spType);
auto idx = lex.eat(tNumberInt).valueInt;
auto body = parseTerm(lex);
spEnd = body->span;
auto e = Exp::make(eiGet, idx, { body }, spStart + spEnd);
e->setType(ty);
return e;
}
TypeDecl parseTypeDecl (Lexer& lex)
{
Span spStart, spEnd, spType;
std::vector<FuncDecl> ctors;
spStart = lex.eat(tType).span;
auto ty = parseType(lex, spType);
if (ty->kind != tyConcrete)
throw spType.die("invalid form of type in declaration");
for (auto sub = ty->subtypes; !sub.nil(); ++sub)
{
auto sty = sub.head();
if (sty->kind != tyPoly)
throw spType.die("arguments to type must be polytypes");
for (auto sty2 : sub.tail())
if (sty2->kind == tyPoly &&
sty->name == sty2->name)
throw spType.die("identical polytypes in declaration");
}
lex.eat(tEqual);
for (;;)
{
auto ctor = parseConstructor(lex);
spEnd = ctor.span;
ctors.push_back(ctor);
if (lex.current() == tComma)
lex.advance();
else
break;
}
return TypeDecl
{
ty->name,
ty->subtypes,
std::move(ctors),
spStart + spEnd
};
}
ExpPtr parseiCall (Lexer& lex)
{
// ^call <ty> <name>
Span spStart, spEnd, spType;
spStart = lex.eat(tiCall).span;
auto ty = parseType(lex, spType);
spEnd = lex.current().span;
auto tag = lex.eat(tString).str;
if (ty->kind != tyConcrete || ty->name != "Fn")
throw spType.die("^call expects function type");
auto e = Exp::make(eiCall, tag, {}, spStart + spEnd);
e->setType(ty);
return e;
}
ExpPtr parseLet (Lexer& lex)
{
Span spStart, spEnd;
std::string name;
TyPtr ty;
ExpPtr init;
// 'let' <var> '=' <exp> ';'
spStart = lex.eat(tLet).span;
parseVar(lex, name, ty, spEnd);
lex.eat(tEqual);
init = parseExp(lex);
spEnd = init->span;
auto e = Exp::make(eLet, ty, name, { init }, spStart + spEnd);
return e;
}
static void parseTypeTupleRaw (Lexer& lex, std::vector<TyPtr>& out, Span& sp)
{
Span spStart, spEnd;
spStart = lex.eat(tLParen).span;
while (lex.current() != tRParen)
{
out.push_back(parseType(lex, spEnd));
if (lex.current() == tComma)
lex.advance();
else
break;
}
spEnd = lex.eat(tRParen).span;
sp = spStart + spEnd;
}
ExpPtr parseList (Lexer& lex)
{
Span spStart, spEnd;
ExpList vals;
spStart = lex.eat(tLBrack).span;
while (lex.current() != tRBrack)
{
vals.push_back(parseExp(lex));
if (lex.current() == tComma)
lex.advance();
else
break;
}
spEnd = lex.eat(tRBrack).span;
return Exp::make(eList, std::move(vals), spStart + spEnd);
}
ExpPtr parseAssign (Lexer& lex, ExpPtr left)
{
if (lex.current() != tEqual)
return left;
lex.eat(tEqual);
auto right = parseExp(lex);
lex.expect(tSemicolon);
return Exp::make(eAssign, { left, right }, left->span + right->span);
}
ExpPtr parseTuple (Lexer& lex)
{
ExpList exps;
ExpPtr e;
Span spStart, spEnd;
spStart = lex.eat(tLParen).span;
while (lex.current() != tRParen)
{
e = parseExp(lex);
exps.push_back(e);
if (lex.current() == tComma)
lex.advance();
else
break;
}
spEnd = lex.eat(tRParen).span;
return Exp::make(eTuple, exps, spStart + spEnd);
}
ExpPtr parseCond (Lexer& lex)
{
Span spStart, spEnd;
ExpPtr expCond, expThen, expElse;
bool blockRequired = true;
spStart = lex.eat(tIf).span;
expCond = parseExp(lex);
if (lex.current() == tThen)
{
lex.advance();
blockRequired = false;
}
expThen = blockRequired ?
parseBlock(lex) :
parseExp(lex);
if (lex.current() == tElse)
{
lex.advance();
if (blockRequired)
{
// if {} else {}
// if {} else if ...
if (lex.current() != tIf)
lex.expect(tLCurl);
expElse = parseExp(lex);
}
else
expElse = parseExp(lex);
spEnd = expElse->span;
}
else if (!blockRequired)
lex.expect(tElse);
else
{
auto expUnit = Exp::make(eTuple, {}, spStart);
expThen->subexps.push_back(expUnit);
expElse = expUnit; // no else => unit (zero tuple)
spEnd = expThen->span;
}
return Exp::make(eCond, { expCond, expThen, expElse }, spStart + spEnd);
}
TyPtr parseTypeConcrete (Lexer& lex, Span& sp)
{
Span spStart, spEnd;
std::vector<TyPtr> sub;
spStart = lex.current().span;
auto kind = lex.eat(tIdent).str;
if (lex.current() == tLParen)
parseTypeTupleRaw(lex, sub, spEnd);
sp = spStart + spEnd;
return Ty::makeConcrete(kind, sub);
}
void parseVar (Lexer& lex, std::string& name, TyPtr& ty, Span& sp)
{
sp = lex.current().span;
name = lex.eat(tIdent).str;
if (lex.current() == tColon)
{
Span spEnd;
lex.advance();
ty = parseType(lex, spEnd);
sp = sp + spEnd;
}
else
ty = Ty::makeWildcard();
}
void parsePublic (Lexer& lex, GlobProto& proto)
{
Span span = lex.eat(tPub).span;
if (lex.current() == tFunc)
{
auto fn = parseFuncDecl(lex);
fn.isPublic = true;
fn.span = span + fn.span;
proto.funcs.push_back(fn);
}
else if (lex.current() == tImport)
{
auto imp = parseImportDecl(lex);
imp.isPublic = true;
imp.span = span + imp.span;
proto.imports.push_back(imp);
}
else
lex.unexpect();
}
ExpPtr parseLoop (Lexer& lex)
{
Span spStart, spEnd;
ExpPtr body;
spStart = lex.eat(tLoop).span;
ExpList sub;
sub.reserve(2);
if (lex.current() == tLCurl)
{
sub.push_back(body = parseBlock(lex));
}
else
{
sub.push_back(parseExp(lex));
sub.push_back(body = parseBlock(lex));
}
spEnd = body->span;
return Exp::make(eLoop, std::move(sub), spStart + spEnd);
}
