X-Git-Url: http://git.scottworley.com/uniqt/blobdiff_plain/63eb64b5b01920499f9ea1a1d16a59eda6f716e3..48964c1ecbda205291cd3b6aba15bb89a1e66716:/uniqt.c

diff --git a/uniqt.c b/uniqt.c
index 2b7d229..1c7ee3f 100644
--- a/uniqt.c
+++ b/uniqt.c
@@ -1,4 +1,4 @@
-#define _POSIX_C_SOURCE 199309L
+#define _POSIX_C_SOURCE 200809L
 
 #include <stdio.h>
 #include <stdlib.h>
@@ -26,6 +26,10 @@ char *encode_time(time_t t) {
   return out;
 }
 
+typedef struct {
+  char *time_format;
+} conf_t;
+
 typedef struct {
   time_t start, end;
 } time_range_t;
@@ -44,19 +48,41 @@ static char *read_line() {
   int newline = getchar();
   if (newline != EOF && newline != (int)'\n')
     die("Expected newline");
+  if (scanf_ret == 0 && newline == (int)'\n')
+    return strdup("\n");
   return line;
 }
 
-static void write_line(time_range_t *range, char *line) {
+static void format_time(conf_t *conf, char *buf, size_t size, time_t *t) {
+  struct tm *tm = localtime(t);
+  if (tm == NULL)
+    die_err("Couldn't unpack time");
+  if (strftime(buf, size, conf->time_format, tm) == 0)
+    die_err("Couldn't format time");
+}
+
+static void write_line(conf_t *conf, time_range_t *range, char *line) {
   if (line == NULL)
     return;
-  if (printf("%ld %ld %s\n", range->start, range->end, line) < 0)
-    die("Couldn't write");
+  if (conf->time_format) {
+    const size_t MAX_TIMESTAMP_LENGTH = 1024;
+    char a[MAX_TIMESTAMP_LENGTH];
+    char b[MAX_TIMESTAMP_LENGTH];
+    format_time(conf, a, MAX_TIMESTAMP_LENGTH, &range->start);
+    format_time(conf, b, MAX_TIMESTAMP_LENGTH, &range->end);
+    if (printf("%s %s %s\n", a, b, line) < 0)
+      die("Couldn't write");
+  } else {
+    if (printf("%ld %ld %s\n", range->start, range->end, line) < 0)
+      die("Couldn't write");
+  }
+  if (fflush(stdout) == EOF)
+    die_err("Couldn't flush");
 }
 
 static int same(char *a, char *b) { return a && b && strcmp(a, b) == 0; }
 
-static void uniqt() {
+static void uniqt(conf_t *conf) {
   char *current_line = NULL;
   time_range_t current_time_range = make_time_range(NULL_TIME);
   for (;;) {
@@ -68,17 +94,37 @@ static void uniqt() {
       free(new_line);
       extend_time_range(&current_time_range, now);
     } else {
-      write_line(&current_time_range, current_line);
+      write_line(conf, &current_time_range, current_line);
       free(current_line);
       current_line = new_line;
       current_time_range = make_time_range(now);
     }
   }
-  write_line(&current_time_range, current_line);
+  write_line(conf, &current_time_range, current_line);
   free(current_line);
 }
 
-int main() {
-  uniqt();
+void usage() { die("usage: uniqt [-f time_format]"); }
+
+conf_t parse_command_line(int argc, char *argv[]) {
+  conf_t conf;
+  conf.time_format = NULL;
+
+  for (int i = 1; i < argc; i++) {
+    if (strcmp(argv[i], "-f") == 0) {
+      if (i + 1 >= argc)
+        die("-f requires an argument");
+      i++;
+      conf.time_format = argv[i];
+    } else
+      usage();
+  }
+
+  return conf;
+}
+
+int main(int argc, char *argv[]) {
+  conf_t conf = parse_command_line(argc, argv);
+  uniqt(&conf);
   return 0;
 }