From 07db4c530e1eae85781dcb4eef29911a8c8d85c9 Mon Sep 17 00:00:00 2001
From: Andrew Chambers <ac@acha.ninja>
Date: Thu, 28 Nov 2019 18:31:01 +1300
Subject: [PATCH] Add os/environ.

---
 src/core/os.c     | 29 +++++++++++++++++++++++++++++
 test/suite7.janet |  6 ++++++
 2 files changed, 35 insertions(+)

diff --git a/src/core/os.c b/src/core/os.c
index 8d114786..bd2755c7 100644
--- a/src/core/os.c
+++ b/src/core/os.c
@@ -372,6 +372,30 @@ static Janet os_shell(int32_t argc, Janet *argv) {
            : janet_wrap_boolean(stat);
 }
 
+static Janet os_environ(int32_t argc, Janet *argv) {
+    (void) argv;
+    janet_fixarity(argc, 0);
+    size_t nenv = 0;
+    char **env = environ;
+    while (*env++)
+        nenv += 1;
+    JanetTable *t = janet_table(nenv);
+    for (size_t i = 0; i < nenv; i++) {
+        char *e = environ[i];
+        char *eq = strchr(e, '=');
+        if (!eq) janet_panic("no '=' in environ");
+        char *v = eq + 1;
+        size_t full_len = strlen(e);
+        size_t val_len = strlen(v);
+        janet_table_put(
+            t,
+            janet_stringv((const uint8_t*)e, full_len - val_len - 1),
+            janet_stringv((const uint8_t*)v, val_len)
+        );
+    }
+    return janet_wrap_table(t);
+}
+
 static Janet os_getenv(int32_t argc, Janet *argv) {
     janet_fixarity(argc, 1);
     const char *cstr = janet_getcstring(argv, 0);
@@ -831,6 +855,11 @@ static const JanetReg os_cfuns[] = {
              "\t:netbsd\n"
              "\t:posix - A POSIX compatible system (default)")
     },
+    {
+        "os/environ", os_environ,
+        JDOC("(os/environ)\n\n"
+             "Get a copy of the os environment table.")
+    },
     {
         "os/getenv", os_getenv,
         JDOC("(os/getenv variable)\n\n"
diff --git a/test/suite7.janet b/test/suite7.janet
index 4234b722..28c89d74 100644
--- a/test/suite7.janet
+++ b/test/suite7.janet
@@ -229,4 +229,10 @@
   (prin (dyn :out))
   (assert (deep= (dyn :out) @"abcdabcdabcdabcd") "print buffer to self"))
 
+(os/setenv "TESTENV1" "v1")
+(os/setenv "TESTENV2" "v2")
+(assert (= (os/getenv "TESTENV1") "v1") "getenv works")
+(def environ (os/environ))
+(assert (= [(environ "TESTENV1") (environ "TESTENV2")] ["v1" "v2"]) "environ works")
+
 (end-suite)