Allowed parsing of variable defenitions to AST.

src/main.rs

@@ -5,6 +5,8 @@ pub mod codegen;
 use crate::codegen::fasm::*;
 use crate::lex::tok::*;
 use crate::parse::ast::*;
+use crate::parse::parse::*;
+use logos::Logos;
 
 pub mod parse;
 
@@ -43,4 +45,8 @@ fn main() {
 
     
     // println!("{}", fc);
+
+    let mut lexer = Token::lexer("let a = b;");
+    println!("{:?}", lex_str("let a = b;"));
+    println!("{:?}", parse_var_declaration(lexer));
 }

src/parse/mod.rs

@@ -1 +1,2 @@
+pub mod parse;
 pub mod ast;

src/parse/parse.rs

@@ -2,33 +2,32 @@ use super::ast::*;
 use crate::lex::tok::*;
 use logos::Lexer;
 
-fn parse_var_declaration(tokens: Lexer) -> Option<Expr> {
+macro_rules! unwrap {
-    let tok = None;
+    ($var:expr) => {
-    match tokens.next() {
+	$var.next().unwrap().unwrap()
-	Let => {
+    }
-	    match tokens.next() {
+}
-		Identifier => {
+
+pub fn parse_var_declaration(mut tokens: Lexer<Token>) -> Option<Expr> {
+    let mut tok = None;
+    if unwrap!(tokens) == Let {
+	if unwrap!(tokens) == Identifier {
 	    let name = tokens.slice();
+
 	    println!("{:?}", name);
-		    match tokens.next() {
+	    if unwrap!(tokens) == Equal {
-			Equal => {
+		let temp_token = unwrap!(tokens);
-			    match tokens.next() {
+		
-				Number(n) => {
+		if let Number(n) = temp_token {
 		    let value = Value::Number(n);
 		    println!("{:?}", value);
-				    return Expr::VarDefenition {name, value};
+		    tok = Some(Expr::VarDefinition(VarDefinition {name, value}));
-				},
+		} else if temp_token == Identifier {
-
-				Identifier => {
 		    let value = Value::Var(VarReference { name: tokens.slice() });
-				    return Expr::Vardefenition {name, value};
+		    tok = Some(Expr::VarDefinition(VarDefinition {name, value}));
 		}
 	    }
 	}
     }
-		}
+    tok
-	    }
-	}
-	_ => None
-    }
 }