Working mapgen and fixed indentation

crates/codegen/src/lib.rs

@@ -41,7 +41,7 @@ impl RustCodegen {
             output: String::new(),
             indent: 0,
             mapping: HashSet::new(),
-            position: RustMap(0, 0),
+            position: RustMap(1, 0),
         }
     }
 
@@ -50,15 +50,12 @@ impl RustCodegen {
     }
 
     fn add(&mut self, s: &str) {
-        for c in s.chars() {
-            match c {
-                '\n' => {
-                    self.position.0 += 1;
-                    self.position.1 = 0;
-                }
-                _ => {
-                    self.position.1 += 1;
-                }
+        for b in s.bytes() {
+            if b == b'\n' {
+                self.position.0 += 1;
+                self.position.1 = 0;
+            } else {
+                self.position.1 += 1;
             }
         }
 

crates/codegen/src/statement.rs

@@ -11,12 +11,14 @@ impl GenRust for Block {
 
         for stmt in &self.statements {
             ctx.expr_ensure_semicolon = true;
+            cg.add_indented("");
             stmt.gen_rust(ctx, cg);
             cg.addln("");
         }
 
         if let Some(soft_return) = &self.soft_return {
             ctx.expr_ensure_semicolon = false;
+            cg.add_indented("");
             soft_return.gen_rust(ctx, cg);
             cg.addln("");
         }

crates/codegen/src/top_level.rs

@@ -73,7 +73,7 @@ impl GenRust for ImplDecl {
 
 impl GenRust for FunctionDecl {
     fn gen_rust(&self, ctx: &mut Context, cg: &mut RustCodegen) {
-        cg.add(&format!(
+        cg.add_indented(&format!(
             "{}fn {}{}(",
             self.visibility.get_rust(),
             self.name.get_rust(),
@@ -97,8 +97,9 @@ impl GenRust for FunctionDecl {
         if let Some(body) = &self.body {
             cg.add(" ");
             body.gen_rust(ctx, cg);
+            cg.addln("");
         } else {
-            cg.add(";");
+            cg.addln(";");
         }
     }
 }