From f1de0cfb8b2c2d71d1776032b63e33090664ec84 Mon Sep 17 00:00:00 2001
From: Shav Kinderlehrer <shav@trinket.icu>
Date: Thu, 13 Apr 2023 09:24:41 -0400
Subject: [PATCH] Add stdin support

---
 include/arg.h |  1 +
 src/lib/arg.c |  4 ++++
 src/main.c    | 24 +++++++++++++++++++-----
 3 files changed, 24 insertions(+), 5 deletions(-)

diff --git a/include/arg.h b/include/arg.h
index 02ca360..985daa4 100644
--- a/include/arg.h
+++ b/include/arg.h
@@ -7,6 +7,7 @@
 struct config {
   bool color;
   bool lines;
+  bool has_read_stdin;
 };
 
 extern struct config conf;
diff --git a/src/lib/arg.c b/src/lib/arg.c
index 4b98f2b..e004540 100644
--- a/src/lib/arg.c
+++ b/src/lib/arg.c
@@ -89,6 +89,10 @@ int parseargs(int argc, char *argv[]) {
   for (i = 1; i < argc; i++) { // offset for argv[0]
     char *arg = argv[i];
 
+    if (*arg == '-') {
+      return i;
+    }
+
     if (arg[0] == '-') {
       if (arg[1] == '-') {
         parselongarg(arg);
diff --git a/src/main.c b/src/main.c
index 59b0849..19b3984 100644
--- a/src/main.c
+++ b/src/main.c
@@ -13,7 +13,7 @@
 #define GREY "\x1b[90m"
 #define RESET "\x1b[0m"
 
-void run(FILE *fp, char *filename, int tty) {
+void run(FILE *fp, char *filename, bool tty) {
   const char *invert_t = conf.color ? INVERT_T : "";
   const char *uinvert_t = conf.color ? UINVERT_T : "";
   const char *grey = conf.color ? GREY : "";
@@ -68,6 +68,13 @@ void run(FILE *fp, char *filename, int tty) {
 void initconf(void) {
   conf.color = true;
   conf.lines = true;
+  conf.has_read_stdin = false;
+}
+
+void clearstdin() {
+  // from
+  // https://stackoverflow.com/questions/7898215/how-to-clear-input-buffer-in-c
+  fseek(stdin, 0, SEEK_END);
 }
 
 int main(int argc, char *argv[]) {
@@ -80,14 +87,21 @@ int main(int argc, char *argv[]) {
     conf.color = false;
   }
 
+  bool tty = isatty(STDOUT_FILENO);
   if (argc > 1) {
     int offset = parseargs(argc, argv);
     for (int i = offset; i < argc; i++) {
+      if (*argv[i] == '-') {
+        if (conf.has_read_stdin)
+          clearstdin();
+        conf.has_read_stdin = true;
+        run(stdin, "stdin", tty);
+        continue;
+      }
+
       FILE *fp = fopen(argv[i], "rb");
       if (fp == NULL)
         die(argv[i]);
-
-      bool tty = isatty(STDOUT_FILENO);
       run(fp, argv[i], tty);
       fclose(fp);
 
@@ -97,10 +111,10 @@ int main(int argc, char *argv[]) {
     }
 
     if (offset == argc) {
-      run(stdin, "stdin", 1);
+      run(stdin, "stdin", tty);
     }
   } else {
-    run(stdin, "stdin", 1); // for piped-input or repl-like behavior
+    run(stdin, "stdin", tty); // for piped-input or repl-like behavior
   }
 
   return EXIT_SUCCESS;