From f9a56c7d8fd9b4f8461dd26f10e939da75deee1c Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Fri, 27 Mar 2020 23:47:40 +0800
Subject: [PATCH] compile statements (WIP)

---
 src/main.rs | 38 +++++++++++++++++++++++++-------------
 1 file changed, 25 insertions(+), 13 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index b0af373..9d41801 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -19,18 +19,21 @@ use std::path::Path;
 enum CompileErrorKind {
     Unsupported(&'static str),
     MissingTypeAnnotation,
-    UnknownTypeAnnotation
+    UnknownTypeAnnotation,
+    Internal(&'static str)
 }
 
 impl fmt::Display for CompileErrorKind {
     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
         match self {
             CompileErrorKind::Unsupported(feature)
-                => write!(f, "The following Python feature is not supported by NAC3 {}: ", feature),
+                => write!(f, "The following Python feature is not supported by NAC3: {}", feature),
             CompileErrorKind::MissingTypeAnnotation
                 => write!(f, "Missing type annotation"),
             CompileErrorKind::UnknownTypeAnnotation
                 => write!(f, "Unknown type annotation"),
+            CompileErrorKind::Internal(details)
+                => write!(f, "Internal compiler error: {}", details),
         }
     }
 }
@@ -148,6 +151,9 @@ impl<'ctx> CodeGen<'ctx> {
         let y = function.get_nth_param(1).unwrap().into_int_value();
         let sum = self.builder.build_int_add(x, y, "sum");
         self.builder.build_return(Some(&sum));
+        for statement in body.iter() {
+            self.compile_statement(statement)?;
+        }
         Ok(())
     }
 
@@ -156,22 +162,28 @@ impl<'ctx> CodeGen<'ctx> {
 
         use ast::StatementType::*;
         match &statement.node {
-            FunctionDef {
-                is_async,
-                name,
-                args,
-                body,
-                decorator_list,
-                returns,
-            } => {
-                self.compile_function_def(name, args, body, decorator_list, returns, *is_async)?;
-            },
             Pass => (),
             _ => return Err(self.compile_error(CompileErrorKind::Unsupported("special statement"))),
         }
         Ok(())
     }
 
+    fn compile_toplevel(&mut self, statement: &ast::Statement) -> CompileResult<()> {
+        self.set_source_location(statement.location);
+        if let ast::StatementType::FunctionDef {
+                    is_async,
+                    name,
+                    args,
+                    body,
+                    decorator_list,
+                    returns,
+                } = &statement.node {
+            self.compile_function_def(name, args, body, decorator_list, returns, *is_async)
+        } else {
+            Err(self.compile_error(CompileErrorKind::Internal("top-level is not a function definition")))
+        }
+    }
+
     fn output(&self) {
         let triple = TargetTriple::create("riscv32-none-linux-gnu");
         let target = Target::from_triple(&triple)
@@ -204,7 +216,7 @@ fn main() {
 
     let context = Context::create();
     let mut codegen = CodeGen::new(&context);
-    match codegen.compile_statement(&ast.statements[0]) {
+    match codegen.compile_toplevel(&ast.statements[0]) {
         Ok(_) => (),
         Err(err) => { println!("{}", err); return; }
     }