cground/src/main.c

#include "parser.h"
#include "interpreter.h"
#include "compiler.h"
#include "types.h"
#include "serialize.h"
#include "repl.h"
#include <stdio.h>
#include <string.h>

char* getFileContents(const char* filename);

int main(int argc, char** argv) {
    if (argc == 1) {
        exit(repl());
    }

    bool compile = false;
    bool writeBytecode = false;
    bool readBytecode = false;
    char* fileName = NULL;
    char* outFileName = NULL;
    List groundArgs = createList();
    for (int i = 1; i < argc; i++) {
        if (strcmp("--compile", argv[i]) == 0 || strcmp("-c", argv[i]) == 0) {
            if (writeBytecode) {
                printf("Cannot choose both bytecode and compilation");
                exit(1);
            }
            compile = true;
        }
        else if (strcmp("--help", argv[i]) == 0 || strcmp("-h", argv[i]) == 0) {
            printf("GroundVM help\n");
            printf("Usage: %s <file> [-c] [--compile] [-h] [--help] [-w <output>] [--writeBytecode <output>] [-b] [--bytecode]\n", argv[0]);
            printf("Options:\n");
            printf("  -c or --compile: Outputs Linux x86_64 assembly instead of interpreting (WIP)\n");
            printf("  -h or --help: Shows this help message\n");
            printf("  -w <output> or --writebytecode <output>: Outputs binary Ground bytecode");
            printf("  -b <output> or --bytecode <output>: Inputs binary Ground bytecode");
            exit(0);
        } else if (strcmp("--writeBytecode", argv[i]) == 0 || strcmp("-w", argv[i]) == 0) {
            if (compile) {
                printf("Cannot choose both bytecode and compilation");
                exit(1);
            }
            writeBytecode = true;
            if (i + 1 >= argc) {
                printf("Usage: %s %s <output>", argv[0], argv[i]);
                exit(1);
            }
            i++;
            outFileName = argv[i];
        } else if (strcmp("--bytecode", argv[i]) == 0 || strcmp("-b", argv[i]) == 0) {
            readBytecode = true;
        } else {
            if (fileName == NULL) {
                fileName = argv[i];
            } else {
                appendToList(&groundArgs, createStringGroundValue(argv[i]));
            }
        }
    }

    if (fileName == NULL) {
        printf("Usage: %s <file> [-c] [--compile] [-h] [--help] [-w <output>] [--writeBytecode <output>] [-b] [--bytecode]\n", argv[0]);
        printf("Error: No file name provided\n");
        exit(1);
    }

    GroundProgram program;

    if (readBytecode) {
        deserializeProgramFromFile(fileName, &program);
    } else {
        char* file = getFileContents(fileName);
        program = parseFile(file);
        free(file);
    }

    if (compile) {
#ifdef GROUND_COMPILE_WITH_TRAM
        compileGroundProgram(&program, "program.gexe");
#else
        printf("This version of Ground has been compiled without Tram, so compilation is not supported.\n");
#endif
    } else if (writeBytecode) {
        serializeProgramToFile(outFileName, &program);
    } else {
        GroundVariable* variables = NULL;
        GroundLabel* labels = NULL;
        GroundCatch* catches = NULL;

        GroundScope scope;
        scope.variables = &variables;
        scope.labels = &labels;
        scope.catches = &catches;
        scope.isMainScope = true;
        addVariable(scope.variables, "CMDLINE_ARGS", createListGroundValue(groundArgs));
        interpretGroundProgram(&program, &scope);
    }
}