moka/lib/builtins.c

96 lines
2.1 KiB
C

#include "builtins.h"
#include "node.h"
#include "path.h"
#include "module.h"
#include "moka.h"
void register_builtins(struct moka* moka)
{
assert(moka);
moka_decl_native(moka, "println", mk_println);
moka_decl_native(moka, "define", mk_define);
moka_decl_native(moka, "array", mk_array);
moka_decl_native(moka, "assert-eqv!", mk_assert_eqv_mut);
}
MOKA mk_println(struct moka* moka, struct vec* args)
{
assert(moka);
assert(args);
int line = 0;
for (size_t i=0; i<args->size; i++)
{
MOKA val = (MOKA) args->data[i];
if (i > 0) { printf(" "); }
else { line = moka_get_lazy(moka, val)->line; }
moka_dump(moka, MK_EVAL(val));
}
printf("\n");
return moka_push_int(moka, 0, line);
}
MOKA mk_define(struct moka* moka, struct vec* args)
{
assert(moka); assert(args);
assert(args->size == 2);
MOKA target = (MOKA) args->data[0];
MOKA value = MK_EVAL((MOKA) args->data[1]);
struct node* node = moka_get_lazy(moka, target);
moka_push(moka, value);
assert(node->token);
moka_decl_var(moka, node->token->value, value);
return value;
}
MOKA mk_array(struct moka* moka, struct vec* args)
{
assert(moka);
assert(args);
for (size_t i=0; i<args->size; i++)
{
MOKA arg = (MOKA) args->data[i];
MOKA val = MK_EVAL(arg);
moka_push(moka, val);
}
moka_make_array(moka, args->size);
return moka_top(moka);
}
MOKA mk_assert_eqv_mut(struct moka* moka, struct vec* args)
{
assert(moka);
assert(args);
MOKA lhs = MK_EVAL((MOKA) args->data[0]);
MOKA rhs = MK_EVAL((MOKA) args->data[1]);
int line = moka_line(moka, lhs);
char lhs_buf[MK_STRLEN];
moka_str(moka, lhs, lhs_buf, MK_STRLEN);
char rhs_buf[MK_STRLEN];
moka_str(moka, rhs, rhs_buf, MK_STRLEN);
if (!moka_is_eqv(moka, lhs, rhs))
{
status_push(moka->status, STATUS_ERROR, line,
"assertion failed\nlhs = %s\nrhs = %s",
lhs_buf, rhs_buf);
return moka_push_bool(moka, false, line);
}
return moka_push_bool(moka, true, line);
}