setEnv

Author: Ricardicus

README.md

@@ -794,7 +794,7 @@ This is in it's very early stage... it only supports one liners!
 
 Some short tips:
 help() gives you a list of available functions
-printEnv() lists the names of variables defined in this session
+printVars() lists the names of variables defined in this session
 quit() quits this interpreter interactive mode
 
 >> a = 1337
@@ -839,7 +839,7 @@ This is in it's very early stage... it only supports one liners!
 
 Some short tips:
 help() gives you a list of available functions
-printEnv() lists the names of variables defined in this session
+printVars() lists the names of variables defined in this session
 quit() quits this interpreter interactive mode
 
 >> help()

samples/ric_lib.ric

@@ -59,8 +59,17 @@ print("'1336' as a string + 1 is: '" + stringed + "'")
 testString = "The length of this text before the column including the space behind and the column itself is: "
 print(testString + len(testString))
 
-exitCode = 2
 
+
+# Manipulating the environment
+# Setting a variable
+setEnv("HELLO_WORLD", "HI")
+environment = listEnv()
+? [ environment.contains("HELLO_WORLD") ] {
+  print(getEnv("HELLO_WORLD"))
+}
+
+exitCode = 2
 print ( "I will exit with exit code: " + exitCode )
 
 exit ( exitCode )

src/lib.c

@@ -135,6 +135,7 @@ libFunction_t ric_library[] = {
     DECLARE_LIB_FUNCTION("find", 1, ric_find_files),
     DECLARE_LIB_FUNCTION("cd", 1, ric_cd),
     DECLARE_LIB_FUNCTION("osName", 0, ric_os_name),
+    DECLARE_LIB_FUNCTION("setEnv", 2, ric_set_env),
     // libctx
     DECLARE_LIB_FUNCTION("setTimeout", 2, ric_set_timeout),
     DECLARE_LIB_FUNCTION("setInterval", 2, ric_set_interval),

src/library/libos.c

@@ -513,3 +513,39 @@ int ric_env_keys(LIBRARY_PARAMS()) {
   PUSH_VECTOR(stv.vec, sp, sc);
   return 0;
 }
+
+int ric_set_env(LIBRARY_PARAMS()) {
+  stackval_t stv_key, stv_val;
+  heapval_t *hpv;
+  int dummy;
+  void *sp = PROVIDE_CONTEXT()->sp;
+  size_t *sc = PROVIDE_CONTEXT()->sc;
+  void *hp = PROVIDE_CONTEXT()->hp;
+
+  // Pop value first, then key (stack order)
+  POP_VAL(&stv_key, sp, sc);
+  POP_VAL(&stv_val, sp, sc);
+
+  if (stv_key.type != TEXT || stv_val.type != TEXT) {
+    fprintf(stderr, "error: function '%s' expected (string, string) as arguments.\n",
+            LIBRARY_FUNC_NAME());
+    exit(1);
+  }
+
+  char *key = stv_key.t;
+  char *val = stv_val.t;
+
+  if (setenv(key, val, 1) != 0) {
+    perror("setenv failed");
+    exit(1);
+  }
+
+  // Return empty string as success indicator
+  stackval_t stv;
+  stv.type = TEXT;
+  stv.t = strdup("");
+  ALLOC_HEAP(&stv, hp, &hpv, &dummy);
+  PUSH_STRING(stv.t, sp, sc);
+
+  return 0;
+}

src/library/libos.h

@@ -22,5 +22,6 @@ int ric_os_name(LIBRARY_PARAMS());
 int ric_mkdir(LIBRARY_PARAMS());
 int ric_find_files(LIBRARY_PARAMS());
 int ric_env_keys(LIBRARY_PARAMS());
+int ric_set_env(LIBRARY_PARAMS());
 
 #endif

src/library/libos_nt.c

@@ -509,3 +509,40 @@ int ric_env_keys(LIBRARY_PARAMS()) {
   PUSH_VECTOR(stv.vec, sp, sc);
   return 0;
 }
+
+int ric_set_env(LIBRARY_PARAMS()) {
+  stackval_t stv_key, stv_val;
+  heapval_t *hpv;
+  int dummy;
+  expr_t *ret = NULL;
+  void *sp = PROVIDE_CONTEXT()->sp;
+  size_t *sc = PROVIDE_CONTEXT()->sc;
+  void *hp = PROVIDE_CONTEXT()->hp;
+
+  // Pop value first, then key (stack order)
+  POP_VAL(&stv_key, sp, sc);
+  POP_VAL(&stv_val, sp, sc);
+
+  if (stv_key.type != TEXT || stv_val.type != TEXT) {
+    fprintf(stderr, "error: function '%s' expected (string, string) as arguments.\n",
+            LIBRARY_FUNC_NAME());
+    exit(1);
+  }
+
+  char *key = stv_key.t;
+  char *val = stv_val.t;
+
+  if (_putenv_s(key, val) != 0) {
+    fprintf(stderr, "error: _putenv_s failed in '%s' for key='%s'\n", LIBRARY_FUNC_NAME(), key);
+    exit(1);
+  }
+
+  // Return empty string as success indicator
+  stackval_t stv;
+  stv.type = TEXT;
+  stv.t = strdup("");
+  ALLOC_HEAP(&stv, hp, &hpv, &dummy);
+  PUSH_STRING(stv.t, sp, sc);
+
+  return 0;
+}

tests/test_ric_lib.py

@@ -54,6 +54,7 @@ def test_output():
 "0",
 "'1336' as a string + 1 is: '1337'",
 "The length of this text before the column including the space behind and the column itself is: 95",
+"HI",
 "I will exit with exit code: 2"
 ]