fmt

rustfmt.toml

@@ -0,0 +1,2 @@
+max_width = 80
+tab_spaces = 2

src/ast.rs

@@ -1,58 +0,0 @@
-use anyhow::Result;
-use inkwell::{context::Context, module::Module};
-
-#[derive(Debug)]
-pub enum Decl {
-    Func(Func),
-}
-
-impl Decl {
-    pub fn unwrap_func(&self) -> Option<&Func> {
-        match self {
-            Decl::Func(func) => Some(func),
-            _ => None,
-        }
-    }
-}
-
-#[derive(Debug)]
-pub struct Func {
-    pub name: String,
-    pub return_ty: Type,
-    pub stmts: Vec<Stmt>,
-}
-
-#[derive(Debug)]
-pub enum Stmt {
-    Return(Option<Expr>),
-}
-
-#[derive(Debug)]
-pub enum Expr {
-    Int(i64),
-}
-
-#[derive(Debug)]
-pub enum Type {
-    Int,
-}
-
-pub fn convert(context: &mut Context, program: Vec<Decl>) -> Result<Module> {
-    let module = context.create_module("program");
-    let builder = context.create_builder();
-
-    for func in program.iter().filter_map(Decl::unwrap_func) {
-        // TODO: hardcoded void -> int function
-        let i64_ty = context.i64_type();
-        let llvm_func_ty = i64_ty.fn_type(&[], false);
-
-        let llvm_func = module.add_function(&func.name, llvm_func_ty, None);
-        let entry_block = context.append_basic_block(llvm_func, "entry");
-
-        builder.position_at_end(entry_block);
-
-        builder.build_return(None);
-    }
-
-    Ok(module)
-}

src/ast/llvm.rs

@@ -0,0 +1,38 @@
+use anyhow::Result;
+use inkwell::{
+  context::Context,
+  module::Module,
+  types::{AnyType, AnyTypeEnum},
+};
+
+use super::{Decl, Type};
+
+impl Type {
+  pub fn into_llvm_type<'ctx>(
+    &self,
+    context: &'ctx Context,
+  ) -> AnyTypeEnum<'ctx> {
+    match self {
+      Type::Int => AnyTypeEnum::IntType(context.i64_type()),
+    }
+  }
+}
+
+pub fn convert(context: &mut Context, program: Vec<Decl>) -> Result<Module> {
+  let module = context.create_module("program");
+  let builder = context.create_builder();
+
+  for func in program.iter().filter_map(Decl::unwrap_func) {
+    let return_ty = func.return_ty.into_llvm_type(context);
+    let llvm_func_ty = return_ty.fn_type();
+
+    let llvm_func = module.add_function(&func.name, llvm_func_ty, None);
+    let entry_block = context.append_basic_block(llvm_func, "entry");
+
+    builder.position_at_end(entry_block);
+
+    builder.build_return(None);
+  }
+
+  Ok(module)
+}

src/ast/mod.rs

@@ -0,0 +1,37 @@
+pub mod llvm;
+
+#[derive(Debug)]
+pub enum Decl {
+  Func(Func),
+}
+
+impl Decl {
+  pub fn unwrap_func(&self) -> Option<&Func> {
+    match self {
+      Decl::Func(func) => Some(func),
+      _ => None,
+    }
+  }
+}
+
+#[derive(Debug)]
+pub struct Func {
+  pub name: String,
+  pub return_ty: Type,
+  pub stmts: Vec<Stmt>,
+}
+
+#[derive(Debug)]
+pub enum Stmt {
+  Return(Option<Expr>),
+}
+
+#[derive(Debug)]
+pub enum Expr {
+  Int(i64),
+}
+
+#[derive(Debug)]
+pub enum Type {
+  Int,
+}

src/main.rs

@@ -34,7 +34,7 @@ fn main() -> Result<()> {
   println!("AST: {ast:?}");
 
   let mut context = Context::create();
-    let module = ast::convert(&mut context, ast)?;
+  let module = ast::llvm::convert(&mut context, ast)?;
 
   {
     let file = File::create(&opts.out_path)?;