X-Git-Url: http://git.scottworley.com/tl-append/blobdiff_plain/b3d5ed960aa9a43df2b48894867ec3481cee77fc..4b4feda25fa71fa0e695206bff02c90613b1f7a8:/tl-append.c?ds=sidebyside

diff --git a/tl-append.c b/tl-append.c
index 21e23a6..3e60a24 100644
--- a/tl-append.c
+++ b/tl-append.c
@@ -1,21 +1,39 @@
+#define _POSIX_C_SOURCE 199309L
+
+#include <errno.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
+#include <time.h>
+#include <unistd.h>
 
-const char *FILENAME = "tl.log";
-const size_t BUF_SIZE = 1024;
+#include "common.h"
 
-static void die(const char *message) {
-  fputs(message, stderr);
-  fputc('\n', stderr);
-  exit(1);
-}
+const int BUF_SIZE = 1024;
+
+const char PROMPT[] = "\33[H" /* Move cursor 'home' */
+                      "\33[J" /* Clear screen */
+                      "> ";
+const char ACKNOWLEDGE[] = "[OK]";
+const struct timespec ACKNOWLEDGE_DELAY = {0, 300000000};
+
+typedef struct {
+  int interactive;
+} conf_t;
+
+conf_t parse_command_line(int argc, char *argv[]) {
+  conf_t conf = {0};
 
-static void die_err(const char *message) {
-  perror(message);
-  exit(1);
+  if (argc == 2 && strcmp(argv[1], "-i") == 0 && isatty(2))
+    conf.interactive = 1;
+
+  return conf;
 }
 
-static void read_line(char *buf) {
+static void read_line(conf_t conf, char *buf) {
+  if (conf.interactive)
+    if (fputs(PROMPT, stderr) == EOF)
+      die("I/O error writing prompt");
   if (fgets(buf, BUF_SIZE, stdin) == NULL) {
     if (ferror(stdin))
       die("I/O error reading line");
@@ -27,20 +45,32 @@ static void read_line(char *buf) {
   }
 }
 
-static void write_line(const char *line) {
+static void write_line(conf_t conf, const char *line) {
+  const char *now = encode_time(time(NULL));
   FILE *f = fopen(FILENAME, "a");
   if (f == NULL)
     die_err("Error opening output file");
+  if (fputs(now, f) == EOF)
+    die("Error writing to output file");
+  if (fputc(' ', f) == EOF)
+    die("Error writing to output file");
   if (fputs(line, f) == EOF)
     die("Error writing to output file");
   if (fclose(f) != 0)
     die_err("Error closing output file");
+  if (conf.interactive) {
+    if (fputs(ACKNOWLEDGE, stderr) == EOF)
+      die("Error writing acknowledgment");
+    if (nanosleep(&ACKNOWLEDGE_DELAY, NULL) == -1 && errno != EINTR)
+      die_err("Error sleeping");
+  }
 }
 
-int main() {
+int main(int argc, char *argv[]) {
+  conf_t conf = parse_command_line(argc, argv);
   char buf[BUF_SIZE];
-  for (read_line(buf); buf[0]; read_line(buf)) {
-    write_line(buf);
+  for (read_line(conf, buf); buf[0]; read_line(conf, buf)) {
+    write_line(conf, buf);
   }
   return 0;
 }