From f82c6199c0b7fe2c8dc4c046c542dfa24f5cf616 Mon Sep 17 00:00:00 2001
From: greg <greg.shuflin@protonmail.com>
Date: Tue, 10 Oct 2017 21:23:24 -0700
Subject: [PATCH] Change around some stuff

---
 src/schala_lang/type_check.rs | 35 ++++++++++++++++++++++++++---------
 1 file changed, 26 insertions(+), 9 deletions(-)

diff --git a/src/schala_lang/type_check.rs b/src/schala_lang/type_check.rs
index 01af834..ad774b1 100644
--- a/src/schala_lang/type_check.rs
+++ b/src/schala_lang/type_check.rs
@@ -201,19 +201,36 @@ impl TypeContext {
     use self::TypeConst::*;
 
     Ok(match (&expr.0, &expr.1) {
-      (ref _t, &Some(ref anno)) => {
-        self.from_anno(anno)// TODO make this better,
+      (&IntLiteral(_), anno) => {
+        match *anno {
+          None => TConst(Integer),
+          Some(ref t) => self.from_anno(t)
+        }
+      }
+      (&FloatLiteral(_), anno) => {
+        match *anno {
+          None => TConst(Float),
+          Some(ref t) => self.from_anno(t),
+        }
       },
-      (&IntLiteral(_), _) => TConst(Integer),
-      (&FloatLiteral(_), _) => TConst(Float),
-      (&StringLiteral(_), _) => TConst(StringT),
-      (&BoolLiteral(_), _) => TConst(Boolean),
-      (&Value(ref name), _) => {
+      (&StringLiteral(_), anno) => {
+        match *anno {
+          None => TConst(StringT),
+          Some(ref t) => self.from_anno(t),
+        }
+      },
+      (&BoolLiteral(_), anno) => {
+        match *anno {
+          None => TConst(Boolean),
+          Some(ref t) => self.from_anno(t),
+        }
+      },
+      (&Value(ref name), ref _anno) => {
         self.lookup(name)
           .map(|entry| entry.type_var)
           .ok_or(format!("Couldn't find {}", name))?
       },
-      (&BinExp(ref op, box ref lhs, box ref rhs), _) => {
+      (&BinExp(ref op, box ref lhs, box ref rhs), ref _anno) => {
         let op_type = self.infer_op(op)?;
         let lhs_type = self.infer(&lhs)?;
 
@@ -232,7 +249,7 @@ impl TypeContext {
           _ => return Err(format!("Bad type for operator")),
         }
       },
-      (&Call { ref f, ref arguments }, _) => {
+      (&Call { ref f, ref arguments }, ref _anno) => {
         let f_type = self.infer(&*f)?;
         let arg_type = self.infer(arguments.get(0).unwrap())?; // TODO fix later
         match f_type {