68 lines
1.6 KiB
C
68 lines
1.6 KiB
C
#include "ctx.h"
|
|
|
|
#include <ivy/lang/ast.h>
|
|
|
|
bool parse_for(struct ivy_parser *parser, struct ivy_ast_node **out)
|
|
{
|
|
if (!parse_keyword(parser, IVY_KW_FOR)) {
|
|
parse_trace(parser, "expected `for` keyword");
|
|
return false;
|
|
}
|
|
|
|
struct ivy_ast_for_loop_node *for_node = create_node(IVY_AST_FOR_LOOP);
|
|
|
|
if (!parse_expr(parser, EXPR_PARSE_STOP_KW_IN, &for_node->n_iterator)) {
|
|
parse_trace(parser, "invalid for-loop iterator");
|
|
return false;
|
|
}
|
|
|
|
if (!parse_keyword(parser, IVY_KW_IN)) {
|
|
parse_trace(parser, "expected `in` after for-loop iterator");
|
|
return false;
|
|
}
|
|
|
|
if (!parse_expr(parser, EXPR_PARSE_STOP_LEFT_BRACKET, &for_node->n_iterable)) {
|
|
parse_trace(parser, "invalid for-loop iterable");
|
|
return false;
|
|
}
|
|
|
|
if (!parse_block(parser, &for_node->n_body)) {
|
|
parse_trace(parser, "invalid for-loop body");
|
|
return false;
|
|
}
|
|
|
|
*out = generic_node(for_node);
|
|
return true;
|
|
}
|
|
|
|
bool parse_for_inline(
|
|
struct ivy_parser *parser, struct ivy_ast_node *body,
|
|
struct ivy_ast_node **out)
|
|
{
|
|
if (!parse_keyword(parser, IVY_KW_FOR)) {
|
|
parse_trace(parser, "expected `for` keyword");
|
|
return false;
|
|
}
|
|
|
|
struct ivy_ast_for_loop_node *for_node = create_node(IVY_AST_FOR_LOOP);
|
|
for_node->n_body = body;
|
|
|
|
if (!parse_expr(parser, EXPR_PARSE_STOP_KW_IN, &for_node->n_iterator)) {
|
|
parse_trace(parser, "invalid for-loop iterator");
|
|
return false;
|
|
}
|
|
|
|
if (!parse_keyword(parser, IVY_KW_IN)) {
|
|
parse_trace(parser, "expected `in` after for-loop iterator");
|
|
return false;
|
|
}
|
|
|
|
if (!parse_expr(parser, EXPR_PARSE_NORMAL, &for_node->n_iterable)) {
|
|
parse_trace(parser, "invalid for-loop iterable");
|
|
return false;
|
|
}
|
|
|
|
*out = generic_node(for_node);
|
|
return true;
|
|
}
|