Adding basic strings and symbols parsing

1 files changed, 37 insertions, 0 deletions

diff --git a/src/main.c b/src/main.c
index 8892a60..75faaf6 100644
--- a/src/main.c
+++ b/src/main.c
@@ -62,6 +62,8 @@ typedef enum NodeKind {
     NODE_BITLSHIFT,
     NODE_BITRSHIFT,
     // Literals.
+    NODE_STRING,
+    NODE_SYMBOL,
     NODE_NUM_INT,
     NODE_NUM_FLOAT,
     NODE_TRUE,
@@ -93,6 +95,8 @@ Str node_str[] = {
     [NODE_BITLSHIFT] = cstr("BITLSHIFT"),
     [NODE_BITRSHIFT] = cstr("BITRSHIFT"),
     // Literals.
+    [NODE_STRING] = cstr("STRING"),
+    [NODE_SYMBOL] = cstr("SYMBOL"),
     [NODE_NUM_INT] = cstr("INT"),
     [NODE_NUM_FLOAT] = cstr("FLOAT"),
     [NODE_TRUE] = cstr("TRUE"),
@@ -109,6 +113,8 @@ typedef struct Node {
     union {
         f64 d;
         sz i;
+        Str str;
+        Str sym;
     } value;
     struct Node *left;
     struct Node *right;
@@ -176,6 +182,8 @@ void parse_unary(Parser *parser);
 void parse_binary(Parser *parser);
 void parse_number(Parser *parser);
 void parse_literal(Parser *parser);
+void parse_string(Parser *parser);
+void parse_symbol(Parser *parser);
 
 ParseRule parse_rules[] = {
     [TOK_LPAREN] = {parse_grouping, NULL, PREC_NONE},
@@ -202,6 +210,8 @@ ParseRule parse_rules[] = {
     [TOK_BITLSHIFT] = {NULL, parse_binary, PREC_BITSHIFT},
     [TOK_BITRSHIFT] = {NULL, parse_binary, PREC_BITSHIFT},
     // Literals.
+    [TOK_STRING] = {parse_string, NULL, PREC_NONE},
+    [TOK_SYMBOL] = {parse_symbol, NULL, PREC_NONE},
     [TOK_NUM_INT] = {parse_number, NULL, PREC_NONE},
     [TOK_NUM_FLOAT] = {parse_number, NULL, PREC_NONE},
     [TOK_TRUE] = {parse_literal, NULL, PREC_NONE},
@@ -377,6 +387,31 @@ parse_number(Parser *parser) {
 }
 
 void
+parse_string(Parser *parser) {
+    Token prev = parser->previous;
+#if DEBUG == 1
+    print("parsing string ");
+    print_token(prev);
+#endif
+    Node *node = node_alloc(NODE_STRING, prev, parser->storage);
+    node->value.str = str_remove_prefix(prev.val, cstr("\""));
+    node->value.str = str_remove_suffix(node->value.str, cstr("\""));
+    array_push(parser->nodes, node, parser->storage);
+}
+
+void
+parse_symbol(Parser *parser) {
+    Token prev = parser->previous;
+#if DEBUG == 1
+    print("parsing string ");
+    print_token(prev);
+#endif
+    Node *node = node_alloc(NODE_STRING, prev, parser->storage);
+    node->value.sym = prev.val;
+    array_push(parser->nodes, node, parser->storage);
+}
+
+void
 parse_grouping(Parser *parser) {
 #if DEBUG == 1
     print("parsing group  ");
@@ -396,6 +431,8 @@ graph_node(Node *node) {
     switch (node->kind) {
         case NODE_NUM_INT: print("| Value: %d", node->value.i); break;
         case NODE_NUM_FLOAT: print("| Value: %f{2}", node->value.d); break;
+        case NODE_STRING: print("| Value: %s", node->value.str); break;
+        case NODE_SYMBOL: print("| Value: %s", node->value.sym); break;
         default: break;
     }
     print("| Line: %d | Col: %d ", node->line, node->col);