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

reassignment

Browse Source
This commit is contained in:
SpookyDervish
2026-04-28 08:16:03 +10:00
parent 8c8d7f4cd3
commit 6c9d7bd316
2 changed files with 38 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/ast.h
View File

@@ -25,6 +25,7 @@ typedef enum {
// statements
AST_EXPRESSION_STATEMENT,
AST_ASSIGN_STATEMENT,
AST_REASSIGN_STATEMENT,
AST_WHILE_STATEMENT,
AST_FOR_STATEMENT,
@@ -48,6 +49,7 @@ struct CometASTNode {
struct AST_EXPRESSION_STATEMENT { CometASTNode* expression; } AST_EXPRESSION_STATEMENT;
struct AST_ASSIGN_STATEMENT { CometASTNode* ident; CometASTNode* expression; CometASTNode* type; } AST_ASSIGN_STATEMENT;
struct AST_REASSIGN_STATEMENT { CometASTNode* ident; CometASTNode* expression; } AST_REASSIGN_STATEMENT;
struct AST_WHILE_STATEMENT { CometASTNode* expression; CometASTNode* program; } AST_WHILE_STATEMENT;
struct AST_FOR_STATEMENT { CometASTNode* ident; CometASTNode* start; CometASTNode* end; CometASTNode* step; CometASTNode* program; } AST_FOR_STATEMENT;
} data;

36
src/parser.c
View File

@@ -19,12 +19,14 @@ const CometTokenPrecedencePair PRECEDENCES[] = {
};
ResultType(astNodePtr, charptr) parseIntLiteral(CometParser* parser);
ResultType(astNodePtr, charptr) parseIdentifier(CometParser* parser);
ResultType(astNodePtr, charptr) parseFloatLiteral(CometParser* parser);
ResultType(astNodePtr, charptr) parseTypeName(CometParser* parser);
ResultType(astNodePtr, charptr) parseGroupedExpression(CometParser* parser);
const CometPrefixParseFn PREFIX_PARSE_FUNCTIONS[] = {
{CT_INT_LITERAL, parseIntLiteral},
{CT_FLOAT_LITERAL, parseFloatLiteral},
{CT_IDENT, parseIdentifier},
{CT_OPEN_PAREN, parseGroupedExpression},
{CT_TYPE_NAME, parseTypeName},
};
@@ -225,6 +227,11 @@ void printNode(CometASTNode* node) {
printf(" = ");
printNode(node->data.AST_ASSIGN_STATEMENT.expression);
break;
case AST_REASSIGN_STATEMENT:
printNode(node->data.AST_REASSIGN_STATEMENT.ident);
printf(" = ");
printNode(node->data.AST_REASSIGN_STATEMENT.expression);
break;
case AST_WHILE_STATEMENT:
printf("while ");
printNode(node->data.AST_WHILE_STATEMENT.expression);
@@ -322,6 +329,10 @@ ResultType(astNodePtr, charptr) parseTypeName(CometParser* parser) {
return Success(astNodePtr, charptr, AST_NODE(AST_TYPE_NAME, parser->currentToken->value.literal));
}
ResultType(astNodePtr, charptr) parseIdentifier(CometParser* parser) {
return Success(astNodePtr, charptr, AST_NODE(AST_IDENTIFIER, parser->currentToken->value.literal));
}
// -- STATEMENT METHODS -- //
ResultType(astNodePtr, charptr) parseExpressionStatement(CometParser* parser) {
ResultType(astNodePtr, charptr) expr = parseExpression(parser, PRECEDENCE_LOWEST);
@@ -492,6 +503,28 @@ ResultType(astNodePtr, charptr) parseForStatement(CometParser* parser) {
return Success(astNodePtr, charptr, stmt);
}
ResultType(astNodePtr, charptr) parseReassignStatement(CometParser* parser) {
// basic format
// x = x + 1
CometASTNode* ident = AST_NODE(AST_IDENTIFIER, parser->currentToken->value.literal);
ResultType(int, charptr) expectEq = expectPeek(parser, CT_EQ);
if (expectEq.error) {
return Error(astNodePtr, charptr, expectEq.as.error);
}
parserNextToken(parser);
ResultType(astNodePtr, charptr) expr = parseExpression(parser, PRECEDENCE_LOWEST);
if (expr.error) {
return expr;
}
CometASTNode* stmt = AST_NODE(AST_REASSIGN_STATEMENT, ident, expr.as.success);
return Success(astNodePtr, charptr, stmt);
}
ResultType(astNodePtr, charptr) parseKeyword(CometParser* parser) {
char* keyword = parser->currentToken->value.literal;
@@ -512,6 +545,9 @@ ResultType(astNodePtr, charptr) parseStatement(CometParser* parser) {
case CT_TYPE_NAME:
return parseAssignmentStatement(parser);
case CT_IDENT:
return parseReassignStatement(parser);
case CT_KEYWORD:
return parseKeyword(parser);