1 files changed, 60 insertions, 0 deletions
diff --git a/src/source.c b/src/source.c
new file mode 100644
index 0000000..b098751
--- /dev/null
+++ b/src/source.c
@@ -0,0 +1,60 @@
+#include "include/source.h"
+#include "include/util.h"
+#include <stdio.h>
+
+char* source_get(char* arg) {
+   return arg? 
+      source_get_from_fpath(arg):
+      source_get_from_stdin();
+
+}
+
+char* source_get_from_fpath(char* path) {
+   FILE* f;       // the file to read from
+   long f_size;   // the size of the file
+   char* src;     // the source code to return
+
+   f = fopen(path, "rb");
+   if (!f) { die("source file not found: %s", path); }
+
+   fseek(f, 0L, SEEK_END);
+   f_size = ftell(f);
+   rewind(f);
+
+   src = calloc(1, f_size + 1);
+
+   if ((fread(src, f_size, 1, f) != 1) || !src) {
+      fclose(f);
+      free(src);
+      die("could not read source file: %s", path);
+   }
+
+   return src;
+}
+
+char* source_get_from_stdin() {
+   size_t len; // the length of the given source
+   char* src;  // the source code to return
+
+   len = 0;
+   src = calloc(len, sizeof(char));
+
+   printf("> ");
+
+   while (src[len - 1] != EOF) {
+      char c;  // the character being read
+
+      if (src[len - 1] == '\n') { printf("> "); }
+
+      c = getchar();
+      src = realloc(src, (len + sizeof(char)));
+      memcpy(src + len, &c, sizeof(char));
+      len += sizeof(char);
+   }
+
+   src[len - 1] = '\0'; // null terminate
+
+   putchar('\n');
+
+   return src;
+}