cli: add option for timestamp line header format

1 files changed, 12 insertions, 1 deletions

diff --git a/main.nim b/main.nim
index 3128bac..cca5528 100644
--- a/main.nim
+++ b/main.nim
@@ -37,6 +37,7 @@ proc main() =
   var
     scale: Scale = log2
     symbols = UNICODE_SYMBOLS
+    timestampHeader = formatTimestampDateTime
     (min, max) = (0, 1000)
     probe: (timeout: Duration) -> int
 
@@ -66,6 +67,12 @@ proc main() =
       of "unicode", "utf8", "utf-8": symbols = UNICODE_SYMBOLS
       of "numeric", "ascii": symbols = NUMERIC_SYMBOLS
 
+    of "timestamp":
+      case val:
+      of "datetime", "date-time": timestampHeader = formatTimestampDateTime
+      of "unix", "epoch": timestampHeader = formatTimestampUnix
+      of "none", "false": timestampHeader = formatTimestampNone
+
     of "range":
       let parts = val.split(':', 1)
       if parts.len != 2: raise newException(ValueError, "Invalid range")
@@ -87,7 +94,11 @@ proc main() =
   if probe == nil:
     raise newException(ValueError, "Missing monitor argument")
 
-  loop(probe, (val: int) => symbols.indicator(min, max, val, scale))
+  loop(
+    probe,
+    (val: int) => symbols.indicator(min, max, val, scale),
+    timestampHeader,
+  )
 
 
 when not defined(test):