cli: add option for timestamp line header format

1 files changed, 15 insertions, 2 deletions

diff --git a/logger.nim b/logger.nim
index a4d90a0..48aedf2 100644
--- a/logger.nim
+++ b/logger.nim
@@ -40,9 +40,22 @@ proc puts(f: File, s: string) =
 proc puts(f: File, r: Rune) =
   f.puts r.toUTF8
 
-proc loop*(probe: (timeout: Duration) -> int, getSymbol: (int) -> Rune) =
+func formatTimestampDateTime*(dt: DateTime): string =
+  dt.format(TIMESTAMP_FORMAT)
+
+func formatTimestampUnix*(dt: DateTime): string =
+  ($dt.toTime.toUnix) & " "
+
+func formatTimestampNone*(dt: DateTime): string =
+  ""
+
+proc loop*(
+  probe: (timeout: Duration) -> int,
+  getSymbol: (int) -> Rune,
+  timestampHeader: (DateTime) -> string,
+) =
   while true:
-    stdout.puts now().format(TIMESTAMP_FORMAT)
+    stdout.puts timestampHeader(now())
 
     for tick in 0..<60:
       let t = now()