Use stdin if the source file name is a single dash

1 files changed, 2 insertions, 1 deletions

diff --git a/src/main.c b/src/main.c
index 36841a8..fad7f53 100644
--- a/src/main.c
+++ b/src/main.c
@@ -38,7 +38,8 @@ int main (int argc, char **argv)
     }
 
     const char *source_name = argv[1];
-    FILE *source_handle = fopen(source_name, "r");
+    FILE *source_handle =
+        strcmp(source_name, "-") == 0 ? stdin : fopen(source_name, "r");
     if (source_handle == NULL) {
         fprintf(stderr, "%s: %s: %s\n", prog_name, source_name, ERR_MSG);
         exit(1);