Basic variable implementation

examples/variables.e0

@@ -0,0 +1,4 @@
+fn main() -> int {
+  let x: int = 42;
+  return x;
+}

src/ast/llvm.rs

@@ -1,23 +1,33 @@
+use std::{collections::HashMap, marker::PhantomData};
+
 use anyhow::Result;
 use inkwell::{
   builder::Builder,
   context::Context,
   module::Module,
-  types::{AnyTypeEnum, BasicMetadataTypeEnum, BasicTypeEnum, FunctionType},
-  values::{
-    AnyValue, AnyValueEnum, BasicValue, BasicValueEnum, FunctionValue, IntValue,
-  },
+  types::{BasicMetadataTypeEnum, BasicTypeEnum, FunctionType},
+  values::{AnyValueEnum, BasicValueEnum, PointerValue},
 };
 
 use super::{Decl, Expr, Stmt, Type};
 
 impl Expr {
-  pub fn into_llvm<'ctx>(
+  fn into_llvm<'ctx>(
     &self,
     context: &'ctx Context,
     builder: &Builder,
+    env: &Env<'_, 'ctx>,
   ) -> Result<BasicValueEnum<'ctx>> {
     match self {
+      Expr::Var(name) => {
+        let (_, value) = match env.lookup(&name) {
+          Some(v) => v,
+          None => bail!("Unbound name {name:?}"),
+        };
+
+        Ok(BasicValueEnum::PointerValue(*value))
+      }
+
       Expr::Int(n) => Ok(BasicValueEnum::IntValue(
         context.i64_type().const_int(*n as u64, false),
       )),
@@ -51,6 +61,65 @@ fn fn_type_basic<'ctx>(
   }
 }
 
+type EnvValue<'a, 'ctx> = (&'a Type, PointerValue<'ctx>);
+
+#[derive(Default)]
+struct Env<'a, 'ctx> {
+  parent: Option<Box<Env<'a, 'ctx>>>,
+  local_type_map: HashMap<String, EnvValue<'a, 'ctx>>,
+}
+
+impl<'a, 'ctx> Env<'a, 'ctx> {
+  pub fn lookup(&self, name: impl AsRef<str>) -> Option<&EnvValue<'a, 'ctx>> {
+    match self.local_type_map.get(name.as_ref()) {
+      Some(v) => Some(v),
+      None => match &self.parent {
+        Some(p) => p.lookup(name),
+        None => None,
+      },
+    }
+  }
+}
+
+fn convert_lexical_block(
+  context: &Context,
+  builder: &Builder,
+  parent_env: Env,
+  stmts: impl AsRef<[Stmt]>,
+) -> Result<()> {
+  let stmts = stmts.as_ref();
+  let mut scope_env = Env {
+    parent: Some(Box::new(parent_env)),
+    ..Default::default()
+  };
+
+  for stmt in stmts.iter() {
+    match stmt {
+      Stmt::Let(name, ty, expr) => {
+        let llvm_ty = ty.into_llvm_basic_type(context);
+        let alloca = builder.build_alloca(llvm_ty, name);
+
+        let expr_val = expr.into_llvm(context, builder, &scope_env)?;
+        builder.build_store(alloca, expr_val);
+
+        scope_env.local_type_map.insert(name.clone(), (ty, alloca));
+      }
+
+      Stmt::Return(ret_val) => {
+        if let Some(ret_val) = ret_val {
+          let value = ret_val.into_llvm(context, builder, &scope_env)?;
+          builder.build_return(Some(&value));
+        } else {
+          builder.build_return(None);
+        }
+        println!("Emitted return.");
+      }
+    }
+  }
+
+  Ok(())
+}
+
 pub fn convert(context: &mut Context, program: Vec<Decl>) -> Result<Module> {
   let module = context.create_module("program");
   let builder = context.create_builder();
@@ -64,19 +133,8 @@ pub fn convert(context: &mut Context, program: Vec<Decl>) -> Result<Module> {
 
     builder.position_at_end(entry_block);
 
-    for stmt in func.stmts.iter() {
-      match stmt {
-        Stmt::Return(ret_val) => {
-          if let Some(ret_val) = ret_val {
-            let value = ret_val.into_llvm(context, &builder)?;
-            builder.build_return(Some(&value));
-          } else {
-            builder.build_return(None);
-          }
-          println!("Emitted return.");
-        }
-      }
-    }
+    let env = Env::default();
+    convert_lexical_block(context, &builder, env, &func.stmts)?;
   }
 
   Ok(module)

src/ast/mod.rs

@@ -23,12 +23,14 @@ pub struct Func {
 
 #[derive(Debug)]
 pub enum Stmt {
+  Let(String, Type, Expr),
   Return(Option<Expr>),
 }
 
 #[derive(Debug)]
 pub enum Expr {
   Int(i64),
+  Var(String),
 }
 
 #[derive(Debug)]

src/main.rs

@@ -20,8 +20,13 @@ use crate::parser::ProgramParser;
 struct Opt {
   path: PathBuf,
 
-  #[clap(short = 'o', long = "out")]
+  /// Path to output the bitcode to.
+  #[clap(short = 'o', long = "out", name = "bitcode-out")]
   out_path: PathBuf,
+
+  /// Path to output the AST to, if at all.
+  #[clap(long = "emit-ast", name = "ast-out")]
+  emit_ast: Option<PathBuf>,
 }
 
 fn main() -> Result<()> {

src/parser.lalrpop

@@ -14,6 +14,8 @@ Func: Func = {
 };
 
 Stmt: Stmt = {
+  "let" <name:Ident> ":" <ty:Type> "=" <expr:Expr> ";" =>
+    Stmt::Let(name, ty, expr),
   "return" <expr:Expr?> ";" => Stmt::Return(expr),
 };
 
@@ -23,6 +25,7 @@ Expr: Expr = {
 
 Expr1: Expr = {
   r"[0-9]+" => Expr::Int(<>.parse::<i64>().unwrap()),
+  Ident => Expr::Var(<>),
   "(" <expr:Expr> ")" => expr,
 };