lemonlang 0.0.3

an experimental, modern, purely safe, programming language.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use crate::{ast, ir};

use super::Builder;

impl Builder<'_> {
	pub fn build_fn_stmt(&mut self, fn_stmt: &mut ast::FnStmt) {
		let ret_type = self.build_type(fn_stmt.ret_id, fn_stmt.get_range());
		self.ctx.push_function_scope(ret_type);
		let name = fn_stmt.name.lexeme().to_owned();
		let args: Vec<_> = fn_stmt.params.iter_mut().map(|arg| self.build_bind(arg)).collect();
		let ret = self.build_type(fn_stmt.ret_id, fn_stmt.get_range());
		let comptime = false;
		let func = ir::Function::new(name, comptime, args, ret);
		self.build_fn_body(&mut fn_stmt.body);
		self.ctx.pop_scope();
		self.push_function_with_blocks(func);
	}

	pub fn build_bind(&mut self, bind: &mut ast::Binding) -> ir::IrBasicValue {
		let lexeme = bind.lexeme().to_owned();
		let kind = self.build_type(bind.type_id, bind.get_range());
		let value = self.ctx.new_register(kind);
		// fix: remove this
		//
		if !kind.is_str() {
			self.ctx.add_dont_load(value.get_value().as_str());
		}

		self.ctx.add_local(lexeme, value.clone());
		value
	}

	pub fn build_fn_body(&mut self, body: &mut ast::FnBody) {
		if let ast::FnBody::Expr(expr) = body {
			self.build_expr(expr);
			if !self.ctx.block.is_returned() {
				self.drop_local_function_values(None);
			}
		}

		if let ast::FnBody::Block(block) = body {
			for stmt in block.stmts.iter_mut() {
				self.build_stmt(stmt);
			}
			if !self.ctx.block.is_returned() {
				self.drop_local_function_values(None);
			}
		}
	}
}