wash/ivy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

lang: ast: implement parsing of subscript operator

Browse Source
This commit is contained in:
max 🍓
2024-12-08 17:08:29 +00:00
parent 753afd7d87
commit 8b8a97c9a6
3 changed files with 23 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
lang/ast/expr/arith.c
View File

@@ -696,10 +696,23 @@ struct token_parse_result arith_parse_left_bracket(
return PARSE_RESULT(IVY_ERR_BAD_SYNTAX, 0);
}
if (state->s_prev_component == EXPR_CMP_OPERAND) {
return PARSE_RESULT(IVY_ERR_BAD_SYNTAX, 0);
if (state->s_prev_component == EXPR_CMP_OPERAND || state->s_prev_component == EXPR_CMP_MSG) {
/* subscript operator */
state->s_prev_token = IVY_SYM_LEFT_BRACKET;
struct ivy_ast_op_node *subscript = (struct ivy_ast_op_node *)ast_node_create(IVY_AST_OP);
subscript->n_op = ivy_operator_get_by_id(IVY_OP_SUBSCRIPT);
arith_push_operator(state, (struct ivy_ast_node *)subscript);
struct expr_parser_state *index = (struct expr_parser_state *)parser_push_state(ctx, IVY_AST_EXPR, 0);
expr_add_terminator(index, IVY_SYM_RIGHT_BRACKET);
index->s_subexpr_depth = state->s_subexpr_depth + 1;
return PARSE_RESULT(IVY_OK, 0);
}
/* lambda */
state->s_prev_token = IVY_SYM_LEFT_BRACKET;
state->s_type = EXPR_TYPE_ARITH;
@@ -713,6 +726,11 @@ struct token_parse_result arith_parse_right_bracket(
struct expr_parser_state *state
= parser_get_state(ctx, struct expr_parser_state);
if (state->s_prev_token == IVY_SYM_LEFT_BRACKET) {
/* we've just parsed a subscript operator. nothing further to do here. */
return PARSE_RESULT(IVY_OK, 0);
}
state->s_prev_token = IVY_SYM_RIGHT_BRACKET;
struct token_parse_result result = expr_finalise_and_return(ctx, state);
result.r_flags = PARSE_REPEAT_TOKEN;