Modify main function to allow multiple file processing

1 files changed, 36 insertions, 16 deletions

diff --git a/src/bootstrap/main.c b/src/bootstrap/main.c
index c98f60c..2847397 100755
--- a/src/bootstrap/main.c
+++ b/src/bootstrap/main.c
@@ -3,11 +3,11 @@
 #include <stdio.h>
 #include <stdlib.h>
 
+#include "string_view.c"
+
 void
-process_input(FILE *file) {
-    // TODO: Implement.
-    getchar();
-    (void)file;
+process_source(const StringView *source) {
+    sv_write(source, stdout);
 }
 
 #define REPL_PROMPT "bdl> "
@@ -17,19 +17,37 @@ run_repl(void) {
     printf("BDL REPL (Press Ctrl-C to exit)\n");
     while (true) {
         printf(REPL_PROMPT);
-        process_input(stdin);
+        getchar();
+        process_source(NULL);
     }
 }
 
 void
 run_file(char *file_name) {
-    FILE *fd = fopen(file_name, "r");
-    if (!fd) {
+    FILE *file = fopen(file_name, "r");
+    if (!file) {
         fprintf(stderr, "error: couldn't open input file: %s\n", file_name);
         exit(EXIT_FAILURE);
     }
-    process_input(fd);
-    fclose(fd);
+
+    // Read entire file into memory.
+    fseek(file, 0, SEEK_END);
+    size_t file_size = ftell(file);
+    fseek(file, 0, SEEK_SET);
+
+    char *source = malloc(file_size + 1);
+    fread(source, 1, file_size, file);
+    source[file_size] = 0;
+
+    StringView sv = (StringView){
+        .start = source,
+        .n = file_size,
+    };
+
+    process_source(&sv);
+
+    free(source);
+    fclose(file);
 }
 
 #ifndef BIN_NAME
@@ -61,16 +79,18 @@ main(int argc, char *argv[]) {
         }
     }
 
-    // TODO: Run from stdin if no file is given.
+    // Run from stdin.
+    if (optind == argc) {
+        // TODO: Run from stdin if no file is given.
+        return EXIT_SUCCESS;
+    }
 
     // Run from file.
-    if (optind != argc - 1) {
-        fprintf(stderr, "%s: No input file given.\n", BIN_NAME);
-        print_usage();
-        return EXIT_FAILURE;
+    while (optind < argc) {
+        char *file_name = argv[optind];
+        run_file(file_name);
+        optind++;
     }
-    char *file_name = argv[optind];
-    run_file(file_name);
 
     return EXIT_SUCCESS;
 }