From cfb4df66f912fc936e582c93c5350aca67c8ec70 Mon Sep 17 00:00:00 2001
From: cpunch <sethtstubbs@gmail.com>
Date: Wed, 10 Feb 2021 17:10:03 -0600
Subject: [PATCH] Added optional os.* library, fixed cosmoB_loadLibrary

- os.time() returns the UNIX Epoch time in seconds
- os.read() reads file and returns the data as a <string> object
---
 src/cbaselib.c | 83 ++++++++++++++++++++++++++++++++++++++++++++++++--
 src/cbaselib.h |  8 ++++-
 src/main.c     |  2 ++
 3 files changed, 90 insertions(+), 3 deletions(-)

diff --git a/src/cbaselib.c b/src/cbaselib.c
index facdac6..7090735 100644
--- a/src/cbaselib.c
+++ b/src/cbaselib.c
@@ -5,6 +5,7 @@
 #include "cmem.h"
 
 #include <math.h>
+#include <sys/time.h>
 
 // ================================================================ [BASELIB] ================================================================
 
@@ -116,7 +117,7 @@ void cosmoB_loadLibrary(CState *state) {
         "type",
         "pcall",
         "tonumber",
-        "tostring"
+        "tostring",
         "loadstring"
     };
 
@@ -191,7 +192,7 @@ int cosmoB_oisChild(CState *state, int nargs, CValue *args) {
 }
 
 COSMO_API void cosmoB_loadObjLib(CState *state) {
-     const char *identifiers[] = {
+    const char *identifiers[] = {
         "ischild"
     };
 
@@ -234,6 +235,84 @@ COSMO_API void cosmoB_loadObjLib(CState *state) {
     cosmoV_register(state, 1);
 }
 
+// ================================================================ [OS.*] ================================================================
+
+// os.read()
+int cosmoB_osRead(CState *state, int nargs, CValue *args) {
+    if (nargs != 1) {
+        cosmoV_error(state, "os.read() expected 1 argument, got %d!", nargs);
+        return 0;
+    }
+
+    if (!IS_STRING(args[0])) {
+        cosmoV_typeError(state, "os.read()", "<string>", "%s", cosmoV_typeStr(args[0]));
+        return 0;
+    }
+
+    CObjString *str = cosmoV_readString(args[0]);
+
+    // open file
+    FILE *file = fopen(str->str, "rb");
+    char *buf;
+    size_t size, bRead;
+
+    // grab the size of the file
+    fseek(file, 0L, SEEK_END);
+    size = ftell(file);
+    rewind(file);
+
+    buf = cosmoM_xmalloc(state, size + 1); // +1 for the NULL terminator
+    bRead = fread(buf, sizeof(char), size, file); // read the file into the buffer
+
+    if (bRead < size) {
+        // an error occured! we don't need to really throw an error, returning a nil is good enough
+        cosmoV_pushNil(state);
+        return 1;
+    }
+
+    buf[bRead] = '\0'; // place the NULL terminator at the end of the buffer
+
+    // push the string to the stack to return
+    cosmoV_pushValue(state, cosmoV_newRef(cosmoO_takeString(state, buf, bRead)));
+    return 1;
+}
+
+// os.time()
+int cosmoB_osTime(CState *state, int nargs, CValue *args) {
+    struct timeval time;
+    if (nargs > 0) {
+        cosmoV_error(state, "os.time() expected no arguments, got %d!", nargs);
+        return 0;
+    }
+
+    gettimeofday(&time, NULL);
+    cosmoV_pushNumber(state, (time.tv_usec / 1000000.0) + time.tv_sec);
+    return 1;
+}
+
+COSMO_API void cosmoB_loadOSLib(CState *state) {
+    const char *identifiers[] = {
+        "read",
+        "time"
+    };
+
+    CosmoCFunction osLib[] = {
+        cosmoB_osRead,
+        cosmoB_osTime
+    };
+
+    cosmoV_pushString(state, "os");
+
+    int i;
+    for (i = 0; i < sizeof(identifiers)/sizeof(identifiers[0]); i++) {
+        cosmoV_pushString(state, identifiers[i]);
+        cosmoV_pushCFunction(state, osLib[i]);
+    }
+
+    cosmoV_makeObject(state, i);
+    cosmoV_register(state, 1); // register the os.* object to the global table
+}
+
 // ================================================================ [STRING.*] ================================================================
 
 // string.sub
diff --git a/src/cbaselib.h b/src/cbaselib.h
index db49b08..284b2d4 100644
--- a/src/cbaselib.h
+++ b/src/cbaselib.h
@@ -13,10 +13,16 @@ COSMO_API void cosmoB_loadLibrary(CState *state);
 
 /* loads the base object library, including:
     - object.ischild or <obj>:ischild()
-
+    - object.__proto (allows grabbing and setting proto objects)
 */
 COSMO_API void cosmoB_loadObjLib(CState *state);
 
+/* loads the os library, including:
+    - os.read()
+    - os.time()
+*/
+COSMO_API void cosmoB_loadOSLib(CState *state);
+
 /* loads the base string library, including:
     - string.sub & <string>:sub()
     - stirng.find & <string>:find()
diff --git a/src/main.c b/src/main.c
index cb0de1c..0826c9a 100644
--- a/src/main.c
+++ b/src/main.c
@@ -52,6 +52,7 @@ static void repl() {
 
     CState *state = cosmoV_newState();
     cosmoB_loadLibrary(state);
+    cosmoB_loadOSLib(state);
     cosmoB_loadVM(state);
 
     // add our custom REPL functions
@@ -113,6 +114,7 @@ static void runFile(const char* fileName) {
     char* script = readFile(fileName);
     CState *state = cosmoV_newState();
     cosmoB_loadLibrary(state);
+    cosmoB_loadOSLib(state);
 
     // add our input() function to the global table
     cosmoV_pushString(state, "input");