HomePhorge
Diffusion Ant (master) Code

ant/src/modules/symbol.c
8755416ca476master

Last Change
Raw File

symbol.c
View Options

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include "ant.h"
#include "utf8.h"
#include "errors.h"
#include "runtime.h"
#include "internal.h"
#include "silver/engine.h"
#include "modules/symbol.h"
#include "descriptors.h"
static jsval_t g_iterator_proto_obj = {0};
static jsval_t g_array_iterator_proto_obj = {0};
static jsval_t g_string_iterator_proto_obj = {0};
#define DECL_SYM(name, _desc) static jsval_t g_##name = {0};
WELLKNOWN_SYMBOLS(DECL_SYM)
#undef DECL_SYM
#define DEF_GET_SYM(name, _desc) jsval_t get_##name##_sym(void) { return g_##name; }
WELLKNOWN_SYMBOLS(DEF_GET_SYM)
#undef DEF_GET_SYM
static jsval_t builtin_Symbol(ant_t *js, jsval_t *args, int nargs) {
const char *desc = NULL;
if (nargs > 0 && vtype(args[0]) == T_STR) {
desc = js_getstr(js, args[0], NULL);
}
return js_mksym(js, desc);
}
static jsval_t builtin_Symbol_for(ant_t *js, jsval_t *args, int nargs) {
if (nargs < 1 || vtype(args[0]) != T_STR) {
return js_mkerr(js, "Symbol.for requires a string argument");
}
char *key = js_getstr(js, args[0], NULL);
if (!key) return js_mkerr(js, "Invalid key");
return js_mksym_for(js, key);
}
static jsval_t builtin_Symbol_keyFor(ant_t *js, jsval_t *args, int nargs) {
if (nargs < 1 || vtype(args[0]) != T_SYMBOL) {
return js_mkundef();
}
const char *key = js_sym_key(args[0]);
if (!key) return js_mkundef();
return js_mkstr(js, key, strlen(key));
}
static jsval_t builtin_Symbol_toString(ant_t *js, jsval_t *args, int nargs) {
jsval_t this_val = js_getthis(js);
if (vtype(this_val) != T_SYMBOL) {
return js_mkerr(js, "Symbol.prototype.toString requires a symbol");
}
return js_symbol_to_string(js, this_val);
}
static jsval_t iterator_next(ant_t *js, jsval_t *args, int nargs) {
jsval_t this_val = js_getthis(js);
// migrate to slots
jsval_t arr = js_get(js, this_val, "__arr");
jsval_t idx_val = js_get(js, this_val, "__idx");
jsval_t len_val = js_get(js, this_val, "__len");
int idx = (int)js_getnum(idx_val);
int len = (int)js_getnum(len_val);
jsval_t result = js_mkobj(js);
if (idx >= len) {
js_set(js, result, "done", js_true);
js_set(js, result, "value", js_mkundef());
} else {
char idxstr[16];
snprintf(idxstr, sizeof(idxstr), "%d", idx);
jsval_t value = js_get(js, arr, idxstr);
js_set(js, result, "value", value);
js_set(js, result, "done", js_false);
js_set(js, this_val, "__idx", js_mknum(idx + 1));
}
return result;
}
static jsval_t get_iterator_prototype(ant_t *js) {
if (vtype(g_iterator_proto_obj) == T_OBJ) return g_iterator_proto_obj;
g_iterator_proto_obj = js_mkobj(js);
js_set_proto(js, g_iterator_proto_obj, js->object);
js_set_sym(js, g_iterator_proto_obj, g_iterator, js_mkfun(sym_this_cb));
return g_iterator_proto_obj;
}
static jsval_t get_array_iterator_prototype(ant_t *js) {
if (vtype(g_array_iterator_proto_obj) == T_OBJ) return g_array_iterator_proto_obj;
jsval_t iterator_proto = get_iterator_prototype(js);
g_array_iterator_proto_obj = js_mkobj(js);
js_set(js, g_array_iterator_proto_obj, "next", js_mkfun(iterator_next));
js_set_proto(js, g_array_iterator_proto_obj, iterator_proto);
return g_array_iterator_proto_obj;
}
static jsval_t array_iterator(ant_t *js, jsval_t *args, int nargs) {
jsval_t arr = js_getthis(js);
jsval_t len_val = js_get(js, arr, "length");
int len = vtype(len_val) == T_NUM ? (int)js_getnum(len_val) : 0;
jsval_t iter = js_mkobj(js);
js_set(js, iter, "__arr", arr);
js_set(js, iter, "__idx", js_mknum(0));
js_set(js, iter, "__len", js_mknum(len));
js_set_proto(js, iter, get_array_iterator_prototype(js));
return iter;
}
static jsval_t string_iterator_next(ant_t *js, jsval_t *args, int nargs) {
jsval_t this_val = js_getthis(js);
jsval_t str = js_get(js, this_val, "__str");
jsval_t idx_val = js_get(js, this_val, "__idx");
jsval_t len_val = js_get(js, this_val, "__len");
int idx = (int)js_getnum(idx_val);
int len = (int)js_getnum(len_val);
jsval_t result = js_mkobj(js);
if (idx >= len) {
js_set(js, result, "done", js_true);
js_set(js, result, "value", js_mkundef());
} else {
char *s = js_getstr(js, str, NULL);
unsigned char c = (unsigned char)s[idx];
int char_bytes = utf8_sequence_length(c);
if (char_bytes < 1) char_bytes = 1;
if (idx + char_bytes > len) char_bytes = len - idx;
js_set(js, result, "value", js_mkstr(js, s + idx, (jsoff_t)char_bytes));
js_set(js, result, "done", js_false);
js_set(js, this_val, "__idx", js_mknum(idx + char_bytes));
}
return result;
}
static jsval_t get_string_iterator_prototype(ant_t *js) {
if (vtype(g_string_iterator_proto_obj) == T_OBJ) return g_string_iterator_proto_obj;
jsval_t iterator_proto = get_iterator_prototype(js);
g_string_iterator_proto_obj = js_mkobj(js);
js_set(js, g_string_iterator_proto_obj, "next", js_mkfun(string_iterator_next));
js_set_proto(js, g_string_iterator_proto_obj, iterator_proto);
return g_string_iterator_proto_obj;
}
static jsval_t string_iterator(ant_t *js, jsval_t *args, int nargs) {
jsval_t str = js_getthis(js);
size_t len; js_getstr(js, str, &len);
jsval_t iter = js_mkobj(js);
js_set(js, iter, "__str", str);
js_set(js, iter, "__idx", js_mknum(0));
js_set(js, iter, "__len", js_mknum((double)len));
js_set_proto(js, iter, get_string_iterator_prototype(js));
return iter;
}
jsval_t maybe_call_symbol_method(
ant_t *js, jsval_t target,
jsval_t sym,
jsval_t this_arg, jsval_t *args,
int nargs, bool *called
) {
*called = false;
if (vtype(sym) != T_SYMBOL || !is_object_type(target)) return js_mkundef();
jsval_t method = js_get_sym(js, target, sym);
if (is_err(method)) return method;
uint8_t mt = vtype(method);
if (mt == T_UNDEF || mt == T_NULL) return js_mkundef();
if (mt != T_FUNC && mt != T_CFUNC) {
return js_mkerr_typed(js, JS_ERR_TYPE, "Symbol method is not callable");
}
*called = true;
return sv_vm_call(js->vm, js, method, this_arg, args, nargs, NULL, false);
}
void js_define_species_getter(ant_t *js, jsval_t ctor) {
if (vtype(ctor) == T_FUNC) ctor = js_func_obj(ctor);
if (vtype(ctor) != T_OBJ || vtype(g_species) != T_SYMBOL) return;
js_set_sym_getter_desc(js, ctor, g_species, js_mkfun(sym_this_cb), JS_DESC_C);
}
void init_symbol_module(void) {
ant_t *js = rt->js;
g_iterator_proto_obj = js_mkundef();
g_array_iterator_proto_obj = js_mkundef();
g_string_iterator_proto_obj = js_mkundef();
#define INIT_SYM(name, desc) g_##name = js_mksym(js, desc);
WELLKNOWN_SYMBOLS(INIT_SYM)
#undef INIT_SYM
jsval_t symbol_proto = js_mkobj(js);
js_set(js, symbol_proto, "toString", js_mkfun(builtin_Symbol_toString));
jsval_t symbol_ctor = js_mkobj(js);
js_set_slot(js, symbol_ctor, SLOT_CFUNC, js_mkfun(builtin_Symbol));
js_setprop(js, symbol_ctor, js_mkstr(js, "for", 3), js_mkfun(builtin_Symbol_for));
js_set(js, symbol_ctor, "keyFor", js_mkfun(builtin_Symbol_keyFor));
js_set(js, symbol_ctor, "prototype", symbol_proto);
#define SET_CTOR_SYM(name, _desc) js_set(js, symbol_ctor, #name, g_##name);
WELLKNOWN_SYMBOLS(SET_CTOR_SYM)
#undef SET_CTOR_SYM
jsval_t func_symbol = js_obj_to_func(symbol_ctor);
js_set(js, js_glob(js), "Symbol", func_symbol);
// set internal types before ant module snapshot
js_set_sym(js, rt->ant_obj, g_toStringTag, js_mkstr(js, "Ant", 3));
jsval_t array_ctor = js_get(js, js_glob(js), "Array");
jsval_t array_proto = js_get(js, array_ctor, "prototype");
(void)get_array_iterator_prototype(js);
(void)get_string_iterator_prototype(js);
js_set_sym(js, array_proto, g_iterator, js_mkfun(array_iterator));
jsval_t array_unscopables = js_mkobj(js);
js_set(js, array_unscopables, "find", js_true);
js_set(js, array_unscopables, "findIndex", js_true);
js_set(js, array_unscopables, "fill", js_true);
js_set(js, array_unscopables, "copyWithin", js_true);
js_set(js, array_unscopables, "entries", js_true);
js_set(js, array_unscopables, "keys", js_true);
js_set(js, array_unscopables, "values", js_true);
js_set(js, array_unscopables, "flat", js_true);
js_set(js, array_unscopables, "flatMap", js_true);
js_set_sym(js, array_proto, g_unscopables, array_unscopables);
jsval_t string_ctor = js_get(js, js_glob(js), "String");
jsval_t string_proto = js_get(js, string_ctor, "prototype");
js_set_sym(js, string_proto, g_iterator, js_mkfun(string_iterator));
jsval_t promise_ctor = js_get(js, js_glob(js), "Promise");
jsval_t promise_proto = js_get(js, promise_ctor, "prototype");
js_set_sym(js, promise_proto, g_toStringTag, js_mkstr(js, "Promise", 7));
jsval_t math_obj = js_get(js, js_glob(js), "Math");
if (is_object_type(math_obj)) js_set_sym(js, math_obj, g_toStringTag, js_mkstr(js, "Math", 4));
js_define_species_getter(js, promise_ctor);
js_define_species_getter(js, array_ctor);
}
void symbol_gc_update_roots(GC_OP_VAL_ARGS) {
op_val(ctx, &g_iterator_proto_obj);
op_val(ctx, &g_array_iterator_proto_obj);
op_val(ctx, &g_string_iterator_proto_obj);
#define GC_SYM(name, _desc) op_val(ctx, &g_##name);
WELLKNOWN_SYMBOLS(GC_SYM)
#undef GC_SYM
sym_gc_update_all(op_val, ctx);
}

Owner Packages

  • No Owners