X-Git-Url: http://git.scottworley.com/tl-append/blobdiff_plain/b3d5ed960aa9a43df2b48894867ec3481cee77fc..4b4feda25fa71fa0e695206bff02c90613b1f7a8:/tl-append.c 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 #include #include +#include +#include +#include -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; }