Add initial custom logging functions

1 files changed, 48 insertions, 0 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