Add initial custom logging functions

2 files changed, 84 insertions, 6 deletions

diff --git a/src/badlib.h b/src/badlib.h
index f7a7974..2bc1807 100644
--- a/src/badlib.h
+++ b/src/badlib.h
@@ -873,4 +873,52 @@ platform_time(void) {
     return ts.tv_sec * 1000000000 + ts.tv_nsec;
 }
 
+//
+// Custom logger.
+//
+
+#define LOG_BUF_SIZE KB(4)
+
+typedef struct Logger {
+    Buf buf;
+    FILE *dest;
+    Arena *storage;
+} Logger;
+
+void
+log_init(Logger *l) {
+    buf_reserve(&l->buf, LOG_BUF_SIZE, l->storage);
+}
+
+void
+log_flush(Logger *l) {
+    if (l->buf.size) {
+        fprintf(l->dest, "%.*s", (int)l->buf.size, l->buf.mem);
+    }
+}
+
+void
+log_str(Logger *l, Str str) {
+    assert(l);
+    Buf *buf = &l->buf;
+    if (buf->size == 0) log_init(l);
+    while (str.size > 0) {
+        sz avail = buf->cap - buf->size;
+        if (avail == 0) {
+            log_flush(l);
+            avail = buf->cap - buf->size;
+            assert(avail > 0);
+        }
+        if (str.size < avail) {
+            buf_insert(&l->buf, str.mem, str.size, l->storage);
+            str.size = 0;
+            return;
+        }
+        buf_insert(&l->buf, str.mem, avail, l->storage);
+        str.mem += avail;
+        str.size -= avail;
+    }
+}
+
+
 #endif  // BADLIB_H
diff --git a/src/main.c b/src/main.c
index db833e4..e65c9e7 100644
--- a/src/main.c
+++ b/src/main.c
@@ -15,6 +15,7 @@ typedef enum ExecMode {
 static ExecMode mode = RUN_NORMAL;
 
 #define LEXER_MEM GB(2)
+#define LOGGER_MEM MB(1)
 
 void
 process_file(Str path) {
@@ -31,13 +32,42 @@ process_file(Str path) {
     printf("reading file...\n");
     FileContents file = platform_read_file(path, &lexer_arena);
 
-    // NOTE: Testing file read line by line.
     Str scanner = file.data;
-    for (sz i = 0; scanner.size != 0; i++) {
-        Str line = str_split(&scanner, cstr("\n"));
-        printf("%04ld  ", i);
-        printstrln(&line);
-    }
+    Str line = str_split(&scanner, cstr("\n"));
+    printf("%04d  ", 1);
+    printstrln(&line);
+    // NOTE: Testing file read line by line.
+    // for (sz i = 0; scanner.size != 0; i++) {
+    //     Str line = str_split(&scanner, cstr("\n"));
+    //     printf("%04ld  ", i + 1);
+    //     printstrln(&line);
+    // }
+
+    // Testing custom logger functions.
+    Arena logger_arena = arena_create(LOGGER_MEM, os_allocator);
+    Logger logger_info = {
+        .dest = stdout,
+        .storage = &logger_arena,
+    };
+    log_str(&logger_info, cstr("<<< "));
+    log_str(&logger_info, cstr("processing file: "));
+    log_str(&logger_info, path);
+    log_str(&logger_info, cstr(" >>>"));
+    log_str(&logger_info, cstr("\n"));
+
+    Logger logger_err = {
+        .dest = stderr,
+        .storage = &logger_arena,
+    };
+    log_str(&logger_err, path);
+    log_str(&logger_err, cstr(":1:1:"));
+    log_str(&logger_err, cstr("error: testing string logger\n"));
+    log_flush(&logger_info);
+    log_flush(&logger_err);
+
+    // TODO: add numeric logging
+    // TODO: add custom struct logging
+    // TODO: add a vaargs function with a little dsl
 
     // TODO: run lexer.
 }