enums

1 files changed, 31 insertions, 0 deletions

diff --git a/src/main.c b/src/main.c
index 10a78fc..a022d75 100644
--- a/src/main.c
+++ b/src/main.c
@@ -423,6 +423,7 @@ new_num(int val, Token *tok)
 static Node *stmt(Token **rest, Token *tok);
 static Node *function(Token **rest, Token *tok);
 static void define(Token **rest, Token *tok);
+static void denum(Token **rest, Token *tok);
 static Node *global(Token **rest, Token *tok);
 static Node *local(Token **rest, Token *tok);
 static Node *expr(Token **rest, Token *tok);
@@ -492,6 +493,31 @@ define(Token **rest, Token *tok)
 	defines = store;
 }
 
+static void
+denum(Token **rest, Token *tok)
+{
+	tok = skip(tok, "enum");
+	Node *store = new_word(tok);
+	int v = 0;
+	store->rhs = new_num(v++, tok);
+	store->type = NOD_DEFINE;
+	store->next = defines;
+	defines = store;
+	tok = tok->next;
+
+	while (tok->type != TOK_EOF && !equal(tok, ";")) {
+		tok = skip(tok, ",");
+		store = new_word(tok);
+		store->rhs = new_num(v++, tok);
+		store->type = NOD_DEFINE;
+		store->next = defines;
+		defines = store;
+		tok = tok->next;
+	}
+
+	*rest = skip(tok, ";");
+}
+
 static int const_expr(Node *node);
 
 static Node *
@@ -1494,6 +1520,11 @@ parse(Token *tok)
 			continue;
 		}
 
+		if (equal(tok, "enum")) {
+			denum(&tok, tok);
+			continue;
+		}
+
 		if (equal(tok, "global")) {
 			cur = cur->next = global(&tok, tok);
 			while (cur->next != NULL)