From c389b44bf8dd6fa644948a204edda113b399d6b6 Mon Sep 17 00:00:00 2001
From: greg <greg.shuflin@protonmail.com>
Date: Tue, 27 Dec 2016 17:33:24 -0800
Subject: [PATCH] Compile statements with variables

---
 main.schala        |  4 +++-
 src/compilation.rs | 20 ++++++++++++++++++--
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/main.schala b/main.schala
index 45daf6c..eed299c 100644
--- a/main.schala
+++ b/main.schala
@@ -1,5 +1,7 @@
 
 
 fn main()
-1 + 2 + 3 + 4 * 2
+a = 4
+b = 20
+a + b
 end
diff --git a/src/compilation.rs b/src/compilation.rs
index daa6124..6218b3e 100644
--- a/src/compilation.rs
+++ b/src/compilation.rs
@@ -109,8 +109,12 @@ impl CodeGen for ASTNode {
 impl CodeGen for Function {
     fn codegen(&self, data: &mut CompilationData) -> LLVMValueRef {
         let ref body = self.body;
-        let first = body.get(0).unwrap();
-        first.codegen(data)
+        let int_type = LLVMWrap::Int64TypeInContext(data.context);
+        let mut ret = LLVMWrap::ConstInt(int_type, 0, false);
+        for expr in body {
+            ret = expr.codegen(data);
+        }
+        ret
     }
 }
 
@@ -121,6 +125,18 @@ impl CodeGen for Expression {
         let int_type = LLVMWrap::Int64TypeInContext(data.context);
 
         match self {
+            &Variable(ref name) => {
+                *data.variables.get(name).unwrap()
+            },
+            &BinExp(ref op, ref left, ref right) if op == "=" => {
+                if let Variable(ref name) = **left {
+                    let new_value = right.codegen(data);
+                    data.variables.insert(name.clone(), new_value);
+                    new_value
+                } else {
+                    panic!("Bad variable assignment")
+                }
+            },
             &BinExp(ref op, ref left, ref right) => {
                 let lhs = left.codegen(data);
                 let rhs = right.codegen(data);