not-python/src/obj/builtin.rs

use crate::{obj::{prelude::*, reserved::*}, vm::signal::*};
use maplit::btreemap;
use once_cell::sync::Lazy;
use std::collections::BTreeMap;

pub static PRINTLN_BUILTIN_FUN: Lazy<NativeFunRef> = Lazy::new(|| NativeFun::new_obj(|_, vm, args| {
    // TODO : use __get_attr__ when it gets added
    let to_string = {
        let obj_ref = &args[0];
        read_obj!(let obj = obj_ref);
        obj.get_attr(STR_MEMBER_NAME.sym)
            .or_else(|| obj.get_attr(REPR_MEMBER_NAME.sym))
            .expect("no __str__ or __repr__ member")
    };
    let return_value = vm.call(to_string, vec![]);
    {
        read_obj!(let str_obj = return_value);
        let str_obj: &Str = str_obj.as_any().downcast_ref().unwrap();
        println!("{}", str_obj.value());
    }

    vm.push(NIL_NAME.sym_ref());
    Signal::Return
}));

pub static PRINT_BUILTIN_FUN: Lazy<NativeFunRef> = Lazy::new(|| NativeFun::new_obj(|_, vm, args| {
    // TODO : use __get_attr__ when it gets added
    let to_string = {
        let obj_ref = &args[0];
        read_obj!(let obj = obj_ref);
        obj.get_attr(STR_MEMBER_NAME.sym)
            .or_else(|| obj.get_attr(REPR_MEMBER_NAME.sym))
            .expect("no __str__ or __repr__ member")
    };
    let return_value = vm.call(to_string, vec![]);
    {
        read_obj!(let str_obj = return_value);
        let str_obj: &Str = str_obj.as_any().downcast_ref().unwrap();
        print!("{}", str_obj.value());
    }

    Signal::Return
}));

pub static BUILTIN_OBJS: Lazy<BTreeMap<Sym, ObjRef>> = Lazy::new(|| btreemap! {
    PRINTLN_BUILTIN_NAME.sym => PRINTLN_BUILTIN_FUN.clone() as _,
    PRINT_BUILTIN_NAME.sym => PRINT_BUILTIN_FUN.clone() as _,
});