Added support for comments using //

1 files changed, 14 insertions, 1 deletions

diff --git a/lib/Lexer/Lexer.cc b/lib/Lexer/Lexer.cc
index 9ac01d9..c50897b 100644
--- a/lib/Lexer/Lexer.cc
+++ b/lib/Lexer/Lexer.cc
@@ -67,7 +67,20 @@ int Lexer::gettok() {
         return tok_eq;
     }
 
-    // for comments
+    if (LastChar == '/') {
+        int NextChar = getchar();
+        if (NextChar == '/') {
+            do {
+                LastChar = getchar();
+            } while (LastChar != EOF && LastChar != '\n' && LastChar != '\r');
+            
+            if (LastChar != EOF)
+                return gettok();
+        } else {
+            LastChar = NextChar;
+            return '/';
+        }
+    }
     
     if (LastChar == ':') {
         LastChar = getchar();