env.c

#include "lib.h"
#include <unistd.h>

/* ext map - funcs */
static void at_env_str(lk_obj_t *self, lk_scope_t *local) {
    const char *k = darray_str_tocstr(DARRAY(ARG(0)));
    const char *v = getenv(k);
    RETURN(v != NULL ? LK_OBJ(lk_str_new_fromcstr(VM, v)) : NIL);
}
extern char** environ;
static void size_env(lk_obj_t *self, lk_scope_t *local) {
    int c;
    for(c = 0; environ[c] != NULL; c ++) { }
    RETURN(lk_num_new(VM, c));
}
static void keys_env(lk_obj_t *self, lk_scope_t *local) {
    lk_list_t *keys = lk_list_new(VM);
    int i, j;
    const char *v;
    for(i = 0; (v = environ[i]) != NULL; i ++) {
        for(j = 0; v[j] != '\0' && v[j] != '='; j ++) { }
        darray_ptr_push(DARRAY(keys), lk_str_new_fromdata(VM, v, j));
    }
    RETURN(keys);
}
void lk_env_extinit(lk_vm_t *vm) {
    lk_obj_t *obj = vm->t_obj, *str = vm->t_str;
    lk_obj_t *env = lk_obj_alloc_withsize(obj, sizeof(lk_env_t));
    lk_global_set("Environment", env);
    lk_obj_set_cfunc_lk(env, "at", at_env_str, str, NULL);
    lk_obj_set_cfunc_lk(env, "size", size_env, NULL);
    lk_obj_set_cfunc_lk(env, "keys", keys_env, NULL);
}