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

use arret_syntax::span::Span;

use arret_runtime::boxed;

use crate::mir::builder::Builder;
use crate::mir::error::Result;
use crate::mir::eval_hir::EvalHirCtx;
use crate::mir::intrinsic::BuildOutcome;
use crate::mir::value::Value;

use crate::mir::intrinsic::partial_print::{partial_pretty_print, PartialPrint};

// `print!` & `println!`
pub fn print(
    ehx: &mut EvalHirCtx,
    b: &mut Builder,
    span: Span,
    arg_list_value: &Value,
) -> Result<BuildOutcome> {
    match partial_pretty_print(ehx, b, span, arg_list_value) {
        Some(partial_print) => Ok(BuildOutcome::SimplifiedArgs(
            partial_print.into_arg_list_value(ehx),
        )),
        None => Ok(BuildOutcome::None),
    }
}

// `print-str`
pub fn print_str(
    ehx: &mut EvalHirCtx,
    b: &mut Builder,
    span: Span,
    arg_list_value: &Value,
) -> Result<BuildOutcome> {
    match partial_pretty_print(ehx, b, span, arg_list_value) {
        Some(PartialPrint::Constant(literal_str)) => Ok(BuildOutcome::ReturnValue(
            boxed::Str::new(ehx, &literal_str).into(),
        )),
        Some(partial_print @ PartialPrint::SimplifiedArgs(_)) => Ok(BuildOutcome::SimplifiedArgs(
            partial_print.into_arg_list_value(ehx),
        )),
        None => Ok(BuildOutcome::None),
    }
}