rue_compiler/compile/expr/
if_expr.rs

1use log::debug;
2use rue_ast::{AstIfExpr, AstNode};
3use rue_hir::{Hir, Value};
4use rue_types::{Type, TypeId, Union};
5
6use crate::{Compiler, compile_expr};
7
8pub fn compile_if_expr(
9    ctx: &mut Compiler,
10    expr: &AstIfExpr,
11    expected_type: Option<TypeId>,
12) -> Value {
13    let condition = if let Some(condition) = expr.condition() {
14        let value = compile_expr(ctx, &condition, None);
15        ctx.assign_type(condition.syntax(), value.ty, ctx.builtins().types.bool);
16        value
17    } else {
18        debug!("Unresolved if condition");
19        ctx.builtins().unresolved.clone()
20    };
21
22    let then_expr = if let Some(then_expr) = expr.then_expr() {
23        let index = ctx.push_mappings(condition.then_map.clone());
24        let value = compile_expr(ctx, &then_expr, expected_type);
25        ctx.revert_mappings(index);
26        value
27    } else {
28        debug!("Unresolved if then expr");
29        ctx.builtins().unresolved.clone()
30    };
31
32    let else_expr = if let Some(else_expr) = expr.else_expr() {
33        let index = ctx.push_mappings(condition.else_map.clone());
34        let value = compile_expr(ctx, &else_expr, expected_type);
35        ctx.revert_mappings(index);
36        value
37    } else {
38        debug!("Unresolved if else expr");
39        ctx.builtins().unresolved.clone()
40    };
41
42    let ty = ctx.alloc_type(Type::Union(Union::new(vec![then_expr.ty, else_expr.ty])));
43    let hir = ctx.alloc_hir(Hir::If(condition.hir, then_expr.hir, else_expr.hir));
44
45    Value::new(hir, ty)
46}
rue_compiler/compile/expr/if_expr.rs

rue_compiler/compile/expr/
if_expr.rs
