From 3c4cce0883abbfee6b851514472b906493458834 Mon Sep 17 00:00:00 2001
From: Garritt McCune <glmccune@gmail.com>
Date: Mon, 16 May 2022 14:27:49 +0000
Subject: [PATCH] Added assembler directives as a class of token.

---
 includes/token.h | 3 ++-
 src/token.c      | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/includes/token.h b/includes/token.h
index 792420a..44e6efe 100644
--- a/includes/token.h
+++ b/includes/token.h
@@ -34,7 +34,8 @@ typedef enum {
     None = 0,
     Reg = 1,
     Immediate16 = 2,
-    Opcode = 4
+    Opcode = 4,
+    Directive = 8
 } TokenClass;
 
 typedef struct {
diff --git a/src/token.c b/src/token.c
index 03f7c15..5c1d1fb 100644
--- a/src/token.c
+++ b/src/token.c
@@ -22,6 +22,7 @@ Token* CreateToken(char* lexeme, void* value, int lineNumber, TokenType type) {
 TokenClass GetTokenClass(TokenType type) {
     if (type >= R1 && type <= R8) return Reg;
     if (type >= COPY && type <= OUT) return Opcode;
+    if (type >= DB && type <= ORG) return Directive;
     
     switch(type) {
         case STRING: